fix: add UserID to link message and enforce default storage setting in silent handler

Co-authored-by: AHCorn <42889600+AHCorn@users.noreply.github.com>

README.md

@@ -2,6 +2,8 @@
 
 # <img src="docs/logo.jpg" width="45" align="center"> Save Any Bot
 
+**简体中文** | [English](README_EN.md) 
+
 把 Telegram 的文件保存到各类存储端.
 
 > _就像 PikPak Bot 一样_
@@ -60,9 +62,9 @@ systemctl enable --now saveany-bot
 
 #### Docker Compose
 
-下载 [docker-compose.yml](https://github.com/krau/SaveAny-Bot/blob/main/docker-compose.yml) 文件, 并修改其中的配置.
+下载 [docker-compose.yml](https://github.com/krau/SaveAny-Bot/blob/main/docker-compose.yml) 文件, 在同目录下新建 `config.toml` 文件, 参考 [config.toml.example](https://github.com/krau/SaveAny-Bot/blob/main/config.example.toml) 编辑配置文件.
 
-运行:
+启动:
 
 ```bash
 docker compose up -d
@@ -94,7 +96,7 @@ docker restart saveany-bot
 
 ## 使用
 
-向 Bot 发送(转发)文件, 按照提示操作.
+向 Bot 发送(转发)文件, 或发送公开频道的消息链接, 按照提示操作.
 
 ---
 

README_EN.md

@@ -0,0 +1,108 @@
+<div align="center">
+
+# <img src="docs/logo.jpg" width="45" align="center"> Save Any Bot
+
+[简体中文](README.md) | **English**
+
+Save Telegram files to various storage endpoints.
+
+> _Just like PikPak Bot_
+
+</div>
+
+Demo Video:
+
+<div align="center">
+
+[SaveAny-Bot Demo Video.webm](https://github.com/user-attachments/assets/a0de2453-a4d1-4a12-81fb-9d84856dce09)
+
+</div>
+
+## Deployment
+
+### Deploy from Binary
+
+Download the binary file for your platform from the [Release](https://github.com/krau/SaveAny-Bot/releases) page.
+
+Create a `config.toml` file in the extracted directory, refer to [config.toml.example](https://github.com/krau/SaveAny-Bot/blob/main/config.example.toml) for configuration.
+
+Run:
+
+```bash
+chmod +x saveany-bot
+./saveany-bot
+```
+
+#### Add as systemd Service
+
+Create file `/etc/systemd/system/saveany-bot.service` and write the following content:
+
+```
+[Unit]
+Description=SaveAnyBot
+After=systemd-user-sessions.service
+
+[Service]
+Type=simple
+WorkingDirectory=/yourpath/
+ExecStart=/yourpath/saveany-bot
+Restart=on-failure
+
+[Install]
+WantedBy=multi-user.target
+```
+
+Enable auto-start and start the service:
+
+```bash
+systemctl enable --now saveany-bot
+```
+
+### Deploy with Docker
+
+#### Docker Compose
+
+Download [docker-compose.yml](https://github.com/krau/SaveAny-Bot/blob/main/docker-compose.yml) file and create a `config.toml` file in the same directory, refer to [config.toml.example](https://github.com/krau/SaveAny-Bot/blob/main/config.example.toml) for configuration.
+
+Run:
+
+```bash
+docker compose up -d
+```
+
+#### Docker
+
+```shell
+docker run -d --name saveany-bot \
+    -v /path/to/config.toml:/app/config.toml \
+    -v /path/to/downloads:/app/downloads \
+    ghcr.io/krau/saveany-bot:latest
+```
+
+## Update
+
+Use `upgrade` or `up` command to upgrade to the latest version:
+
+```bash
+./saveany-bot upgrade
+```
+
+If deployed with Docker, use the following commands to update:
+
+```bash
+docker pull ghcr.io/krau/saveany-bot:latest
+docker restart saveany-bot
+```
+
+## Usage
+
+Send (forward) files to the Bot and follow the prompts.
+
+---
+
+## Thanks
+
+- [gotd](https://github.com/gotd/td)
+- [TG-FileStreamBot](https://github.com/EverythingSuckz/TG-FileStreamBot)
+- [gotgproto](https://github.com/celestix/gotgproto)
+- All the dependencies

bootstrap/init.go

@@ -1,6 +1,9 @@
 package bootstrap
 
 import (
+	"fmt"
+	"os"
+
 	"github.com/krau/SaveAny-Bot/bot"
 	"github.com/krau/SaveAny-Bot/common"
 	"github.com/krau/SaveAny-Bot/config"
@@ -10,12 +13,14 @@ import (
 )
 
 func InitAll() {
-	config.Init()
+	if err := config.Init(); err != nil {
+		fmt.Println("加载配置文件失败: ", err)
+		os.Exit(1)
+	}
 	logger.InitLogger()
-	logger.L.Info("Running...")
-
+	logger.L.Info("正在启动 SaveAny-Bot...")
+	storage.LoadStorages()
 	common.Init()
-	storage.Init()
 	dao.Init()
 	bot.Init()
 }

bot/bot.go

@@ -27,7 +27,7 @@ func newProxyDialer(proxyUrl string) (proxy.Dialer, error) {
 }
 
 func Init() {
-	logger.L.Info("Initializing client...")
+	logger.L.Info("初始化 Telegram 客户端...")
 	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
 	defer cancel()
 	resultChan := make(chan struct {
@@ -76,7 +76,6 @@ func Init() {
 				{Command: "silent", Description: "开启/关闭静默模式"},
 				{Command: "storage", Description: "设置默认存储端"},
 				{Command: "save", Description: "保存所回复的文件"},
-				{Command: "path", Description: "更改保存路径配置"},
 			},
 		})
 		resultChan <- struct {
@@ -87,15 +86,15 @@ func Init() {
 
 	select {
 	case <-ctx.Done():
-		logger.L.Fatal("Failed to initialize client: timeout")
+		logger.L.Fatal("初始化客户端失败: 超时")
 		os.Exit(1)
 	case result := <-resultChan:
 		if result.err != nil {
-			logger.L.Fatalf("Failed to initialize client: %s", result.err)
+			logger.L.Fatalf("初始化客户端失败: %s", result.err)
 			os.Exit(1)
 		}
 		Client = result.client
 		RegisterHandlers(Client.Dispatcher)
-		logger.L.Info("Client initialized")
+		logger.L.Info("客户端初始化完成")
 	}
 }

bot/handle_link.go

@@ -0,0 +1,111 @@
+package bot
+
+import (
+	"fmt"
+	"regexp"
+	"strconv"
+	"strings"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/storage"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+var (
+	linkRegexString = `t.me/.*/\d+`
+	linkRegex       = regexp.MustCompile(linkRegexString)
+)
+
+func handleLinkMessage(ctx *ext.Context, update *ext.Update) error {
+	logger.L.Trace("Got link message")
+	link := linkRegex.FindString(update.EffectiveMessage.Text)
+	if link == "" {
+		return dispatcher.ContinueGroups
+	}
+	strSlice := strings.Split(link, "/")
+	if len(strSlice) < 3 {
+		return dispatcher.ContinueGroups
+	}
+	messageID, err := strconv.Atoi(strSlice[2])
+	if err != nil {
+		logger.L.Errorf("解析消息 ID 失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("无法解析消息 ID"), nil)
+		return dispatcher.EndGroups
+	}
+	chatUsername := strSlice[1]
+	linkChat, err := ctx.ResolveUsername(chatUsername)
+	if err != nil {
+		logger.L.Errorf("解析 Chat ID 失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("无法解析 Chat ID"), nil)
+		return dispatcher.EndGroups
+	}
+	if linkChat == nil {
+		logger.L.Errorf("无法找到聊天: %s", chatUsername)
+		ctx.Reply(update, ext.ReplyTextString("无法找到聊天"), nil)
+		return dispatcher.EndGroups
+	}
+	user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
+	if err != nil {
+		logger.L.Errorf("获取用户失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
+		return dispatcher.EndGroups
+	}
+	storages := storage.GetUserStorages(user.ChatID)
+
+	if len(storages) == 0 {
+		ctx.Reply(update, ext.ReplyTextString("无可用的存储"), nil)
+		return dispatcher.EndGroups
+	}
+	replied, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件..."), nil)
+	if err != nil {
+		logger.L.Errorf("回复失败: %s", err)
+		return dispatcher.EndGroups
+	}
+
+	file, err := FileFromMessage(ctx, linkChat.GetID(), messageID, "")
+	if err != nil {
+		logger.L.Errorf("获取文件失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取文件失败: "+err.Error()), nil)
+		return dispatcher.EndGroups
+	}
+	// TODO: Better file name
+	if file.FileName == "" {
+		logger.L.Warnf("文件名为空，使用生成的名称")
+		file.FileName = fmt.Sprintf("%d_%d_%s", linkChat.GetID(), messageID, file.Hash())
+	}
+
+	receivedFile := &types.ReceivedFile{
+		Processing:     false,
+		FileName:       file.FileName,
+		ChatID:         linkChat.GetID(),
+		MessageID:      messageID,
+		ReplyMessageID: replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+	}
+	if err := dao.SaveReceivedFile(receivedFile); err != nil {
+		logger.L.Errorf("保存接收的文件失败: %s", err)
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "无法保存文件: " + err.Error(),
+			ID:      replied.ID,
+		})
+		return dispatcher.EndGroups
+	}
+	if !user.Silent || user.DefaultStorage == "" {
+		return ProvideSelectMessage(ctx, update, file, linkChat.GetID(), messageID, replied.ID)
+	}
+	return HandleSilentAddTask(ctx, update, user, &types.Task{
+		Ctx:            ctx,
+		Status:         types.Pending,
+		File:           file,
+		StorageName:    user.DefaultStorage,
+		UserID:         user.ChatID,
+		FileChatID:     linkChat.GetID(),
+		FileMessageID:  messageID,
+		ReplyMessageID: replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+	})
+}

bot/handler_conversation.go

@@ -0,0 +1,75 @@
+package bot
+
+import (
+	"sync"
+)
+
+type ConversationType string
+
+type ConversationState struct {
+	sync.Mutex
+	conversationType ConversationType
+	InConversation   bool
+	data             map[ConversationType]map[string]interface{}
+}
+
+func (c *ConversationState) Reset() {
+	c.Lock()
+	defer c.Unlock()
+	c.InConversation = false
+	c.conversationType = ""
+	c.data = make(map[ConversationType]map[string]interface{})
+}
+
+func (c *ConversationState) SetConversationType(t ConversationType) {
+	c.Lock()
+	defer c.Unlock()
+	c.conversationType = t
+}
+
+func (c *ConversationState) GetData(key string) interface{} {
+	if c.data == nil || c.data[c.conversationType] == nil {
+		return nil
+	}
+	return c.data[c.conversationType][key]
+}
+
+func (c *ConversationState) SetData(key string, value interface{}) {
+	c.Lock()
+	defer c.Unlock()
+	if c.data == nil {
+		c.data = make(map[ConversationType]map[string]interface{})
+	}
+	if c.data[c.conversationType] == nil {
+		c.data[c.conversationType] = make(map[string]interface{})
+	}
+	c.data[c.conversationType][key] = value
+}
+
+// TODO: Implement conversation handling
+// var userConversationState = make(map[int64]*ConversationState)
+
+// func handleConversation(ctx *ext.Context, update *ext.Update) error {
+// 	userID := update.EffectiveUser().GetID()
+// 	state, ok := userConversationState[userID]
+// 	if !ok {
+// 		return dispatcher.ContinueGroups
+// 	}
+// 	if update.EffectiveMessage.Text == "/cancel" {
+// 		state.Reset()
+// 		ctx.Reply(update, ext.ReplyTextString("已取消"), nil)
+// 		return dispatcher.EndGroups
+// 	}
+// 	if !state.InConversation {
+// 		return dispatcher.ContinueGroups
+// 	}
+// 	return handleConversationState(ctx, update, state)
+// }
+
+// func handleConversationState(ctx *ext.Context, update *ext.Update, state *ConversationState) error {
+// 	switch state.conversationType {
+// 	default:
+// 		logger.L.Errorf("Unknown conversation type: %s", state.conversationType)
+// 	}
+// 	return dispatcher.EndGroups
+// }

bot/handlers.go

@@ -1,13 +1,11 @@
 package bot
 
 import (
-	"errors"
 	"fmt"
 	"strconv"
 	"strings"
 
 	"github.com/duke-git/lancet/v2/slice"
-	"github.com/gookit/goutil/maputil"
 	"github.com/gotd/td/telegram/message/entity"
 	"github.com/gotd/td/telegram/message/styling"
 	"github.com/gotd/td/tg"
@@ -29,21 +27,26 @@ func RegisterHandlers(dispatcher dispatcher.Dispatcher) {
 	dispatcher.AddHandler(handlers.NewCommand("start", start))
 	dispatcher.AddHandler(handlers.NewCommand("help", help))
 	dispatcher.AddHandler(handlers.NewCommand("silent", silent))
-	dispatcher.AddHandler(handlers.NewCommand("storage", setDefaultStorage))
+	dispatcher.AddHandler(handlers.NewCommand("storage", storageCmd))
 	dispatcher.AddHandler(handlers.NewCommand("save", saveCmd))
-	dispatcher.AddHandler(handlers.NewCommand("path", setPath))
+	linkRegexFilter, err := filters.Message.Regex(linkRegexString)
+	if err != nil {
+		logger.L.Panicf("创建正则表达式过滤器失败: %s", err)
+	}
+	dispatcher.AddHandler(handlers.NewMessage(linkRegexFilter, handleLinkMessage))
 	dispatcher.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix("add"), AddToQueue))
+	dispatcher.AddHandler(handlers.NewCallbackQuery(filters.CallbackQuery.Prefix("set_default"), setDefaultStorage))
 	dispatcher.AddHandler(handlers.NewMessage(filters.Message.Media, handleFileMessage))
 }
 
 const noPermissionText string = `
-本 Bot 仅限个人使用.
+您不在白名单中, 无法使用此 Bot.
 您可以部署自己的实例: https://github.com/krau/SaveAny-Bot
 `
 
 func checkPermission(ctx *ext.Context, update *ext.Update) error {
 	userID := update.GetUserChat().GetID()
-	if !slice.Contain(config.Cfg.Telegram.Admins, userID) {
+	if !slice.Contain(config.Cfg.GetUsersID(), userID) {
 		ctx.Reply(update, ext.ReplyTextString(noPermissionText), nil)
 		return dispatcher.EndGroups
 	}
@@ -52,7 +55,7 @@ func checkPermission(ctx *ext.Context, update *ext.Update) error {
 
 func start(ctx *ext.Context, update *ext.Update) error {
 	if err := dao.CreateUser(update.GetUserChat().GetID()); err != nil {
-		logger.L.Errorf("Failed to create user: %s", err)
+		logger.L.Errorf("创建用户失败: %s", err)
 		return dispatcher.EndGroups
 	}
 	return help(ctx, update)
@@ -63,12 +66,15 @@ Save Any Bot - 转存你的 Telegram 文件
 命令:
 /start - 开始使用
 /help - 显示帮助
-/silent - 静默模式
+/silent - 开关静默模式
 /storage - 设置默认存储位置
 /save [自定义文件名] - 保存文件
-/path <存储类型> <路径> - 更改文件保存路径
 
 静默模式: 开启后 Bot 直接保存到收到的文件到默认位置, 不再询问
+
+默认存储位置: 在静默模式下保存到的位置
+
+向 Bot 发送(转发)文件, 或发送一个公开频道的消息链接以保存文件
 `
 
 func help(ctx *ext.Context, update *ext.Update) error {
@@ -77,58 +83,25 @@ func help(ctx *ext.Context, update *ext.Update) error {
 }
 
 func silent(ctx *ext.Context, update *ext.Update) error {
-	user, err := dao.GetUserByUserID(update.GetUserChat().GetID())
+	user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
 	if err != nil {
-		logger.L.Errorf("Failed to get user: %s", err)
+		logger.L.Errorf("获取用户失败: %s", err)
+		return dispatcher.EndGroups
+	}
+	if !user.Silent && user.DefaultStorage == "" {
+		ctx.Reply(update, ext.ReplyTextString("请先使用 /storage 设置默认存储位置"), nil)
 		return dispatcher.EndGroups
 	}
 	user.Silent = !user.Silent
 	if err := dao.UpdateUser(user); err != nil {
-		logger.L.Errorf("Failed to update user: %s", err)
+		logger.L.Errorf("更新用户失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("更新用户失败"), nil)
 		return dispatcher.EndGroups
 	}
 	ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("已%s静默模式", map[bool]string{true: "开启", false: "关闭"}[user.Silent])), nil)
 	return dispatcher.EndGroups
 }
 
-func setDefaultStorage(ctx *ext.Context, update *ext.Update) error {
-	if len(storage.Storages) == 0 {
-		ctx.Reply(update, ext.ReplyTextString("未配置存储"), nil)
-		return dispatcher.EndGroups
-	}
-	args := strings.Split(update.EffectiveMessage.Text, " ")
-	avaliableStorages := maputil.Keys(storage.Storages)
-	if len(args) < 2 {
-		text := []styling.StyledTextOption{
-			styling.Plain("请提供存储位置名称, 可用项:"),
-		}
-		for _, name := range avaliableStorages {
-			text = append(text, styling.Plain("\n"))
-			text = append(text, styling.Code(name))
-		}
-		text = append(text, styling.Plain("\n示例: /storage local"))
-		ctx.Reply(update, ext.ReplyTextStyledTextArray(text), nil)
-		return dispatcher.EndGroups
-	}
-	storageName := args[1]
-	if !slice.Contain(avaliableStorages, storageName) {
-		ctx.Reply(update, ext.ReplyTextString("存储位置不存在"), nil)
-		return dispatcher.EndGroups
-	}
-	user, err := dao.GetUserByUserID(update.GetUserChat().GetID())
-	if err != nil {
-		logger.L.Errorf("Failed to get user: %s", err)
-		return dispatcher.EndGroups
-	}
-	user.DefaultStorage = storageName
-	if err := dao.UpdateUser(user); err != nil {
-		logger.L.Errorf("Failed to update user: %s", err)
-		return dispatcher.EndGroups
-	}
-	ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("已设置默认存储位置为 %s", storageName)), nil)
-	return dispatcher.EndGroups
-}
-
 func saveCmd(ctx *ext.Context, update *ext.Update) error {
 	res, ok := update.EffectiveMessage.GetReplyTo()
 	if !ok || res == nil {
@@ -146,9 +119,23 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
 		return dispatcher.EndGroups
 	}
 
-	msg, err := GetTGMessage(ctx, Client, replyToMsgID)
+	user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
 	if err != nil {
-		logger.L.Errorf("Failed to get message: %s", err)
+		logger.L.Errorf("获取用户失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
+		return dispatcher.EndGroups
+	}
+
+	storages := storage.GetUserStorages(user.ChatID)
+
+	if len(storages) == 0 {
+		ctx.Reply(update, ext.ReplyTextString("无可用的存储"), nil)
+		return dispatcher.EndGroups
+	}
+
+	msg, err := GetTGMessage(ctx, update.EffectiveChat().GetID(), replyToMsgID)
+	if err != nil {
+		logger.L.Errorf("获取消息失败: %s", err)
 		ctx.Reply(update, ext.ReplyTextString("无法获取消息"), nil)
 		return dispatcher.EndGroups
 	}
@@ -159,148 +146,136 @@ func saveCmd(ctx *ext.Context, update *ext.Update) error {
 		return dispatcher.EndGroups
 	}
 
-	user, err := dao.GetUserByUserID(update.GetUserChat().GetID())
-	if err != nil {
-		logger.L.Errorf("Failed to get user: %s", err)
-		return dispatcher.EndGroups
-	}
-
 	replied, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件信息..."), nil)
 	if err != nil {
-		logger.L.Errorf("Failed to reply: %s", err)
+		logger.L.Errorf("回复失败: %s", err)
 		return dispatcher.EndGroups
 	}
 
 	cmdText := update.EffectiveMessage.Text
 	customFileName := strings.TrimSpace(strings.TrimPrefix(cmdText, "/save"))
 
-	file, err := FileFromMessage(ctx, Client, update.EffectiveChat().GetID(), msg.ID, customFileName)
+	file, err := FileFromMessage(ctx, update.EffectiveChat().GetID(), msg.ID, customFileName)
 	if err != nil {
-		logger.L.Errorf("Failed to get file from message: %s", err)
+		logger.L.Errorf("获取文件失败: %s", err)
 		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message: "无法获取文件",
+			Message: fmt.Sprintf("获取文件失败: %s", err),
 			ID:      replied.ID,
 		})
 		return dispatcher.EndGroups
 	}
 
