mirror-ollama-twitch-bot/main.go

package main

import (
	"os"
	"strings"

	"github.com/gempir/go-twitch-irc/v4"
	"github.com/joho/godotenv"
	"go.uber.org/zap"
)

type config struct {
	twitchUsername string
	twitchOauth    string
	commandPrefix  string
}

type application struct {
	TwitchClient  *twitch.Client
	Log           *zap.SugaredLogger
	OllamaModel   string
	OllamaContext string
	OllamaSystem  string
	Config        config
	UserMsgStore  map[string][]ollamaMessage
	MsgStore      []ollamaMessage
}

func main() {
	var cfg config

	logger := zap.NewExample()
	defer func() {
		if err := logger.Sync(); err != nil {
			logger.Sugar().Errorw("error syncing logger",
				"error", err,
			)
		}
	}()
	sugar := logger.Sugar()

	err := godotenv.Load()
	if err != nil {
		sugar.Fatal("Error loading .env")
	}

	// Twitch account config
	cfg.twitchUsername = os.Getenv("TWITCH_USERNAME")
	cfg.twitchOauth = os.Getenv("TWITCH_OAUTH")
	tc := twitch.NewClient(cfg.twitchUsername, cfg.twitchOauth)

	userMsgStore := make(map[string][]ollamaMessage)

	app := &application{
		TwitchClient:  tc,
		Log:           sugar,
		OllamaModel:   os.Getenv("OLLAMA_MODEL"),
		OllamaContext: os.Getenv("OLLAMA_CONTEXT"),
		OllamaSystem:  os.Getenv("OLLAMA_SYSTEM"),
		Config:        cfg,
		UserMsgStore:  userMsgStore,
	}

	// Received a PrivateMessage (normal chat message).
	app.TwitchClient.OnPrivateMessage(func(message twitch.PrivateMessage) {
		// roomId is the Twitch UserID of the channel the message originated from.
		// If there is no roomId something went really wrong.
		roomId := message.Tags["room-id"]
		if roomId == "" {
			return
		}

		// Message was shorter than our prefix is therefore it's irrelevant for us.
		if len(message.Message) >= 2 {
			// Check if the first 2 characters of the mesage were our prefix.
			// if they were forward the message to the command handler.
			if message.Message[:2] == "()" {
				go app.handleCommand(message)
				return
			}
		}
	})

	app.TwitchClient.OnConnect(func() {
		app.Log.Info("Successfully connected to Twitch Servers")
		app.Log.Info("Ollama Context: ", app.OllamaContext)
		app.Log.Info("Ollama System: ", app.OllamaSystem)

	})

	channels := os.Getenv("TWITCH_CHANNELS")
	channel := strings.Split(channels, ",")
	for i := 0; i < len(channel); i++ {
		app.TwitchClient.Join(channel[i])
		app.TwitchClient.Say(channel[i], "MrDestructoid")
		app.Log.Infof("Joining channel: %s", channel[i])
	}

	// Actually connect to chat.
	err = app.TwitchClient.Connect()
	if err != nil {
		panic(err)
	}
}