Files
dislate/main.go
Gustavo L de Mello (Guz) 0566b1624a refactor: rewrite, a lot
The last implementation was sketchy, and since I have been weeks without
interacting with the codebase, I don't have the pacience to know what
the fuck my mind was thinking before
2024-11-04 19:21:01 -03:00

88 lines
2.0 KiB
Go

package main
import (
"database/sql"
"flag"
"log/slog"
"os"
"os/signal"
"syscall"
"time"
"forge.capytal.company/capytal/dislate/bot"
"forge.capytal.company/capytal/dislate/translator"
_ "github.com/tursodatabase/go-libsql"
"github.com/charmbracelet/log"
)
type TranslationProvider string
const (
GOOGLE_TRANSLATE TranslationProvider = "google-translate"
)
// var translation_provider = flag.String("tprovider", string(GOOGLE_TRANSLATE), "Translation provider")
var (
database_file = flag.String("db", "file:./guild.db", "SQLite database file/location")
discord_token = flag.String(
"token",
os.Getenv("DISCORD_TOKEN"),
"Discord bot authentication token",
)
)
func init() {
flag.Parse()
}
func main() {
logger := slog.New(log.NewWithOptions(os.Stderr, log.Options{
TimeFormat: time.DateTime,
ReportTimestamp: true,
ReportCaller: true,
}))
db, err := sql.Open("libsql", "file://sqlite.db")
if err != nil {
logger.Error("Failed to start SQLite database", slog.String("error", err.Error()))
return
}
logger.Info("Connection to database started", slog.String("file", *database_file))
defer func() {
err := db.Close()
if err != nil {
logger.Error("Failed to close database connection", slog.String("err", err.Error()))
return
}
logger.Info("Connection to database closed", slog.String("file", *database_file))
}()
bot, err := bot.NewBot(*discord_token, db, translator.NewMockTranslator(), logger)
if err != nil {
logger.Error("Failed to create discord bot", slog.String("err", err.Error()))
return
}
if err := bot.Start(); err != nil {
logger.Error("Failed to start discord bot", slog.String("err", err.Error()))
return
}
logger.Info("Discord bot started")
defer func() {
if err := bot.Stop(); err != nil {
logger.Error("Failed to stop discord bot", slog.String("err", err.Error()))
return
}
logger.Info("Discord bot stopped")
}()
sig := make(chan os.Signal, 1)
signal.Notify(sig, os.Interrupt, syscall.SIGINT)
<-sig
}