mirror-nourybot/cmd/nourybot/timer.go

package main

import (
	"fmt"
	"regexp"
	"strings"

	"github.com/gempir/go-twitch-irc/v4"
	"github.com/google/uuid"
	"github.com/nouryxd/nourybot/internal/data"
)

// AddTimer slices the message into relevant parts, adding the values onto a
// new data.Timer struct so that the timer can be inserted into the database.
func (app *application) AddTimer(name, repeat string, message twitch.PrivateMessage) {
	cmdParams := strings.SplitN(message.Message, " ", 500)
	prefix := "()add timer"
	prefixLength := len("()add timer")
	nameLength := len(name)
	repeatLength := len(repeat)
	app.Log.Infow("Lengths",
		"prefix", prefixLength,
		"name", nameLength,
		"repeat", repeatLength,
		"repeat2", len(cmdParams[2]),
		"sum", prefixLength+nameLength+repeatLength,
	)

	// Split the message into the parts we need.
	//
	// message:  ()timer add sponsor    20m  hecking love my madmonq pills BatChest
	// parts:    | prefix  |  |name | |repeat | <----------- text ------------->   |
	text := message.Message[3+len(prefix)+len(name)+len(repeat) : len(message.Message)]

	// validateTimeFormat will be true if the repeat parameter is in
	// the format of either 30m, 10h, or 10h30m.
	validateTimeFormat, err := regexp.MatchString(`^(\d{1,2}[h])$|^(\d+[m])$|^(\d+[s])$|((\d{1,2}[h])((([0]?|[1-5]{1})[0-9])[m]))$`, repeat)
	if err != nil {
		app.Log.Errorw("Received malformed time format in timer",
			"repeat", repeat,
			"error", err,
		)
		return
	}
	id := uuid.NewString()
	timer := &data.Timer{
		Name:       name,
		Text:       text,
		Identifier: id,
		Channel:    message.Channel,
		Repeat:     repeat,
	}

	// Check if the time string we got is valid, this is important because the Scheduler
	// panics instead of erroring out if an invalid time format string is supplied.
	if validateTimeFormat {
		timer := &data.Timer{
			Name:       name,
			Text:       text,
			Identifier: id,
			Channel:    message.Channel,
			Repeat:     repeat,
		}

		err = app.Models.Timers.Insert(timer)
		if err != nil {
			app.Log.Errorw("Error inserting new timer into database",
				"timer", timer,
				"error", err,
			)
			reply := fmt.Sprintln("Something went wrong FeelsBadMan")
			app.Send(message.Channel, reply, message)
			return
		} else {
			app.Scheduler.AddFunc(fmt.Sprintf("@every %s", repeat), func() { app.newPrivateMessageTimer(message.Channel, text) }, id)
			app.Log.Infow("Added new timer",
				"timer", timer,
			)
			reply := fmt.Sprintf("Successfully added timer %s repeating every %s", name, repeat)
			app.Send(message.Channel, reply, message)
			return
		}
	} else {
		app.Log.Errorw("Received malformed time format in timer",
			"timer", timer,
			"error", err,
		)
		reply := "Something went wrong FeelsBadMan received wrong time format. Allowed formats: 30m, 10h, 10h30m"
		app.Send(message.Channel, reply, message)
		return
	}
}

