twitch-bot-streak/main.go

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)
	}
}