go-dette/main.go

package main

import (
	"context"
	"log"
	"os"

	"git.nefrace.ru/nefrace/nechotron"
	"git.nefrace.ru/nefrace/tongo"
	"github.com/NicoNex/echotron/v3"
	"github.com/joho/godotenv"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

var BuildTime string
var db *tongo.Database

func main() {
	log.Println("Starting Go-Dette")
	godotenv.Load()
	var err error
	db, err = tongo.NewConnection(os.Getenv("MONGO_URI"), "godette")
	if err != nil {
		log.Fatalf("Can't connect to database: %v", err)
	}
	createIndexes(db)
	token := os.Getenv("TELEGRAM_TOKEN")
	neco := nechotron.NewTron(token, &MainState)
	neco.
		Use(UserDBUpdater(db)).
		Use(ChatUpdate(db)).
		Use(ErrorLogger).
		Use(ExecTimeLogger)
	api := echotron.NewAPI(token)
	nechotron.SetMyCommands(api, "", echotron.BotCommandScope{Type: echotron.Any}, defaultCommands...)
	nechotron.SetMyCommands(api, "", echotron.BotCommandScope{Type: echotron.BCSTAllChatAdministrators}, allCommands...)
	go RSSTask(&api)
	for {
		log.Println(neco.DispatchPoll())
	}
}

func createIndexes(db *tongo.Database) error {
	userStore := tongo.NewStore[User](db)
	_, err := userStore.Coll.Indexes().CreateOne(context.Background(), userIndex, options.CreateIndexes())
	karmaShotStore := tongo.NewStore[KarmaShot](db)
	_, err = karmaShotStore.Coll.Indexes().CreateMany(context.Background(), []mongo.IndexModel{karmaFromIndex, karmaToIndex}, options.CreateIndexes())
	return err
}