// EditTimer just contains the logic for deleting a timer, and then adding a new one
// with the same name. It is technically not editing the timer.
func (app *application) EditTimer(name, repeat string, message twitch.PrivateMessage) {
	// Check if a timer with that name is in the database.
	old, err := app.Models.Timers.Get(name)
	if err != nil {
		app.Log.Errorw("Could not get timer",
			"timer", old,
			"error", err,
		)
		reply := "Something went wrong FeelsBadMan"
		app.Send(message.Channel, reply, message)
		return
	}

	// Delete the old timer
	identifier := old.Identifier
	app.Scheduler.RemoveJob(identifier)

	err = app.Models.Timers.Delete(identifier)
	if err != nil {
		app.Log.Errorw("Error deleting timer from database",
			"name", name,
			"identifier", identifier,
			"error", err,
		)

		reply := fmt.Sprintln("Something went wrong FeelsBadMan")
		app.Send(message.Channel, reply, message)
		return
	}

	// Add the new timer
	prefix := "()edit timer"

	// Split the message into the parts we need.
	//
	// message:  ()addtimer   sponsor    20m  hecking love my madmonq pills BatChest
	// parts:    | prefix  |  |name | |repeat | <----------- text ------------->   |
	text := message.Message[3+len(prefix)+len(name)+len(repeat) : len(message.Message)]

	// validateTimeFormat will be true if the repeat parameter is in the format
	// of either 30m, 10h, or 10h30m.
	validateTimeFormat, err := regexp.MatchString(`^(\d{1,2}[h])$|^(\d+[m])$|^(\d+[s])$|((\d{1,2}[h])((([0]?|[1-5]{1})[0-9])[m]))$`, repeat)
	if err != nil {
		app.Log.Errorw("Received malformed time format in timer",
			"repeat", repeat,
			"error", err,
		)
		return
	}

	id := uuid.NewString()
	timer := &data.Timer{
		Name:       name,
		Text:       text,
		Identifier: id,
		Channel:    message.Channel,
		Repeat:     repeat,
	}

	// Check if the time string we got is valid, this is important because the
	// Scheduler panics instead of erroring out if an invalid time format string is supplied.
	if validateTimeFormat {
		timer := &data.Timer{
			Name:       name,
			Text:       text,
			Identifier: id,
			Channel:    message.Channel,
			Repeat:     repeat,
		}

		err = app.Models.Timers.Insert(timer)
		if err != nil {
			app.Log.Errorw("Error inserting new timer into database",
				"timer", timer,
				"error", err,
			)
			reply := fmt.Sprintln("Something went wrong FeelsBadMan")
			app.Send(message.Channel, reply, message)
			return
		} else { // this is a bit scuffed. The else here is the end of a successful call.
			app.Scheduler.AddFunc(fmt.Sprintf("@every %s", repeat), func() { app.newPrivateMessageTimer(message.Channel, text) }, id)

			app.Log.Infow("Updated a timer",
				"Name", name,
				"Channel", message.Channel,
				"Old timer", old,
				"New timer", timer,
			)

			reply := fmt.Sprintf("Successfully updated timer %s", name)
			app.Send(message.Channel, reply, message)
			return
		}
	} else {
		app.Log.Errorw("Received malformed time format in timer",
			"timer", timer,
			"error", err,
		)
		reply := "Something went wrong FeelsBadMan received wrong time format. Allowed formats: 30s, 30m, 10h, 10h30m"
		app.Send(message.Channel, reply, message)
		return
	}
}

// ListTimers queries the database for all timers and uploads the contents as a paste.
func (app *application) ListTimers() string {
	timer, err := app.Models.Timers.GetAll()
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return ""
	}

	// The slice of timers is only used to log them at
	// the start so it looks a bit nicer.
	var ts []string

	// Iterate over all timers and then add them onto the scheduler.
	for i, v := range timer {
		// idk why this works but it does so no touchy touchy.
		// https://github.com/robfig/cron/issues/420#issuecomment-940949195
		i, v := i, v
		_ = i

		// cronName is the internal, unique tag/name for the timer.
		// A timer named "sponsor" in channel "forsen" will be named "forsensponsor"
		t := fmt.Sprintf(
			"ID: \t\t%v\n"+
				"Name: \t\t%v\n"+
				"Identifier: \t%v\n"+
				"Text: \t\t%v\n"+
				"Channel: \t%v\n"+
				"Repeat: \t%v\n"+
				"\n\n",
			v.ID, v.Name, v.Identifier, v.Text, v.Channel, v.Repeat,
		)
		ts = append(ts, t)
	}

	reply, err := app.uploadPaste(strings.Join(ts, ""))
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return ""
	}

	return reply
}

