first

internal/telegram/bot.go

@@ -0,0 +1,81 @@
+package telegram
+
+import (
+	"log"
+	"time"
+
+	"tgchanbot/internal/config"
+	"tgchanbot/internal/storage"
+	"tgchanbot/internal/toc"
+
+	tele "gopkg.in/telebot.v3"
+)
+
+type Bot struct {
+	bot     *tele.Bot
+	cfg     *config.Config
+	storage *storage.Storage
+	toc     *toc.Manager
+	states  *StateManager
+}
+
+func New(cfg *config.Config, store *storage.Storage) (*Bot, error) {
+	pref := tele.Settings{
+		Token:  cfg.Bot.Token,
+		Poller: &tele.LongPoller{Timeout: 10 * time.Second},
+	}
+
+	b, err := tele.NewBot(pref)
+	if err != nil {
+		return nil, err
+	}
+
+	bot := &Bot{
+		bot:     b,
+		cfg:     cfg,
+		storage: store,
+		states:  NewStateManager(),
+	}
+
+	bot.toc = toc.NewManager(store, b, cfg.Channel.ID, time.Duration(cfg.TOC.DebounceSeconds)*time.Second)
+
+	bot.setupRoutes()
+
+	return bot, nil
+}
+
+func (b *Bot) setupRoutes() {
+	adminOnly := b.bot.Group()
+	adminOnly.Use(b.AdminMiddleware())
+
+	// Category management
+	adminOnly.Handle("/cat_add", b.handleCatAdd)
+	adminOnly.Handle("/cat_del", b.handleCatDel)
+	adminOnly.Handle("/cat_list", b.handleCatList)
+
+	// Post flow
+	adminOnly.Handle("/post", b.handlePost)
+	adminOnly.Handle(tele.OnText, b.handleTextOrForwarded)
+
+	// Entry management
+	adminOnly.Handle("/del", b.handleEntryDel)
+	adminOnly.Handle("/edit", b.handleEntryEdit)
+	adminOnly.Handle("/move", b.handleEntryMove)
+	adminOnly.Handle("/list", b.handleEntryList)
+
+	// TOC
+	adminOnly.Handle("/refresh", b.handleRefresh)
+
+	// Callbacks
+	b.bot.Handle(tele.OnCallback, b.handleCallback)
+}
+
+func (b *Bot) Start() {
+	log.Println("Bot started...")
+	b.bot.Start()
+}
+
+func (b *Bot) Stop() {
+	b.bot.Stop()
+	b.toc.Stop()
+}