+	// TODO: better file name
 	if file.FileName == "" {
-		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message: "无法获取文件名",
-			ID:      replied.ID,
-		})
-		return dispatcher.EndGroups
+		file.FileName = fmt.Sprintf("%d_%d_%s", update.EffectiveChat().GetID(), replyToMsgID, file.Hash())
 	}
-
 	receivedFile := &types.ReceivedFile{
 		Processing:     false,
 		FileName:       file.FileName,
 		ChatID:         update.EffectiveChat().GetID(),
 		MessageID:      replyToMsgID,
 		ReplyMessageID: replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
 	}
 
 	if err := dao.SaveReceivedFile(receivedFile); err != nil {
-		logger.L.Errorf("Failed to save received file: %s", err)
+		logger.L.Errorf("保存接收的文件失败: %s", err)
 		if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message: fmt.Sprintf("Failed to save received file: %s", err),
+			Message: fmt.Sprintf("保存接收的文件失败: %s", err),
 			ID:      replied.ID,
 		}); err != nil {
-			logger.L.Errorf("Failed to edit message: %s", err)
+			logger.L.Errorf("编辑消息失败: %s", err)
 		}
 		return dispatcher.EndGroups
 	}
-
-	if !user.Silent {
-		entityBuilder := entity.Builder{}
-		var entities []tg.MessageEntityClass
-		text := fmt.Sprintf("文件名: %s\n请选择存储位置", file.FileName)
-		if err := styling.Perform(&entityBuilder,
-			styling.Plain("文件名: "),
-			styling.Code(file.FileName),
-			styling.Plain("\n请选择存储位置"),
-		); err != nil {
-			logger.L.Errorf("Failed to build entity: %s", err)
-		} else {
-			text, entities = entityBuilder.Complete()
-		}
-		_, err = ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message:     text,
-			Entities:    entities,
-			ReplyMarkup: getAddTaskMarkup(msg.ID),
-			ID:          replied.ID,
-		})
-		if err != nil {
-			logger.L.Errorf("Failed to reply: %s", err)
-		}
-		return dispatcher.EndGroups
+	if !user.Silent || user.DefaultStorage == "" {
+		return ProvideSelectMessage(ctx, update, file, update.EffectiveChat().GetID(), msg.ID, replied.ID)
 	}
-
-	if user.DefaultStorage == "" {
-		ctx.Reply(update, ext.ReplyTextString("请先使用 /storage 设置默认存储位置"), nil)
-		return dispatcher.EndGroups
-	}
-	queue.AddTask(types.Task{
+	return HandleSilentAddTask(ctx, update, user, &types.Task{
 		Ctx:            ctx,
 		Status:         types.Pending,
 		File:           file,
-		Storage:        types.StorageType(user.DefaultStorage),
-		ChatID:         update.EffectiveChat().GetID(),
+		StorageName:    user.DefaultStorage,
+		FileChatID:     update.EffectiveChat().GetID(),
 		ReplyMessageID: replied.ID,
-		MessageID:      msg.ID,
-	})
-	_, err = ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-		Message: fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", file.FileName, queue.Len()),
-		ID:      replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+		FileMessageID:  msg.ID,
+		UserID:         user.ChatID,
 	})
+}
+
+func storageCmd(ctx *ext.Context, update *ext.Update) error {
+	user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
 	if err != nil {
-		logger.L.Errorf("Failed to edit message: %s", err)
+		logger.L.Errorf("获取用户失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
+		return dispatcher.EndGroups
 	}
+	storages := storage.GetUserStorages(user.ChatID)
+	if len(storages) == 0 {
+		ctx.Reply(update, ext.ReplyTextString("无可用的存储"), nil)
+		return dispatcher.EndGroups
+	}
+
+	ctx.Reply(update, ext.ReplyTextString("请选择要设为默认的存储位置"), &ext.ReplyOpts{
+		Markup: getSetDefaultStorageMarkup(user.ChatID, storages),
+	})
+
 	return dispatcher.EndGroups
 }
 
-func setPath(ctx *ext.Context, update *ext.Update) error {
-	if len(storage.Storages) == 0 {
-		ctx.Reply(update, ext.ReplyTextString("未配置存储"), nil)
+func setDefaultStorage(ctx *ext.Context, update *ext.Update) error {
+	args := strings.Split(string(update.CallbackQuery.Data), " ")
+	userID, _ := strconv.Atoi(args[1])
+	storageNameHash := args[2]
+	if userID != int(update.CallbackQuery.GetUserID()) {
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "你没有权限",
+			CacheTime: 5,
+		})
 		return dispatcher.EndGroups
 	}
-	if update.EffectiveMessage == nil {
-		logger.L.Error("No effective message")
+	storageName := storageHashName[storageNameHash]
+	selectedStorage, err := storage.GetStorageByName(storageName)
+
+	if err != nil {
+		logger.L.Errorf("获取指定存储失败: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "获取指定存储失败",
+			CacheTime: 5,
+		})
 		return dispatcher.EndGroups
 	}
-	args := strings.Split(update.EffectiveMessage.Text, " ")
-	if len(args) < 3 {
-		text := []styling.StyledTextOption{
-			styling.Plain("请提供存储位置名称和路径, 可用项:"),
-		}
-		for name := range storage.Storages {
-			text = append(text, styling.Plain("\n"))
-			text = append(text, styling.Code(string(name)))
-		}
-		text = append(text, styling.Plain("\n示例: /path local /path/to/save"))
-		ctx.Reply(update, ext.ReplyTextStyledTextArray(text), nil)
+	user, err := dao.GetUserByChatID(int64(userID))
+	if err != nil {
+		logger.L.Errorf("Failed to get user: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "获取用户失败",
+			CacheTime: 5,
+		})
 		return dispatcher.EndGroups
 	}
-	storageName := args[1]
-	if _, ok := storage.Storages[types.StorageType(storageName)]; !ok {
-		ctx.Reply(update, ext.ReplyTextString("存储位置不存在"), nil)
+	user.DefaultStorage = storageName
+	if err := dao.UpdateUser(user); err != nil {
+		logger.L.Errorf("Failed to update user: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "更新用户失败",
+			CacheTime: 5,
+		})
 		return dispatcher.EndGroups
 	}
-	path := strings.Join(args[2:], " ")
-	switch storageName {
-	case "local":
-		config.Set("storage.local.base_path", path)
-	case "webdav":
-		config.Set("storage.webdav.base_path", path)
-	case "alist":
-		config.Set("storage.alist.base_path", path)
-	}
-	if err := config.ReloadConfig(); err != nil {
-		logger.L.Errorf("Failed to reload config: %s", err)
-		ctx.Reply(update, ext.ReplyTextString("设置失败: "+err.Error()), nil)
-		return dispatcher.EndGroups
-	}
-	ctx.Reply(update, ext.ReplyTextString("设置成功"), nil)
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message: fmt.Sprintf("已将 %s (%s) 设为默认存储位置", selectedStorage.Name(), selectedStorage.Type()),
+		ID:      update.CallbackQuery.GetMsgID(),
+	})
 	return dispatcher.EndGroups
 }
 
@@ -314,31 +289,32 @@ func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
 		return dispatcher.EndGroups
 	}
 
-	user, err := dao.GetUserByUserID(update.GetUserChat().GetID())
+	user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
 	if err != nil {
-		logger.L.Errorf("Failed to get user: %s", err)
+		logger.L.Errorf("获取用户失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取用户失败"), nil)
+		return dispatcher.EndGroups
+	}
+	storages := storage.GetUserStorages(user.ChatID)
+	if len(storages) == 0 {
+		ctx.Reply(update, ext.ReplyTextString("无可用的存储"), nil)
 		return dispatcher.EndGroups
 	}
 
 	msg, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件信息..."), nil)
 	if err != nil {
-		logger.L.Errorf("Failed to reply: %s", err)
+		logger.L.Errorf("回复失败: %s", err)
 		return dispatcher.EndGroups
 	}
 	media := update.EffectiveMessage.Media
 	file, err := FileFromMedia(media, "")
 	if err != nil {
-		logger.L.Errorf("Failed to get file from media: %s", err)
-		if errors.Is(err, ErrEmptyFileName) {
-			ctx.Reply(update, ext.ReplyTextString("无法获取文件名, 请使用 /save <自定义文件名> 回复此文件"), nil)
-		} else {
-			ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("获取文件失败: %s", err)), nil)
-		}
+		logger.L.Errorf("获取文件失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("获取文件失败: %s", err)), nil)
 		return dispatcher.EndGroups
 	}
 	if file.FileName == "" {
-		ctx.Reply(update, ext.ReplyTextString("无法获取文件名"), nil)
-		return dispatcher.EndGroups
+		file.FileName = fmt.Sprintf("%d_%d_%s", update.EffectiveChat().GetID(), update.EffectiveMessage.ID, file.Hash())
 	}
 
 	if err := dao.SaveReceivedFile(&types.ReceivedFile{
@@ -347,69 +323,36 @@ func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
 		ChatID:         update.EffectiveChat().GetID(),
 		MessageID:      update.EffectiveMessage.ID,
 		ReplyMessageID: msg.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
 	}); err != nil {
-		logger.L.Errorf("Failed to add received file: %s", err)
+		logger.L.Errorf("添加接收的文件失败: %s", err)
 		if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message: fmt.Sprintf("Failed to add received file: %s", err),
+			Message: fmt.Sprintf("添加接收的文件失败: %s", err),
 			ID:      msg.ID,
 		}); err != nil {
-			logger.L.Errorf("Failed to edit message: %s", err)
+			logger.L.Errorf("编辑消息失败: %s", err)
 		}
 		return dispatcher.EndGroups
 	}
 
-	if !user.Silent {
-		entityBuilder := entity.Builder{}
-		var entities []tg.MessageEntityClass
-		text := fmt.Sprintf("文件名: %s\n请选择存储位置", file.FileName)
-		if err := styling.Perform(&entityBuilder,
-			styling.Plain("文件名: "),
-			styling.Code(file.FileName),
-			styling.Plain("\n请选择存储位置"),
-		); err != nil {
-			logger.L.Errorf("Failed to build entity: %s", err)
-		} else {
-			text, entities = entityBuilder.Complete()
-		}
-		_, err = ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message:     text,
-			Entities:    entities,
-			ReplyMarkup: getAddTaskMarkup(update.EffectiveMessage.ID),
-			ID:          msg.ID,
-		})
-		if err != nil {
-			logger.L.Errorf("Failed to edit message: %s", err)
-		}
-		return dispatcher.EndGroups
+	if !user.Silent || user.DefaultStorage == "" {
+		return ProvideSelectMessage(ctx, update, file, update.EffectiveChat().GetID(), update.EffectiveMessage.ID, msg.ID)
 	}
