refactor: migrate to gotd (wip)

bot/bot.go

@@ -1,59 +1,54 @@
 package bot
 
 import (
+	"context"
 	"os"
+	"time"
 
-	"github.com/amarnathcjd/gogram/telegram"
+	"github.com/celestix/gotgproto"
+	"github.com/celestix/gotgproto/sessionMaker"
+	"github.com/glebarez/sqlite"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
 )
 
-var (
+var Client *gotgproto.Client
-	Client *telegram.Client
-)
 
 func Init() {
-	logger.L.Debug("Initializing bot...")
+	logger.L.Info("Initializing client...")
-	var err error
+	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
-	Client, err = telegram.NewClient(telegram.ClientConfig{
+	defer cancel()
-		AppID:    config.Cfg.Telegram.AppID,
+
-		AppHash:  config.Cfg.Telegram.AppHash,
+	resultChan := make(chan struct {
-		LogLevel: telegram.LogInfo,
+		client *gotgproto.Client
+		err    error
 	})
-	if err != nil {
+
-		logger.L.Fatal("Failed to create telegram client: ", err)
+	go func() {
+		client, err := gotgproto.NewClient(int(config.Cfg.Telegram.AppID), config.Cfg.Telegram.AppHash, gotgproto.ClientTypeBot(config.Cfg.Telegram.Token),
+			&gotgproto.ClientOpts{
+				Session:          sessionMaker.SqlSession(sqlite.Open("data/session.db")),
+				DisableCopyright: true,
+				Middlewares:      FloodWaitMiddleware(),
+			},
+		)
+		resultChan <- struct {
+			client *gotgproto.Client
+			err    error
+		}{client, err}
+	}()
+
+	select {
+	case <-ctx.Done():
+		logger.L.Fatal("Failed to initialize client")
 		os.Exit(1)
+	case result := <-resultChan:
+		if result.err != nil {
+			logger.L.Fatalf("Failed to initialize client: %s", result.err)
+			os.Exit(1)
+		}
+		Client = result.client
+		RegisterHandlers(Client.Dispatcher)
+		logger.L.Info("Client initialized")
 	}
-	if err := Client.LoginBot(config.Cfg.Telegram.Token); err != nil {
-		logger.L.Fatal("Failed to login bot: ", err)
-		os.Exit(1)
-	}
-	logger.L.Info("Bot logged in")
-	_, err = Client.BotsSetBotCommands(&telegram.BotCommandScopeDefault{}, "", []*telegram.BotCommand{
-		{Command: "start", Description: "开始使用"},
-		{Command: "help", Description: "显示帮助"},
-		{Command: "silent", Description: "静默模式"},
-		{Command: "storage", Description: "设置默认存储位置"},
-		{Command: "save", Description: "保存所回复文件"},
-	})
-	if err != nil {
-		logger.L.Errorf("Failed to set bot commands: ", err)
-	}
-	logger.L.Info("Bot initialized")
-}
-
-func Run() {
-	if Client == nil {
-		Init()
-	}
-
-	Client.On("command:start", Start, telegram.FilterPrivate, telegram.FilterChats(config.Cfg.Telegram.Admins...))
-	Client.On("command:help", Help, telegram.FilterPrivate, telegram.FilterChats(config.Cfg.Telegram.Admins...))
-	Client.On("command:silent", ChangeSilentMode, telegram.FilterPrivate, telegram.FilterChats(config.Cfg.Telegram.Admins...))
-	Client.On("command:storage", SetDefaultStorage, telegram.FilterPrivate, telegram.FilterChats(config.Cfg.Telegram.Admins...))
-	Client.On("command:save", SaveCmd, telegram.FilterPrivate, telegram.FilterChats(config.Cfg.Telegram.Admins...))
-	Client.On(telegram.OnMessage, HandleFileMessage, telegram.FilterPrivate, telegram.FilterChats(config.Cfg.Telegram.Admins...), telegram.FilterMedia)
-	Client.On("callback:add", AddToQueue)
-
-	Client.Idle()
 }

bot/handlers.go

@@ -1,33 +1,62 @@
 package bot
 
 import (
-	"context"
 	"fmt"
 	"strconv"
 	"strings"
 
-	"github.com/amarnathcjd/gogram/telegram"
 	"github.com/duke-git/lancet/v2/slice"
 	"github.com/gookit/goutil/maputil"
+	"github.com/gotd/td/telegram/message/styling"
+	"github.com/gotd/td/tg"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/dispatcher/handlers"
+	"github.com/celestix/gotgproto/dispatcher/handlers/filters"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/krau/SaveAny-Bot/common"
+	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/dao"
 	"github.com/krau/SaveAny-Bot/logger"
 	"github.com/krau/SaveAny-Bot/model"
 	"github.com/krau/SaveAny-Bot/queue"
 	"github.com/krau/SaveAny-Bot/storage"
 	"github.com/krau/SaveAny-Bot/types"
-	"github.com/mymmrac/telego/telegoutil"
 )
 
-func Start(message *telegram.NewMessage) error {
+func RegisterHandlers(dispatcher dispatcher.Dispatcher) {
-	if err := dao.CreateUser(message.ChatID()); err != nil {
+	dispatcher.AddHandler(handlers.NewAnyUpdate(checkPermission))
-		logger.L.Errorf("Failed to create user: %s", err)
+	dispatcher.AddHandler(handlers.NewCommand("start", start))
-		return err
+	dispatcher.AddHandler(handlers.NewCommand("help", help))
-	}
+	dispatcher.AddHandler(handlers.NewCommand("silent", silent))
-	return Help(message)
+	dispatcher.AddHandler(handlers.NewCommand("storage", setDefaultStorage))
+	dispatcher.AddHandler(handlers.NewCommand("save", saveCmd))
+	dispatcher.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix("add"), AddToQueue))
+	dispatcher.AddHandler(handlers.NewMessage(filters.Message.Media, handleFileMessage))
 }
 
-func Help(message *telegram.NewMessage) error {
+const noPermissionText string = `
-	helpText := `
+本 Bot 仅限个人使用.
+您可以部署自己的实例: https://github.com/krau/SaveAny-Bot
+`
+
+func checkPermission(ctx *ext.Context, update *ext.Update) error {
+	if !slice.Contain(config.Cfg.Telegram.Admins, update.EffectiveUser().ID) {
+		ctx.Reply(update, noPermissionText, nil)
+		return dispatcher.EndGroups
+	}
+	return dispatcher.ContinueGroups
+}
+
+func start(ctx *ext.Context, update *ext.Update) error {
+	if err := dao.CreateUser(update.EffectiveUser().ID); err != nil {
+		logger.L.Errorf("Failed to create user: %s", err)
+		return dispatcher.EndGroups
+	}
+	return help(ctx, update)
+}
+
+const helpText string = `
 SaveAny Bot - 转存你的 Telegram 文件
 命令:
 /start - 开始使用
@@ -37,235 +66,191 @@ SaveAny Bot - 转存你的 Telegram 文件
 /save - 保存文件
 
 静默模式: 开启后 Bot 直接保存到收到的文件到默认位置, 不再询问
-	`
+`
-	if _, err := message.Reply(helpText); err != nil {
+
-		logger.L.Errorf("Failed to send help message: %s", err)
+func help(ctx *ext.Context, update *ext.Update) error {
-		return err
+	ctx.Reply(update, helpText, nil)
-	}
+	return dispatcher.EndGroups
-	return nil
 }
 
-func ChangeSilentMode(message *telegram.NewMessage) error {
+func silent(ctx *ext.Context, update *ext.Update) error {
-	user, err := dao.GetUserByUserID(message.ChatID())
+	user, err := dao.GetUserByUserID(update.EffectiveUser().ID)
 	if err != nil {
-		logger.L.Error(err)
+		logger.L.Errorf("Failed to get user: %s", err)
-		return err
+		return dispatcher.EndGroups
 	}
 	user.Silent = !user.Silent
-	err = dao.UpdateUser(user)
+	if err := dao.UpdateUser(user); err != nil {
-	if err != nil {
+		logger.L.Errorf("Failed to update user: %s", err)
-		logger.L.Error(err)
+		return dispatcher.EndGroups
-		return err
 	}
-	if _, err := message.Reply(fmt.Sprintf("已%s静默模式", map[bool]string{true: "开启", false: "关闭"}[user.Silent])); err != nil {
+	ctx.Reply(update, fmt.Sprintf("已%s静默模式", func() string {
-		return err
+		if user.Silent {
-	}
+			return "开启"
-	return nil
+		}
+		return "关闭"
+	}()), nil)
+	return dispatcher.EndGroups
 }
 
-func SetDefaultStorage(message *telegram.NewMessage) error {
+func setDefaultStorage(ctx *ext.Context, update *ext.Update) error {
 	if len(storage.Storages) == 0 {
-		message.Reply("当前无可用存储端, 请检查配置.")
+		ctx.Reply(update, "未配置存储", nil)
-		return nil
+		return dispatcher.EndGroups
 	}
-	_, _, args := telegoutil.ParseCommand(message.Text())
+	args := strings.Split(update.EffectiveMessage.Text, " ")
-	availableStorages := maputil.Keys(storage.Storages)
+	avaliableStorages := maputil.Keys(storage.Storages)
-	if len(args) == 0 {
+	if len(args) < 2 {
-		text := "请提供存储位置名称, 可用项:"
+		text := []styling.StyledTextOption{
-		for _, name := range availableStorages {
+			styling.Plain("请提供存储位置名称, 可用项:"),
-			text += fmt.Sprintf("\n`%s`", name)
 		}
-		text += fmt.Sprintf("\n`all`")
+		for _, name := range avaliableStorages {
-		message.Reply(text, telegram.SendOptions{ParseMode: telegram.MarkDown})
+			text = append(text, styling.Plain("\n"))
-		return nil
+			text = append(text, styling.Code(name))
+		}
+		ctx.Reply(update, text, nil)
+		return dispatcher.EndGroups
 	}
-	storageName := args[0]
+	storageName := args[1]
-	if !slice.Contain(availableStorages, storageName) {
+	if !slice.Contain(avaliableStorages, storageName) {
-		message.Reply("参数错误")
+		ctx.Reply(update, "存储位置不存在", nil)
-		return nil
+		return dispatcher.EndGroups
 	}
-	user, err := dao.GetUserByUserID(message.ChatID())
+	user, err := dao.GetUserByUserID(update.EffectiveUser().ID)
 	if err != nil {
-		logger.L.Error(err)
+		logger.L.Errorf("Failed to get user: %s", err)
-		return err
+		return dispatcher.EndGroups
 	}
 	user.DefaultStorage = storageName
-	err = dao.UpdateUser(user)
+	if err := dao.UpdateUser(user); err != nil {
-	if err != nil {
+		logger.L.Errorf("Failed to update user: %s", err)
-		logger.L.Error(err)
+		return dispatcher.EndGroups
-		return err
 	}
-	if _, err := message.Reply(fmt.Sprintf("已设置默认存储位置为: %s", storageName)); err != nil {
+	ctx.Reply(update, fmt.Sprintf("已设置默认存储位置为 %s", storageName), nil)
-		return err
+	return dispatcher.EndGroups
-	}
-	return nil
 }
 
-func SaveCmd(message *telegram.NewMessage) error {
+func saveCmd(ctx *ext.Context, update *ext.Update) error {
-	targetMessage, err := message.GetReplyMessage()
+	// TODO: Implement save command
-	if err != nil {
+	return dispatcher.EndGroups
-		message.Reply("请回复要保存的文件")
+}
-		return nil
-	}
-	if !targetMessage.IsMedia() {
-		message.Reply("回复的消息不包含文件")
-		return nil
-	}
 
-	msg, err := targetMessage.Reply("正在获取文件信息...")
+func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
+	logger.L.Debug("Got media: ", update.EffectiveMessage.Media.TypeName())
+	supported, err := supportedMediaFilter(update.EffectiveMessage)
 	if err != nil {
-		logger.L.Error(err)
-		message.Reply("获取文件信息失败")
 		return err
 	}
-
+	if !supported {
-	_, _, _, fileName, err := telegram.GetFileLocation(targetMessage.Media())
+		return dispatcher.EndGroups
-	if err != nil {
-		logger.L.Error(err)
-		targetMessage.Reply("获取文件信息失败")
-		return err
 	}
-	if fileName == "" {
+
-		logger.L.Error("Empty file name")
+	user, err := dao.GetUserByUserID(update.EffectiveUser().ID)
-		targetMessage.Reply("文件名为空")
+	if err != nil {
-		return nil
+		logger.L.Errorf("Failed to get user: %s", err)
+		return dispatcher.EndGroups
+	}
+
+	msg, err := ctx.Reply(update, "正在获取文件信息...", nil)
+	if err != nil {
+		logger.L.Errorf("Failed to reply: %s", err)
+		return dispatcher.EndGroups
+	}
+	media := update.EffectiveMessage.Media
+	file, err := common.FileFromMedia(media)
+	if err != nil {
+		logger.L.Errorf("Failed to get file from media: %s", err)
+		ctx.Reply(update, "无法获取文件", nil)
+		return dispatcher.EndGroups
+	}
+	if file.FileName == "" {
+		ctx.Reply(update, "无法获取文件名", nil)
+		return dispatcher.EndGroups
 	}
 
 	if err := dao.AddReceivedFile(&model.ReceivedFile{
 		Processing:     false,
-		FileName:       fileName,
+		FileName:       file.FileName,
-		ChatID:         targetMessage.ChatID(),
+		ChatID:         update.EffectiveChat().GetID(),
-		MessageID:      targetMessage.Message.ID,
+		MessageID:      update.EffectiveMessage.ID,
 		ReplyMessageID: msg.ID,
 	}); err != nil {
-		logger.L.Error(err)
+		logger.L.Errorf("Failed to add received file: %s", err)
-		msg.Edit("保存文件信息失败")
+		if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-		return err
+			Message: "无法保存文件",
-	}
+			ID:      msg.ID,
+		}); err != nil {
+			logger.L.Errorf("Failed to edit message: %s", err)
+		}
 
-	user, err := dao.GetUserByUserID(message.ChatID())
+		return dispatcher.EndGroups
-	if err != nil {
-		logger.L.Error(err)
-		msg.Edit("获取用户信息失败")
-		return err
 	}
 
 	if !user.Silent {
-		msg.Edit("请选择要保存的位置:", telegram.SendOptions{
+		text := "请选择存储位置"
-			ReplyMarkup: AddTaskReplyMarkup(targetMessage.Message.ID),
+		_, err = ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message:     text,
+			ReplyMarkup: getAddTaskMarkup(update.EffectiveMessage.ID),
+			ID:          msg.ID,
 		})
-		return nil
+		if err != nil {
+			logger.L.Errorf("Failed to edit message: %s", err)
+		}
+		return dispatcher.EndGroups
 	}
 
 	if user.DefaultStorage == "" {
-		msg.Edit("请先使用 /storage 命令设置默认存储位置, 或者关闭静默模式")
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-		return nil
+			Message: "请先使用 /storage 设置默认存储位置",
-	}
+			ID:      msg.ID,
-
-	queue.AddTask(types.Task{
-		Ctx:            context.TODO(),
-		Status:         types.Pending,
-		FileName:       fileName,
-		Storage:        types.StorageType(user.DefaultStorage),
-		ChatID:         targetMessage.ChatID(),
-		MessageID:      targetMessage.Message.ID,
-		ReplyMessageID: msg.ID,
-	})
-
-	msg.Edit(fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", fileName, queue.Len()))
-
-	return nil
-}
-
-func HandleFileMessage(message *telegram.NewMessage) error {
-	if !message.IsMedia() {
-		return nil
-	}
-
-	user, err := dao.GetUserByUserID(message.ChatID())
-	if err != nil {
-		logger.L.Error(err)
-		return nil
-	}
-
-	msg, err := message.Reply("正在获取文件信息...")
-	if err != nil {
-		logger.L.Error(err)
-		return err
-	}
-
-	_, _, _, fileName, err := telegram.GetFileLocation(message.Media())
-	if err != nil {
-		logger.L.Error(err)
-		message.Reply("获取文件信息失败")
-		return err
-	}
-	if fileName == "" {
-		logger.L.Error("Empty file name")
-		message.Reply("文件名为空")
-		return nil
-	}
-
-	if err := dao.AddReceivedFile(&model.ReceivedFile{
-		Processing:     false,
-		FileName:       fileName,
-		ChatID:         message.ChatID(),
-		MessageID:      message.Message.ID,
-		ReplyMessageID: msg.ID,
-	}); err != nil {
-		logger.L.Error(err)
-		msg.Edit("保存文件信息失败")
-		return err
-	}
-
-	if !user.Silent {
-		msg.Edit("请选择要保存的位置:", telegram.SendOptions{
-			ReplyMarkup: AddTaskReplyMarkup(message.Message.ID),
 		})
-		return nil
+		return dispatcher.EndGroups
-	}
-
-	if user.DefaultStorage == "" {
-		msg.Edit("请先使用 /storage 命令设置默认存储位置, 或者关闭静默模式")
-		return nil
 	}
 
 	queue.AddTask(types.Task{
-		Ctx:            context.TODO(),
+		Ctx:            ctx,
 		Status:         types.Pending,
-		FileName:       fileName,
+		FileName:       file.FileName,
 		Storage:        types.StorageType(user.DefaultStorage),
-		ChatID:         message.ChatID(),
+		ChatID:         update.EffectiveChat().GetID(),
-		MessageID:      message.Message.ID,
 		ReplyMessageID: msg.ID,
+		MessageID:      update.EffectiveMessage.ID,
 	})
 
-	msg.Edit(fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", fileName, queue.Len()))
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-	return nil
+		Message: fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", file.FileName, queue.Len()),
+		ID:      msg.ID,
+	})
+	return dispatcher.EndGroups
 }
 
-func AddToQueue(query *telegram.CallbackQuery) error {
+func AddToQueue(ctx *ext.Context, update *ext.Update) error {
-	args := strings.Split(query.DataString(), " ")
+	args := strings.Split(string(update.CallbackQuery.Data), " ")
 	messageID, _ := strconv.Atoi(args[1])
-	logger.L.Debug(query.ChatID, messageID)
+	logger.L.Debugf("Got add to queue: chatID: %d, messageID: %d, storage: %s", update.EffectiveChat().GetID(), messageID, args[2])
-	receivedFile, err := dao.GetReceivedFileByChatAndMessageID(query.ChatID, int32(messageID))
+	record, err := dao.GetReceivedFileByChatAndMessageID(update.EffectiveChat().GetID(), messageID)
 	if err != nil {
-		logger.L.Error(err)
+		logger.L.Errorf("Failed to get received file: %s", err)
-		query.Answer("获取文件信息失败", &telegram.CallbackOptions{
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
 			Alert:     true,
+			Message:   "无法添加到队列",
 			CacheTime: 5,
 		})
-		return err
+		return dispatcher.EndGroups
 	}
 	queue.AddTask(types.Task{
-		Ctx:            context.TODO(),
+		Ctx:            ctx,
 		Status:         types.Pending,
-		FileName:       receivedFile.FileName,
+		FileName:       record.FileName,
 		Storage:        types.StorageType(args[2]),
-		ChatID:         receivedFile.ChatID,
+		ChatID:         update.EffectiveChat().GetID(),
-		MessageID:      receivedFile.MessageID,
+		ReplyMessageID: record.ReplyMessageID,
-		ReplyMessageID: receivedFile.ReplyMessageID,
+		MessageID:      record.MessageID,
 	})
-	query.Edit(fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", receivedFile.FileName, queue.Len()))
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-	return nil
+		Message: fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", record.FileName, queue.Len()),
+		ID:      record.ReplyMessageID,
+	})
+	return dispatcher.EndGroups
 }

bot/middlewares.go

@@ -0,0 +1,19 @@
+package bot
+
+import (
+	"time"
+
+	"github.com/gotd/contrib/middleware/floodwait"
+	"github.com/gotd/contrib/middleware/ratelimit"
+	"github.com/gotd/td/telegram"
+	"golang.org/x/time/rate"
+)
+
+func FloodWaitMiddleware() []telegram.Middleware {
+	waiter := floodwait.NewSimpleWaiter().WithMaxRetries(5)
+	ratelimiter := ratelimit.New(rate.Every(time.Millisecond*100), 5)
+	return []telegram.Middleware{
+		waiter,
+		ratelimiter,
+	}
+}

bot/utils.go

@@ -2,12 +2,29 @@ package bot
 
 import (
 	"fmt"
-	"regexp"
 
-	"github.com/amarnathcjd/gogram/telegram"
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/types"
+	"github.com/gotd/td/tg"
 	"github.com/krau/SaveAny-Bot/storage"
 )
 
+func supportedMediaFilter(m *types.Message) (bool, error) {
+	if not := m.Media == nil; not {
+		return false, dispatcher.EndGroups
+	}
+	switch m.Media.(type) {
+	case *tg.MessageMediaDocument:
+		return true, nil
+	case *tg.MessageMediaWebPage:
+		return false, dispatcher.EndGroups
+	case tg.MessageMediaClass:
+		return false, dispatcher.EndGroups
+	default:
+		return false, nil
+	}
+}
+
 var StorageDisplayNames = map[string]string{
 	"all":    "全部",
 	"local":  "服务器磁盘",
@@ -15,48 +32,40 @@ var StorageDisplayNames = map[string]string{
 	"webdav": "WebDAV",
 }
 
-func AddTaskReplyMarkup(messageID int32) telegram.ReplyMarkup {
+func getAddTaskMarkup(messageID int) *tg.ReplyInlineMarkup {
-	// TODO: sort storage buttons
+	storageButtons := make([]tg.KeyboardButtonClass, 0)
-	storageButtons := make([]telegram.KeyboardButton, 0)
+	for _, name := range storage.StorageKeys {
-	for name := range storage.Storages {
+		storageButtons = append(storageButtons, &tg.KeyboardButtonCallback{
-		storageButtons = append(storageButtons, &telegram.KeyboardButtonCallback{
 			Text: StorageDisplayNames[string(name)],
 			Data: []byte(fmt.Sprintf("add %d %s", messageID, name)),
 		})
 	}
 
-	if len(storageButtons) > 1 {
+	if len(storageButtons) < 1 {
-		return &telegram.ReplyInlineMarkup{
+		return nil
-			Rows: []*telegram.KeyboardButtonRow{
+	}
+	if len(storageButtons) == 1 {
+		return &tg.ReplyInlineMarkup{
+			Rows: []tg.KeyboardButtonRow{
 				{
 					Buttons: storageButtons,
 				},
-				{
+			},
-					Buttons: []telegram.KeyboardButton{
+		}
-						&telegram.KeyboardButtonCallback{
+	}
-							Text: "全部",
+	return &tg.ReplyInlineMarkup{
-							Data: []byte(fmt.Sprintf("add %d all", messageID)),
+		Rows: []tg.KeyboardButtonRow{
-						},
+			{
+				Buttons: storageButtons,
+			},
+			{
+				Buttons: []tg.KeyboardButtonClass{
+					&tg.KeyboardButtonCallback{
+						Text: "全部",
+						Data: []byte(fmt.Sprintf("add %d all", messageID)),
 					},
 				},
 			},
-		}
+		},
 	}
-
-	if len(storageButtons) == 1 {
-		return &telegram.ReplyInlineMarkup{
-			Rows: []*telegram.KeyboardButtonRow{
-				{
-					Buttons: storageButtons,
-				},
-			},
-		}
-	}
-	return nil
-}
-
-var markdownRe = regexp.MustCompile("([" + regexp.QuoteMeta(`\_*[]()~`+"`"+`>#+-=|{}.!`) + "])")
-
-func EscapeMarkdown(text string) string {
-	return markdownRe.ReplaceAllString(text, "\\$1")
 }

cmd/root.go

@@ -9,9 +9,7 @@ import (
 var rootCmd = &cobra.Command{
 	Use:   "saveany-bot",
 	Short: "saveany-bot",
-	Run: func(cmd *cobra.Command, args []string) {
+	Run:   Run,
-		Run()
-	},
 }
 
 func Execute() {

cmd/run.go

@@ -1,13 +1,22 @@
 package cmd
 
 import (
+	"os"
+	"os/signal"
+	"syscall"
+
 	"github.com/krau/SaveAny-Bot/bootstrap"
-	"github.com/krau/SaveAny-Bot/bot"
 	"github.com/krau/SaveAny-Bot/core"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/spf13/cobra"
 )
 
-func Run() {
+func Run(_ *cobra.Command, _ []string) {
 	bootstrap.InitAll()
-	go core.Run()
+	core.Run()
-	bot.Run()
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	sig := <-quit
+	logger.L.Info(sig, "exit")
 }

common/utils.go

@@ -0,0 +1,35 @@
+package common
+
+import (
+	"fmt"
+
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+func FileFromMedia(media tg.MessageMediaClass) (*types.File, error) {
+	logger.L.Debug("FileFromMedia")
+	switch media := media.(type) {
+	case *tg.MessageMediaDocument:
+		document, ok := media.Document.AsNotEmpty()
+		if !ok {
+			return nil, fmt.Errorf("unexpected type %T", media)
+		}
+		var fileName string
+		for _, attribute := range document.Attributes {
+			if name, ok := attribute.(*tg.DocumentAttributeFilename); ok {
+				fileName = name.FileName
+				break
+			}
+		}
+		return &types.File{
+			Location: document.AsInputDocumentFileLocation(),
+			FileSize: document.Size,
+			FileName: fileName,
+			MimeType: document.MimeType,
+			ID:       document.ID,
+		}, nil
+	}
+	return nil, fmt.Errorf("unexpected type %T", media)
+}

core/core.go

@@ -3,55 +3,56 @@ package core
 import (
 	"context"
 	"errors"
-	"os"
 	"time"
 
-	"github.com/amarnathcjd/gogram/telegram"
+	"github.com/celestix/gotgproto/ext"
-	"github.com/krau/SaveAny-Bot/bot"
+	"github.com/gotd/td/tg"
-	"github.com/krau/SaveAny-Bot/common"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
 	"github.com/krau/SaveAny-Bot/queue"
-	"github.com/krau/SaveAny-Bot/storage"
 	"github.com/krau/SaveAny-Bot/types"
 )
 
 func processPendingTask(task types.Task) error {
-	os.MkdirAll(config.Cfg.Temp.BasePath, os.ModePerm)
+	logger.L.Debugf("Start processing task: %s", task.FileName)
+	time.Sleep(10 * time.Second)
+	logger.L.Debugf("Task done: %s", task.FileName)
 
-	message, err := bot.Client.GetMessageByID(task.ChatID, task.MessageID)
+	// os.MkdirAll(config.Cfg.Temp.BasePath, os.ModePerm)
-	if err != nil {
-		return err
-	}
-	logger.L.Debugf("Start downloading file: %s", task.FileName)
-	bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "正在下载文件...")
-	dest, err := message.Download(&telegram.DownloadOptions{
-		FileName:  common.GetCacheFilePath(task.FileName),
-		Threads:   config.Cfg.Threads,
-		ChunkSize: config.Cfg.ChunkSize,
-		// ProgressCallback: func(totalBytes, downloadedBytes int64) {},
-	})
-	if err != nil {
-		return err
-	}
 
-	defer func() {
+	// message, err := bot.Client.GetMessageByID(task.ChatID, task.MessageID)
-		if config.Cfg.Temp.CacheTTL > 0 {
+	// if err != nil {
-			common.RmFileAfter(dest, time.Duration(config.Cfg.Temp.CacheTTL)*time.Second)
+	// 	return err
-		} else {
+	// }
-			if err := os.Remove(dest); err != nil {
+	// logger.L.Debugf("Start downloading file: %s", task.FileName)
-				logger.L.Errorf("Failed to purge file: %s", err)
+	// bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "正在下载文件...")
-			}
+	// dest, err := message.Download(&telegram.DownloadOptions{
-		}
+	// 	FileName:  common.GetCacheFilePath(task.FileName),
-	}()
+	// 	Threads:   config.Cfg.Threads,
-	if task.StoragePath == "" {
+	// 	ChunkSize: config.Cfg.ChunkSize,
-		task.StoragePath = task.FileName
+	// 	// ProgressCallback: func(totalBytes, downloadedBytes int64) {},
-	}
+	// })
+	// if err != nil {
+	// 	return err
+	// }
 
-	bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "下载完成, 正在转存文件...")
+	// defer func() {
-	if err := storage.Save(task.Storage, task.Ctx, dest, task.StoragePath); err != nil {
+	// 	if config.Cfg.Temp.CacheTTL > 0 {
-		return err
+	// 		common.RmFileAfter(dest, time.Duration(config.Cfg.Temp.CacheTTL)*time.Second)
-	}
+	// 	} else {
+	// 		if err := os.Remove(dest); err != nil {
+	// 			logger.L.Errorf("Failed to purge file: %s", err)
+	// 		}
+	// 	}
+	// }()
+	// if task.StoragePath == "" {
+	// 	task.StoragePath = task.FileName
+	// }
+
+	// bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "下载完成, 正在转存文件...")
+	// if err := storage.Save(task.Storage, task.Ctx, dest, task.StoragePath); err != nil {
+	// 	return err
+	// }
 
 	return nil
 }
@@ -69,6 +70,7 @@ func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
 				logger.L.Errorf("Failed to do task: %s", err)
 				task.Error = err
 				if errors.Is(err, context.Canceled) {
+					logger.L.Debugf("Task canceled: %s", task.String())
 					task.Status = types.Canceled
 				} else {
 					task.Status = types.Failed
@@ -79,10 +81,16 @@ func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
 			queue.AddTask(task)
 		case types.Succeeded:
 			logger.L.Infof("Task succeeded: %s", task.String())
-			bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "文件保存成功")
+			task.Ctx.(*ext.Context).EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
+				Message: "文件保存成功",
+				ID:      task.ReplyMessageID,
+			})
 		case types.Failed:
 			logger.L.Errorf("Task failed: %s", task.String())
-			bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "文件保存失败")
+			task.Ctx.(*ext.Context).EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
+				Message: "文件保存失败",
+				ID:      task.ReplyMessageID,
+			})
 		case types.Canceled:
 			logger.L.Infof("Task canceled: %s", task.String())
 		default:
@@ -99,4 +107,5 @@ func Run() {
 	for i := 0; i < config.Cfg.Workers; i++ {
 		go worker(queue.Queue, semaphore)
 	}
+
 }

dao/file.go

@@ -6,7 +6,7 @@ func AddReceivedFile(receivedFile *model.ReceivedFile) error {
 	return db.Create(receivedFile).Error
 }
 
-func GetReceivedFileByChatAndMessageID(chatID int64, messageID int32) (*model.ReceivedFile, error) {
+func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*model.ReceivedFile, error) {
 	var receivedFile model.ReceivedFile
 	err := db.Where("chat_id = ? AND message_id = ?", chatID, messageID).First(&receivedFile).Error
 	if err != nil {

go.mod

@@ -3,26 +3,35 @@ module github.com/krau/SaveAny-Bot
 go 1.23.2
 
 require (
-	github.com/amarnathcjd/gogram v0.0.0-20241030211922-4e5abb2407e9
 	github.com/blang/semver v3.5.1+incompatible
+	github.com/celestix/gotgproto v1.0.0-beta18
 	github.com/gookit/slog v0.5.7
+	github.com/gotd/contrib v0.20.0
+	github.com/gotd/td v0.111.2
 	github.com/imroc/req/v3 v3.48.0
-	github.com/mymmrac/telego v0.31.4
 	github.com/rhysd/go-github-selfupdate v1.2.3
 	github.com/spf13/cobra v1.8.1
 	github.com/spf13/viper v1.19.0
 	github.com/studio-b12/gowebdav v0.9.0
+	golang.org/x/time v0.5.0
 )
 
 require (
+	github.com/AnimeKaizoku/cacher v1.0.2 // indirect
+	github.com/cenkalti/backoff/v4 v4.3.0 // indirect
 	github.com/cloudflare/circl v1.5.0 // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/glebarez/go-sqlite v1.22.0 // indirect
+	github.com/go-faster/errors v0.7.1 // indirect
+	github.com/go-faster/jx v1.1.0 // indirect
+	github.com/go-faster/xor v1.0.0 // indirect
 	github.com/go-task/slim-sprig/v3 v3.0.0 // indirect
 	github.com/google/go-github/v30 v30.1.0 // indirect
 	github.com/google/go-querystring v1.1.0 // indirect
 	github.com/google/pprof v0.0.0-20241101162523-b92577c0c142 // indirect
 	github.com/google/uuid v1.6.0 // indirect
+	github.com/gotd/ige v0.2.2 // indirect
+	github.com/gotd/neo v0.1.5 // indirect
 	github.com/hashicorp/errwrap v1.1.0 // indirect
 	github.com/hashicorp/go-multierror v1.1.1 // indirect
 	github.com/inconshreveable/go-update v0.0.0-20160112193335-8152e7eb6ccf // indirect
@@ -36,10 +45,15 @@ require (
 	github.com/quic-go/quic-go v0.48.1 // indirect
 	github.com/refraction-networking/utls v1.6.7 // indirect
 	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+	github.com/segmentio/asm v1.2.0 // indirect
 	github.com/tcnksm/go-gitconfig v0.1.2 // indirect
 	github.com/ulikunitz/xz v0.5.12 // indirect
+	go.opentelemetry.io/otel v1.31.0 // indirect
+	go.opentelemetry.io/otel/trace v1.31.0 // indirect
+	go.uber.org/atomic v1.11.0 // indirect
 	go.uber.org/mock v0.5.0 // indirect
-	golang.org/x/crypto v0.28.0 // indirect
+	go.uber.org/zap v1.27.0 // indirect
+	golang.org/x/crypto v0.29.0 // indirect
 	golang.org/x/mod v0.21.0 // indirect
 	golang.org/x/net v0.30.0 // indirect
 	golang.org/x/oauth2 v0.23.0 // indirect
@@ -49,48 +63,38 @@ require (
 	modernc.org/mathutil v1.6.0 // indirect
 	modernc.org/memory v1.8.0 // indirect
 	modernc.org/sqlite v1.33.1 // indirect
+	nhooyr.io/websocket v1.8.17 // indirect
+	rsc.io/qr v0.2.0 // indirect
 )
 
 require (
 	github.com/andybalholm/brotli v1.1.1 // indirect
-	github.com/bytedance/sonic v1.12.3 // indirect
-	github.com/bytedance/sonic/loader v0.2.1 // indirect
-	github.com/cloudwego/base64x v0.1.4 // indirect
-	github.com/cloudwego/iasm v0.2.0 // indirect
 	github.com/duke-git/lancet/v2 v2.3.3
-	github.com/fasthttp/router v1.5.2 // indirect
 	github.com/fsnotify/fsnotify v1.8.0 // indirect
 	github.com/glebarez/sqlite v1.11.0
 	github.com/gookit/color v1.5.4 // indirect
 	github.com/gookit/goutil v0.6.17
 	github.com/gookit/gsr v0.1.0 // indirect
-	github.com/grbit/go-json v0.11.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/klauspost/compress v1.17.11 // indirect
-	github.com/klauspost/cpuid/v2 v2.2.8 // indirect
 	github.com/magiconair/properties v1.8.7 // indirect
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
 	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
 	github.com/sagikazarmark/locafero v0.6.0 // indirect
 	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
-	github.com/savsgio/gotils v0.0.0-20240704082632-aef3928b8a38 // indirect
 	github.com/sourcegraph/conc v0.3.0 // indirect
 	github.com/spf13/afero v1.11.0 // indirect
 	github.com/spf13/cast v1.7.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
-	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/valyala/bytebufferpool v1.0.0 // indirect
-	github.com/valyala/fasthttp v1.57.0 // indirect
-	github.com/valyala/fastjson v1.6.4 // indirect
 	github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
 	go.uber.org/multierr v1.11.0 // indirect
-	golang.org/x/arch v0.11.0 // indirect
 	golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c // indirect
-	golang.org/x/sync v0.8.0 // indirect
+	golang.org/x/sync v0.9.0 // indirect
-	golang.org/x/sys v0.26.0 // indirect
+	golang.org/x/sys v0.27.0 // indirect
-	golang.org/x/text v0.19.0 // indirect
+	golang.org/x/text v0.20.0 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 	gorm.io/gorm v1.25.12

go.sum

@@ -1,22 +1,16 @@
-github.com/amarnathcjd/gogram v0.0.0-20241030211922-4e5abb2407e9 h1:ahjtb5R//A+BA88qQ/847Qzq4hl1ajgCANXV4fjixRA=
+github.com/AnimeKaizoku/cacher v1.0.2 h1:7Bf5qRylWb7q2Evib0OXlhG37/t7BP2HK/7IyPvSmGQ=
-github.com/amarnathcjd/gogram v0.0.0-20241030211922-4e5abb2407e9/go.mod h1:MPzWyqnIwVK/tji8O5pmumdoPoGHqemsIJyr7xedoE8=
+github.com/AnimeKaizoku/cacher v1.0.2/go.mod h1:jw0de/b0K6W7Y3T9rHCMGVKUf6oG7hENNcssxYcZTCc=
 github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
 github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
 github.com/blang/semver v3.5.1+incompatible h1:cQNTCjp13qL8KC3Nbxr/y2Bqb63oX6wdnnjpJbkM4JQ=
 github.com/blang/semver v3.5.1+incompatible/go.mod h1:kRBLl5iJ+tD4TcOOxsy/0fnwebNt5EWlYSAyrTnjyyk=
-github.com/bytedance/sonic v1.12.3 h1:W2MGa7RCU1QTeYRTPE3+88mVC0yXmsRQRChiyVocVjU=
+github.com/celestix/gotgproto v1.0.0-beta18 h1:7884H/il+mzNreOQ4SqoMa4S5njt3UmGPKZTxPu38fU=
-github.com/bytedance/sonic v1.12.3/go.mod h1:B8Gt/XvtZ3Fqj+iSKMypzymZxw/FVwgIGKzMzT9r/rk=
+github.com/celestix/gotgproto v1.0.0-beta18/go.mod h1:osZOlN5irPByA0+3IPsZOH+Ibs0tOMSKmIdgGYEBRgE=
-github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
+github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
-github.com/bytedance/sonic/loader v0.2.1 h1:1GgorWTqf12TA8mma4DDSbaQigE2wOgQo7iCjjJv3+E=
+github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
-github.com/bytedance/sonic/loader v0.2.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
 github.com/cloudflare/circl v1.5.0 h1:hxIWksrX6XN5a1L2TI/h53AGPhNHoUBo+TD1ms9+pys=
 github.com/cloudflare/circl v1.5.0/go.mod h1:uddAzsPgqdMAYatqJ0lsjX1oECcQLIlRpzZh3pJrofs=
-github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
-github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
-github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
-github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
 github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
-github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
@@ -24,8 +18,6 @@ github.com/duke-git/lancet/v2 v2.3.3 h1:OhqzNzkbJBS9ZlWLo/C7g+WSAOAAyNj7p9CAiEHu
 github.com/duke-git/lancet/v2 v2.3.3/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
 github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
 github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
-github.com/fasthttp/router v1.5.2 h1:ckJCCdV7hWkkrMeId3WfEhz+4Gyyf6QPwxi/RHIMZ6I=
-github.com/fasthttp/router v1.5.2/go.mod h1:C8EY53ozOwpONyevc/V7Gr8pqnEjwnkFFqPo1alAGs0=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
 github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
@@ -35,6 +27,13 @@ github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec
 github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc=
 github.com/glebarez/sqlite v1.11.0 h1:wSG0irqzP6VurnMEpFGer5Li19RpIRi2qvQz++w0GMw=
 github.com/glebarez/sqlite v1.11.0/go.mod h1:h8/o8j5wiAsqSPoWELDUdJXhjAhsVliSn7bWZjOhrgQ=
+github.com/go-faster/errors v0.7.1 h1:MkJTnDoEdi9pDabt1dpWf7AA8/BaSYZqibYyhZ20AYg=
+github.com/go-faster/errors v0.7.1/go.mod h1:5ySTjWFiphBs07IKuiL69nxdfd5+fzh1u7FPGZP2quo=
+github.com/go-faster/jx v1.1.0 h1:ZsW3wD+snOdmTDy9eIVgQdjUpXRRV4rqW8NS3t+20bg=
+github.com/go-faster/jx v1.1.0/go.mod h1:vKDNikrKoyUmpzaJ0OkIkRQClNHFX/nF3dnTJZb3skg=
+github.com/go-faster/xor v0.3.0/go.mod h1:x5CaDY9UKErKzqfRfFZdfu+OSTfoZny3w5Ak7UxcipQ=
+github.com/go-faster/xor v1.0.0 h1:2o8vTOgErSGHP3/7XwA5ib1FTtUsNtwCoLLBjl31X38=
+github.com/go-faster/xor v1.0.0/go.mod h1:x5CaDY9UKErKzqfRfFZdfu+OSTfoZny3w5Ak7UxcipQ=
 github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=
 github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
 github.com/go-task/slim-sprig/v3 v3.0.0 h1:sUs3vkvUymDpBKi3qH1YSqBQk9+9D/8M2mN1vB6EwHI=
@@ -61,8 +60,14 @@ github.com/gookit/gsr v0.1.0 h1:0gadWaYGU4phMs0bma38t+Do5OZowRMEVlHv31p0Zig=
 github.com/gookit/gsr v0.1.0/go.mod h1:7wv4Y4WCnil8+DlDYHBjidzrEzfHhXEoFjEA0pPPWpI=
 github.com/gookit/slog v0.5.7 h1:n3Dhgmr3NP+KppkNg95+vpFcI4YD1csu9VTQwgcEYTs=
 github.com/gookit/slog v0.5.7/go.mod h1:uWCRB4YO+FmgwXEq3s8U7ob1wWP7RStOuY/2a4yC/8o=
-github.com/grbit/go-json v0.11.0 h1:bAbyMdYrYl/OjYsSqLH99N2DyQ291mHy726Mx+sYrnc=
+github.com/gotd/contrib v0.20.0 h1:1Wc4+HMQiIKYQuGHVwVksIx152HFTP6B5n88dDe0ZYw=
-github.com/grbit/go-json v0.11.0/go.mod h1:IYpHsdybQ386+6g3VE6AXQ3uTGa5mquBme5/ZWmtzek=
+github.com/gotd/contrib v0.20.0/go.mod h1:P6o8W4niqhDPHLA0U+SA/L7l3BQHYLULpeHfRSePn9o=
+github.com/gotd/ige v0.2.2 h1:XQ9dJZwBfDnOGSTxKXBGP4gMud3Qku2ekScRjDWWfEk=
+github.com/gotd/ige v0.2.2/go.mod h1:tuCRb+Y5Y3eNTo3ypIfNpQ4MFjrnONiL2jN2AKZXmb0=
+github.com/gotd/neo v0.1.5 h1:oj0iQfMbGClP8xI59x7fE/uHoTJD7NZH9oV1WNuPukQ=
+github.com/gotd/neo v0.1.5/go.mod h1:9A2a4bn9zL6FADufBdt7tZt+WMhvZoc5gWXihOPoiBQ=
+github.com/gotd/td v0.111.2 h1:f1u3FueE1QXr6n0WzE5k4tOJOjn5oFaSeF71ai9OE/8=
+github.com/gotd/td v0.111.2/go.mod h1:zzgUtTDJD4TVaCpKfCD0rxazQxPhSlPzx/CVBpqsx1g=
 github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
 github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I=
 github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
@@ -83,10 +88,6 @@ github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
 github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
 github.com/klauspost/compress v1.17.11 h1:In6xLpyWOi1+C7tXUUWv2ot1QvBjxevKAaI6IXrJmUc=
 github.com/klauspost/compress v1.17.11/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0=
-github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
-github.com/klauspost/cpuid/v2 v2.2.8 h1:+StwCXwm9PdpiEkPyzBXIy+M9KUb4ODm0Zarf1kS5BM=
-github.com/klauspost/cpuid/v2 v2.2.8/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
-github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
 github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
 github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
@@ -100,8 +101,6 @@ github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWE
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/mitchellh/mapstructure v1.5.0 h1:jeMsZIYE/09sWLaz43PL7Gy6RuMjD2eJVyuac5Z2hdY=
 github.com/mitchellh/mapstructure v1.5.0/go.mod h1:bFUtVrKA4DC2yAKiSyO/QUcy7e+RRV2QTWOzhPopBRo=
-github.com/mymmrac/telego v0.31.4 h1:NpiNl0P/8eydknka/k6XaaaWVj5BKMlM3Ibba63QTBU=
-github.com/mymmrac/telego v0.31.4/go.mod h1:T12js1PgbYDYznvoN05MSMuPMfWTYo7D9LKl5cPFWiI=
 github.com/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
 github.com/ncruces/go-strftime v0.1.9/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
 github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
@@ -114,7 +113,6 @@ github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNH
 github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
-github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/quic-go/qpack v0.5.1 h1:giqksBPnT/HDtZ6VhtFKgoLOWmlyo9Ei6u9PqzIMbhI=
@@ -127,15 +125,15 @@ github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
 github.com/rhysd/go-github-selfupdate v1.2.3 h1:iaa+J202f+Nc+A8zi75uccC8Wg3omaM7HDeimXA22Ag=
 github.com/rhysd/go-github-selfupdate v1.2.3/go.mod h1:mp/N8zj6jFfBQy/XMYoWsmfzxazpPAODuqarmPDe2Rg=
-github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
+github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8=
-github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 github.com/sagikazarmark/locafero v0.6.0 h1:ON7AQg37yzcRPU69mt7gwhFEBwxI6P9T4Qu3N51bwOk=
 github.com/sagikazarmark/locafero v0.6.0/go.mod h1:77OmuIc6VTraTXKXIs/uvUxKGUXjE1GbemJYHqdNjX0=
 github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
 github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
-github.com/savsgio/gotils v0.0.0-20240704082632-aef3928b8a38 h1:D0vL7YNisV2yqE55+q0lFuGse6U8lxlg7fYTctlT5Gc=
+github.com/segmentio/asm v1.2.0 h1:9BQrFxC+YOHJlTlHGkTrFWf59nbL3XnCoFLTwDCI7ys=
-github.com/savsgio/gotils v0.0.0-20240704082632-aef3928b8a38/go.mod h1:sM7Mt7uEoCeFSCBM+qBrqvEo+/9vdmj19wzp3yzUhmg=
+github.com/segmentio/asm v1.2.0/go.mod h1:BqMnlJP91P8d+4ibuonYZw9mfnzI9HfxselHZr5aAcs=
 github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
 github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
 github.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8=
@@ -148,13 +146,6 @@ github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
 github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 github.com/spf13/viper v1.19.0 h1:RWq5SEjt8o25SROyN3z2OrDB9l7RPd3lwTWU8EcEdcI=
 github.com/spf13/viper v1.19.0/go.mod h1:GQUN9bilAbhU/jgc1bKs99f/suXKeUMct8Adx5+Ntkg=
-github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
-github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
-github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
-github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
-github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
 github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/studio-b12/gowebdav v0.9.0 h1:1j1sc9gQnNxbXXM4M/CebPOX4aXYtr7MojAVcN4dHjU=
@@ -163,31 +154,33 @@ github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8
 github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
 github.com/tcnksm/go-gitconfig v0.1.2 h1:iiDhRitByXAEyjgBqsKi9QU4o2TNtv9kPP3RgPgXBPw=
 github.com/tcnksm/go-gitconfig v0.1.2/go.mod h1:/8EhP4H7oJZdIPyT+/UIsG87kTzrzM4UsLGSItWYCpE=
-github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
-github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
 github.com/ulikunitz/xz v0.5.9/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=
 github.com/ulikunitz/xz v0.5.12 h1:37Nm15o69RwBkXM0J6A5OlE67RZTfzUxTj8fB3dfcsc=
 github.com/ulikunitz/xz v0.5.12/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=
 github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
 github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
-github.com/valyala/fasthttp v1.57.0 h1:Xw8SjWGEP/+wAAgyy5XTvgrWlOD1+TxbbvNADYCm1Tg=
-github.com/valyala/fasthttp v1.57.0/go.mod h1:h6ZBaPRlzpZ6O3H5t2gEk1Qi33+TmLvfwgLLp0t9CpE=
-github.com/valyala/fastjson v1.6.4 h1:uAUNq9Z6ymTgGhcm0UynUAB6tlbakBrz6CQFax3BXVQ=
-github.com/valyala/fastjson v1.6.4/go.mod h1:CLCAqky6SMuOcxStkYQvblddUtoRxhYMGLrsQns1aXY=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
 github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
 github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
+go.opentelemetry.io/otel v1.31.0 h1:NsJcKPIW0D0H3NgzPDHmo0WW6SptzPdqg/L1zsIm2hY=
+go.opentelemetry.io/otel v1.31.0/go.mod h1:O0C14Yl9FgkjqcCZAsE053C13OaddMYr/hz6clDkEJE=
+go.opentelemetry.io/otel/trace v1.31.0 h1:ffjsj1aRouKewfr85U2aGagJ46+MvodynlQ1HYdmJys=
+go.opentelemetry.io/otel/trace v1.31.0/go.mod h1:TXZkRk7SM2ZQLtR6eoAWQFIHPvzQ06FJAsO1tJg480A=
+go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE=
+go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=
+go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
+go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
 go.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=
 go.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=
 go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
 go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
-golang.org/x/arch v0.11.0 h1:KXV8WWKCXm6tRpLirl2szsO5j/oOODwZf4hATmGVNs4=
+go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8=
-golang.org/x/arch v0.11.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
+go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20201221181555-eec23a3978ad/go.mod h1:jdWPYTVW3xRLrWPugEBEK3UY2ZEsg3UU495nc5E+M+I=
-golang.org/x/crypto v0.28.0 h1:GBDwsMXVQi34v5CCYUm2jkJvu4cbtru2U4TN2PSyQnw=
+golang.org/x/crypto v0.29.0 h1:L5SG1JTTXupVV3n6sUqMTeWbjAyfPwoda2DLX8J8FrQ=
-golang.org/x/crypto v0.28.0/go.mod h1:rmgy+3RHxRZMyY0jjAJShp2zgEdOqj2AO7U0pYmeQ7U=
+golang.org/x/crypto v0.29.0/go.mod h1:+F4F4N5hv6v38hfeYwTdx20oUvLLc+QfrE9Ax9HtgRg=
 golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c h1:7dEasQXItcW1xKJ2+gg5VOiBnqWrJc+rq0DPKyvvdbY=
 golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c/go.mod h1:NQtJDoLvd6faHhE7m4T/1IY708gDefGGjR/iUW8yQQ8=
 golang.org/x/mod v0.21.0 h1:vvrHzRwRfVKSiLrG+d4FMl/Qi4ukBCE6kZlTUkDYRT0=
@@ -203,22 +196,22 @@ golang.org/x/oauth2 v0.0.0-20181106182150-f42d05182288/go.mod h1:N/0e6XlmueqKjAG
 golang.org/x/oauth2 v0.23.0 h1:PbgcYx2W7i4LvjJWEbf0ngHV6qJYr86PkAV3bXdLEbs=
 golang.org/x/oauth2 v0.23.0/go.mod h1:XYTD2NtWslqkgxebSiOHnXEap4TF09sJSc7H1sXbhtI=
 golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
+golang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.9.0 h1:fEo0HyrW1GIgZdpbhCRO0PkJajUS5H9IFUztCgEo2jQ=
+golang.org/x/sync v0.9.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
-golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.26.0 h1:KHjCJyddX0LoSTb3J+vWpupP9p0oznkqVk/IfjymZbo=
+golang.org/x/sys v0.27.0 h1:wBqf8DvsY9Y/2P8gAfPDEYNuS30J4lPHJxXSb/nJZ+s=
-golang.org/x/sys v0.26.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.27.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXRh80K7BxOlk5/8JxuGnuhpl+muw=
-golang.org/x/term v0.25.0 h1:WtHI/ltw4NvSUig5KARz9h521QvRC8RmF/cuYqifU24=
+golang.org/x/term v0.26.0 h1:WEQa6V3Gja/BhNxg540hBip/kkaYtRg3cxg4oXSw4AU=
-golang.org/x/term v0.25.0/go.mod h1:RPyXicDX+6vLxogjjRxjgD2TKtmAO6NZBsBRfrOLu7M=
+golang.org/x/term v0.26.0/go.mod h1:Si5m1o57C5nBNQo5z1iq+XDijt21BDBDp2bK0QI8e3E=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
-golang.org/x/text v0.19.0 h1:kTxAhCbGbxhK0IwgSKiMO5awPoDQ0RpfiVYBfK860YM=
+golang.org/x/text v0.20.0 h1:gK/Kv2otX8gz+wn7Rmb3vT96ZwuoxnQlY+HlJVj7Qug=
-golang.org/x/text v0.19.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
+golang.org/x/text v0.20.0/go.mod h1:D4IsuqiFMhST5bX19pQ9ikHC2GsaKyk/oF+pn3ducp4=
 golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
 golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
@@ -238,7 +231,6 @@ gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=
 gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
 gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
-gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gorm.io/gorm v1.25.12 h1:I0u8i2hWQItBq1WfE0o2+WuL9+8L21K9e2HHSTE/0f8=
@@ -267,4 +259,7 @@ modernc.org/strutil v1.2.0 h1:agBi9dp1I+eOnxXeiZawM8F4LawKv4NzGWSaLfyeNZA=
 modernc.org/strutil v1.2.0/go.mod h1:/mdcBmfOibveCTBxUl5B5l6W+TTH1FXPLHZE6bTosX0=
 modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y=
 modernc.org/token v1.1.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM=
-nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
+nhooyr.io/websocket v1.8.17 h1:KEVeLJkUywCKVsnLIDlD/5gtayKp8VoCkksHCGGfT9Y=
+nhooyr.io/websocket v1.8.17/go.mod h1:rN9OFWIUwuxg4fR5tELlYC04bXYowCP9GX47ivo2l+c=
+rsc.io/qr v0.2.0 h1:6vBLea5/NRMVTz8V66gipeLycZMl/+UlFmk8DvqQ6WY=
+rsc.io/qr v0.2.0/go.mod h1:IF+uZjkb9fqyeF/4tlBoynqmQxUoPfWEKh921coOuXs=

model/model.go

@@ -9,8 +9,8 @@ type ReceivedFile struct {
 	Processing     bool
 	FileName       string
 	ChatID         int64
-	MessageID      int32
+	MessageID      int
-	ReplyMessageID int32
+	ReplyMessageID int
 }
 
 type User struct {

storage/storage.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"sync"
 
+	"github.com/duke-git/lancet/v2/slice"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
 	"github.com/krau/SaveAny-Bot/storage/alist"
@@ -19,6 +20,7 @@ type Storage interface {
 }
 
 var Storages = make(map[types.StorageType]Storage)
+var StorageKeys = make([]types.StorageType, 0)
 
 func Init() {
 	logger.L.Debug("Initializing storage...")
@@ -35,6 +37,12 @@ func Init() {
 		Storages[types.Webdav].Init()
 	}
 
+	for k := range Storages {
+		StorageKeys = append(StorageKeys, k)
+	}
+
+	slice.Sort(StorageKeys)
+
 	logger.L.Debug("Storage initialized")
 }
 

types/types.go

@@ -1,6 +1,10 @@
 package types
 
-import "context"
+import (
+	"context"
+
+	"github.com/gotd/td/tg"
+)
 
 type TaskStatus string
 
@@ -30,11 +34,19 @@ type Task struct {
 	Storage     StorageType
 	StoragePath string
 
-	MessageID      int32
+	MessageID      int
 	ChatID         int64
-	ReplyMessageID int32
+	ReplyMessageID int
 }
 
-func (t *Task) String() string {
+func (t Task) String() string {
 	return t.FileName
 }
+
+type File struct {
+	Location *tg.InputDocumentFileLocation
+	FileSize int64
+	FileName string
+	MimeType string
+	ID       int64
+}