twitch-bot-streak/main.go

115 lines
2.7 KiB
Go
Raw Normal View History

2024-03-24 01:57:46 +00:00
package main
import (
"bytes"
_ "embed"
"encoding/json"
"os"
"regexp"
"text/template"
_ "github.com/go-sql-driver/mysql"
"github.com/jmoiron/sqlx"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
"github.com/Luzifer/rconfig/v2"
)
var (
cfg = struct {
Action string `flag:"action,a" description:"Which action to perform"`
TwitchID uint64 `flag:"twitch-id" description:"ID of the user to access the streak of"`
TwitchUsername string `flag:"twitch-username" description:"Username of the user to update in streaks table"`
StorageConnString string `flag:"storage-conn-string" description:"How to connect to the database"`
LogLevel string `flag:"log-level" default:"info" description:"Log level (debug, info, warn, error, fatal)"`
VersionAndExit bool `flag:"version" default:"false" description:"Prints current version and exits"`
}{}
version = "dev"
//go:embed message.tpl
messageTemplate string
)
func initApp() error {
rconfig.AutoEnv(true)
if err := rconfig.ParseAndValidate(&cfg); err != nil {
return errors.Wrap(err, "parsing cli options")
}
l, err := logrus.ParseLevel(cfg.LogLevel)
if err != nil {
return errors.Wrap(err, "parsing log-level")
}
logrus.SetLevel(l)
return nil
}
func main() {
var err error
if err = initApp(); err != nil {
logrus.WithError(err).Fatal("initializing app")
}
if cfg.VersionAndExit {
logrus.WithField("version", version).Info("twitch-bot-streak")
os.Exit(0)
}
tpl, err := template.New("tpl").Parse(messageTemplate)
if err != nil {
logrus.WithError(err).Fatal("parsing message template")
}
db, err := sqlx.Open("mysql", cfg.StorageConnString)
if err != nil {
logrus.WithError(err).Fatal("connecting to database")
}
switch cfg.Action {
case "stream_start":
if err = startStream(db); err != nil {
logrus.WithError(err).Fatal("starting stream")
}
case "stream_offline":
if err = setStreamOffline(db); err != nil {
logrus.WithError(err).Fatal("stopping stream")
}
case "count_stream":
if err = startStream(db); err != nil {
logrus.WithError(err).Fatal("starting stream")
}
user, err := countStreak(db, cfg.TwitchID, cfg.TwitchUsername)
if err != nil {
logrus.WithError(err).Fatal("counting streak")
}
buf := new(bytes.Buffer)
if err = tpl.Execute(buf, user); err != nil {
logrus.WithError(err).Fatal("rendering template")
}
msg := regexp.MustCompile(`\s+`).ReplaceAllString(buf.String(), " ")
if err = json.NewEncoder(os.Stdout).Encode([]botResponse{
{
Type: "respond",
Attributes: map[string]any{
"message": msg,
},
},
}); err != nil {
logrus.WithError(err).Fatal("encoding bot response")
}
default:
logrus.Fatalf("unknown action %q", cfg.Action)
}
}