simpbot/main.go

package main

import (
	"fmt"
	"net/url"
	"os"
	"os/signal"
	"sync"

	"github.com/charmbracelet/log"
	"github.com/fsnotify/fsnotify"
	"github.com/spf13/viper"
)

func main() {
	simp := newSimp()
	logger := log.NewWithOptions(os.Stderr, log.Options{
		ReportCaller: true,
	})
	log.SetDefault(logger)
	viper.SetConfigName("config")
	viper.AddConfigPath(".")
	viper.AddConfigPath("/etc/simpbot")
	err := viper.ReadInConfig()
	if err != nil {
		log.Fatalf("Fatal error config file: %v \n", err)
	}
	viper.SetConfigType("yaml")
	Homeserver := viper.GetString("homeserver")
	viper.SetDefault("domain", Homeserver)
	Username := viper.GetString("username")
	Password := viper.GetString("password")
	Token := viper.GetString("access_token")
	simp.holodexToken = viper.GetString("api_token")
	HomeserverDomain := viper.GetString("domain")
	DimensionServer := viper.GetString("dimension")

	log.Info("logging into matrix", "homeserver", Homeserver, "username", Username)
	err = simp.SetupMatrix(Username, Password, Token, Homeserver, HomeserverDomain, DimensionServer)
	if err != nil {
		log.Fatal(err)
	}
	if simp.holodexToken == "" {
		log.Warn("No holodex API token provided, unlikely to be able to get streams")
	}
	simp.vtubers = loadVtubers()
	viper.WatchConfig()
	viper.OnConfigChange(func(e fsnotify.Event) {
		fmt.Println("Config file changed,reloading vtubers:", e.Name)
		simp.vtubers = loadVtubers()
	})
	var wg sync.WaitGroup
	stop := make(chan bool)
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt)
	go func() {
		<-c
		log.Info("trying to shutdown cleanly")
		simp.Stop()
		stop <- true
	}()

	wg.Add(1)
	go func() {
		err = simp.client.Sync()
		if err != nil {
			log.Error(err)
		}
		wg.Done()
	}()

	wg.Add(1)
	go func() {
		simp.Run()
		wg.Done()
		log.Info("simpbot shutdown")
	}()
	log.Info("simpbot running")
	wg.Wait()
	log.Info("shutting down")
}

func isValidURL(toTest string) bool {
	_, err := url.ParseRequestURI(toTest)
	if err != nil {
		return false
	}

	u, err := url.Parse(toTest)
	if err != nil || u.Scheme == "" || u.Host == "" {
		return false
	}

	return true
}