matrix-gptzoo/main.go

47 lines
933 B
Go
Raw Normal View History

2023-05-12 13:59:20 +02:00
package main
import (
2023-05-16 20:07:05 +02:00
"os"
"os/signal"
2023-05-17 19:51:45 +02:00
"ewintr.nl/matrix-bots/bot"
2023-06-09 14:28:32 +02:00
"github.com/BurntSushi/toml"
2023-05-15 12:28:12 +02:00
_ "github.com/mattn/go-sqlite3"
2023-05-12 13:59:20 +02:00
"golang.org/x/exp/slog"
)
func main() {
2023-05-15 12:28:12 +02:00
logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
2023-06-09 14:28:32 +02:00
var config bot.Config
if _, err := toml.DecodeFile(getParam("CONFIG_PATH", "conf.toml"), &config); err != nil {
logger.Error(err.Error())
os.Exit(1)
}
logger.Info("loaded config", slog.Int("bots", len(config.Bots)))
for _, bc := range config.Bots {
b := bot.New(config.OpenAI.APIKey, bc, logger)
2023-06-08 19:18:38 +02:00
if err := b.Init(); err != nil {
logger.Error(err.Error())
os.Exit(1)
}
go b.Run()
2023-06-09 14:28:32 +02:00
logger.Info("started bot", slog.String("name", bc.UserDisplayName))
2023-05-15 12:28:12 +02:00
}
2023-05-12 15:07:16 +02:00
done := make(chan os.Signal)
signal.Notify(done, os.Interrupt)
<-done
2023-06-09 14:28:32 +02:00
logger.Info("service stopped")
2023-05-16 20:07:05 +02:00
}
func getParam(name, def string) string {
val, ok := os.LookupEnv(name)
if !ok {
return def
2023-05-15 12:28:12 +02:00
}
2023-05-16 20:07:05 +02:00
return val
2023-05-12 13:59:20 +02:00
}