// InitialTimers is called on startup and queries the database for a list of
// timers and then adds each onto the scheduler.
func (app *application) InitialTimers() {
	timer, err := app.Models.Timers.GetAll()
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return
	}

	var ts []*data.Timer

	// Iterate over all timers and then add them onto the scheduler.
	for i, v := range timer {
		// idk why this works but it does so no touchy touchy.
		// https://github.com/robfig/cron/issues/420#issuecomment-940949195
		i, v := i, v
		_ = i

		// cronName is the internal, unique tag/name for the timer.
		// A timer named "sponsor" in channel "forsen" will be named "forsensponsor"

		// Repeating is at what times the timer should repeat.
		// 2 minute timer is @every 2m
		repeating := fmt.Sprintf("@every %s", v.Repeat)

		// Add new value to the slice
		ts = append(ts, v)

		app.Scheduler.AddFunc(repeating, func() { app.newPrivateMessageTimer(v.Channel, v.Text) }, v.Identifier)
	}

	app.Log.Infow("Initial timers",
		"timer", ts,
	)
}

// newPrivateMessageTimer is a helper function to set timers which trigger
// into sending a twitch PrivateMessage.
func (app *application) newPrivateMessageTimer(channel, text string) {
	app.SendNoContext(channel, text)
}

// DeleteTimer takes in the name of a timer and tries to delete the timer from the database.
func (app *application) DeleteTimer(name string, message twitch.PrivateMessage) {
	identifier, err := app.Models.Timers.GetIdentifier(name)
	if err != nil {
		app.Log.Errorw("Error retrieving identifier rom database",
			"name", name,
			"identifier", identifier,
			"error", err,
		)
	}

	app.Scheduler.RemoveJob(identifier)

	app.Log.Infow("Deleting timer",
		"name", name,
		"identifier", identifier,
		"message.Channel", message.Channel,
	)

	err = app.Models.Timers.Delete(identifier)
	if err != nil {
		app.Log.Errorw("Error deleting timer from database",
			"name", name,
			"identifier", identifier,
			"error", err,
		)

		reply := fmt.Sprintln("Something went wrong FeelsBadMan")
		app.Send(message.Channel, reply, message)
		return
	}

	reply := fmt.Sprintf("Deleted timer with name %s", name)
	app.Send(message.Channel, reply, message)
}

// DebugChannelTimers queries the database for all timers in channel
// and uploads the contents as a paste.
func (app *application) DebugChannelTimers(channel string) string {
	timer, err := app.Models.Timers.GetChannelTimer(channel)
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return ""
	}

	// The slice of timers is only used to log them at
	// the start so it looks a bit nicer.
	var ts []string

	// Iterate over all timers and then add them onto the scheduler.
	for i, v := range timer {
		// idk why this works but it does so no touchy touchy.
		// https://github.com/robfig/cron/issues/420#issuecomment-940949195
		i, v := i, v
		_ = i

		t := fmt.Sprintf(
			"Name: \t%v\n"+
				"ID: \t%v\n"+
				"Identifier: \t%v\n"+
				"Text: \t%v\n"+
				"Repeat: \t%v\n"+
				"\n",
			v.Name, v.ID, v.Identifier, v.Text, v.Repeat,
		)

		// Add new value to the slice
		ts = append(ts, t)

	}

	reply, err := app.uploadPaste(strings.Join(ts, ""))
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return ""
	}

	return reply
}

// ListchannelTimer queries the database for all timers in channel and
// uploads the contents as a paste.
func (app *application) ListChannelTimer(channel string) string {
	timer, err := app.Models.Timers.GetChannelTimer(channel)
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return ""
	}

	var ts []string

	// Iterate over all timers and then add them onto the scheduler.
	for i, v := range timer {
		// idk why this works but it does so no touchy touchy.
		// https://github.com/robfig/cron/issues/420#issuecomment-940949195
		i, v := i, v
		_ = i

		t := fmt.Sprintf(
			"Name: %v\n"+
				"Text: %v\n"+
				"Repeat: %v\n"+
				"\n",
			v.Name, v.Text, v.Repeat,
		)

		ts = append(ts, t)

	}

	reply, err := app.uploadPaste(strings.Join(ts, ""))
	if err != nil {
		app.Log.Errorw("Error trying to retrieve all timers from database", err)
		return ""
	}

	return reply
}