mirror of https://github.com/velour/catbase.git
89 lines
1.9 KiB
Go
89 lines
1.9 KiB
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)
|
|
}
|
|
}
|