mirror-nourybot/cmd/bot/timer.go

package main

import (
	"fmt"
	"strings"
	"time"

	"github.com/gempir/go-twitch-irc/v3"
	"github.com/lyx0/nourybot/internal/data"
	"github.com/lyx0/nourybot/pkg/common"
)

// AddCommand takes in a name parameter and a twitch.PrivateMessage. It slices the
// twitch.PrivateMessage after the name parameter and adds everything after to a text
// value. Then it calls the app.Models.Commands.Insert method with both name, and text
// values adding them to the database.
func (app *Application) AddTimer(name string, message twitch.PrivateMessage) {
	// prefixLength is the length of `()addtimer` plus +2 (for the space and zero based)
	cmdParams := strings.SplitN(message.Message, " ", 500)
	prefixLength := 12
	repeat := cmdParams[2]

	// Split the twitch message at the length of the prefix + the length of the name of the command.
	//      prefixLength |name| text
	//      0123456789012|4567|
	// e.g. ()addcommand dank FeelsDankMan
	//      |   part1    snip ^  part2   |
	text := message.Message[prefixLength+len(name)+len(cmdParams[2]) : len(message.Message)]

	// ()addtimer gfuel 5m Yo buy my cool gfuel cause its cool n shit
	// |         | name |
	timer := &data.Timer{
		Name:    name,
		Text:    text,
		Channel: message.Channel,
		Repeat:  repeat,
	}

	app.Logger.Infow("timer", timer)
	err := app.Models.Timers.Insert(timer)
	if err != nil {
		reply := fmt.Sprintf("Something went wrong FeelsBadMan %s", err)
		common.Send(message.Channel, reply, app.TwitchClient)
		return
	} else {
		app.Scheduler.Tag("pipelines", fmt.Sprintf("%s-%s", message.Channel, name)).Every(repeat).StartAt(time.Now()).Do(app.newTimer, message.Channel, text)
		reply := fmt.Sprintf("Successfully added timer %s repeating every %s", name, repeat)
		common.Send(message.Channel, reply, app.TwitchClient)
		return
	}
}

// InitialJoin is called on startup and queries the database for a list of
// channels which the TwitchClient then joins.
func (app *Application) InitialTimers() {
	// GetJoinable returns a slice of channel names.
	timer, err := app.Models.Timers.GetAll()
	if err != nil {
		app.Logger.Error(err)
		return
	}

	app.Logger.Info(timer)

	// Iterate over the slice of channels and join each.
	for _, v := range timer {
		app.Logger.Infow("Initial timers:",
			"Name", v.Name,
			"Channel", v.Channel,
			"Text", v.Text,
			"Repeat", v.Repeat,
			"V", v,
		)

		app.Scheduler.Tag("pipelines", fmt.Sprintf("%s-%s", v.Channel, v.Name)).Every(v.Repeat).StartAt(time.Now()).Do(app.newTimer, v.Channel, v.Text)

	}
}

func (app *Application) newTimer(channel, text string) {
	common.Send(channel, text, app.TwitchClient)
}