-
-	if user.DefaultStorage == "" {
-		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-			Message: "请先使用 /storage 设置默认存储位置",
-			ID:      msg.ID,
-		})
-		return dispatcher.EndGroups
-	}
-
-	queue.AddTask(types.Task{
+	return HandleSilentAddTask(ctx, update, user, &types.Task{
 		Ctx:            ctx,
 		Status:         types.Pending,
 		File:           file,
-		Storage:        types.StorageType(user.DefaultStorage),
-		ChatID:         update.EffectiveChat().GetID(),
+		StorageName:    user.DefaultStorage,
+		FileChatID:     update.EffectiveChat().GetID(),
 		ReplyMessageID: msg.ID,
-		MessageID:      update.EffectiveMessage.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+		FileMessageID:  update.EffectiveMessage.ID,
+		UserID:         user.ChatID,
 	})
-
-	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
-		Message: fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", file.FileName, queue.Len()),
-		ID:      msg.ID,
-	})
-	return dispatcher.EndGroups
 }
 
 func AddToQueue(ctx *ext.Context, update *ext.Update) error {
-	if !slice.Contain(config.Cfg.Telegram.Admins, update.CallbackQuery.UserID) {
+	if !slice.Contain(config.Cfg.GetUsersID(), update.CallbackQuery.UserID) {
 		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
 			QueryID:   update.CallbackQuery.QueryID,
 			Alert:     true,
@@ -419,11 +362,24 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
 		return dispatcher.EndGroups
 	}
 	args := strings.Split(string(update.CallbackQuery.Data), " ")
-	messageID, _ := strconv.Atoi(args[1])
-	logger.L.Tracef("Got add to queue: chatID: %d, messageID: %d, storage: %s", update.EffectiveChat().GetID(), messageID, args[2])
-	record, err := dao.GetReceivedFileByChatAndMessageID(update.EffectiveChat().GetID(), messageID)
+	fileChatID, _ := strconv.Atoi(args[1])
+	fileMessageID, _ := strconv.Atoi(args[2])
+	storageNameHash := args[3]
+	storageName := storageHashName[storageNameHash]
+	if storageName == "" {
+		logger.L.Errorf("未知存储位置哈希: %d", storageNameHash)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "未知存储位置",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+	logger.L.Tracef("Got add to queue: chatID: %d, messageID: %d, storage: %s", fileChatID, fileMessageID, storageName)
+	record, err := dao.GetReceivedFileByChatAndMessageID(int64(fileChatID), fileMessageID)
 	if err != nil {
-		logger.L.Errorf("Failed to get received file: %s", err)
+		logger.L.Errorf("获取记录失败: %s", err)
 		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
 			QueryID:   update.CallbackQuery.QueryID,
 			Alert:     true,
@@ -435,13 +391,12 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
 	if update.CallbackQuery.MsgID != record.ReplyMessageID {
 		record.ReplyMessageID = update.CallbackQuery.MsgID
 		if err := dao.SaveReceivedFile(record); err != nil {
-			logger.L.Errorf("Failed to update received file: %s", err)
+			logger.L.Errorf("更新接收的文件失败: %s", err)
 		}
 	}
-
-	file, err := FileFromMessage(ctx, Client, record.ChatID, record.MessageID, record.FileName)
+	file, err := FileFromMessage(ctx, record.ChatID, record.MessageID, record.FileName)
 	if err != nil {
-		logger.L.Errorf("Failed to get file from message: %s", err)
+		logger.L.Errorf("获取消息中的文件失败: %s", err)
 		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
 			QueryID:   update.CallbackQuery.QueryID,
 			Alert:     true,
@@ -455,10 +410,12 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
 		Ctx:            ctx,
 		Status:         types.Pending,
 		File:           file,
-		Storage:        types.StorageType(args[2]),
-		ChatID:         record.ChatID,
+		StorageName:    storageName,
+		FileChatID:     record.ChatID,
 		ReplyMessageID: record.ReplyMessageID,
-		MessageID:      record.MessageID,
+		FileMessageID:  record.MessageID,
+		ReplyChatID:    record.ReplyChatID,
+		UserID:         update.EffectiveUser().GetID(),
 	})
 
 	entityBuilder := entity.Builder{}

bot/utils.go

@@ -1,26 +1,29 @@
 package bot
 
 import (
-	"context"
 	"errors"
 	"fmt"
 	"time"
 
-	"github.com/celestix/gotgproto"
 	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/telegram/message/entity"
+	"github.com/gotd/td/telegram/message/styling"
 	"github.com/gotd/td/tg"
 	"github.com/krau/SaveAny-Bot/common"
+	"github.com/krau/SaveAny-Bot/dao"
 	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/queue"
 	"github.com/krau/SaveAny-Bot/storage"
 	"github.com/krau/SaveAny-Bot/types"
 )
 
 var (
-	ErrEmptyFileName   = errors.New("file name is empty")
 	ErrEmptyDocument   = errors.New("document is empty")
 	ErrEmptyPhoto      = errors.New("photo is empty")
 	ErrEmptyPhotoSize  = errors.New("photo size is empty")
 	ErrEmptyPhotoSizes = errors.New("photo size slice is empty")
+	ErrNoStorages      = errors.New("no available storage")
 )
 
 func supportedMediaFilter(m *tg.Message) (bool, error) {
@@ -37,49 +40,52 @@ func supportedMediaFilter(m *tg.Message) (bool, error) {
 	}
 }
 
-var StorageDisplayNames = map[string]string{
-	"all":    "全部",
-	"local":  "服务器磁盘",
-	"alist":  "Alist",
-	"webdav": "WebDAV",
-}
+// for callback data
+var storageHashName = map[string]string{}
 
-func getAddTaskMarkup(messageID int) *tg.ReplyInlineMarkup {
-	storageButtons := make([]tg.KeyboardButtonClass, 0)
-	for _, name := range storage.StorageKeys {
-		storageButtons = append(storageButtons, &tg.KeyboardButtonCallback{
-			Text: StorageDisplayNames[string(name)],
-			Data: []byte(fmt.Sprintf("add %d %s", messageID, name)),
+func getSelectStorageMarkup(userChatID int64, fileChatID, fileMessageID int) (*tg.ReplyInlineMarkup, error) {
+	user, err := dao.GetUserByChatID(userChatID)
+	if err != nil {
+		return nil, err
+	}
+	storages := storage.GetUserStorages(user.ChatID)
+
+	buttons := make([]tg.KeyboardButtonClass, 0)
+	for _, storage := range storages {
+		nameHash := common.HashString(storage.Name())
+		storageHashName[nameHash] = storage.Name()
+		buttons = append(buttons, &tg.KeyboardButtonCallback{
+			Text: storage.Name(),
+			Data: []byte(fmt.Sprintf("add %d %d %s", fileChatID, fileMessageID, nameHash)),
 		})
 	}
+	markup := &tg.ReplyInlineMarkup{}
+	for i := 0; i < len(buttons); i += 3 {
+		row := tg.KeyboardButtonRow{}
+		row.Buttons = buttons[i:min(i+3, len(buttons))]
+		markup.Rows = append(markup.Rows, row)
+	}
+	return markup, nil
+}
 
-	if len(storageButtons) < 1 {
-		return nil
+func getSetDefaultStorageMarkup(userChatID int64, storages []storage.Storage) *tg.ReplyInlineMarkup {
+	buttons := make([]tg.KeyboardButtonClass, 0)
+	for _, storage := range storages {
+		nameHash := common.HashString(storage.Name())
+		storageHashName[nameHash] = storage.Name()
+		buttons = append(buttons, &tg.KeyboardButtonCallback{
+			Text: storage.Name(),
+			Data: []byte(fmt.Sprintf("set_default %d %s", userChatID, nameHash)),
+		})
 	}
-	if len(storageButtons) == 1 {
-		return &tg.ReplyInlineMarkup{
-			Rows: []tg.KeyboardButtonRow{
-				{
-					Buttons: storageButtons,
-				},
-			},
-		}
-	}
-	return &tg.ReplyInlineMarkup{
-		Rows: []tg.KeyboardButtonRow{
-			{
-				Buttons: storageButtons,
-			},
-			{
-				Buttons: []tg.KeyboardButtonClass{
-					&tg.KeyboardButtonCallback{
-						Text: "全部",
-						Data: []byte(fmt.Sprintf("add %d all", messageID)),
-					},
-				},
-			},
-		},
+	markup := &tg.ReplyInlineMarkup{}
+	for i := 0; i < len(buttons); i += 3 {
+		row := tg.KeyboardButtonRow{}
+		row.Buttons = buttons[i:min(i+3, len(buttons))]
+		markup.Rows = append(markup.Rows, row)
 	}
+	return markup
+
 }
 
 func FileFromMedia(media tg.MessageMediaClass, customFileName string) (*types.File, error) {
@@ -103,9 +109,6 @@ func FileFromMedia(media tg.MessageMediaClass, customFileName string) (*types.Fi
 				break
 			}
 		}
-		if fileName == "" {
-			return nil, ErrEmptyFileName
-		}
 		return &types.File{
 			Location: document.AsInputDocumentFileLocation(),
 			FileSize: document.Size,
@@ -144,7 +147,7 @@ func FileFromMedia(media tg.MessageMediaClass, customFileName string) (*types.Fi
 	return nil, fmt.Errorf("unexpected type %T", media)
 }
 
-func FileFromMessage(ctx context.Context, client *gotgproto.Client, chatID int64, messageID int, customFileName string) (*types.File, error) {
+func FileFromMessage(ctx *ext.Context, chatID int64, messageID int, customFileName string) (*types.File, error) {
 	key := fmt.Sprintf("file:%d:%d", chatID, messageID)
 	logger.L.Debugf("Getting file: %s", key)
 	var cachedFile types.File
@@ -152,8 +155,7 @@ func FileFromMessage(ctx context.Context, client *gotgproto.Client, chatID int64
 	if err == nil {
 		return &cachedFile, nil
 	}
-
-	message, err := GetTGMessage(ctx, client, messageID)
+	message, err := GetTGMessage(ctx, chatID, messageID)
 	if err != nil {
 		return nil, err
 	}
@@ -167,20 +169,76 @@ func FileFromMessage(ctx context.Context, client *gotgproto.Client, chatID int64
 	return file, nil
 }
 
-func GetTGMessage(ctx context.Context, client *gotgproto.Client, messageID int) (*tg.Message, error) {
+func GetTGMessage(ctx *ext.Context, chatId int64, messageID int) (*tg.Message, error) {
 	logger.L.Debugf("Fetching message: %d", messageID)
-	res, err := client.API().MessagesGetMessages(ctx, []tg.InputMessageClass{
-		&tg.InputMessageID{
-			ID: messageID,
-		},
-	})
+	messages, err := ctx.GetMessages(chatId, []tg.InputMessageClass{&tg.InputMessageID{ID: messageID}})
 	if err != nil {
 		return nil, err
 	}
-	messages := res.(*tg.MessagesMessages)
-	msg := messages.Messages[0]
-	if _, ok := msg.(*tg.Message); !ok {
-		return nil, fmt.Errorf("unexpected type %T, this file may be deleted", msg)
+	if len(messages) == 0 {
+		return nil, errors.New("no messages found")
 	}
-	return msg.(*tg.Message), nil
+	msg := messages[0]
+	tgMessage, ok := msg.(*tg.Message)
+	if !ok {
+		return nil, fmt.Errorf("unexpected message type: %T", msg)
+	}
+	return tgMessage, nil
+}
+
+func ProvideSelectMessage(ctx *ext.Context, update *ext.Update, file *types.File, chatID int64, fileMsgID, toEditMsgID int) error {
+	entityBuilder := entity.Builder{}
+	var entities []tg.MessageEntityClass
+	text := fmt.Sprintf("文件名: %s\n请选择存储位置", file.FileName)
+	if err := styling.Perform(&entityBuilder,
+		styling.Plain("文件名: "),
+		styling.Code(file.FileName),
+		styling.Plain("\n请选择存储位置"),
+	); err != nil {
+		logger.L.Errorf("Failed to build entity: %s", err)
+	} else {
+		text, entities = entityBuilder.Complete()
+	}
+	markup, err := getSelectStorageMarkup(update.EffectiveUser().GetID(), int(chatID), fileMsgID)
+	if errors.Is(err, ErrNoStorages) {
+		logger.L.Errorf("Failed to get select storage markup: %s", err)
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "无可用存储",
+			ID:      toEditMsgID,
+		})
+		return dispatcher.EndGroups
+	} else if err != nil {
+		logger.L.Errorf("Failed to get select storage markup: %s", err)
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "无法获取存储",
+			ID:      toEditMsgID,
+		})
+		return dispatcher.EndGroups
+	}
+	_, err = ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message:     text,
+		Entities:    entities,
+		ReplyMarkup: markup,
+		ID:          toEditMsgID,
+	})
+	if err != nil {
+		logger.L.Errorf("Failed to reply: %s", err)
+	}
+	return dispatcher.EndGroups
+}
+
+func HandleSilentAddTask(ctx *ext.Context, update *ext.Update, user *types.User, task *types.Task) error {
+	if user.DefaultStorage == "" {
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "请先使用 /storage 设置默认存储位置",
+			ID:      task.ReplyMessageID,
+		})
+		return dispatcher.EndGroups
+	}
+	queue.AddTask(*task)
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message: fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", task.FileName(), queue.Len()),
+		ID:      task.ReplyMessageID,
+	})
+	return dispatcher.EndGroups
 }

common/utils.go

@@ -0,0 +1,12 @@
+package common
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+)
+
+func HashString(s string) string {
+	hash := md5.New()
+	hash.Write([]byte(s))
+	return hex.EncodeToString(hash.Sum(nil))
+}

config.example.toml

@@ -2,41 +2,83 @@ workers = 4 # 同时下载文件数
 retry = 3   # 下载失败重试次数
 
 [telegram]
-token = ""                                    # Bot Token
-admins = [777000]                             # 你的 user_id
-app_id = 123456                               # Telegram API ID
-app_hash = "0123456789abcdef0123456789abcdef" # Telegram API Hash
+# Bot Token
+token = ""
+# Telegram API 配置, 若不配置也可运行, 将使用默认的 API ID 和 API HASH
+# 推荐使用自己的 API ID 和 API HASH (https://my.telegram.org)
+# app_id = 123456
+# app_hash = "0123456789abcdef0123456789abcdef"
 
 [telegram.proxy]
+# 启用代理连接 telegram, 只支持 socks5
 enable = false
-url = "socks5://127.0.0.1:7890" # 代理地址
+url = "socks5://127.0.0.1:7890"
 
-[log]
-level = "DEBUG" # 日志等级
 
-[temp]
-base_path = "cache/" # 下载文件临时目录, 请不要在此目录下存放任何其他文件
-cache_ttl = 30       # 临时文件保存时间, 单位: 秒
-
-[db]
-path = "data/data.db" # 数据库文件路径
-
-[storage]
-[storage.alist] # Alist
+# 存储配置列表
+[[storages]]
+# 标识名, 需要唯一
+name = "本机1"
+# 存储类型, 目前可用: local , alist , webdav
+type = "local"
 enable = true
-base_path = "/telegram"   # 保存路径
-username = "admin"        # 用户名
-password = "password"     # 密码
-url = "https://alist.com" # Alist 地址
-token_exp = 86400         # token 过期时间, 单位: 秒
+base_path = "./downloads"
 
-[storage.local] # 本地磁盘
+[[storages]]
+name = "本机2"
+type = "local"
 enable = true
-base_path = "downloads/" # 保存路径
+base_path = "./downloads/2"
 
-[storage.webdav] # WebDav
+[[storages]]
+name = "MyAlist"
+type = "alist"
 enable = true
-base_path = "/telegram"
-username = "admin"
-password = "password"
-url = "https://alist.com/dav"
+base_path = '/'
+url = 'https://alist.com'
+username = 'admin'
+password = 'password'
+token_exp = 86400
+# alist 可直接使用 token 登录, 此时 username, password, token_exp 将被忽略
+# 请自行在 alist 侧配置合理的 token 过期时间
+# token = ""
+
+
+[[storages]]
+name = "MyWebdav"
+type = "webdav"
+base_path = '/path/telegram'
+enable = true
+url = 'https://example.com/dav'
+username = 'username'
+password = 'password'
+
+
+# 用户列表
+[[users]]
+# user id
+id = 123456
+# 存储名称过滤列表
+storages = ["MyAlist", "本机1"]
+# 开启黑名单模式, 过滤列表中的存储将无法使用, 默认为白名单模式
+blacklist = false
+
+[[users]]
+id = 114514
+# 将列表留空并开启黑名单模式以允许使用所有存储
+storages = []
+blacklist = true
+
+
+# [log]
+# # 日志等级
+# level = "DEBUG"
+
+# [temp]
+# # 下载文件临时目录, 请不要在此目录下存放任何其他文件
+# base_path = "cache/"
+# # 临时文件保存时间, 单位: 秒
+# cache_ttl = 30
+
+# [db]
+# path = "data/data.db" # 数据库文件路径

config/deprecated.go

@@ -0,0 +1,95 @@
+package config
+
+import (
+	"strconv"
+
+	"github.com/krau/SaveAny-Bot/types"
+	"gorm.io/datatypes"
+)
+
+// for compatibility
+type deprecatedStorageConfig struct {
+	Alist  alistConfig  `toml:"alist" mapstructure:"alist"`
+	Local  localConfig  `toml:"local" mapstructure:"local"`
+	Webdav webdavConfig `toml:"webdav" mapstructure:"webdav"`
+}
+
+type alistConfig struct {
+	Enable   bool   `toml:"enable" mapstructure:"enable" json:"enable"`
+	URL      string `toml:"url" mapstructure:"url" json:"url"`
+	Username string `toml:"username" mapstructure:"username" json:"username"`
+	Password string `toml:"password" mapstructure:"password" json:"password"`
+	Token    string `toml:"token" mapstructure:"token" json:"token"`
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+	TokenExp int64  `toml:"token_exp" mapstructure:"token_exp" json:"token_exp"`
+}
+
+func (a *alistConfig) ToJSON() datatypes.JSON {
+	tokenExp := strconv.FormatInt(a.TokenExp, 10)
+	return datatypes.JSON([]byte(`{"url":"` + a.URL + `","username":"` + a.Username + `","password":"` + a.Password + `","token":"` + a.Token + `","base_path":"` + a.BasePath + `","token_exp":` + tokenExp + `}`))
+}
+
+type localConfig struct {
+	Enable   bool   `toml:"enable" mapstructure:"enable" json:"enable"`
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+}
+
+func (l *localConfig) ToJSON() datatypes.JSON {
+	return datatypes.JSON([]byte(`{"base_path":"` + l.BasePath + `"}`))
+}
+
+type webdavConfig struct {
+	Enable   bool   `toml:"enable" mapstructure:"enable" json:"enable"`
+	URL      string `toml:"url" mapstructure:"url" json:"url"`
+	Username string `toml:"username" mapstructure:"username" json:"username"`
+	Password string `toml:"password" mapstructure:"password" json:"password"`
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+}
+
+func (w *webdavConfig) ToJSON() datatypes.JSON {
+	return datatypes.JSON([]byte(`{"url":"` + w.URL + `","username":"` + w.Username + `","password":"` + w.Password + `","base_path":"` + w.BasePath + `"}`))
+}
+
+func transformDeprecatedStorageConfig() {
+	if Cfg.DeprecatedStorage.Alist.Enable {
+		alistStorage := &AlistStorageConfig{
+			NewStorageConfig: NewStorageConfig{
+				Name:   "Alist",
+				Enable: true,
+				Type:   string(types.StorageTypeAlist),
+			},
+			URL:      Cfg.DeprecatedStorage.Alist.URL,
+			Username: Cfg.DeprecatedStorage.Alist.Username,
+			Password: Cfg.DeprecatedStorage.Alist.Password,
+			Token:    Cfg.DeprecatedStorage.Alist.Token,
+			BasePath: Cfg.DeprecatedStorage.Alist.BasePath,
+			TokenExp: Cfg.DeprecatedStorage.Alist.TokenExp,
+		}
+		Cfg.Storages = append(Cfg.Storages, alistStorage)
+	}
+	if Cfg.DeprecatedStorage.Local.Enable {
+		localStorage := &LocalStorageConfig{
+			NewStorageConfig: NewStorageConfig{
+				Name:   "Local",
+				Enable: true,
+				Type:   string(types.StorageTypeLocal),
+			},
+			BasePath: Cfg.DeprecatedStorage.Local.BasePath,
+		}
+		Cfg.Storages = append(Cfg.Storages, localStorage)
+	}
+	if Cfg.DeprecatedStorage.Webdav.Enable {
+		webdavStorage := &WebdavStorageConfig{
+			NewStorageConfig: NewStorageConfig{
+				Name:   "Webdav",
+				Enable: true,
+				Type:   string(types.StorageTypeWebdav),
+			},
+			URL:      Cfg.DeprecatedStorage.Webdav.URL,
+			Username: Cfg.DeprecatedStorage.Webdav.Username,
+			Password: Cfg.DeprecatedStorage.Webdav.Password,
+			BasePath: Cfg.DeprecatedStorage.Webdav.BasePath,
+		}
+		Cfg.Storages = append(Cfg.Storages, webdavStorage)
+	}
+}

config/storage_factory.go

@@ -0,0 +1,104 @@
+// storage_config.go
+
+package config
+
+import (
+	"fmt"
+
+	"github.com/krau/SaveAny-Bot/types"
+	"github.com/mitchellh/mapstructure"
+	"github.com/spf13/viper"
+)
+
+type StorageConfig interface {
+	Validate() error
+	GetType() types.StorageType
+	GetName() string
+}
+
+// Base storage config
+type NewStorageConfig struct {
+	Name      string                 `toml:"name" mapstructure:"name" json:"name"`
+	Type      string                 `toml:"type" mapstructure:"type" json:"type"`
+	Enable    bool                   `toml:"enable" mapstructure:"enable" json:"enable"`
+	RawConfig map[string]interface{} `toml:"-" mapstructure:",remain"`
+}
+
+type StorageConfigFactory func(cfg *NewStorageConfig) (StorageConfig, error)
+
+var storageFactories = make(map[string]StorageConfigFactory)
+
+func RegisterStorageFactory(storageType string, factory StorageConfigFactory) {
+	storageFactories[storageType] = factory
+}
+
+func init() {
+	RegisterStorageFactory(string(types.StorageTypeLocal), newLocalStorageConfig)
+	RegisterStorageFactory(string(types.StorageTypeAlist), newAlistStorageConfig)
+	RegisterStorageFactory(string(types.StorageTypeWebdav), newWebdavStorageConfig)
+}
+
+func newLocalStorageConfig(cfg *NewStorageConfig) (StorageConfig, error) {
+	var localCfg LocalStorageConfig
+	localCfg.NewStorageConfig = *cfg
+
+	if err := mapstructure.Decode(cfg.RawConfig, &localCfg); err != nil {
+		return nil, fmt.Errorf("failed to decode local storage config: %w", err)
+	}
+
+	return &localCfg, nil
+}
+
+func newAlistStorageConfig(cfg *NewStorageConfig) (StorageConfig, error) {
+	var alistCfg AlistStorageConfig
+	alistCfg.NewStorageConfig = *cfg
+
+	if err := mapstructure.Decode(cfg.RawConfig, &alistCfg); err != nil {
+		return nil, fmt.Errorf("failed to decode alist storage config: %w", err)
+	}
+
+	return &alistCfg, nil
+}
+
+func newWebdavStorageConfig(cfg *NewStorageConfig) (StorageConfig, error) {
+	var webdavCfg WebdavStorageConfig
+	webdavCfg.NewStorageConfig = *cfg
+
+	if err := mapstructure.Decode(cfg.RawConfig, &webdavCfg); err != nil {
+		return nil, fmt.Errorf("failed to decode webdav storage config: %w", err)
+	}
+
+	return &webdavCfg, nil
+}
+
+func LoadStorageConfigs(v *viper.Viper) ([]StorageConfig, error) {
+	var baseConfigs []NewStorageConfig
+	if err := v.UnmarshalKey("storages", &baseConfigs); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal storage configs: %w", err)
+	}
+
+	var configs []StorageConfig
+	for _, baseCfg := range baseConfigs {
+		if !baseCfg.Enable {
+			continue
+		}
+
+		factory, ok := storageFactories[baseCfg.Type]
+		if !ok {
+			return nil, fmt.Errorf("unsupported storage type: %s", baseCfg.Type)
+		}
+
+		cfg, err := factory(&baseCfg)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create storage config for %s: %w", baseCfg.Name, err)
+		}
+
+		if err := cfg.Validate(); err != nil {
+			return nil, fmt.Errorf("invalid storage config for %s: %w", baseCfg.Name, err)
+		}
+
+		configs = append(configs, cfg)
+	}
+
+	return configs, nil
+}

config/storages.go

@@ -0,0 +1,106 @@
+package config
+
+import (
+	"fmt"
+
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+func (c *Config) GetStoragesByType(storageType types.StorageType) []StorageConfig {
+	var storages []StorageConfig
+	for _, storage := range c.Storages {
+		if storage.GetType() == storageType {
+			storages = append(storages, storage)
+		}
+	}
+	return storages
+}
+
+func (c *Config) GetStorageByName(name string) StorageConfig {
+	for _, storage := range c.Storages {
+		if storage.GetName() == name {
+			return storage
+		}
+	}
+	return nil
+}
+
+type LocalStorageConfig struct {
+	NewStorageConfig
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+}
+
+func (l *LocalStorageConfig) Validate() error {
+	if l.BasePath == "" {
+		return fmt.Errorf("path is required for local storage")
+	}
+	return nil
+}
+
+func (l *LocalStorageConfig) GetType() types.StorageType {
+	return types.StorageTypeLocal
+}
+
+func (l *LocalStorageConfig) GetName() string {
+	return l.Name
+}
+
+type AlistStorageConfig struct {
+	NewStorageConfig
+	URL      string `toml:"url" mapstructure:"url" json:"url"`
+	Username string `toml:"username" mapstructure:"username" json:"username"`
+	Password string `toml:"password" mapstructure:"password" json:"password"`
+	Token    string `toml:"token" mapstructure:"token" json:"token"`
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+	TokenExp int64  `toml:"token_exp" mapstructure:"token_exp" json:"token_exp"`
+}
+
+func (a *AlistStorageConfig) Validate() error {
+	if a.URL == "" {
+		return fmt.Errorf("url is required for alist storage")
+	}
+	if a.Token == "" && (a.Username == "" || a.Password == "") {
+		return fmt.Errorf("username and password or token is required for alist storage")
+	}
+	if a.BasePath == "" {
+		return fmt.Errorf("base_path is required for alist storage")
+	}
+	return nil
+}
+
+func (a *AlistStorageConfig) GetType() types.StorageType {
+	return types.StorageTypeAlist
+}
+
+func (a *AlistStorageConfig) GetName() string {
+	return a.Name
+}
+
+type WebdavStorageConfig struct {
+	NewStorageConfig
+	URL      string `toml:"url" mapstructure:"url" json:"url"`
+	Username string `toml:"username" mapstructure:"username" json:"username"`
+	Password string `toml:"password" mapstructure:"password" json:"password"`
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+}
+
+func (w *WebdavStorageConfig) Validate() error {
+	if w.URL == "" {
+		return fmt.Errorf("url is required for webdav storage")
+	}
+	if w.Username == "" || w.Password == "" {
+		return fmt.Errorf("username and password is required for webdav storage")
+	}
+	if w.BasePath == "" {
+		return fmt.Errorf("base_path is required for webdav storage")
+	}
+	return nil
+}
+
+func (w *WebdavStorageConfig) GetType() types.StorageType {
+	return types.StorageTypeWebdav
+}
+
+func (w *WebdavStorageConfig) GetName() string {
+	return w.Name
+}

config/user.go

@@ -0,0 +1,49 @@
+package config
+
+import (
+	"github.com/duke-git/lancet/v2/slice"
+)
+
+type userConfig struct {
+	ID        int64    `toml:"id" mapstructure:"id" json:"id"`                      // telegram user id
+	Storages  []string `toml:"storages" mapstructure:"storages" json:"storages"`    // storage names
+	Blacklist bool     `toml:"blacklist" mapstructure:"blacklist" json:"blacklist"` // 黑名单模式, storage names 中的存储将不会被使用, 默认为白名单模式
+}
+
+func (c *Config) GetStorageNamesByUserID(userID int64) []string {
+	for _, user := range c.Users {
+		if user.ID == userID {
+			if user.Blacklist {
+				allStorages := make([]string, 0, len(c.Storages))
+				for _, storage := range c.Storages {
+					allStorages = append(allStorages, storage.GetName())
+				}
+				return slice.Compact(slice.Difference(allStorages, user.Storages))
+			} else {
+				return user.Storages
+			}
+		}
+	}
+	return nil
+}
+
+func (c *Config) GetUsersID() []int64 {
+	var ids []int64
+	for _, user := range c.Users {
+		ids = append(ids, user.ID)
+	}
+	return ids
+}
+
+func (c *Config) HasStorage(userID int64, storageName string) bool {
+	for _, user := range c.Users {
+		if user.ID == userID {
+			if user.Blacklist {
+				return !slice.Contain(user.Storages, storageName)
+			} else {
+				return slice.Contain(user.Storages, storageName)
+			}
+		}
+	}
+	return false
+}

config/viper.go

@@ -9,26 +9,29 @@ import (
 )
 
 type Config struct {
-	Workers      int  `toml:"workers" mapstructure:"workers"`
-	Retry        int  `toml:"retry" mapstructure:"retry"`
-	NoCleanCache bool `toml:"no_clean_cache" mapstructure:"no_clean_cache"`
+	Workers      int          `toml:"workers" mapstructure:"workers"`
+	Retry        int          `toml:"retry" mapstructure:"retry"`
+	NoCleanCache bool         `toml:"no_clean_cache" mapstructure:"no_clean_cache" json:"no_clean_cache"`
+	Users        []userConfig `toml:"users" mapstructure:"users" json:"users"`
 
-	Temp     tempConfig     `toml:"temp" mapstructure:"temp"`
-	Log      logConfig      `toml:"log" mapstructure:"log"`
-	DB       dbConfig       `toml:"db" mapstructure:"db"`
-	Telegram telegramConfig `toml:"telegram" mapstructure:"telegram"`
-	Storage  storageConfig  `toml:"storage" mapstructure:"storage"`
+	Temp     tempConfig      `toml:"temp" mapstructure:"temp"`
+	Log      logConfig       `toml:"log" mapstructure:"log"`
+	DB       dbConfig        `toml:"db" mapstructure:"db"`
+	Telegram telegramConfig  `toml:"telegram" mapstructure:"telegram"`
+	Storages []StorageConfig `toml:"-" mapstructure:"-" json:"storages"`
+	// Deprecated
+	DeprecatedStorage deprecatedStorageConfig `toml:"storage" mapstructure:"storage"`
 }
 
 type tempConfig struct {
-	BasePath string `toml:"base_path" mapstructure:"base_path"`
-	CacheTTL int64  `toml:"cache_ttl" mapstructure:"cache_ttl"`
+	BasePath string `toml:"base_path" mapstructure:"base_path" json:"base_path"`
+	CacheTTL int64  `toml:"cache_ttl" mapstructure:"cache_ttl" json:"cache_ttl"`
 }
 
 type logConfig struct {
 	Level       string `toml:"level" mapstructure:"level"`
 	File        string `toml:"file" mapstructure:"file"`
-	BackupCount uint   `toml:"backup_count" mapstructure:"backup_count"`
+	BackupCount uint   `toml:"backup_count" mapstructure:"backup_count" json:"backup_count"`
 }
 
 type dbConfig struct {
@@ -37,10 +40,12 @@ type dbConfig struct {
 
 type telegramConfig struct {
 	Token   string      `toml:"token" mapstructure:"token"`
-	AppID   int         `toml:"app_id" mapstructure:"app_id"`
-	AppHash string      `toml:"app_hash" mapstructure:"app_hash"`
-	Admins  []int64     `toml:"admins" mapstructure:"admins"`
+	AppID   int         `toml:"app_id" mapstructure:"app_id" json:"app_id"`
+	AppHash string      `toml:"app_hash" mapstructure:"app_hash" json:"app_hash"`
 	Proxy   proxyConfig `toml:"proxy" mapstructure:"proxy"`
+
+	// Deprecated
+	Admins []int64 `toml:"admins" mapstructure:"admins"`
 }
 
 type proxyConfig struct {
@@ -48,37 +53,9 @@ type proxyConfig struct {
 	URL    string `toml:"url" mapstructure:"url"`
 }
 
-type storageConfig struct {
-	Alist  alistConfig  `toml:"alist" mapstructure:"alist"`
-	Local  localConfig  `toml:"local" mapstructure:"local"`
-	Webdav webdavConfig `toml:"webdav" mapstructure:"webdav"`
-}
-
-type alistConfig struct {
-	Enable   bool   `toml:"enable" mapstructure:"enable"`
-	URL      string `toml:"url" mapstructure:"url"`
-	Username string `toml:"username" mapstructure:"username"`
-	Password string `toml:"password" mapstructure:"password"`
-	BasePath string `toml:"base_path" mapstructure:"base_path"`
-	TokenExp int64  `toml:"token_exp" mapstructure:"token_exp"`
-}
-
-type localConfig struct {
-	Enable   bool   `toml:"enable" mapstructure:"enable"`
-	BasePath string `toml:"base_path" mapstructure:"base_path"`
-}
-
-type webdavConfig struct {
-	Enable   bool   `toml:"enable" mapstructure:"enable"`
-	URL      string `toml:"url" mapstructure:"url"`
-	Username string `toml:"username" mapstructure:"username"`
-	Password string `toml:"password" mapstructure:"password"`
-	BasePath string `toml:"base_path" mapstructure:"base_path"`
-}
-
 var Cfg *Config
 
-func Init() {
+func Init() error {
 	viper.SetConfigName("config")
 	viper.AddConfigPath(".")
 	viper.AddConfigPath("/etc/saveany/")
@@ -103,8 +80,11 @@ func Init() {
 
 	viper.SetDefault("db.path", "data/saveany.db")
 
-	viper.SetDefault("storage.alist.base_path", "/")
-	viper.SetDefault("storage.alist.token_exp", 3600)
+	if err := viper.SafeWriteConfigAs("config.toml"); err != nil {
+		if _, ok := err.(viper.ConfigFileAlreadyExistsError); !ok {
+			return fmt.Errorf("error saving default config: %w", err)
+		}
+	}
 
 	if err := viper.ReadInConfig(); err != nil {
 		fmt.Println("Error reading config file, ", err)
@@ -112,14 +92,62 @@ func Init() {
 	}
 
 	Cfg = &Config{}
+
 	if err := viper.Unmarshal(Cfg); err != nil {
 		fmt.Println("Error unmarshalling config file, ", err)
 		os.Exit(1)
 	}
-	if Cfg.Workers < 1 || Cfg.Retry < 1 {
-		fmt.Println("Invalid workers or retry value")
-		os.Exit(1)
+
+	if Cfg.Telegram.Admins != nil {
+		fmt.Println("警告: 你正在使用旧版 Telegram 管理员配置, 该配置下的用户将可用所有存储.\ntelegram.admins 未来版本将会被废弃, 请参考新的配置文件模板, 使用 users 配置替代.")
+		for _, admin := range Cfg.Telegram.Admins {
+			found := false
+			for _, user := range Cfg.Users {
+				if user.ID == admin {
+					found = true
+					break
+				}
+			}
+			if found {
+				continue
+			}
+			Cfg.Users = append(Cfg.Users, userConfig{
+				ID:        admin,
+				Storages:  []string{},
+				Blacklist: true,
+			})
+		}
 	}
+
+	storagesConfig, err := LoadStorageConfigs(viper.GetViper())
+	if err != nil {
+		return fmt.Errorf("error loading storage configs: %w", err)
+	}
+	Cfg.Storages = storagesConfig
+
+	if Cfg.DeprecatedStorage != (deprecatedStorageConfig{}) {
+		fmt.Println("\n警告: 你正在使用旧版存储配置, 未来版本将会被废弃.\n请参考新的配置文件模板.")
+		transformDeprecatedStorageConfig()
+	}
+
+	storageNames := make(map[string]struct{})
+	for _, storage := range Cfg.Storages {
+		if _, ok := storageNames[storage.GetName()]; ok {
+			return fmt.Errorf("重复的存储名: %s", storage.GetName())
+		}
+		storageNames[storage.GetName()] = struct{}{}
+	}
+
+	fmt.Printf("已加载 %d 个存储:\n", len(Cfg.Storages))
+	for _, storage := range Cfg.Storages {
+		fmt.Printf("  - %s (%s)\n", storage.GetName(), storage.GetType())
+	}
+
+	if Cfg.Workers < 1 || Cfg.Retry < 1 {
+		return fmt.Errorf("workers 和 retry 必须大于 0, 当前值: workers=%d, retry=%d", Cfg.Workers, Cfg.Retry)
+	}
+
+	return nil
 }
 
 func Set(key string, value any) {

core/core.go

@@ -10,6 +10,8 @@ import (
 	"path/filepath"
 	"time"
 
+	"github.com/gabriel-vasile/mimetype"
+
 	"github.com/celestix/gotgproto/ext"
 	"github.com/duke-git/lancet/v2/fileutil"
 	"github.com/gotd/td/tg"
@@ -17,34 +19,36 @@ import (
 	"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 {
 	logger.L.Debugf("Start processing task: %s", task.String())
+	if task.FileName() == "" {
+		task.File.FileName = fmt.Sprintf("%d_%d_%s", task.FileChatID, task.FileMessageID, task.File.Hash())
+	}
 	cacheDestPath := filepath.Join(config.Cfg.Temp.BasePath, task.FileName())
 	cacheDestPath, err := filepath.Abs(cacheDestPath)
 	if err != nil {
-		return fmt.Errorf("failed to get absolute path: %w", err)
+		return fmt.Errorf("处理路径失败: %w", err)
 	}
 	if err := fileutil.CreateDir(filepath.Dir(cacheDestPath)); err != nil {
-		return fmt.Errorf("failed to create directory: %w", err)
+		return fmt.Errorf("创建目录失败: %w", err)
 	}
 
 	if task.StoragePath == "" {
 		task.StoragePath = task.File.FileName
 	}
-	switch task.Storage {
-	case types.Local:
-		task.StoragePath = filepath.Join(config.Cfg.Storage.Local.BasePath, task.StoragePath)
-	case types.Webdav:
-		task.StoragePath = path.Join(config.Cfg.Storage.Webdav.BasePath, task.StoragePath)
-	case types.Alist:
-		task.StoragePath = path.Join(config.Cfg.Storage.Alist.BasePath, task.StoragePath)
+
+	taskStorage, err := storage.GetStorageByUserIDAndName(task.UserID, task.StorageName)
+	if err != nil {
+		return err
 	}
+	task.StoragePath = taskStorage.JoinStoragePath(*task)
 
 	if task.File.FileSize == 0 {
-		return processPhoto(task, cacheDestPath)
+		return processPhoto(task, taskStorage, cacheDestPath)
 	}
 
 	ctx := task.Ctx.(*ext.Context)
@@ -58,7 +62,7 @@ func processPendingTask(task *types.Task) error {
 			return
 		}
 		text, entities := buildProgressMessageEntity(task, barTotalCount, bytesRead, task.StartTime, progress)
-		ctx.EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
+		ctx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
 			Message:  text,
 			Entities: entities,
 			ID:       task.ReplyMessageID,
@@ -66,38 +70,47 @@ func processPendingTask(task *types.Task) error {
 	}
 
 	text, entities := buildProgressMessageEntity(task, barTotalCount, 0, task.StartTime, 0)
-	ctx.EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
+	ctx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
 		Message:  text,
 		Entities: entities,
 		ID:       task.ReplyMessageID,
 	})
+
 	readCloser, err := NewTelegramReader(task.Ctx, bot.Client, &task.File.Location,
 		0, task.File.FileSize-1, task.File.FileSize,
 		progressCallback, task.File.FileSize/100)
 	if err != nil {
-		return fmt.Errorf("failed to create reader: %w", err)
+		return fmt.Errorf("创建下载失败: %w", err)
 	}
 	defer readCloser.Close()
 
 	dest, err := os.Create(cacheDestPath)
 	if err != nil {
-		return fmt.Errorf("failed to create file: %w", err)
+		return fmt.Errorf("创建文件失败: %w", err)
 	}
 	defer dest.Close()
 	task.StartTime = time.Now()
 	if _, err := io.CopyN(dest, readCloser, task.File.FileSize); err != nil {
-		return fmt.Errorf("failed to download file: %w", err)
+		return fmt.Errorf("下载文件失败: %w", err)
+	}
+	defer cleanCacheFile(cacheDestPath)
+	if path.Ext(task.FileName()) == "" {
+		mimeType, err := mimetype.DetectFile(cacheDestPath)
+		if err != nil {
+			logger.L.Errorf("Failed to detect mime type: %s", err)
+		} else {
+			task.File.FileName = fmt.Sprintf("%s%s", task.FileName(), mimeType.Extension())
+			task.StoragePath = fmt.Sprintf("%s%s", task.StoragePath, mimeType.Extension())
+		}
 	}
 
-	defer cleanCacheFile(cacheDestPath)
-
 	logger.L.Infof("Downloaded file: %s", cacheDestPath)
-	ctx.EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
+	ctx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
 		Message: fmt.Sprintf("下载完成: %s\n正在转存文件...", task.FileName()),
 		ID:      task.ReplyMessageID,
 	})
 
-	return saveFileWithRetry(task, cacheDestPath)
+	return saveFileWithRetry(task, taskStorage, cacheDestPath)
 }
 
 func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
@@ -124,13 +137,13 @@ func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
 			queue.AddTask(task)
 		case types.Succeeded:
 			logger.L.Infof("Task succeeded: %s", task.String())
-			task.Ctx.(*ext.Context).EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
-				Message: fmt.Sprintf("文件保存成功\n [%s]: %s", task.Storage, task.StoragePath),
+			task.Ctx.(*ext.Context).EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
+				Message: fmt.Sprintf("文件保存成功\n [%s]: %s", task.StorageName, task.StoragePath),
 				ID:      task.ReplyMessageID,
 			})
 		case types.Failed:
 			logger.L.Errorf("Task failed: %s", task.String())
-			task.Ctx.(*ext.Context).EditMessage(task.ChatID, &tg.MessagesEditMessageRequest{
+			task.Ctx.(*ext.Context).EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
 				Message: "文件保存失败\n" + task.Error.Error(),
 				ID:      task.ReplyMessageID,
 			})

core/utils.go

@@ -16,9 +16,9 @@ import (
 	"github.com/krau/SaveAny-Bot/types"
 )
 
-func saveFileWithRetry(task *types.Task, localFilePath string) error {
+func saveFileWithRetry(task *types.Task, taskStorage storage.Storage, localFilePath string) error {
 	for i := 0; i <= config.Cfg.Retry; i++ {
-		if err := storage.Save(task.Storage, task.Ctx, localFilePath, task.StoragePath); err != nil {
+		if err := taskStorage.Save(task.Ctx, localFilePath, task.StoragePath); err != nil {
 			if i == config.Cfg.Retry {
 				return fmt.Errorf("failed to save file: %w", err)
 			}
@@ -30,7 +30,7 @@ func saveFileWithRetry(task *types.Task, localFilePath string) error {
 	return nil
 }
 
-func processPhoto(task *types.Task, cachePath string) error {
+func processPhoto(task *types.Task, taskStorage storage.Storage, cachePath string) error {
 	res, err := bot.Client.API().UploadGetFile(task.Ctx, &tg.UploadGetFileRequest{
 		Location: task.File.Location,
 		Offset:   0,
@@ -53,7 +53,7 @@ func processPhoto(task *types.Task, cachePath string) error {
 
 	logger.L.Infof("Downloaded file: %s", cachePath)
 
-	return saveFileWithRetry(task, cachePath)
+	return saveFileWithRetry(task, taskStorage, cachePath)
 }
 
 func getProgressBar(progress float64, totalCount int) string {
@@ -104,7 +104,7 @@ func buildProgressMessageEntity(task *types.Task, barTotalCount int, bytesRead i
 	entityBuilder := entity.Builder{}
 	text := fmt.Sprintf("正在处理下载任务\n文件名: %s\n保存路径: %s\n平均速度: %s\n当前进度: [%s] %.2f%%",
 		task.FileName(),
-		fmt.Sprintf("[%s]:%s", task.Storage, task.StoragePath),
+		fmt.Sprintf("[%s]:%s", task.StorageName, task.StoragePath),
 		getSpeed(bytesRead, startTime),
 		getProgressBar(progress, barTotalCount),
 		progress,
@@ -114,7 +114,7 @@ func buildProgressMessageEntity(task *types.Task, barTotalCount int, bytesRead i
 		styling.Plain("正在处理下载任务\n文件名: "),
 		styling.Code(task.FileName()),
 		styling.Plain("\n保存路径: "),
-		styling.Code(fmt.Sprintf("[%s]:%s", task.Storage, task.StoragePath)),
+		styling.Code(fmt.Sprintf("[%s]:%s", task.StorageName, task.StoragePath)),
 		styling.Plain("\n平均速度: "),
 		styling.Bold(getSpeed(bytesRead, task.StartTime)),
 		styling.Plain("\n当前进度:\n "),

dao/db.go

@@ -1,6 +1,7 @@
 package dao
 
 import (
+	"fmt"
 	"os"
 	"path/filepath"
 	"time"
@@ -16,7 +17,7 @@ import (
 var db *gorm.DB
 
 func Init() {
-	if err := os.MkdirAll(filepath.Dir(config.Cfg.DB.Path), 755); err != nil {
+	if err := os.MkdirAll(filepath.Dir(config.Cfg.DB.Path), 0755); err != nil {
 		logger.L.Fatal("Failed to create data directory: ", err)
 		os.Exit(1)
 	}
@@ -36,9 +37,48 @@ func Init() {
 		os.Exit(1)
 	}
 	logger.L.Debug("Database connected")
-	db.AutoMigrate(&types.ReceivedFile{}, &types.User{})
+	if err := db.AutoMigrate(&types.ReceivedFile{}, &types.User{}); err != nil {
+		logger.L.Fatal("迁移数据库失败, 如果您从旧版本升级, 建议手动删除数据库文件后重试: ", err)
+	}
 
-	for _, admin := range config.Cfg.Telegram.Admins {
-		CreateUser(int64(admin))
+	if err := syncUsers(); err != nil {
+		logger.L.Fatal("Failed to sync users:", err)
 	}
 }
+
+func syncUsers() error {
+	dbUsers, err := GetAllUsers()
+	if err != nil {
+		return fmt.Errorf("failed to get users: %w", err)
+	}
+
+	dbUserMap := make(map[int64]types.User)
+	for _, u := range dbUsers {
+		dbUserMap[u.ChatID] = u
+	}
+
+	cfgUserMap := make(map[int64]struct{})
+	for _, u := range config.Cfg.Users {
+		cfgUserMap[u.ID] = struct{}{}
+	}
+
+	for cfgID := range cfgUserMap {
+		if _, exists := dbUserMap[cfgID]; !exists {
+			if err := CreateUser(cfgID); err != nil {
+				return fmt.Errorf("failed to create user %d: %w", cfgID, err)
+			}
+			logger.L.Infof("创建用户: %d", cfgID)
+		}
+	}
+
+	for dbID, dbUser := range dbUserMap {
+		if _, exists := cfgUserMap[dbID]; !exists {
+			if err := DeleteUser(&dbUser); err != nil {
+				return fmt.Errorf("failed to delete user %d: %w", dbID, err)
+			}
+			logger.L.Infof("删除用户: %d", dbID)
+		}
+	}
+
+	return nil
+}

dao/file.go

@@ -20,5 +20,5 @@ func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*types.Rece
 }
 
 func DeleteReceivedFile(receivedFile *types.ReceivedFile) error {
-	return db.Delete(receivedFile).Error
+	return db.Unscoped().Delete(receivedFile).Error
 }

dao/user.go

@@ -4,19 +4,29 @@ import (
 	"github.com/krau/SaveAny-Bot/types"
 )
 
-func CreateUser(userID int64) error {
-	if _, err := GetUserByUserID(userID); err == nil {
+func CreateUser(chatID int64) error {
+	if _, err := GetUserByChatID(chatID); err == nil {
 		return nil
 	}
-	return db.Create(&types.User{UserID: userID}).Error
+	return db.Create(&types.User{ChatID: chatID}).Error
 }
 
-func GetUserByUserID(userID int64) (*types.User, error) {
+func GetAllUsers() ([]types.User, error) {
+	var users []types.User
+	err := db.Find(&users).Error
+	return users, err
+}
+
+func GetUserByChatID(chatID int64) (*types.User, error) {
 	var user types.User
-	err := db.Where("user_id = ?", userID).First(&user).Error
+	err := db.Where("chat_id = ?", chatID).First(&user).Error
 	return &user, err
 }
 
 func UpdateUser(user *types.User) error {
 	return db.Save(user).Error
 }
+
+func DeleteUser(user *types.User) error {
+	return db.Unscoped().Delete(user).Error
+}

docker-compose.yml

@@ -3,30 +3,8 @@ services:
     image: ghcr.io/krau/saveany-bot:latest
     container_name: saveany-bot
     restart: unless-stopped
-    environment:
-      - SAVEANY_TELEGRAM_TOKEN=bot_token
-      - SAVEANY_TELEGRAM_ADMINS=admin_id1,admin_id2
-      # 推荐使用自己的 API ID 和 API HASH (https://my.telegram.org)
-      # 若不配置将使用默认的 API ID 和 API HASH
-      # - SAVEANY_TELEGRAM_APP_ID=app_id
-      # - SAVEANY_TELEGRAM_APP_HASH=app_hash
-      
-      # 本地存储
-      - SAVEANY_STORAGE_LOCAL_ENABLE=true
-      - SAVEANY_STORAGE_LOCAL_BASE_PATH=/app/downloads
-      # Alist
-      - SAVEANY_STORAGE_ALIST_ENABLE=true
-      - SAVEANY_STORAGE_ALIST_BASE_PATH=/saveany
-      - SAVEANY_STORAGE_ALIST_URL=https://example.com
-      - SAVEANY_STORAGE_ALIST_USERNAME=username
-      - SAVEANY_STORAGE_ALIST_PASSWORD=password
-      # webdav
-      - SAVEANY_STORAGE_WEBDAV_ENABLE=true
-      - SAVEANY_STORAGE_WEBDAV_BASE_PATH=/saveany
-      - SAVEANY_STORAGE_WEBDAV_URL=https://example.com
-      - SAVEANY_STORAGE_WEBDAV_USERNAME=username
-      - SAVEANY_STORAGE_WEBDAV_PASSWORD=password
     volumes:
       - ./data:/app/data
+      - ./config.toml:/app/config.toml
       - ./downloads:/app/downloads
       - ./cache:/app/cache

go.mod

@@ -5,6 +5,7 @@ go 1.23.5
 require (
 	github.com/blang/semver v3.5.1+incompatible
 	github.com/celestix/gotgproto v1.0.0-beta20.2
+	github.com/gabriel-vasile/mimetype v1.4.8
 	github.com/gookit/slog v0.5.7
 	github.com/gotd/contrib v0.21.0
 	github.com/gotd/td v0.120.0
@@ -17,6 +18,7 @@ require (
 )
 
 require (
+	filippo.io/edwards25519 v1.1.0 // indirect
 	github.com/AnimeKaizoku/cacher v1.0.2 // indirect
 	github.com/cenkalti/backoff/v4 v4.3.0 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
@@ -30,6 +32,7 @@ require (
 	github.com/go-faster/jx v1.1.0 // indirect
 	github.com/go-faster/xor v1.0.0 // indirect
 	github.com/go-faster/yaml v0.4.6 // indirect
+	github.com/go-sql-driver/mysql v1.8.1 // 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-20250128161936-077ca0a936bf // indirect
@@ -59,6 +62,7 @@ require (
 	golang.org/x/oauth2 v0.26.0 // indirect
 	golang.org/x/tools v0.30.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
+	gorm.io/driver/mysql v1.5.6 // indirect
 	modernc.org/libc v1.61.13 // indirect
 	modernc.org/mathutil v1.7.1 // indirect
 	modernc.org/memory v1.8.2 // indirect
@@ -72,13 +76,13 @@ require (
 	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.18
+	github.com/gookit/goutil v0.6.18 // indirect
 	github.com/gookit/gsr v0.1.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/magiconair/properties v1.8.9 // indirect
-	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/mitchellh/mapstructure v1.5.0
 	github.com/pelletier/go-toml/v2 v2.2.3 // indirect
 	github.com/sagikazarmark/locafero v0.7.0 // indirect
 	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
@@ -96,5 +100,6 @@ require (
 	golang.org/x/text v0.22.0 // indirect
 	gopkg.in/ini.v1 v1.67.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
+	gorm.io/datatypes v1.2.5
 	gorm.io/gorm v1.25.12
 )

go.sum

@@ -1,9 +1,9 @@
+filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
+filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
 github.com/AnimeKaizoku/cacher v1.0.2 h1:7Bf5qRylWb7q2Evib0OXlhG37/t7BP2HK/7IyPvSmGQ=
 github.com/AnimeKaizoku/cacher v1.0.2/go.mod h1:jw0de/b0K6W7Y3T9rHCMGVKUf6oG7hENNcssxYcZTCc=
 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/celestix/gotgproto v1.0.0-beta20.1 h1:F7H08CuSiHP0YlZqATBi2wJvg7dxXFvFbpauWFd0IbI=
-github.com/celestix/gotgproto v1.0.0-beta20.1/go.mod h1:j42ZhBMUke6QyBLvCgx8tA+TL9L3+pq/Q46B+b5+3aU=
 github.com/celestix/gotgproto v1.0.0-beta20.2 h1:+WcsKdsyj4xy+TAV+4Sw6zp1xiQrIr4dMnM31+k8NYM=
 github.com/celestix/gotgproto v1.0.0-beta20.2/go.mod h1:j42ZhBMUke6QyBLvCgx8tA+TL9L3+pq/Q46B+b5+3aU=
 github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
@@ -19,8 +19,6 @@ github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46t
 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=
-github.com/dlclark/regexp2 v1.11.4 h1:rPYF9/LECdNymJufQKmri9gV604RvvABwgOA8un7yAo=
-github.com/dlclark/regexp2 v1.11.4/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
 github.com/dlclark/regexp2 v1.11.5 h1:Q/sSnsKerHeCkc/jSTNq1oCm7KiVgUMZRDUoRu0JQZQ=
 github.com/dlclark/regexp2 v1.11.5/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
 github.com/duke-git/lancet/v2 v2.3.4 h1:8XGI7P9w+/GqmEBEXYaH/XuNiM0f4/90Ioti0IvYJls=
@@ -34,6 +32,8 @@ github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7z
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
 github.com/fsnotify/fsnotify v1.8.0 h1:dAwr6QBTBZIkG8roQaJjGof0pp0EeF+tNV7YBP3F/8M=
 github.com/fsnotify/fsnotify v1.8.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
+github.com/gabriel-vasile/mimetype v1.4.8 h1:FfZ3gj38NjllZIeJAmMhr+qKL8Wu+nOoI3GqacKw1NM=
+github.com/gabriel-vasile/mimetype v1.4.8/go.mod h1:ByKUIKGjh1ODkGM1asKUbQZOLGrPjydw3hYPU2YU9t8=
 github.com/ghodss/yaml v1.0.0 h1:wQHKEahhL6wmXdzwWG11gIVCkOv05bNOh+Rxn0yngAk=
 github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=
 github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec+ruQ=
@@ -53,6 +53,13 @@ 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-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
 github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
+github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
+github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=
+github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
+github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9 h1:au07oEsX2xN0ktxqI+Sida1w446QrXBRJ0nee3SNZlA=
+github.com/golang-sql/civil v0.0.0-20220223132316-b832511892a9/go.mod h1:8vg3r2VgvsThLBIFL93Qb5yWzgyZWhEmBwUJWevAkK0=
+github.com/golang-sql/sqlexp v0.1.0 h1:ZCD6MBpcuOVfGVqsEmY5/4FtYiKz6tSyUv9LPEDei6A=
+github.com/golang-sql/sqlexp v0.1.0/go.mod h1:J4ad9Vo8ZCWQ2GMrC4UCQy1JpCbwU9m3EOqtpKwwwHI=
 github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
@@ -81,8 +88,6 @@ 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.118.0 h1:iPGkaOAd3QO72TcvzNJGKGpLDzYOW8GIz+Va2upxBbY=
-github.com/gotd/td v0.118.0/go.mod h1:FUNVeJB9Id2Vqps9yF+8kmBNNyCGO6VXDyO8Ah7bVSw=
 github.com/gotd/td v0.120.0 h1:XeiafJM82/9SaB+ZMjMm/dnUx5+avINwVZOEsnV0zMo=
 github.com/gotd/td v0.120.0/go.mod h1:BCc2jFj1l5zP9Trk4J7nxeqW0KBGl6K95eXMgszkbOI=
 github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
@@ -92,6 +97,14 @@ github.com/inconshreveable/go-update v0.0.0-20160112193335-8152e7eb6ccf h1:WfD7V
 github.com/inconshreveable/go-update v0.0.0-20160112193335-8152e7eb6ccf/go.mod h1:hyb9oH7vZsitZCiBt0ZvifOrB+qc8PS5IiilCIb87rg=
 github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
 github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
+github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
+github.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9 h1:L0QtFUgDarD7Fpv9jeVMgy/+Ec0mtnmYuImjTz6dtDA=
+github.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=
+github.com/jackc/pgx/v5 v5.5.5 h1:amBjrZVmksIdNjxGW/IiIMzxMKZFelXbUoPNb+8sjQw=
+github.com/jackc/pgx/v5 v5.5.5/go.mod h1:ez9gk+OAat140fv9ErkZDYFWmXLfV+++K0uAOiwgm1A=
+github.com/jackc/puddle/v2 v2.2.1 h1:RhxXJtFG022u4ibrCSMSiu5aOq1i77R3OHKNJj77OAk=
+github.com/jackc/puddle/v2 v2.2.1/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
 github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
 github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
 github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
@@ -111,12 +124,14 @@ github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHP
 github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
 github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
 github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-sqlite3 v1.14.17 h1:mCRHCLDUBXgpKAqIKsaAaAsrAlbkeomtRFKXh2L6YIM=
+github.com/mattn/go-sqlite3 v1.14.17/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
+github.com/microsoft/go-mssqldb v1.7.2 h1:CHkFJiObW7ItKTJfHo1QX7QBBD1iV+mn1eOyRP3b/PA=
+github.com/microsoft/go-mssqldb v1.7.2/go.mod h1:kOvZKUdrhhFQmxLZqbwUV0rHkNkZpthMITIb2Ko1IoA=
 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/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
 github.com/ncruces/go-strftime v0.1.9/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
-github.com/ogen-go/ogen v1.9.0 h1:n+lDQpiSFYC9G4hTvuNVWnqmIP0LR8ws0faDn9jX3hU=
-github.com/ogen-go/ogen v1.9.0/go.mod h1:vkHpuRyzjdfuRCy81EShi4t9sIgZDcNPGmiDKipRloc=
 github.com/ogen-go/ogen v1.10.0 h1:x3ukRtq/pdn/k8+pYBtqWceVASiSmgK9M5lrH89Q+04=
 github.com/ogen-go/ogen v1.10.0/go.mod h1:WExXrswerPzGWD0NpzBFsz+5eQIbP7HAtZUmpV8dqqI=
 github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
@@ -188,63 +203,42 @@ go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8=
 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.32.0 h1:euUpcYgM8WcP71gNpTqQCn6rC2t6ULUPiOzfWaXVVfc=
-golang.org/x/crypto v0.32.0/go.mod h1:ZnnJkOaASj8g0AjIduWNlq2NRxL0PlBrbKVyZ6V/Ugc=
 golang.org/x/crypto v0.33.0 h1:IOBPskki6Lysi0lo9qQvbxiQ+FvsCC/YWOecCHAixus=
 golang.org/x/crypto v0.33.0/go.mod h1:bVdXmD7IV/4GdElGPozy6U7lWdRXA4qyRVGJV57uQ5M=
-golang.org/x/exp v0.0.0-20250128182459-e0ece0dbea4c h1:KL/ZBHXgKGVmuZBZ01Lt57yE5ws8ZPSkkihmEyq7FXc=
-golang.org/x/exp v0.0.0-20250128182459-e0ece0dbea4c/go.mod h1:tujkw807nyEEAamNbDrEGzRav+ilXA7PCRAd6xsmwiU=
 golang.org/x/exp v0.0.0-20250210185358-939b2ce775ac h1:l5+whBCLH3iH2ZNHYLbAe58bo7yrN4mVcnkHDYz5vvs=
 golang.org/x/exp v0.0.0-20250210185358-939b2ce775ac/go.mod h1:hH+7mtFmImwwcMvScyxUhjuVHR3HGaDPMn9rMSUUbxo=
-golang.org/x/mod v0.22.0 h1:D4nJWe9zXqHOmWqj4VMOJhvzj7bEZg4wEYa759z1pH4=
-golang.org/x/mod v0.22.0/go.mod h1:6SkKJ3Xj0I0BrPOZoBy3bdMptDDU9oJrpohJ3eWZ1fY=
 golang.org/x/mod v0.23.0 h1:Zb7khfcRGKk+kqfxFaP5tZqCnDZMjC5VtUBs87Hr6QM=
 golang.org/x/mod v0.23.0/go.mod h1:6SkKJ3Xj0I0BrPOZoBy3bdMptDDU9oJrpohJ3eWZ1fY=
 golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
 golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
 golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
-golang.org/x/net v0.34.0 h1:Mb7Mrk043xzHgnRM88suvJFwzVrRfHEHJEl5/71CKw0=
-golang.org/x/net v0.34.0/go.mod h1:di0qlW3YNM5oh6GqDGQr92MyTozJPmybPK4Ev/Gm31k=
 golang.org/x/net v0.35.0 h1:T5GQRQb2y08kTAByq9L4/bz8cipCdA8FbRTXewonqY8=
 golang.org/x/net v0.35.0/go.mod h1:EglIi67kWsHKlRzzVMUD93VMSWGFOMSZgxFjparz1Qk=
 golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
 golang.org/x/oauth2 v0.0.0-20181106182150-f42d05182288/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
-golang.org/x/oauth2 v0.25.0 h1:CY4y7XT9v0cRI9oupztF8AgiIu99L/ksR/Xp/6jrZ70=
-golang.org/x/oauth2 v0.25.0/go.mod h1:XYTD2NtWslqkgxebSiOHnXEap4TF09sJSc7H1sXbhtI=
 golang.org/x/oauth2 v0.26.0 h1:afQXWNNaeC4nvZ0Ed9XvCCzXM6UHJG7iCg0W4fPqSBE=
 golang.org/x/oauth2 v0.26.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.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ=
-golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 golang.org/x/sync v0.11.0 h1:GGz8+XQP4FvTTrjZPzNKTMFtSXH80RAzG+5ghFPgK9w=
 golang.org/x/sync v0.11.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.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU=
-golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc=
 golang.org/x/sys v0.30.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.28.0 h1:/Ts8HFuMR2E6IP/jlo7QVLZHggjKQbhu/7H0LJFr3Gg=
-golang.org/x/term v0.28.0/go.mod h1:Sw/lC2IAUZ92udQNf3WodGtn4k/XoLyZoh8v/8uiwek=
 golang.org/x/term v0.29.0 h1:L6pJp37ocefwRRtYPKSWOWzOtWSxVajvz2ldH/xi3iU=
+golang.org/x/term v0.29.0/go.mod h1:6bl4lRlvVuDgSf3179VpIxBF0o10JUpXWOnI7nErv7s=
 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.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
-golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
 golang.org/x/text v0.22.0 h1:bofq7m3/HAFvbF51jz3Q9wLg3jkvSPuiZu/pD1XwgtM=
 golang.org/x/text v0.22.0/go.mod h1:YRoo4H8PVmsu+E3Ou7cqLVH8oXWIHVoX0jqUWALQhfY=
-golang.org/x/time v0.9.0 h1:EsRrnYcQiGH+5FfbgvV4AP7qEZstoyrHB0DzarOQ4ZY=
-golang.org/x/time v0.9.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
 golang.org/x/time v0.10.0 h1:3usCWA8tQn0L8+hFJQNgzpWbd89begxN66o1Ojdn5L4=
 golang.org/x/time v0.10.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
-golang.org/x/tools v0.29.0 h1:Xx0h3TtM9rzQpQuR4dKLrdglAmCEN5Oi+P74JdhdzXE=
-golang.org/x/tools v0.29.0/go.mod h1:KMQVMRsVxU6nHCFXrBPhDB8XncLNLM0lIy/F14RP588=
 golang.org/x/tools v0.30.0 h1:BgcpHewrV5AUp2G9MebG4XPFI1E2W41zU1SaqVA9vJY=
 golang.org/x/tools v0.30.0/go.mod h1:c347cR/OJfw5TI+GfX7RUPNMdDRRbjvYTS0jPyvsVtY=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
@@ -263,20 +257,27 @@ gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
 gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
 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/datatypes v1.2.5 h1:9UogU3jkydFVW1bIVVeoYsTpLRgwDVW3rHfJG6/Ek9I=
+gorm.io/datatypes v1.2.5/go.mod h1:I5FUdlKpLb5PMqeMQhm30CQ6jXP8Rj89xkTeCSAaAD4=
+gorm.io/driver/mysql v1.5.6 h1:Ld4mkIickM+EliaQZQx3uOJDJHtrd70MxAUqWqlx3Y8=
+gorm.io/driver/mysql v1.5.6/go.mod h1:sEtPWMiqiN1N1cMXoXmBbd8C6/l+TESwriotuRRpkDM=
+gorm.io/driver/postgres v1.5.0 h1:u2FXTy14l45qc3UeCJ7QaAXZmZfDDv0YrthvmRq1l0U=
+gorm.io/driver/postgres v1.5.0/go.mod h1:FUZXzO+5Uqg5zzwzv4KK49R8lvGIyscBOqYrtI1Ce9A=
+gorm.io/driver/sqlite v1.5.5 h1:7MDMtUZhV065SilG62E0MquljeArQZNfJnjd9i9gx3E=
+gorm.io/driver/sqlite v1.5.5/go.mod h1:6NgQ7sQWAIFsPrJJl1lSNSu2TABh0ZZ/zm5fosATavE=
+gorm.io/driver/sqlserver v1.5.4 h1:xA+Y1KDNspv79q43bPyjDMUgHoYHLhXYmdFcYPobg8g=
+gorm.io/driver/sqlserver v1.5.4/go.mod h1:+frZ/qYmuna11zHPlh5oc2O6ZA/lS88Keb0XSH1Zh/g=
+gorm.io/gorm v1.25.7/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=
 gorm.io/gorm v1.25.12 h1:I0u8i2hWQItBq1WfE0o2+WuL9+8L21K9e2HHSTE/0f8=
 gorm.io/gorm v1.25.12/go.mod h1:xh7N7RHfYlNc5EmcI/El95gXusucDrQnHXe0+CgWcLQ=
 modernc.org/cc/v4 v4.24.4 h1:TFkx1s6dCkQpd6dKurBNmpo+G8Zl4Sq/ztJ+2+DEsh0=
 modernc.org/cc/v4 v4.24.4/go.mod h1:uVtb5OGqUKpoLWhqwNQo/8LwvoiEBLvZXIQ/SmO6mL0=
-modernc.org/ccgo/v4 v4.23.15 h1:wFDan71KnYqeHz4eF63vmGE6Q6Pc0PUGDpP0PRMYjDc=
-modernc.org/ccgo/v4 v4.23.15/go.mod h1:nJX30dks/IWuBOnVa7VRii9Me4/9TZ1SC9GNtmARTy0=
 modernc.org/ccgo/v4 v4.23.16 h1:Z2N+kk38b7SfySC1ZkpGLN2vthNJP1+ZzGZIlH7uBxo=
+modernc.org/ccgo/v4 v4.23.16/go.mod h1:nNma8goMTY7aQZQNTyN9AIoJfxav4nvTnvKThAeMDdo=
 modernc.org/fileutil v1.3.0 h1:gQ5SIzK3H9kdfai/5x41oQiKValumqNTDXMvKo62HvE=
 modernc.org/fileutil v1.3.0/go.mod h1:XatxS8fZi3pS8/hKG2GH/ArUogfxjpEKs3Ku3aK4JyQ=
-modernc.org/gc/v2 v2.6.2 h1:YBXi5Kqp6aCK3fIxwKQ3/fErvawVKwjOLItxj1brGds=
-modernc.org/gc/v2 v2.6.2/go.mod h1:YgIahr1ypgfe7chRuJi2gD7DBQiKSLMPgBQe9oIiito=
 modernc.org/gc/v2 v2.6.3 h1:aJVhcqAte49LF+mGveZ5KPlsp4tdGdAOT4sipJXADjw=
-modernc.org/libc v1.61.11 h1:6sZG8uB6EMMG7iTLPTndi8jyTdgAQNIeLGjCFICACZw=
-modernc.org/libc v1.61.11/go.mod h1:HHX+srFdn839oaJRd0W8hBM3eg+mieyZCAjWwB08/nM=
+modernc.org/gc/v2 v2.6.3/go.mod h1:YgIahr1ypgfe7chRuJi2gD7DBQiKSLMPgBQe9oIiito=
 modernc.org/libc v1.61.13 h1:3LRd6ZO1ezsFiX1y+bHd1ipyEHIJKvuprv0sLTBwLW8=
 modernc.org/libc v1.61.13/go.mod h1:8F/uJWL/3nNil0Lgt1Dpz+GgkApWh04N3el3hxJcA6E=
 modernc.org/mathutil v1.7.1 h1:GCZVGXdaN8gTqB1Mf/usp1Y/hSqgI2vAGGP4jZMCxOU=
@@ -287,8 +288,6 @@ modernc.org/opt v0.1.4 h1:2kNGMRiUjrp4LcaPuLY2PzUfqM/w9N23quVwhKt5Qm8=
 modernc.org/opt v0.1.4/go.mod h1:03fq9lsNfvkYSfxrfUhZCWPk1lm4cq4N+Bh//bEtgns=
 modernc.org/sortutil v1.2.1 h1:+xyoGf15mM3NMlPDnFqrteY07klSFxLElE2PVuWIJ7w=
 modernc.org/sortutil v1.2.1/go.mod h1:7ZI3a3REbai7gzCLcotuw9AC4VZVpYMjDzETGsSMqJE=
-modernc.org/sqlite v1.34.5 h1:Bb6SR13/fjp15jt70CL4f18JIN7p7dnMExd+UFnF15g=
-modernc.org/sqlite v1.34.5/go.mod h1:YLuNmX9NKs8wRNK2ko1LW1NGYcc9FkBO69JOt1AR9JE=
 modernc.org/sqlite v1.35.0 h1:yQps4fegMnZFdphtzlfQTCNBWtS0CZv48pRpW3RFHRw=
 modernc.org/sqlite v1.35.0/go.mod h1:9cr2sicr7jIaWTBKQmAxQLfBv9LL0su4ZTEV+utt3ic=
 modernc.org/strutil v1.2.1 h1:UneZBkQA+DX2Rp35KcM69cSsNES9ly8mQWD71HKlOA0=

storage/alist/alist.go

@@ -1,19 +1,19 @@
 package alist
 
 import (
-	"bytes"
 	"context"
 	"encoding/json"
-	"errors"
 	"fmt"
 	"io"
 	"net/http"
 	"net/url"
 	"os"
+	"path"
 	"time"
 
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/types"
 )
 
 type Alist struct {
@@ -21,107 +21,80 @@ type Alist struct {
 	token     string
 	baseURL   string
 	loginInfo *loginRequest
+	config    config.AlistStorageConfig
 }
 
-var (
-	ErrAlistLoginFailed = errors.New("failed to login to Alist")
-)
-
-type loginRequest struct {
-	Username string `json:"username"`
-	Password string `json:"password"`
-}
-
-type loginResponse struct {
-	Code    int    `json:"code"`
-	Message string `json:"message"`
-	Data    struct {
-		Token string `json:"token"`
-	} `json:"data"`
-}
-
-type putResponse struct {
-	Code    int    `json:"code"`
-	Message string `json:"message"`
-	Data    struct {
-		Task struct {
-			ID       string `json:"id"`
-			Name     string `json:"name"`
-			State    int    `json:"state"`
-			Status   string `json:"status"`
-			Progress int    `json:"progress"`
-			Error    string `json:"error"`
-		} `json:"task"`
-	} `json:"data"`
-}
-
-func (a *Alist) getToken() error {
-	loginBody, err := json.Marshal(a.loginInfo)
-	if err != nil {
-		return fmt.Errorf("failed to marshal login request: %w", err)
+func (a *Alist) Init(cfg config.StorageConfig) error {
+	alistConfig, ok := cfg.(*config.AlistStorageConfig)
+	if !ok {
+		return fmt.Errorf("failed to cast alist config")
 	}
-
-	req, err := http.NewRequest(http.MethodPost, a.baseURL+"/api/auth/login", bytes.NewBuffer(loginBody))
-	if err != nil {
-		return fmt.Errorf("failed to create login request: %w", err)
+	if err := alistConfig.Validate(); err != nil {
+		return err
 	}
-	req.Header.Set("Content-Type", "application/json")
+	a.config = *alistConfig
 
-	resp, err := a.client.Do(req)
-	if err != nil {
-		return fmt.Errorf("failed to send login request: %w", err)
-	}
-	defer resp.Body.Close()
-
-	body, err := io.ReadAll(resp.Body)
-	if err != nil {
-		return fmt.Errorf("failed to read login response: %w", err)
-	}
-
-	var loginResp loginResponse
-	if err := json.Unmarshal(body, &loginResp); err != nil {
-		return fmt.Errorf("failed to unmarshal login response: %w", err)
-	}
-
-	if loginResp.Code != http.StatusOK {
-		return fmt.Errorf("%w: %s", ErrAlistLoginFailed, loginResp.Message)
-	}
-
-	a.token = loginResp.Data.Token
-	return nil
-}
-
-func (a *Alist) refreshToken() {
-	for {
-		time.Sleep(time.Duration(config.Cfg.Storage.Alist.TokenExp) * time.Second)
-		if err := a.getToken(); err != nil {
-			logger.L.Errorf("Failed to refresh jwt token: %v", err)
-			continue
+	a.baseURL = alistConfig.URL
+	a.client = getHttpClient()
+	if alistConfig.Token != "" {
+		a.token = alistConfig.Token
+		ctx, cancel := context.WithTimeout(context.Background(), 1*time.Minute)
+		defer cancel()
+		req, err := http.NewRequestWithContext(ctx, http.MethodGet, a.baseURL+"/api/me", nil)
+		if err != nil {
+			logger.L.Fatalf("Failed to create request: %v", err)
+			return err
 		}
-		logger.L.Info("Refreshed Alist jwt token")
-	}
-}
+		req.Header.Set("Authorization", a.token)
 
-func (a *Alist) Init() {
-	a.baseURL = config.Cfg.Storage.Alist.URL
-	a.client = &http.Client{
-		Timeout: 12 * time.Hour,
-		Transport: &http.Transport{
-			TLSHandshakeTimeout: 10 * time.Second,
-		},
+		resp, err := a.client.Do(req)
+		if err != nil {
+			logger.L.Fatalf("Failed to send request: %v", err)
+			return err
+		}
+		defer resp.Body.Close()
+		if resp.StatusCode != http.StatusOK {
+			logger.L.Fatalf("Failed to get alist user info: %s", resp.Status)
+			return err
+		}
+		body, err := io.ReadAll(resp.Body)
+		if err != nil {
+			logger.L.Fatalf("Failed to read response body: %v", err)
+			return err
+		}
+		var meResp meResponse
+		if err := json.Unmarshal(body, &meResp); err != nil {
+			logger.L.Fatalf("Failed to unmarshal me response: %v", err)
+			return err
+		}
+		if meResp.Code != http.StatusOK {
+			logger.L.Fatalf("Failed to get alist user info: %s", meResp.Message)
+			return err
+		}
+		logger.L.Debugf("Logged in Alist as %s", meResp.Data.Username)
+		return nil
 	}
 	a.loginInfo = &loginRequest{
-		Username: config.Cfg.Storage.Alist.Username,
-		Password: config.Cfg.Storage.Alist.Password,
+		Username: alistConfig.Username,
+		Password: alistConfig.Password,
 	}
 
 	if err := a.getToken(); err != nil {
 		logger.L.Fatalf("Failed to login to Alist: %v", err)
-		os.Exit(1)
+		return err
 	}
 	logger.L.Debug("Logged in to Alist")
 
-	go a.refreshToken()
+	go a.refreshToken(*alistConfig)
+	return nil
+}
+
+func (a *Alist) Type() types.StorageType {
+	return types.StorageTypeAlist
+}
+
+func (a *Alist) Name() string {
+	return a.config.Name
 }
 
 func (a *Alist) Save(ctx context.Context, filePath, storagePath string) error {
@@ -172,3 +145,7 @@ func (a *Alist) Save(ctx context.Context, filePath, storagePath string) error {
 
 	return nil
 }
+
+func (a *Alist) JoinStoragePath(task types.Task) string {
+	return path.Join(a.config.BasePath, task.StoragePath)
+}

storage/alist/token.go

@@ -0,0 +1,65 @@
+package alist
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"time"
+
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/logger"
+)
+
+func (a *Alist) getToken() error {
+	loginBody, err := json.Marshal(a.loginInfo)
+	if err != nil {
+		return fmt.Errorf("failed to marshal login request: %w", err)
+	}
+
+	req, err := http.NewRequest(http.MethodPost, a.baseURL+"/api/auth/login", bytes.NewBuffer(loginBody))
+	if err != nil {
+		return fmt.Errorf("failed to create login request: %w", err)
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	resp, err := a.client.Do(req)
+	if err != nil {
+		return fmt.Errorf("failed to send login request: %w", err)
+	}
+	defer resp.Body.Close()
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return fmt.Errorf("failed to read login response: %w", err)
+	}
+
+	var loginResp loginResponse
+	if err := json.Unmarshal(body, &loginResp); err != nil {
+		return fmt.Errorf("failed to unmarshal login response: %w", err)
+	}
+
+	if loginResp.Code != http.StatusOK {
+		return fmt.Errorf("%w: %s", ErrAlistLoginFailed, loginResp.Message)
+	}
+
+	a.token = loginResp.Data.Token
+	return nil
+}
+
+func (a *Alist) refreshToken(cfg config.AlistStorageConfig) {
+	tokenExp := cfg.TokenExp
+	if tokenExp <= 0 {
+		logger.L.Warn("Invalid token expiration time, using default value")
+		tokenExp = 3600
+	}
+	for {
+		time.Sleep(time.Duration(tokenExp) * time.Second)
+		if err := a.getToken(); err != nil {
+			logger.L.Errorf("Failed to refresh jwt token: %v", err)
+			continue
+		}
+		logger.L.Info("Refreshed Alist jwt token")
+	}
+}

storage/alist/types.go

@@ -0,0 +1,44 @@
+package alist
+
+import "errors"
+
+var (
+	ErrAlistLoginFailed = errors.New("failed to login to Alist")
+)
+
+type loginRequest struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
+type loginResponse struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Data    struct {
+		Token string `json:"token"`
+	} `json:"data"`
+}
+
+type meResponse struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Data    struct {
+		ID       int    `json:"id"`
+		Username string `json:"username"`
+	} `json:"data"`
+}
+
+type putResponse struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Data    struct {
+		Task struct {
+			ID       string `json:"id"`
+			Name     string `json:"name"`
+			State    int    `json:"state"`
+			Status   string `json:"status"`
+			Progress int    `json:"progress"`
+			Error    string `json:"error"`
+		} `json:"task"`
+	} `json:"data"`
+}

storage/alist/utils.go

@@ -0,0 +1,23 @@
+package alist
+
+import (
+	"net/http"
+	"time"
+)
+
+var (
+	httpClient *http.Client
+)
+
+func getHttpClient() *http.Client {
+	if httpClient != nil {
+		return httpClient
+	}
+	httpClient = &http.Client{
+		Timeout: 12 * time.Hour,
+		Transport: &http.Transport{
+			TLSHandshakeTimeout: 10 * time.Second,
+		},
+	}
+	return httpClient
+}

storage/errs.go

@@ -0,0 +1,9 @@
+package storage
+
+import (
+	"errors"
+)
+
+var (
+	ErrStorageNameEmpty = errors.New("storage name is empty")
+)

storage/local/local.go

@@ -2,22 +2,41 @@ package local
 
 import (
 	"context"
+	"fmt"
 	"os"
 	"path/filepath"
 
 	"github.com/duke-git/lancet/v2/fileutil"
 	"github.com/krau/SaveAny-Bot/config"
-	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/types"
 )
 
-type Local struct{}
+type Local struct {
+	config config.LocalStorageConfig
+}
 
-func (l *Local) Init() {
-	err := os.MkdirAll(config.Cfg.Storage.Local.BasePath, os.ModePerm)
-	if err != nil {
-		logger.L.Fatalf("Failed to create local storage directory: %s", err)
-		os.Exit(1)
+func (l *Local) Init(cfg config.StorageConfig) error {
+	localConfig, ok := cfg.(*config.LocalStorageConfig)
+	if !ok {
+		return fmt.Errorf("failed to cast local config")
 	}
+	if err := localConfig.Validate(); err != nil {
+		return err
+	}
+	l.config = *localConfig
+	err := os.MkdirAll(localConfig.BasePath, os.ModePerm)
+	if err != nil {
+		return fmt.Errorf("failed to create local storage directory: %w", err)
+	}
+	return nil
+}
+
+func (l *Local) Type() types.StorageType {
+	return types.StorageTypeLocal
+}
+
+func (l *Local) Name() string {
+	return l.config.Name
 }
 
 func (l *Local) Save(ctx context.Context, filePath, storagePath string) error {
@@ -30,3 +49,7 @@ func (l *Local) Save(ctx context.Context, filePath, storagePath string) error {
 	}
 	return fileutil.CopyFile(filePath, storagePath)
 }
+
+func (l *Local) JoinStoragePath(task types.Task) string {
+	return filepath.Join(l.config.BasePath, task.StoragePath)
+}

storage/storage.go

@@ -2,12 +2,8 @@ package storage
 
 import (
 	"context"
-	"errors"
-	"path"
-	"path/filepath"
-	"sync"
+	"fmt"
 
-	"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"
@@ -17,68 +13,92 @@ import (
 )
 
 type Storage interface {
-	Init()
+	Init(cfg config.StorageConfig) error
+	Type() types.StorageType
+	Name() string
+	JoinStoragePath(task types.Task) string
 	Save(cttx context.Context, localFilePath, storagePath string) error
 }
 
-var Storages = make(map[types.StorageType]Storage)
-var StorageKeys = make([]types.StorageType, 0)
+var Storages = make(map[string]Storage)
 
-func Init() {
-	logger.L.Debug("Initializing storage...")
-	if config.Cfg.Storage.Alist.Enable {
-		Storages[types.Alist] = new(alist.Alist)
-		Storages[types.Alist].Init()
-	}
-	if config.Cfg.Storage.Local.Enable {
-		Storages[types.Local] = new(local.Local)
-		Storages[types.Local].Init()
-	}
-	if config.Cfg.Storage.Webdav.Enable {
-		Storages[types.Webdav] = new(webdav.Webdav)
-		Storages[types.Webdav].Init()
+// GetStorageByName returns storage by name from cache or creates new one
+func GetStorageByName(name string) (Storage, error) {
+	if name == "" {
+		return nil, ErrStorageNameEmpty
 	}
 
-	for k := range Storages {
-		StorageKeys = append(StorageKeys, k)
+	storage, ok := Storages[name]
+	if ok {
+		return storage, nil
+	}
+	cfg := config.Cfg.GetStorageByName(name)
+	if cfg == nil {
+		return nil, fmt.Errorf("未找到存储 %s", name)
 	}
 
-	slice.Sort(StorageKeys)
-
-	logger.L.Debug("Storage initialized")
+	storage, err := NewStorage(cfg)
+	if err != nil {
+		return nil, err
+	}
+	Storages[name] = storage
+	return storage, nil
 }
 
-func Save(storageType types.StorageType, ctx context.Context, filePath, storagePath string) error {
-	logger.L.Debugf("Saving file %s to storage: [%s] %s", filePath, storageType, storagePath)
-	if ctx == nil {
-		ctx = context.Background()
+// 检查 user 是否可用指定的 storage, 若不可用则返回未找到错误
+func GetStorageByUserIDAndName(chatID int64, name string) (Storage, error) {
+	if name == "" {
+		return nil, ErrStorageNameEmpty
 	}
-	if storageType != types.StorageAll {
-		return Storages[storageType].Save(ctx, filePath, storagePath)
+
+	if !config.Cfg.HasStorage(chatID, name) {
+		return nil, fmt.Errorf("没有找到用户 %d 的存储 %s", chatID, name)
 	}
-	errs := make([]error, 0)
-	var wg sync.WaitGroup
-	for _, storage := range Storages {
-		wg.Add(1)
-		go func(storage Storage) {
-			defer wg.Done()
-			storageDestPath := storagePath
-			switch storage.(type) {
-			case *local.Local:
-				storageDestPath = filepath.Join(config.Cfg.Storage.Local.BasePath, storagePath)
-			case *webdav.Webdav:
-				storageDestPath = path.Join(config.Cfg.Storage.Webdav.BasePath, storagePath)
-			case *alist.Alist:
-				storageDestPath = path.Join(config.Cfg.Storage.Alist.BasePath, storagePath)
-			}
-			if err := storage.Save(ctx, filePath, storageDestPath); err != nil {
-				errs = append(errs, err)
-			}
-		}(storage)
-	}
-	wg.Wait()
-	if len(errs) > 0 {
-		return errors.Join(errs...)
-	}
-	return nil
+
+	return GetStorageByName(name)
+}
+
+func GetUserStorages(chatID int64) []Storage {
+	var storages []Storage
+	for _, name := range config.Cfg.GetStorageNamesByUserID(chatID) {
+		storage, err := GetStorageByName(name)
+		if err != nil {
+			continue
+		}
+		storages = append(storages, storage)
+	}
+	return storages
+}
+
+type StorageConstructor func() Storage
+
+var storageConstructors = map[string]StorageConstructor{
+	string(types.StorageTypeAlist):  func() Storage { return new(alist.Alist) },
+	string(types.StorageTypeLocal):  func() Storage { return new(local.Local) },
+	string(types.StorageTypeWebdav): func() Storage { return new(webdav.Webdav) },
+}
+
+func NewStorage(cfg config.StorageConfig) (Storage, error) {
+	constructor, ok := storageConstructors[string(cfg.GetType())]
+	if !ok {
+		return nil, fmt.Errorf("不支持的存储类型: %s", cfg.GetType())
+	}
+
+	storage := constructor()
+	if err := storage.Init(cfg); err != nil {
+		return nil, fmt.Errorf("初始化 %s 存储失败: %w", cfg.GetName(), err)
+	}
+
+	return storage, nil
+}
+
+func LoadStorages() {
+	logger.L.Info("加载存储...")
+	for _, storage := range config.Cfg.Storages {
+		_, err := GetStorageByName(storage.GetName())
+		if err != nil {
+			logger.L.Errorf("加载存储 %s 失败: %v", storage.GetName(), err)
+		}
+	}
+	logger.L.Infof("成功加载 %d 个存储", len(Storages))
 }

storage/webdav/webdav.go

@@ -2,33 +2,50 @@ package webdav
 
 import (
 	"context"
+	"fmt"
 	"os"
 	"path"
 	"time"
 
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/types"
 	"github.com/studio-b12/gowebdav"
 )
 
-type Webdav struct{}
+type Webdav struct {
+	config config.WebdavStorageConfig
+	client *gowebdav.Client
+}
 
-var (
-	Client *gowebdav.Client
-)
-
-func (w *Webdav) Init() {
-	webdavConfig := config.Cfg.Storage.Webdav
-	Client = gowebdav.NewClient(webdavConfig.URL, webdavConfig.Username, webdavConfig.Password)
-	if err := Client.Connect(); err != nil {
-		logger.L.Fatalf("Failed to connect to webdav server: %v", err)
-		os.Exit(1)
+func (w *Webdav) Init(cfg config.StorageConfig) error {
+	webdavConfig, ok := cfg.(*config.WebdavStorageConfig)
+	if !ok {
+		return fmt.Errorf("failed to cast webdav config")
 	}
-	Client.SetTimeout(24 * time.Hour)
+	if err := webdavConfig.Validate(); err != nil {
+		return err
+	}
+	w.config = *webdavConfig
+	client := gowebdav.NewClient(webdavConfig.URL, webdavConfig.Username, webdavConfig.Password)
+	if err := client.Connect(); err != nil {
+		return fmt.Errorf("failed to connect to webdav server: %w", err)
+	}
+	client.SetTimeout(12 * time.Hour)
+	w.client = client
+	return nil
+}
+
+func (w *Webdav) Type() types.StorageType {
+	return types.StorageTypeWebdav
+}
+
+func (w *Webdav) Name() string {
+	return w.config.Name
 }
 
 func (w *Webdav) Save(ctx context.Context, filePath, storagePath string) error {
-	if err := Client.MkdirAll(path.Dir(storagePath), os.ModePerm); err != nil {
+	if err := w.client.MkdirAll(path.Dir(storagePath), os.ModePerm); err != nil {
 		logger.L.Errorf("Failed to create directory %s: %v", path.Dir(storagePath), err)
 		return ErrFailedToCreateDirectory
 	}
@@ -39,9 +56,13 @@ func (w *Webdav) Save(ctx context.Context, filePath, storagePath string) error {
 	}
 	defer file.Close()
 
-	if err := Client.WriteStream(storagePath, file, os.ModePerm); err != nil {
+	if err := w.client.WriteStream(storagePath, file, os.ModePerm); err != nil {
 		logger.L.Errorf("Failed to write file %s: %v", storagePath, err)
 		return ErrFailedToWriteFile
 	}
 	return nil
 }
+
+func (w *Webdav) JoinStoragePath(task types.Task) string {
+	return path.Join(w.config.BasePath, task.StoragePath)
+}

types/model.go

@@ -6,16 +6,19 @@ import (
 
 type ReceivedFile struct {
 	gorm.Model
-	Processing     bool
-	ChatID         int64 `gorm:"uniqueIndex:idx_chat_id_message_id;not null"`
-	MessageID      int   `gorm:"uniqueIndex:idx_chat_id_message_id;not null"`
+	Processing bool
+	// Which chat the file is from
+	ChatID int64 `gorm:"uniqueIndex:idx_chat_id_message_id;not null"`
+	// Which message the file is from
+	MessageID      int `gorm:"uniqueIndex:idx_chat_id_message_id;not null"`
 	ReplyMessageID int
+	ReplyChatID    int64
 	FileName       string
 }
 
 type User struct {
 	gorm.Model
-	UserID         int64 `gorm:"uniqueIndex"`
+	ChatID         int64 `gorm:"uniqueIndex;not null"`
 	Silent         bool
-	DefaultStorage string
+	DefaultStorage string // Default storage name
 }

types/types.go

@@ -2,6 +2,8 @@ package types
 
 import (
 	"context"
+	"crypto/md5"
+	"encoding/hex"
 	"fmt"
 	"time"
 
@@ -20,30 +22,38 @@ var (
 type StorageType string
 
 var (
-	StorageAll StorageType = "all"
-	Local      StorageType = "local"
-	Webdav     StorageType = "webdav"
-	Alist      StorageType = "alist"
+	StorageTypeLocal  StorageType = "local"
+	StorageTypeWebdav StorageType = "webdav"
+	StorageTypeAlist  StorageType = "alist"
 )
 
-var StorageTypes = []StorageType{Local, Alist, Webdav, StorageAll}
+var StorageTypes = []StorageType{StorageTypeLocal, StorageTypeAlist, StorageTypeWebdav}
+var StorageTypeDisplay = map[StorageType]string{
+	StorageTypeLocal:  "本地磁盘",
+	StorageTypeWebdav: "WebDAV",
+	StorageTypeAlist:  "Alist",
+}
 
 type Task struct {
 	Ctx         context.Context
 	Error       error
 	Status      TaskStatus
 	File        *File
-	Storage     StorageType
+	StorageName string
 	StoragePath string
 	StartTime   time.Time
 
-	MessageID      int
-	ChatID         int64
+	FileMessageID int
+	FileChatID    int64
+	// to track the reply message
 	ReplyMessageID int
+	ReplyChatID    int64
+	// to track the user
+	UserID int64
 }
 
 func (t Task) String() string {
-	return fmt.Sprintf("[%d:%d]:%s", t.ChatID, t.MessageID, t.File.FileName)
+	return fmt.Sprintf("[%d:%d]:%s", t.FileChatID, t.FileMessageID, t.File.FileName)
 }
 
 func (t Task) FileName() string {
@@ -55,3 +65,18 @@ type File struct {
 	FileSize int64
 	FileName string
 }
+
+func (f File) Hash() string {
+	locationBytes := []byte(f.Location.String())
+	fileSizeBytes := []byte(fmt.Sprintf("%d", f.FileSize))
+	fileNameBytes := []byte(f.FileName)
+
+	structBytes := append(locationBytes, fileSizeBytes...)
+	structBytes = append(structBytes, fileNameBytes...)
+
+	hash := md5.New()
+	hash.Write(structBytes)
+	hashBytes := hash.Sum(nil)
+
+	return hex.EncodeToString(hashBytes)
+}