catbase/plugins/countdown/countdown.go

// Countdown is a plugin to do tasks on various milestones.
package countdown

import (
	"fmt"
	bh "github.com/timshannon/bolthold"
	"time"

	"github.com/rs/zerolog/log"

	"github.com/velour/catbase/bot"
	"github.com/velour/catbase/config"
)

var nextYear = time.Date(time.Now().Year()+1, time.January, 1, 0, 0, 0, 1, time.Local)
var thisYear = time.Now().Year()

type CountdownPlugin struct {
	b     bot.Bot
	store *bh.Store
	c     *config.Config
}

func New(b bot.Bot) *CountdownPlugin {
	p := &CountdownPlugin{
		b:     b,
		store: b.Store(),
		c:     b.Config(),
	}

	p.scheduleNYE(p.newYearRollover)

	return p
}

func (p *CountdownPlugin) scheduleNYE(f func()) {
	time.AfterFunc(nextYear.Sub(time.Now()), f)
	for i := 10; i > 0; i-- {
		mkFunc := func(i int) func() {
			return func() {
				p.nyeCountdown(i)
			}
		}
		time.AfterFunc(nextYear.Add(time.Duration(10-i)*time.Second).Sub(time.Now()), mkFunc(i))
	}
}

func (p *CountdownPlugin) newYearRollover() {
	log.Debug().Msgf("Happy new year!")

	// todo: something about rolling over
	//tx, err := p.db.Beginx()
	//if err != nil {
	//	logError(err, "error getting transaction")
	//	return
	//}
	//q := fmt.Sprintf(`create table counter_%d as select * from counter`, thisYear)
	//_, err = tx.Exec(q)
	//if err != nil {
	//	logError(err, "error running insert into")
	//	logError(tx.Rollback(), "error with insert rollback")
	//	return
	//}
	//q = `delete from counter`
	//_, err = tx.Exec(q)
	//if err != nil {
	//	logError(err, "error running delete")
	//	logError(tx.Rollback(), "error with delete rollback")
	//	return
	//}
	//err = tx.Commit()
	//logError(err, "error committing transaction")
}

func logError(err error, msg string) {
	if err == nil {
		return
	}
	log.Error().Err(err).Msgf(msg)
}

func (p *CountdownPlugin) nyeCountdown(i int) {
	chs := p.c.GetArray("channels", []string{})
	for _, ch := range chs {
		msg := fmt.Sprintf("%d!", i)
		p.b.Send(p.b.DefaultConnector(), bot.Message, ch, msg, msg)
	}
}