Initial commit

tg_bot/main.go

@@ -0,0 +1,152 @@
+package main
+
+import (
+	"os"
+	"regexp"
+	"strings"
+	"time"
+
+	"github.com/NicoNex/echotron/v3"
+	"github.com/TelegramExchange/tgex-backend/tg_bot/backend"
+	"github.com/TelegramExchange/tgex-backend/tg_bot/bot"
+	"github.com/TelegramExchange/tgex-backend/tg_bot/screens"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+)
+
+var botCommands = []echotron.BotCommand{
+	{Command: "start", Description: "Главное меню"},
+	{Command: "projects", Description: "Мои проекты"},
+	{Command: "placements", Description: "Размещения"},
+	{Command: "platform", Description: "Веб-платформа"},
+	{Command: "help", Description: "Помощь"},
+}
+
+func main() {
+	initLogger()
+
+	botToken := mustEnv("TELEGRAM__TOKEN")
+
+	backendClient := backend.New(backend.Config{
+		BaseURL:  mustEnv("BACKEND__BASE_URL"),
+		LoginURL: mustEnv("LOGIN_URL"),
+	})
+
+	// Регистрируем команды бота в Telegram
+	api := echotron.NewAPI(botToken)
+	if me, err := api.GetMe(); err != nil {
+		log.Error().Err(err).Msg("Failed to get bot profile")
+	} else if me.Result != nil {
+		bot.SetUsername(me.Result.Username)
+		log.Info().Str("username", me.Result.Username).Msg("Telegram bot authorized")
+	}
+
+	if _, err := api.SetMyCommands(nil, botCommands...); err != nil {
+		log.Error().Err(err).Msg("Failed to set bot commands")
+	} else {
+		log.Info().Msg("Bot commands registered successfully")
+	}
+
+	var commandRouter = func(command string) bot.State {
+		createCreativeRe := regexp.MustCompile(`^/start project_(.+)_createcreative$`)
+
+		switch {
+		case command == "/start":
+			return &screens.MainMenu{}
+		case command == "/start login":
+			return &screens.Login{}
+		case command == "/projects":
+			return &screens.MyProjects{BackState: &screens.MainMenu{}}
+		case command == "/placements":
+			return &screens.MyProjects{BackState: &screens.MainMenu{}, OpenPlacements: true}
+		case command == "/platform":
+			return &screens.PlatformLink{}
+		case command == "/help":
+			return &screens.Help{}
+		case createCreativeRe.MatchString(command):
+			projectID := createCreativeRe.FindStringSubmatch(command)[1]
+			return &screens.AddCreativeStart{Ctx: &screens.AddCreativeCtx{
+				ProjectID: projectID,
+				BackState: &screens.MainMenu{},
+			}}
+		}
+		panic("Unknown command: " + command)
+	}
+
+	var handleGlobalCallback = func(callbackData string) bot.State {
+		id, ok := strings.CutPrefix(callbackData, "pending_channel:")
+		if ok {
+			return &screens.SelectWorkspace{ChannelID: id, BackState: &screens.MainMenu{}}
+		}
+
+		id, ok = strings.CutPrefix(callbackData, "workspace_invite_accept:")
+		if ok {
+			return &screens.AcceptWorkspaceInvite{InviteID: id}
+		}
+
+		return nil
+	}
+
+	newBot := func(chatID int64) echotron.Bot {
+		return bot.NewBot(chatID, botToken, commandRouter, backendClient, handleGlobalCallback)
+	}
+
+	dsp := echotron.NewDispatcher(botToken, newBot)
+
+	updateOpts := echotron.UpdateOptions{
+		AllowedUpdates: []echotron.UpdateType{
+			echotron.MessageUpdate,
+			echotron.CallbackQueryUpdate,
+			echotron.MyChatMemberUpdate,
+			echotron.ChatMemberUpdate,
+			echotron.UpdateType("chat_join_request"),
+		},
+	}
+
+	echotron.SetChatRequestLimit(0, 0)
+
+	for {
+		err := dsp.PollOptions(false, updateOpts)
+		if err != nil {
+			log.Error().Err(err).Msg("dsp.Poll failed, retrying in 5 seconds...")
+			time.Sleep(5 * time.Second)
+			continue
+		}
+		break
+	}
+}
+
+func initLogger() {
+	zerolog.TimeFieldFormat = time.RFC3339
+
+	level := zerolog.InfoLevel
+	if parsedLevel, err := zerolog.ParseLevel(os.Getenv("LOGGER__LEVEL")); err == nil {
+		level = parsedLevel
+	}
+	zerolog.SetGlobalLevel(level)
+
+	log.Logger = zerolog.New(os.Stdout).With().Timestamp().Logger().Level(level)
+
+	prettyConsole := os.Getenv("LOGGER__PRETTY_CONSOLE")
+	if prettyConsole == "" {
+		prettyConsole = "true"
+	}
+	if prettyConsole == "true" {
+		log.Logger = zerolog.New(zerolog.ConsoleWriter{Out: os.Stdout, TimeFormat: "15:04:05"}).
+			With().
+			Timestamp().
+			Logger().
+			Level(level)
+	}
+
+	log.Info().Msg("Logger initialized")
+}
+
+func mustEnv(key string) string {
+	v := os.Getenv(key)
+	if v == "" {
+		log.Fatal().Str("env", key).Msg("missing required environment variable")
+	}
+
+	return v
+}