Merge pull request #30 from krau/dev

feat: cancel task

.github/workflows/build-docker.yml

@@ -0,0 +1,60 @@
+name: Build and Publish Docker Image
+
+on:
+  push:
+    tags:
+      - "v*"
+
+env:
+  REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository }}
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Extract metadata for Docker
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+          tags: |
+            type=semver,pattern={{version}}
+            type=semver,pattern={{major}}.{{minor}}
+            type=sha
+            type=raw,value=latest
+            type=ref,event=branch
+            type=ref,event=tag
+          labels: |
+            org.opencontainers.image.title=${{ env.IMAGE_NAME }}
+            org.opencontainers.image.source=https://github.com/krau/SaveAny-Bot
+            org.opencontainers.image.url=https://github.com/krau/SaveAny-Bot
+
+      - name: Build and push Docker image
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          platforms: linux/amd64,linux/arm64
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}

.github/workflows/build-release.yml

@@ -23,7 +23,7 @@ jobs:
       - name: Setup node
         uses: actions/setup-node@v4
         with:
-          node-version: 20
+          node-version: 22
 
       - run: npx changelogithub
         env:
@@ -49,7 +49,7 @@ jobs:
       - name: Release Go Binary
         uses: wangyoucao577/go-release-action@v1
         with:
-          pre_command: export
+          pre_command: export CGO_ENABLED=0
           goos: ${{ matrix.goos }}
           goarch: ${{ matrix.goarch }}
           github_token: ${{ secrets.GITHUB_TOKEN }}

.gitignore

@@ -3,4 +3,6 @@ logs/
 tmp/
 data/
 downloads/
-cache/
+cache/
+session.*
+cache.db

Dockerfile

@@ -0,0 +1,18 @@
+FROM golang:alpine AS builder
+
+WORKDIR /app
+
+COPY go.* ./
+RUN go mod download
+
+COPY . .
+
+RUN CGO_ENABLED=0 GOOS=linux go build -o saveany-bot .
+
+FROM alpine:latest
+
+WORKDIR /app
+
+COPY --from=builder /app/saveany-bot .
+
+CMD ["./saveany-bot"]

LICENSE

@@ -1,5 +1,5 @@
-                    GNU GENERAL PUBLIC LICENSE
-                       Version 3, 29 June 2007
+                    GNU AFFERO GENERAL PUBLIC LICENSE
+                       Version 3, 19 November 2007
 
  Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
  Everyone is permitted to copy and distribute verbatim copies
@@ -7,17 +7,15 @@
 
                             Preamble
 
-  The GNU General Public License is a free, copyleft license for
-software and other kinds of works.
+  The GNU Affero General Public License is a free, copyleft license for
+software and other kinds of works, specifically designed to ensure
+cooperation with the community in the case of network server software.
 
   The licenses for most software and other practical works are designed
 to take away your freedom to share and change the works.  By contrast,
-the GNU General Public License is intended to guarantee your freedom to
+our General Public Licenses are intended to guarantee your freedom to
 share and change all versions of a program--to make sure it remains free
-software for all its users.  We, the Free Software Foundation, use the
-GNU General Public License for most of our software; it applies also to
-any other work released this way by its authors.  You can apply it to
-your programs, too.
+software for all its users.
 
   When we speak of free software, we are referring to freedom, not
 price.  Our General Public Licenses are designed to make sure that you
@@ -26,44 +24,34 @@ them if you wish), that you receive source code or can get it if you
 want it, that you can change the software or use pieces of it in new
 free programs, and that you know you can do these things.
 
-  To protect your rights, we need to prevent others from denying you
-these rights or asking you to surrender the rights.  Therefore, you have
-certain responsibilities if you distribute copies of the software, or if
-you modify it: responsibilities to respect the freedom of others.
+  Developers that use our General Public Licenses protect your rights
+with two steps: (1) assert copyright on the software, and (2) offer
+you this License which gives you legal permission to copy, distribute
+and/or modify the software.
 
-  For example, if you distribute copies of such a program, whether
-gratis or for a fee, you must pass on to the recipients the same
-freedoms that you received.  You must make sure that they, too, receive
-or can get the source code.  And you must show them these terms so they
-know their rights.
+  A secondary benefit of defending all users' freedom is that
+improvements made in alternate versions of the program, if they
+receive widespread use, become available for other developers to
+incorporate.  Many developers of free software are heartened and
+encouraged by the resulting cooperation.  However, in the case of
+software used on network servers, this result may fail to come about.
+The GNU General Public License permits making a modified version and
+letting the public access it on a server without ever releasing its
+source code to the public.
 
-  Developers that use the GNU GPL protect your rights with two steps:
-(1) assert copyright on the software, and (2) offer you this License
-giving you legal permission to copy, distribute and/or modify it.
+  The GNU Affero General Public License is designed specifically to
+ensure that, in such cases, the modified source code becomes available
+to the community.  It requires the operator of a network server to
+provide the source code of the modified version running there to the
+users of that server.  Therefore, public use of a modified version, on
+a publicly accessible server, gives the public access to the source
+code of the modified version.
 
-  For the developers' and authors' protection, the GPL clearly explains
-that there is no warranty for this free software.  For both users' and
-authors' sake, the GPL requires that modified versions be marked as
-changed, so that their problems will not be attributed erroneously to
-authors of previous versions.
-
-  Some devices are designed to deny users access to install or run
-modified versions of the software inside them, although the manufacturer
-can do so.  This is fundamentally incompatible with the aim of
-protecting users' freedom to change the software.  The systematic
-pattern of such abuse occurs in the area of products for individuals to
-use, which is precisely where it is most unacceptable.  Therefore, we
-have designed this version of the GPL to prohibit the practice for those
-products.  If such problems arise substantially in other domains, we
-stand ready to extend this provision to those domains in future versions
-of the GPL, as needed to protect the freedom of users.
-
-  Finally, every program is threatened constantly by software patents.
-States should not allow patents to restrict development and use of
-software on general-purpose computers, but in those that do, we wish to
-avoid the special danger that patents applied to a free program could
-make it effectively proprietary.  To prevent this, the GPL assures that
-patents cannot be used to render the program non-free.
+  An older license, called the Affero General Public License and
+published by Affero, was designed to accomplish similar goals.  This is
+a different license, not a version of the Affero GPL, but Affero has
+released a new version of the Affero GPL which permits relicensing under
+this license.
 
   The precise terms and conditions for copying, distribution and
 modification follow.
@@ -72,7 +60,7 @@ modification follow.
 
   0. Definitions.
 
-  "This License" refers to version 3 of the GNU General Public License.
+  "This License" refers to version 3 of the GNU Affero General Public License.
 
   "Copyright" also means copyright-like laws that apply to other kinds of
 works, such as semiconductor masks.
@@ -549,35 +537,45 @@ to collect a royalty for further conveying from those to whom you convey
 the Program, the only way you could satisfy both those terms and this
 License would be to refrain entirely from conveying the Program.
 
-  13. Use with the GNU Affero General Public License.
+  13. Remote Network Interaction; Use with the GNU General Public License.
+
+  Notwithstanding any other provision of this License, if you modify the
+Program, your modified version must prominently offer all users
+interacting with it remotely through a computer network (if your version
+supports such interaction) an opportunity to receive the Corresponding
+Source of your version by providing access to the Corresponding Source
+from a network server at no charge, through some standard or customary
+means of facilitating copying of software.  This Corresponding Source
+shall include the Corresponding Source for any work covered by version 3
+of the GNU General Public License that is incorporated pursuant to the
+following paragraph.
 
   Notwithstanding any other provision of this License, you have
 permission to link or combine any covered work with a work licensed
-under version 3 of the GNU Affero General Public License into a single
+under version 3 of the GNU General Public License into a single
 combined work, and to convey the resulting work.  The terms of this
 License will continue to apply to the part which is the covered work,
-but the special requirements of the GNU Affero General Public License,
-section 13, concerning interaction through a network will apply to the
-combination as such.
+but the work with which it is combined will remain governed by version
+3 of the GNU General Public License.
 
   14. Revised Versions of this License.
 
   The Free Software Foundation may publish revised and/or new versions of
-the GNU General Public License from time to time.  Such new versions will
-be similar in spirit to the present version, but may differ in detail to
+the GNU Affero General Public License from time to time.  Such new versions
+will be similar in spirit to the present version, but may differ in detail to
 address new problems or concerns.
 
   Each version is given a distinguishing version number.  If the
-Program specifies that a certain numbered version of the GNU General
+Program specifies that a certain numbered version of the GNU Affero General
 Public License "or any later version" applies to it, you have the
 option of following the terms and conditions either of that numbered
 version or of any later version published by the Free Software
 Foundation.  If the Program does not specify a version number of the
-GNU General Public License, you may choose any version ever published
+GNU Affero General Public License, you may choose any version ever published
 by the Free Software Foundation.
 
   If the Program specifies that a proxy can decide which future
-versions of the GNU General Public License can be used, that proxy's
+versions of the GNU Affero General Public License can be used, that proxy's
 public statement of acceptance of a version permanently authorizes you
 to choose that version for the Program.
 
@@ -635,40 +633,29 @@ the "copyright" line and a pointer to where the full notice is found.
     Copyright (C) <year>  <name of author>
 
     This program is free software: you can redistribute it and/or modify
-    it under the terms of the GNU General Public License as published by
-    the Free Software Foundation, either version 3 of the License, or
+    it under the terms of the GNU Affero General Public License as published
+    by the Free Software Foundation, either version 3 of the License, or
     (at your option) any later version.
 
     This program is distributed in the hope that it will be useful,
     but WITHOUT ANY WARRANTY; without even the implied warranty of
     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-    GNU General Public License for more details.
+    GNU Affero General Public License for more details.
 
-    You should have received a copy of the GNU General Public License
+    You should have received a copy of the GNU Affero General Public License
     along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
 Also add information on how to contact you by electronic and paper mail.
 
-  If the program does terminal interaction, make it output a short
-notice like this when it starts in an interactive mode:
-
-    <program>  Copyright (C) <year>  <name of author>
-    This program comes with ABSOLUTELY NO WARRANTY; for details type `show w'.
-    This is free software, and you are welcome to redistribute it
-    under certain conditions; type `show c' for details.
-
-The hypothetical commands `show w' and `show c' should show the appropriate
-parts of the General Public License.  Of course, your program's commands
-might be different; for a GUI interface, you would use an "about box".
+  If your software can interact with users remotely through a computer
+network, you should also make sure that it provides a way for users to
+get its source.  For example, if your program is a web application, its
+interface could display a "Source" link that leads users to an archive
+of the code.  There are many ways you could offer source, and different
+solutions will be better for different programs; see section 13 for the
+specific requirements.
 
   You should also get your employer (if you work as a programmer) or school,
 if any, to sign a "copyright disclaimer" for the program, if necessary.
-For more information on this, and how to apply and follow the GNU GPL, see
+For more information on this, and how to apply and follow the GNU AGPL, see
 <https://www.gnu.org/licenses/>.
-
-  The GNU General Public License does not permit incorporating your program
-into proprietary programs.  If your program is a subroutine library, you
-may consider it more useful to permit linking proprietary applications with
-the library.  If this is what you want to do, use the GNU Lesser General
-Public License instead of this License.  But first, please read
-<https://www.gnu.org/licenses/why-not-lgpl.html>.

README.md

@@ -1,19 +1,32 @@
-# Save Any Bot
+<div align="center">
 
-把 Telegram 的文件保存到各类存储端. 
 
-> *就像 PikPak Bot 一样*
+# <img src="docs/logo.jpg" width="45" align="center"> Save Any Bot
+
+**简体中文** | [English](README_EN.md) 
+
+把 Telegram 的文件保存到各类存储端.
+
+> _就像 PikPak Bot 一样_
+
+</div
+
+Demo Video:
+
+<div align="center">
+
+
+[SaveAny-Bot 演示视频 ｜ The Demo of SaveAny-Bot.webm](https://github.com/user-attachments/assets/a0de2453-a4d1-4a12-81fb-9d84856dce09)
+
+</div>
 
 ## 部署
 
+### 从二进制文件部署
+
 在 [Release](https://github.com/krau/SaveAny-Bot/releases) 页面下载对应平台的二进制文件.
 
-在解压后目录新建 `config.toml` 文件, 参考 [config.toml.example](https://github.com/krau/SaveAny-Bot/blob/main/config.example.toml) 编辑配置文件.
-
-> [!TIP]
-> 由于 Telegram 官方 Bot API 的限制, Bot 无法下载大于 20MB 的文件. 你需要部署一个本地的 Telegram Bot API 来解决这个问题, 然后将配置文件中的 telegram.api 改为你自己的 api 地址.
->
-> 参考: [telegram-bot-api-compose](https://github.com/krau/telegram-bot-api-compose)
+在解压后目录新建 `config.toml` 文件, 参考 [config.example.toml](./config.example.toml) 编辑配置文件.
 
 运行:
 
@@ -22,6 +35,95 @@ chmod +x saveany-bot
 ./saveany-bot
 ```
 
+#### 添加为 systemd 服务
+
+创建文件 `/etc/systemd/system/saveany-bot.service` 并写入以下内容:
+
+```
+[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
+```
+
+设为开机启动并启动服务:
+
+```bash
+systemctl enable --now saveany-bot
+```
+
+#### 为OpenWrt及衍生系统添加开机自启动服务
+
+创建文件 ` /etc/init.d/saveanybot` ，参考[saveanybot](./docs/saveanybot)自行修改.
+
+`chmod +x /etc/init.d/saveanybot`
+
+完成后，将文件复制到 `/etc/rc.d`并重命名为`S99saveanybot`.
+
+`chmod +x /etc/rc.d/S99saveanybot`
+
+#### 为OpenWrt及衍生系统添加快捷指令
+
+创建文件` /usr/bin/sabot` ，参考[sabot](./docs/sabot)自行配置修改，注意此处文件编码仅支持 ANSI 936 .
+
+`chmod +x /usr/bin/sabot`
+
+之后，终端输入`sabot start|stop|restart|status|enable|disable`即可.
+
+
+### 使用 Docker 部署
+
+#### Docker Compose
+
+下载 [docker-compose.yml](./docker-compose.yml) 文件, 在同目录下新建 `config.toml` 文件, 参考 [config.example.toml](./config.example.toml) 编辑配置文件.
+
+启动:
+
+```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
+```
+
+## 更新
+
+使用 `upgrade` 或 `up` 升级到最新版
+
+```bash
+./saveany-bot upgrade
+```
+
+如果是 Docker 部署, 使用以下命令更新:
+
+```bash
+docker pull ghcr.io/krau/saveany-bot:latest
+docker restart saveany-bot
+```
+
 ## 使用
 
-向 Bot 发送(转发)文件, 按照提示操作.
+向 Bot 发送(转发)文件, 或发送公开频道的消息链接, 按照提示操作.
+
+---
+
+## Thanks
+
+- [gotd](https://github.com/gotd/td)
+- [TG-FileStreamBot](https://github.com/EverythingSuckz/TG-FileStreamBot)
+- [gotgproto](https://github.com/celestix/gotgproto)
+- All the dependencies

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,21 +0,0 @@
-package bootstrap
-
-import (
-	"github.com/krau/SaveAny-Bot/bot"
-	"github.com/krau/SaveAny-Bot/common"
-	"github.com/krau/SaveAny-Bot/config"
-	"github.com/krau/SaveAny-Bot/dao"
-	"github.com/krau/SaveAny-Bot/logger"
-	"github.com/krau/SaveAny-Bot/storage"
-)
-
-func InitAll() {
-	config.Init()
-	logger.InitLogger()
-	logger.L.Info("Running...")
-
-	common.Init()
-	storage.Init()
-	dao.Init()
-	bot.Init()
-}

bot/bot.go

@@ -1,110 +1,101 @@
 package bot
 
 import (
+	"context"
+	"net/url"
 	"os"
+	"time"
 
-	"github.com/duke-git/lancet/v2/slice"
+	"github.com/celestix/gotgproto"
+	"github.com/celestix/gotgproto/sessionMaker"
+	"github.com/glebarez/sqlite"
+	"github.com/gotd/td/telegram/dcs"
+	"github.com/gotd/td/tg"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
-	"github.com/mymmrac/telego"
-	"github.com/mymmrac/telego/telegohandler"
-	"github.com/mymmrac/telego/telegoutil"
+	"golang.org/x/net/proxy"
 )
 
-var (
-	Bot *telego.Bot
-)
+var Client *gotgproto.Client
+
+func newProxyDialer(proxyUrl string) (proxy.Dialer, error) {
+	url, err := url.Parse(proxyUrl)
+	if err != nil {
+		return nil, err
+	}
+	return proxy.FromURL(url, proxy.Direct)
+}
 
 func Init() {
-	logger.L.Debug("Initializing bot...")
-	var err error
-	Bot, err = telego.NewBot(
-		config.Cfg.Telegram.Token,
-		telego.WithDefaultLogger(false, true),
-		telego.WithAPIServer(config.Cfg.Telegram.API),
-	)
-	if err != nil {
-		logger.L.Fatal("Failed to create bot: ", err)
-		os.Exit(1)
-	}
-	Bot.SetMyCommands(&telego.SetMyCommandsParams{
-		Commands: []telego.BotCommand{
-			{Command: "start", Description: "开始使用"},
-			{Command: "help", Description: "显示帮助"},
-			{Command: "silent", Description: "静默模式"},
-			{Command: "storage", Description: "设置默认存储位置"},
-			{Command: "save", Description: "保存文件"},
-			{Command: "clean", Description: "清除文件记录"},
-		},
+	logger.L.Info("初始化 Telegram 客户端...")
+	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
+	defer cancel()
+	resultChan := make(chan struct {
+		client *gotgproto.Client
+		err    error
 	})
-	logger.L.Debug("Bot initialized")
-}
-
-func Run() {
-	if Bot == nil {
-		Init()
-	}
-	logger.L.Info("Start polling...")
-	updates, err := Bot.UpdatesViaLongPolling(&telego.GetUpdatesParams{
-		Offset: -1,
-		AllowedUpdates: []string{
-			telego.MessageUpdates,
-			telego.CallbackQueryUpdates,
-		},
-	})
-	if err != nil {
-		logger.L.Fatal("Failed to start polling: ", err)
-		os.Exit(1)
-	}
-	botHandler, err := telegohandler.NewBotHandler(Bot, updates)
-	if err != nil {
-		logger.L.Fatal("Failed to create bot handler: ", err)
-		os.Exit(1)
-	}
-	defer botHandler.Stop()
-	defer Bot.StopLongPolling()
-
-	botHandler.Use(telegohandler.PanicRecovery())
-	baseGroup := botHandler.BaseGroup()
-
-	registerHandlers(baseGroup)
-
-	botHandler.Start()
-}
-
-func registerHandlers(hg *telegohandler.HandlerGroup) {
-	msgGroup := hg.Group(telegohandler.AnyMessage())
-	msgGroup.Use(func(bot *telego.Bot, update telego.Update, next telegohandler.Handler) {
-		if !slice.Contain(config.Cfg.Telegram.Admins, update.Message.From.ID) {
-			bot.SendMessage(telegoutil.Message(update.Message.Chat.ChatID(), "抱歉, 该 Bot 为个人使用设计, 您可以部署自己的 SaveAnyBot 实例: https://github.com/krau/SaveAny-Bot"))
+	go func() {
+		var resolver dcs.Resolver
+		if config.Cfg.Telegram.Proxy.Enable && config.Cfg.Telegram.Proxy.URL != "" {
+			dialer, err := newProxyDialer(config.Cfg.Telegram.Proxy.URL)
+			if err != nil {
+				resultChan <- struct {
+					client *gotgproto.Client
+					err    error
+				}{nil, err}
+				return
+			}
+			resolver = dcs.Plain(dcs.PlainOptions{
+				Dial: dialer.(proxy.ContextDialer).DialContext,
+			})
+		} else {
+			resolver = dcs.DefaultResolver()
+		}
+		client, err := gotgproto.NewClient(config.Cfg.Telegram.AppID,
+			config.Cfg.Telegram.AppHash,
+			gotgproto.ClientTypeBot(config.Cfg.Telegram.Token),
+			&gotgproto.ClientOpts{
+				Session:          sessionMaker.SqlSession(sqlite.Open("data/session.db")),
+				DisableCopyright: true,
+				Middlewares:      FloodWaitMiddleware(),
+				Resolver:         resolver,
+			},
+		)
+		if err != nil {
+			resultChan <- struct {
+				client *gotgproto.Client
+				err    error
+			}{nil, err}
 			return
 		}
-		next(bot, update)
-	})
+		_, err = client.API().BotsSetBotCommands(ctx, &tg.BotsSetBotCommandsRequest{
+			Scope: &tg.BotCommandScopeDefault{},
+			Commands: []tg.BotCommand{
+				{Command: "start", Description: "开始使用"},
+				{Command: "help", Description: "显示帮助"},
+				{Command: "silent", Description: "开启/关闭静默模式"},
+				{Command: "storage", Description: "设置默认存储端"},
+				{Command: "save", Description: "保存所回复的文件"},
+				{Command: "dir", Description: "管理存储文件夹"},
+			},
+		})
+		resultChan <- struct {
+			client *gotgproto.Client
+			err    error
+		}{client, err}
+	}()
 
-	msgGroup.HandleMessageCtx(Start, telegohandler.CommandEqual("start"))
-	msgGroup.HandleMessageCtx(Help, telegohandler.CommandEqual("help"))
-	msgGroup.HandleMessageCtx(ChangeSilentMode, telegohandler.CommandEqual("silent"))
-	msgGroup.HandleMessageCtx(SetDefaultStorage, telegohandler.CommandEqual("storage"))
-	msgGroup.HandleMessageCtx(SaveFile, telegohandler.CommandEqual("save"))
-	msgGroup.HandleMessageCtx(CleanReceivedFile, telegohandler.CommandEqual("clean"))
-
-	msgGroup.HandleMessageCtx(HandleFileMessage, func(update telego.Update) bool {
-		return update.Message.Document != nil || update.Message.Video != nil || update.Message.Audio != nil
-	})
-
-	callbackGroup := hg.Group(telegohandler.AnyCallbackQueryWithMessage())
-	callbackGroup.Use(func(bot *telego.Bot, update telego.Update, next telegohandler.Handler) {
-		if !slice.Contain(config.Cfg.Telegram.Admins, update.CallbackQuery.From.ID) {
-			bot.AnswerCallbackQuery(telegoutil.
-				CallbackQuery(update.CallbackQuery.ID).
-				WithText("抱歉, 该 Bot 为个人使用设计, 您可以部署自己的 SaveAnyBot 实例: https://github.com/krau/SaveAny-Bot").
-				WithShowAlert().
-				WithCacheTime(60))
-			return
+	select {
+	case <-ctx.Done():
+		logger.L.Fatal("初始化客户端失败: 超时")
+		os.Exit(1)
+	case result := <-resultChan:
+		if result.err != nil {
+			logger.L.Fatalf("初始化客户端失败: %s", result.err)
+			os.Exit(1)
 		}
-		next(bot, update)
-	})
-
-	callbackGroup.HandleCallbackQueryCtx(AddToQueue, telegohandler.CallbackDataPrefix("add"))
+		Client = result.client
+		RegisterHandlers(Client.Dispatcher)
+		logger.L.Info("客户端初始化完成")
+	}
 }

bot/handle_add_task.go

@@ -0,0 +1,188 @@
+package bot
+
+import (
+	"errors"
+	"fmt"
+	"path"
+	"strconv"
+	"strings"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/duke-git/lancet/v2/slice"
+	"github.com/gotd/td/telegram/message/entity"
+	"github.com/gotd/td/telegram/message/styling"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/queue"
+	"github.com/krau/SaveAny-Bot/types"
+	"gorm.io/gorm"
+)
+
+func AddToQueue(ctx *ext.Context, update *ext.Update) error {
+	// TODO: 回调数据用户独立鉴权 (处理 bot 在群聊中的情况)
+	if !slice.Contain(config.Cfg.GetUsersID(), update.CallbackQuery.UserID) {
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "你没有权限",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+	args := strings.Split(string(update.CallbackQuery.Data), " ")
+	addToDir := args[0] == "add_to_dir"
+	cbDataId, _ := strconv.Atoi(args[1])
+	cbData, err := dao.GetCallbackData(uint(cbDataId))
+	if err != nil {
+		logger.L.Errorf("获取回调数据失败: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "获取回调数据失败",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+
+	data := strings.Split(cbData, " ")
+	fileChatID, _ := strconv.Atoi(data[0])
+	fileMessageID, _ := strconv.Atoi(data[1])
+	storageName := data[2]
+	dirIdInt, _ := strconv.Atoi(data[3])
+	dirId := uint(dirIdInt)
+
+	user, err := dao.GetUserByChatID(update.CallbackQuery.UserID)
+	if err != nil {
+		logger.L.Errorf("获取用户失败: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "获取用户失败",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+
+	if !addToDir {
+		dirs, err := dao.GetDirsByUserIDAndStorageName(user.ID, storageName)
+		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+			logger.L.Errorf("获取路径失败: %s", err)
+			ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+				QueryID:   update.CallbackQuery.QueryID,
+				Alert:     true,
+				Message:   "获取路径失败",
+				CacheTime: 5,
+			})
+			return dispatcher.EndGroups
+		}
+		if len(dirs) != 0 {
+			markup, err := getSelectDirMarkup(fileChatID, fileMessageID, storageName, dirs)
+			if err != nil {
+				logger.L.Errorf("获取路径失败: %s", err)
+				ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+					QueryID:   update.CallbackQuery.QueryID,
+					Alert:     true,
+					Message:   "获取路径失败",
+					CacheTime: 5,
+				})
+				return dispatcher.EndGroups
+			}
+			_, err = ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+				ID:          update.CallbackQuery.GetMsgID(),
+				Message:     "请选择要保存到的路径",
+				ReplyMarkup: markup,
+			})
+			if err != nil {
+				logger.L.Errorf("编辑消息失败: %s", err)
+			}
+			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("获取记录失败: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "查询记录失败",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+	if update.CallbackQuery.MsgID != record.ReplyMessageID {
+		record.ReplyMessageID = update.CallbackQuery.MsgID
+		if err := dao.SaveReceivedFile(record); err != nil {
+			logger.L.Errorf("更新接收的文件失败: %s", err)
+		}
+	}
+
+	var dir *dao.Dir
+	if addToDir && dirId != 0 {
+		dir, err = dao.GetDirByID(dirId)
+		if err != nil {
+			logger.L.Errorf("获取路径失败: %s", err)
+			ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+				QueryID:   update.CallbackQuery.QueryID,
+				Alert:     true,
+				Message:   "获取路径失败",
+				CacheTime: 5,
+			})
+			return dispatcher.EndGroups
+		}
+	}
+
+	file, err := FileFromMessage(ctx, record.ChatID, record.MessageID, record.FileName)
+	if err != nil {
+		logger.L.Errorf("获取消息中的文件失败: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   fmt.Sprintf("获取消息中的文件失败: %s", err),
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+
+	task := types.Task{
+		Ctx:            ctx,
+		Status:         types.Pending,
+		File:           file,
+		StorageName:    storageName,
+		FileChatID:     record.ChatID,
+		ReplyMessageID: record.ReplyMessageID,
+		FileMessageID:  record.MessageID,
+		ReplyChatID:    record.ReplyChatID,
+		UserID:         update.GetUserChat().GetID(),
+	}
+	if dir != nil {
+		task.StoragePath = path.Join(dir.Path, file.FileName)
+	}
+
+	queue.AddTask(&task)
+
+	entityBuilder := entity.Builder{}
+	var entities []tg.MessageEntityClass
+	text := fmt.Sprintf("已添加到任务队列\n文件名: %s\n当前排队任务数: %d", record.FileName, queue.Len())
+	if err := styling.Perform(&entityBuilder,
+		styling.Plain("已添加到任务队列\n文件名: "),
+		styling.Code(record.FileName),
+		styling.Plain("\n当前排队任务数: "),
+		styling.Bold(strconv.Itoa(queue.Len())),
+	); err != nil {
+		logger.L.Errorf("Failed to build entity: %s", err)
+	} else {
+		text, entities = entityBuilder.Complete()
+	}
+
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message:  text,
+		Entities: entities,
+		ID:       record.ReplyMessageID,
+	})
+	return dispatcher.EndGroups
+}

bot/handle_cancel_task.go

@@ -0,0 +1,27 @@
+package bot
+
+import (
+	"strings"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/queue"
+)
+
+func cancelTask(ctx *ext.Context, update *ext.Update) error {
+	key := strings.Split(string(update.CallbackQuery.Data), " ")[1]
+	ok := queue.CancelTask(key)
+	if ok {
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID: update.CallbackQuery.QueryID,
+			Message: "任务已取消",
+		})
+		return dispatcher.EndGroups
+	}
+	ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+		QueryID: update.CallbackQuery.QueryID,
+		Message: "任务取消失败",
+	})
+	return dispatcher.EndGroups
+}

bot/handle_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/handle_dir.go

@@ -0,0 +1,88 @@
+package bot
+
+import (
+	"strings"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/telegram/message/styling"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/storage"
+)
+
+func dirCmd(ctx *ext.Context, update *ext.Update) error {
+	args := strings.Split(strings.TrimPrefix(update.EffectiveMessage.Text, "/dir "), " ")
+	if len(args) < 3 {
+		dirs, err := dao.GetUserDirsByChatID(update.GetUserChat().GetID())
+		if err != nil {
+			logger.L.Errorf("获取用户路径失败: %s", err)
+			ctx.Reply(update, ext.ReplyTextString("获取用户路径失败"), nil)
+			return dispatcher.EndGroups
+		}
+		ctx.Reply(update, ext.ReplyTextStyledTextArray(
+			[]styling.StyledTextOption{
+				styling.Bold("使用方法: /dir <操作> <存储名> <路径>"),
+				styling.Plain("\n\n可用操作:\n"),
+				styling.Code("add"),
+				styling.Plain(" - 添加路径\n"),
+				styling.Code("del"),
+				styling.Plain(" - 删除路径\n"),
+				styling.Plain("\n示例:\n"),
+				styling.Code("/dir add local1 path/to/dir"),
+				styling.Plain("\n\n当前已添加的路径:\n"),
+				styling.Blockquote(func() string {
+					var sb strings.Builder
+					for _, dir := range dirs {
+						sb.WriteString(dir.StorageName)
+						sb.WriteString(" - ")
+						sb.WriteString(dir.Path)
+						sb.WriteString("\n")
+					}
+					return sb.String()
+				}(), true),
+			},
+		), 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
+	}
+	switch args[0] {
+	case "add":
+		return addDir(ctx, update, user, args[1], args[2])
+	case "del":
+		return delDir(ctx, update, user, args[1], args[2])
+	default:
+		ctx.Reply(update, ext.ReplyTextString("未知操作"), nil)
+		return dispatcher.EndGroups
+	}
+}
+
+func addDir(ctx *ext.Context, update *ext.Update, user *dao.User, storageName, path string) error {
+	if _, err := storage.GetStorageByUserIDAndName(user.ChatID, storageName); err != nil {
+		ctx.Reply(update, ext.ReplyTextString(err.Error()), nil)
+		return dispatcher.EndGroups
+	}
+
+	if err := dao.CreateDirForUser(user.ID, storageName, path); err != nil {
+		logger.L.Errorf("创建路径失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("创建路径失败"), nil)
+		return dispatcher.EndGroups
+	}
+	ctx.Reply(update, ext.ReplyTextString("路径添加成功"), nil)
+	return dispatcher.EndGroups
+}
+
+func delDir(ctx *ext.Context, update *ext.Update, user *dao.User, storageName, path string) error {
+	if err := dao.DeleteDirForUser(user.ID, storageName, path); err != nil {
+		logger.L.Errorf("删除路径失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("删除路径失败"), nil)
+		return dispatcher.EndGroups
+	}
+	ctx.Reply(update, ext.ReplyTextString("路径删除成功"), nil)
+	return dispatcher.EndGroups
+}

bot/handle_file.go

@@ -0,0 +1,85 @@
+package bot
+
+import (
+	"fmt"
+
+	"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"
+)
+
+func handleFileMessage(ctx *ext.Context, update *ext.Update) error {
+	logger.L.Trace("Got media: ", update.EffectiveMessage.Media.TypeName())
+	supported, err := supportedMediaFilter(update.EffectiveMessage.Message)
+	if err != nil {
+		return err
+	}
+	if !supported {
+		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
+	}
+
+	msg, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件信息..."), nil)
+	if err != nil {
+		logger.L.Errorf("回复失败: %s", err)
+		return dispatcher.EndGroups
+	}
+	media := update.EffectiveMessage.Media
+	file, err := FileFromMedia(media, "")
+	if err != nil {
+		logger.L.Errorf("获取文件失败: %s", err)
+		ctx.Reply(update, ext.ReplyTextString(fmt.Sprintf("获取文件失败: %s", err)), nil)
+		return dispatcher.EndGroups
+	}
+	if file.FileName == "" {
+		file.FileName = fmt.Sprintf("%d_%d_%s", update.EffectiveChat().GetID(), update.EffectiveMessage.ID, file.Hash())
+	}
+
+	if err := dao.SaveReceivedFile(&dao.ReceivedFile{
+		Processing:     false,
+		FileName:       file.FileName,
+		ChatID:         update.EffectiveChat().GetID(),
+		MessageID:      update.EffectiveMessage.ID,
+		ReplyMessageID: msg.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+	}); err != nil {
+		logger.L.Errorf("添加接收的文件失败: %s", err)
+		if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: fmt.Sprintf("添加接收的文件失败: %s", err),
+			ID:      msg.ID,
+		}); err != nil {
+			logger.L.Errorf("编辑消息失败: %s", err)
+		}
+		return dispatcher.EndGroups
+	}
+
+	if !user.Silent || user.DefaultStorage == "" {
+		return ProvideSelectMessage(ctx, update, file, update.EffectiveChat().GetID(), update.EffectiveMessage.ID, msg.ID)
+	}
+	return HandleSilentAddTask(ctx, update, user, &types.Task{
+		Ctx:            ctx,
+		Status:         types.Pending,
+		File:           file,
+		StorageName:    user.DefaultStorage,
+		FileChatID:     update.EffectiveChat().GetID(),
+		ReplyMessageID: msg.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+		FileMessageID:  update.EffectiveMessage.ID,
+		UserID:         user.ChatID,
+	})
+}

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 := &dao.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/handle_save.go

@@ -0,0 +1,116 @@
+package bot
+
+import (
+	"fmt"
+	"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"
+)
+
+func saveCmd(ctx *ext.Context, update *ext.Update) error {
+	res, ok := update.EffectiveMessage.GetReplyTo()
+	if !ok || res == nil {
+		ctx.Reply(update, ext.ReplyTextString("请回复要保存的文件"), nil)
+		return dispatcher.EndGroups
+	}
+	replyHeader, ok := res.(*tg.MessageReplyHeader)
+	if !ok {
+		ctx.Reply(update, ext.ReplyTextString("请回复要保存的文件"), nil)
+		return dispatcher.EndGroups
+	}
+	replyToMsgID, ok := replyHeader.GetReplyToMsgID()
+	if !ok {
+		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
+	}
+
+	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
+	}
+
+	supported, _ := supportedMediaFilter(msg)
+	if !supported {
+		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
+	}
+
+	cmdText := update.EffectiveMessage.Text
+	customFileName := strings.TrimSpace(strings.TrimPrefix(cmdText, "/save"))
+
+	file, err := FileFromMessage(ctx, update.EffectiveChat().GetID(), msg.ID, customFileName)
+	if err != nil {
+		logger.L.Errorf("获取文件失败: %s", err)
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: fmt.Sprintf("获取文件失败: %s", err),
+			ID:      replied.ID,
+		})
+		return dispatcher.EndGroups
+	}
+
+	// TODO: better file name
+	if file.FileName == "" {
+		file.FileName = fmt.Sprintf("%d_%d_%s", update.EffectiveChat().GetID(), replyToMsgID, file.Hash())
+	}
+	receivedFile := &dao.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("保存接收的文件失败: %s", err)
+		if _, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: fmt.Sprintf("保存接收的文件失败: %s", err),
+			ID:      replied.ID,
+		}); err != nil {
+			logger.L.Errorf("编辑消息失败: %s", err)
+		}
+		return dispatcher.EndGroups
+	}
+	if !user.Silent || user.DefaultStorage == "" {
+		return ProvideSelectMessage(ctx, update, file, update.EffectiveChat().GetID(), msg.ID, replied.ID)
+	}
+	return HandleSilentAddTask(ctx, update, user, &types.Task{
+		Ctx:            ctx,
+		Status:         types.Pending,
+		File:           file,
+		StorageName:    user.DefaultStorage,
+		FileChatID:     update.EffectiveChat().GetID(),
+		ReplyMessageID: replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+		FileMessageID:  msg.ID,
+		UserID:         user.ChatID,
+	})
+}

bot/handle_silent.go

@@ -0,0 +1,30 @@
+package bot
+
+import (
+	"fmt"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+)
+
+func silent(ctx *ext.Context, update *ext.Update) error {
+	user, err := dao.GetUserByChatID(update.GetUserChat().GetID())
+	if err != nil {
+		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("更新用户失败: %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
+}

bot/handle_start.go

@@ -0,0 +1,37 @@
+package bot
+
+import (
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+)
+
+func start(ctx *ext.Context, update *ext.Update) error {
+	if err := dao.CreateUser(update.GetUserChat().GetID()); err != nil {
+		logger.L.Errorf("创建用户失败: %s", err)
+		return dispatcher.EndGroups
+	}
+	return help(ctx, update)
+}
+
+const helpText string = `
+Save Any Bot - 转存你的 Telegram 文件
+命令:
+/start - 开始使用
+/help - 显示帮助
+/silent - 开关静默模式
+/storage - 设置默认存储位置
+/save [自定义文件名] - 保存文件
+
+静默模式: 开启后 Bot 直接保存到收到的文件到默认位置, 不再询问
+
+默认存储位置: 在静默模式下保存到的位置
+
+向 Bot 发送(转发)文件, 或发送一个公开频道的消息链接以保存文件
+`
+
+func help(ctx *ext.Context, update *ext.Update) error {
+	ctx.Reply(update, ext.ReplyTextString(helpText), nil)
+	return dispatcher.EndGroups
+}

bot/handle_storage.go

@@ -0,0 +1,99 @@
+package bot
+
+import (
+	"fmt"
+	"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"
+)
+
+func storageCmd(ctx *ext.Context, update *ext.Update) error {
+	userChatID := update.GetUserChat().GetID()
+	storages := storage.GetUserStorages(userChatID)
+	if len(storages) == 0 {
+		ctx.Reply(update, ext.ReplyTextString("无可用的存储"), nil)
+		return dispatcher.EndGroups
+	}
+	markup, err := getSetDefaultStorageMarkup(userChatID, storages)
+	if err != nil {
+		logger.L.Errorf("Failed to get markup: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取存储位置失败"), nil)
+		return dispatcher.EndGroups
+	}
+	ctx.Reply(update, ext.ReplyTextString("请选择要设为默认的存储位置"), &ext.ReplyOpts{
+		Markup: markup,
+	})
+	return dispatcher.EndGroups
+}
+
+func setDefaultStorage(ctx *ext.Context, update *ext.Update) error {
+	args := strings.Split(string(update.CallbackQuery.Data), " ")
+	userID, _ := strconv.Atoi(args[1])
+	if userID != int(update.CallbackQuery.GetUserID()) {
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "你没有权限",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+	cbDataId, _ := strconv.Atoi(args[2])
+	storageName, err := dao.GetCallbackData(uint(cbDataId))
+	if err != nil {
+		logger.L.Errorf("获取回调数据失败: %s", err)
+		ctx.AnswerCallback(&tg.MessagesSetBotCallbackAnswerRequest{
+			QueryID:   update.CallbackQuery.QueryID,
+			Alert:     true,
+			Message:   "获取回调数据失败",
+			CacheTime: 5,
+		})
+		return dispatcher.EndGroups
+	}
+
+	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
+	}
+	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
+	}
+	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
+	}
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message: fmt.Sprintf("已将 %s (%s) 设为默认存储位置", selectedStorage.Name(), selectedStorage.Type()),
+		ID:      update.CallbackQuery.GetMsgID(),
+	})
+	return dispatcher.EndGroups
+}

bot/handlers.go

@@ -1,265 +1,27 @@
 package bot
 
 import (
-	"context"
-	"fmt"
-	"path/filepath"
-	"strconv"
-	"strings"
-
-	"github.com/duke-git/lancet/v2/slice"
-	"github.com/gookit/goutil/maputil"
-	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/dispatcher/handlers"
+	"github.com/celestix/gotgproto/dispatcher/handlers/filters"
 	"github.com/krau/SaveAny-Bot/logger"
-	"github.com/krau/SaveAny-Bot/model"
-	"github.com/krau/SaveAny-Bot/queue"
-	"github.com/krau/SaveAny-Bot/storage"
-	"github.com/krau/SaveAny-Bot/types"
-	"github.com/mymmrac/telego"
-	"github.com/mymmrac/telego/telegoutil"
 )
 
-func Start(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	if err := dao.CreateUser(message.From.ID); err != nil {
-		logger.L.Errorf("Failed to create user: %s", err)
-		return
+func RegisterHandlers(dispatcher dispatcher.Dispatcher) {
+	dispatcher.AddHandler(handlers.NewMessage(filters.Message.All, checkPermission))
+	dispatcher.AddHandler(handlers.NewCommand("start", start))
+	dispatcher.AddHandler(handlers.NewCommand("help", help))
+	dispatcher.AddHandler(handlers.NewCommand("silent", silent))
+	dispatcher.AddHandler(handlers.NewCommand("storage", storageCmd))
+	dispatcher.AddHandler(handlers.NewCommand("save", saveCmd))
+	dispatcher.AddHandler(handlers.NewCommand("dir", dirCmd))
+	linkRegexFilter, err := filters.Message.Regex(linkRegexString)
+	if err != nil {
+		logger.L.Panicf("创建正则表达式过滤器失败: %s", err)
 	}
-	Help(ctx, bot, message)
-}
-
-func Help(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	helpText := `
-SaveAny Bot - 转存你的 Telegram 文件
-命令:
-/start - 开始使用
-/help - 显示帮助
-/silent - 静默模式
-/storage - 设置默认存储位置
-/save - 保存文件
-/clean - 清除文件记录
-
-静默模式: 开启后 Bot 直接保存到收到的文件到默认位置, 不再询问
-	`
-	ReplyMessage(message, helpText)
-}
-
-func ChangeSilentMode(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	user, err := dao.GetUserByUserID(message.From.ID)
-	if err != nil {
-		logger.L.Error(err)
-		return
-	}
-	user.Silent = !user.Silent
-	err = dao.UpdateUser(user)
-	if err != nil {
-		logger.L.Error(err)
-		return
-	}
-	ReplyMessage(message, fmt.Sprintf("已%s静默模式", map[bool]string{true: "开启", false: "关闭"}[user.Silent]))
-}
-
-func SetDefaultStorage(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	if len(storage.Storages) == 0 {
-		ReplyMessage(message, "当前无可用存储端, 请检查配置.")
-		return
-	}
-	_, _, args := telegoutil.ParseCommand(message.Text)
-	availableStorages := maputil.Keys(storage.Storages)
-	if len(args) == 0 {
-		text := EscapeMarkdown("请提供存储位置名称, 可用项:")
-		for _, name := range availableStorages {
-			text += fmt.Sprintf("\n`%s`", name)
-		}
-		text += fmt.Sprintf("\n`all`")
-		bot.SendMessage(telegoutil.Message(message.Chat.ChatID(), text).WithParseMode(telego.ModeMarkdownV2))
-		return
-	}
-	storageName := args[0]
-	if !slice.Contain(availableStorages, storageName) {
-		ReplyMessage(message, "参数错误")
-		return
-	}
-	user, err := dao.GetUserByUserID(message.From.ID)
-	if err != nil {
-		logger.L.Error(err)
-		return
-	}
-	user.DefaultStorage = storageName
-	err = dao.UpdateUser(user)
-	if err != nil {
-		logger.L.Error(err)
-		return
-	}
-	ReplyMessage(message, fmt.Sprintf("已设置默认存储位置为: %s", storageName))
-}
-
-func SaveFile(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	targetMessage := message.ReplyToMessage
-	if targetMessage == nil {
-		ReplyMessage(message, "请回复要保存的文件")
-		return
-	}
-	if targetMessage.Document == nil && targetMessage.Video == nil && targetMessage.Audio == nil {
-		ReplyMessage(message, "回复的消息不包含文件")
-		return
-	}
-	ctx = context.WithValue(ctx, "force", true)
-	HandleFileMessage(ctx, bot, *targetMessage)
-}
-
-func CleanReceivedFile(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	targetMessage := message.ReplyToMessage
-	if targetMessage == nil {
-		ReplyMessage(message, "请回复要清除记录的文件")
-		return
-	}
-	if targetMessage.Document == nil && targetMessage.Video == nil && targetMessage.Audio == nil {
-		ReplyMessage(message, "回复的消息不包含文件")
-		return
-	}
-	var fileUniqueID string
-	if targetMessage.Document != nil {
-		fileUniqueID = targetMessage.Document.FileUniqueID
-	} else if targetMessage.Video != nil {
-		fileUniqueID = targetMessage.Video.FileUniqueID
-	} else if targetMessage.Audio != nil {
-		fileUniqueID = targetMessage.Audio.FileUniqueID
-	}
-
-	if fileUniqueID == "" {
-		ReplyMessage(message, "文件信息获取失败")
-		return
-	}
-
-	if err := dao.DeleteReceivedFileByFileUniqueID(fileUniqueID); err != nil {
-		logger.L.Error(err)
-		ReplyMessage(message, "删除记录失败")
-		return
-	}
-	ReplyMessage(message, "记录已删除")
-}
-
-func HandleFileMessage(ctx context.Context, bot *telego.Bot, message telego.Message) {
-	var fileID, fileName string
-	if message.Document != nil {
-		fileID = message.Document.FileID
-		fileName = message.Document.FileName
-	} else if message.Video != nil {
-		fileID = message.Video.FileID
-		fileName = message.Video.FileName
-	} else if message.Audio != nil {
-		fileID = message.Audio.FileID
-		fileName = message.Audio.FileName
-	}
-
-	if fileID == "" {
-		logger.L.Error("File ID is empty")
-		ReplyMessage(message, "文件信息获取失败")
-		return
-	}
-	user, err := dao.GetUserByUserID(message.From.ID)
-	if err != nil {
-		logger.L.Error(err)
-		return
-	}
-	msg, err := ReplyMessage(message, "正在获取文件信息")
-	if err != nil {
-		logger.L.Error(err)
-		return
-	}
-	file, err := bot.GetFile(&telego.GetFileParams{FileID: fileID})
-	if err != nil {
-		logger.L.Error(err)
-		ReplyMessage(message, "获取文件信息失败")
-		return
-	}
-	if ctx.Value("force") == nil {
-		receivedFile, _ := dao.GetReceivedFileByFileID(file.FileID)
-		if receivedFile != nil && receivedFile.Processing {
-			bot.EditMessageText(&telego.EditMessageTextParams{
-				ChatID:    message.Chat.ChatID(),
-				MessageID: msg.MessageID,
-				Text:      "该文件或许正在处理中, 使用 /save 命令回复此文件以强制加入队列\n使用 /clean 命令回复此文件以清除对应的记录",
-			})
-			return
-		}
-	}
-	if fileName == "" {
-		fileName = filepath.Base(file.FilePath)
-	}
-
-	err = dao.AddReceivedFile(&model.ReceivedFile{
-		FileUniqueID:   file.FileUniqueID,
-		FileID:         file.FileID,
-		Processing:     false,
-		FileName:       fileName,
-		FilePath:       file.FilePath,
-		FileSize:       file.FileSize,
-		MediaGroupID:   message.MediaGroupID,
-		ChatID:         message.Chat.ChatID().ID,
-		MessageID:      message.MessageID,
-		ReplyMessageID: msg.MessageID,
-	})
-
-	if err != nil {
-		logger.L.Error(err)
-		ReplyMessage(message, "创建任务失败")
-		return
-	}
-
-	if !user.Silent {
-		bot.EditMessageText(&telego.EditMessageTextParams{
-			ChatID:      message.Chat.ChatID(),
-			MessageID:   msg.MessageID,
-			Text:        "选择要保存的位置",
-			ReplyMarkup: AddTaskReplyMarkup(message.MessageID),
-		})
-		return
-	}
-
-	if user.DefaultStorage == "" {
-		bot.EditMessageText(&telego.EditMessageTextParams{
-			ChatID:    message.Chat.ChatID(),
-			MessageID: msg.MessageID,
-			Text:      "请先使用 /storage 命令设置默认存储位置, 或者关闭静默模式",
-		})
-		return
-	}
-
-	queue.AddTask(types.Task{
-		Ctx:            context.TODO(),
-		FileID:         file.FileID,
-		Status:         types.Pending,
-		FileName:       fileName,
-		FilePath:       file.FilePath,
-		Storage:        types.StorageType(user.DefaultStorage),
-		ChatID:         message.Chat.ChatID().ID,
-		ReplyMessageID: msg.MessageID,
-	})
-}
-
-func AddToQueue(ctx context.Context, bot *telego.Bot, query telego.CallbackQuery) {
-	args := strings.Split(query.Data, " ")
-	messageID, _ := strconv.Atoi(args[1])
-	receivedFile, err := dao.GetReceivedFileByChatAndMessageID(query.Message.GetChat().ID, messageID)
-	if err != nil {
-		logger.L.Error(err)
-		bot.AnswerCallbackQuery(telegoutil.CallbackQuery(query.ID).WithShowAlert().WithText("获取文件信息失败").WithCacheTime(5))
-		return
-	}
-	queue.AddTask(types.Task{
-		Ctx:            context.TODO(),
-		FileID:         receivedFile.FileID,
-		Status:         types.Pending,
-		FileName:       receivedFile.FileName,
-		FilePath:       receivedFile.FilePath,
-		Storage:        types.StorageType(args[2]),
-		ChatID:         receivedFile.ChatID,
-		ReplyMessageID: receivedFile.ReplyMessageID,
-	})
-	bot.EditMessageText(&telego.EditMessageTextParams{
-		Text:      fmt.Sprintf("已添加到队列, 当前排队中的任务数: %d", queue.Len()),
-		MessageID: query.Message.GetMessageID(),
-		ChatID:    telegoutil.ID(receivedFile.ChatID),
-	})
+	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.NewCallbackQuery(filters.CallbackQuery.Prefix("cancel"), cancelTask))
+	dispatcher.AddHandler(handlers.NewMessage(filters.Message.Media, handleFileMessage))
 }

bot/middleware.go

@@ -1 +0,0 @@
-package bot

bot/middlewares.go

@@ -0,0 +1,37 @@
+package bot
+
+import (
+	"time"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/duke-git/lancet/v2/slice"
+	"github.com/gotd/contrib/middleware/floodwait"
+	"github.com/gotd/contrib/middleware/ratelimit"
+	"github.com/gotd/td/telegram"
+	"github.com/krau/SaveAny-Bot/config"
+	"golang.org/x/time/rate"
+)
+
+func FloodWaitMiddleware() []telegram.Middleware {
+	waiter := floodwait.NewSimpleWaiter().WithMaxRetries(5)
+	ratelimiter := ratelimit.New(rate.Every(time.Millisecond*100), 5)
+	return []telegram.Middleware{
+		waiter,
+		ratelimiter,
+	}
+}
+
+const noPermissionText string = `
+您不在白名单中, 无法使用此 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.GetUsersID(), userID) {
+		ctx.Reply(update, ext.ReplyTextString(noPermissionText), nil)
+		return dispatcher.EndGroups
+	}
+	return dispatcher.ContinueGroups
+}

bot/utils.go

@@ -1,52 +1,273 @@
 package bot
 
 import (
+	"errors"
 	"fmt"
-	"regexp"
+	"time"
 
+	"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/mymmrac/telego"
-	"github.com/mymmrac/telego/telegoutil"
+	"github.com/krau/SaveAny-Bot/types"
 )
 
-func FileDownloadURL(filePath string) string {
-	return Bot.FileDownloadURL(filePath)
+var (
+	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")
+	ErrEmptyMessage    = errors.New("message is empty")
+)
+
+func supportedMediaFilter(m *tg.Message) (bool, error) {
+	if not := m.Media == nil; not {
+		return false, dispatcher.EndGroups
+	}
+	switch m.Media.(type) {
+	case *tg.MessageMediaDocument:
+		return true, nil
+	case *tg.MessageMediaPhoto:
+		return true, nil
+	default:
+		return false, nil
+	}
 }
 
-func ReplyMessage(replyTo telego.Message, format string, args ...any) (*telego.Message, error) {
-	return Bot.SendMessage(telegoutil.Messagef(replyTo.Chat.ChatID(), format, args...).
-		WithReplyParameters(&telego.ReplyParameters{
-			MessageID: replyTo.MessageID,
-		}))
-}
-
-var StorageDisplayNames = map[string]string{
-	"all":    "全部",
-	"local":  "服务器磁盘",
-	"alist":  "Alist",
-	"webdav": "WebDAV",
-}
-
-func AddTaskReplyMarkup(messageID int) *telego.InlineKeyboardMarkup {
-	storageButtons := make([]telego.InlineKeyboardButton, 0)
-	for name := range storage.Storages {
-		storageButtons = append(storageButtons, telegoutil.InlineKeyboardButton(StorageDisplayNames[string(name)]).
-			WithCallbackData(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, fmt.Errorf("failed to get user by chat ID: %d, error: %w", userChatID, err)
+	}
+	storages := storage.GetUserStorages(user.ChatID)
+	if len(storages) == 0 {
+		return nil, ErrNoStorages
 	}
 
-	if len(storageButtons) > 1 {
-		return telegoutil.InlineKeyboard(storageButtons, []telego.InlineKeyboardButton{
-			telegoutil.InlineKeyboardButton("全部").WithCallbackData(fmt.Sprintf("add %d all", messageID)),
+	buttons := make([]tg.KeyboardButtonClass, 0)
+	for _, storage := range storages {
+		cbData := fmt.Sprintf("%d %d %s 0", fileChatID, fileMessageID, storage.Name()) // 0 for empty dir id
+		cbDataId, err := dao.CreateCallbackData(cbData)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create callback data: %w", err)
+		}
+		buttons = append(buttons, &tg.KeyboardButtonCallback{
+			Text: storage.Name(),
+			Data: []byte(fmt.Sprintf("add %d", cbDataId)),
 		})
 	}
-	if len(storageButtons) == 1 {
-		return telegoutil.InlineKeyboard(storageButtons)
+	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 nil
+	return markup, nil
 }
 
-var markdownRe = regexp.MustCompile("([" + regexp.QuoteMeta(`\_*[]()~`+"`"+`>#+-=|{}.!`) + "])")
-
-func EscapeMarkdown(text string) string {
-	return markdownRe.ReplaceAllString(text, "\\$1")
+func getSelectDirMarkup(fileChatID, fileMessageID int, storageName string, dirs []dao.Dir) (*tg.ReplyInlineMarkup, error) {
+	buttons := make([]tg.KeyboardButtonClass, 0)
+	for _, dir := range dirs {
+		if dir.ID == 0 || dir.StorageName != storageName {
+			return nil, fmt.Errorf("unexpected dir: %v", dir)
+		}
+		cbDataId, err := dao.CreateCallbackData(fmt.Sprintf("%d %d %s %d", fileChatID, fileMessageID, storageName, dir.ID))
+		if err != nil {
+			return nil, fmt.Errorf("failed to create callback data: %w", err)
+		}
+		buttons = append(buttons, &tg.KeyboardButtonCallback{
+			Text: dir.Path,
+			Data: []byte(fmt.Sprintf("add_to_dir %d", cbDataId)),
+		})
+	}
+	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
+}
+
+func getSetDefaultStorageMarkup(userChatID int64, storages []storage.Storage) (*tg.ReplyInlineMarkup, error) {
+	buttons := make([]tg.KeyboardButtonClass, 0)
+	for _, storage := range storages {
+		cbDataId, err := dao.CreateCallbackData(storage.Name())
+		if err != nil {
+			return nil, fmt.Errorf("failed to create callback data: %w", err)
+		}
+		buttons = append(buttons, &tg.KeyboardButtonCallback{
+			Text: storage.Name(),
+			Data: []byte(fmt.Sprintf("set_default %d %d", userChatID, cbDataId)),
+		})
+	}
+	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
+}
+
+func FileFromMedia(media tg.MessageMediaClass, customFileName string) (*types.File, error) {
+	switch media := media.(type) {
+	case *tg.MessageMediaDocument:
+		document, ok := media.Document.AsNotEmpty()
+		if !ok {
+			return nil, ErrEmptyDocument
+		}
+		if customFileName != "" {
+			return &types.File{
+				Location: document.AsInputDocumentFileLocation(),
+				FileSize: document.Size,
+				FileName: customFileName,
+			}, nil
+		}
+		fileName := ""
+		for _, attribute := range document.Attributes {
+			if name, ok := attribute.(*tg.DocumentAttributeFilename); ok {
+				fileName = name.GetFileName()
+				break
+			}
+		}
+		return &types.File{
+			Location: document.AsInputDocumentFileLocation(),
+			FileSize: document.Size,
+			FileName: fileName,
+		}, nil
+	case *tg.MessageMediaPhoto:
+		photo, ok := media.Photo.AsNotEmpty()
+		if !ok {
+			return nil, ErrEmptyPhoto
+		}
+		sizes := photo.Sizes
+		if len(sizes) == 0 {
+			return nil, ErrEmptyPhotoSizes
+		}
+		photoSize := sizes[len(sizes)-1]
+		size, ok := photoSize.AsNotEmpty()
+		if !ok {
+			return nil, ErrEmptyPhotoSize
+		}
+		location := new(tg.InputPhotoFileLocation)
+		location.ID = photo.GetID()
+		location.AccessHash = photo.GetAccessHash()
+		location.FileReference = photo.GetFileReference()
+		location.ThumbSize = size.GetType()
+		fileName := customFileName
+		if fileName == "" {
+			fileName = fmt.Sprintf("photo_%s_%d.jpg", time.Now().Format("2006-01-02_15-04-05"), photo.GetID())
+		}
+		return &types.File{
+			Location: location,
+			FileSize: 0,
+			FileName: fileName,
+		}, nil
+
+	}
+	return nil, fmt.Errorf("unexpected type %T", media)
+}
+
+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
+	err := common.Cache.Get(key, &cachedFile)
+	if err == nil {
+		return &cachedFile, nil
+	}
+	message, err := GetTGMessage(ctx, chatID, messageID)
+	if err != nil {
+		return nil, err
+	}
+	file, err := FileFromMedia(message.Media, customFileName)
+	if err != nil {
+		return nil, err
+	}
+	if err := common.Cache.Set(key, file, 3600); err != nil {
+		logger.L.Errorf("Failed to cache file: %s", err)
+	}
+	return file, nil
+}
+
+func GetTGMessage(ctx *ext.Context, chatId int64, messageID int) (*tg.Message, error) {
+	logger.L.Debugf("Fetching message: %d", messageID)
+	messages, err := ctx.GetMessages(chatId, []tg.InputMessageClass{&tg.InputMessageID{ID: messageID}})
+	if err != nil {
+		return nil, err
+	}
+	if len(messages) == 0 {
+		return nil, ErrEmptyMessage
+	}
+	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.GetUserChat().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 *dao.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
 }

cmd/root.go

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

cmd/run.go

@@ -1,13 +1,68 @@
 package cmd
 
 import (
-	"github.com/krau/SaveAny-Bot/bootstrap"
+	"fmt"
+	"os"
+	"os/signal"
+	"path/filepath"
+	"syscall"
+
 	"github.com/krau/SaveAny-Bot/bot"
+	"github.com/krau/SaveAny-Bot/common"
+	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/core"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/storage"
+	"github.com/spf13/cobra"
 )
 
-func Run() {
-	bootstrap.InitAll()
-	go core.Run()
-	bot.Run()
+func Run(_ *cobra.Command, _ []string) {
+	InitAll()
+	core.Run()
+
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	sig := <-quit
+	logger.L.Info(sig, ", exitting...")
+	defer logger.L.Info("Bye!")
+	if config.Cfg.NoCleanCache {
+		return
+	}
+	if config.Cfg.Temp.BasePath != "" {
+		for _, path := range []string{"/", ".", "\\", ".."} {
+			if filepath.Clean(config.Cfg.Temp.BasePath) == path {
+				logger.L.Error("Invalid cache dir: ", config.Cfg.Temp.BasePath)
+				return
+			}
+		}
+		currentDir, err := os.Getwd()
+		if err != nil {
+			logger.L.Error("Failed to get current dir: ", err)
+			return
+		}
+		cachePath := filepath.Join(currentDir, config.Cfg.Temp.BasePath)
+		cachePath, err = filepath.Abs(cachePath)
+		if err != nil {
+			logger.L.Error("Failed to get absolute path: ", err)
+			return
+		}
+		logger.L.Info("Cleaning cache dir: ", cachePath)
+		if err := os.RemoveAll(cachePath); err != nil {
+			logger.L.Error("Failed to clean cache dir: ", err)
+		}
+	}
+}
+
+func InitAll() {
+	if err := config.Init(); err != nil {
+		fmt.Println("加载配置文件失败: ", err)
+		os.Exit(1)
+	}
+	logger.InitLogger()
+	logger.L.Info("正在启动 SaveAny-Bot...")
+	dao.Init()
+	storage.LoadStorages()
+	common.Init()
+	bot.Init()
 }

common/cache.go

@@ -0,0 +1,60 @@
+package common
+
+import (
+	"bytes"
+	"encoding/gob"
+	"sync"
+
+	"github.com/coocood/freecache"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+type CommonCache struct {
+	cache *freecache.Cache
+	mu    sync.RWMutex
+}
+
+var Cache *CommonCache
+
+func initCache() {
+	gob.Register(types.File{})
+	gob.Register(tg.InputDocumentFileLocation{})
+	gob.Register(tg.InputPhotoFileLocation{})
+	Cache = &CommonCache{cache: freecache.NewCache(10 * 1024 * 1024)}
+}
+
+func (c *CommonCache) Get(key string, value *types.File) error {
+	c.mu.RLock()
+	defer c.mu.RUnlock()
+	data, err := Cache.cache.Get([]byte(key))
+	if err != nil {
+		return err
+	}
+	dec := gob.NewDecoder(bytes.NewReader(data))
+	err = dec.Decode(&value)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *CommonCache) Set(key string, value *types.File, expireSeconds int) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	var buf bytes.Buffer
+	enc := gob.NewEncoder(&buf)
+	err := enc.Encode(value)
+	if err != nil {
+		return err
+	}
+	Cache.cache.Set([]byte(key), buf.Bytes(), expireSeconds)
+	return nil
+}
+
+func (c *CommonCache) Delete(key string) error {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	Cache.cache.Del([]byte(key))
+	return nil
+}

common/common.go

@@ -1,5 +1,5 @@
 package common
 
 func Init() {
-	initClient()
+	initCache()
 }

common/os.go

@@ -18,11 +18,6 @@ func MkFile(path string, data []byte) error {
 	return os.WriteFile(path, data, os.ModePerm)
 }
 
-func FileExists(path string) bool {
-	_, err := os.Stat(path)
-	return err == nil
-}
-
 // 删除文件, 并清理空目录. 如果文件不存在则返回 nil
 func PurgeFile(path string) error {
 	if err := os.Remove(path); err != nil {
@@ -33,6 +28,18 @@ func PurgeFile(path string) error {
 	return RemoveEmptyDirectories(filepath.Dir(path))
 }
 
+func RmFileAfter(path string, td time.Duration) {
+	_, err := os.Stat(path)
+	if err != nil {
+		logger.L.Errorf("Failed to create timer for %s: %s", path, err)
+		return
+	}
+	logger.L.Debugf("Remove file after %s: %s", td, path)
+	time.AfterFunc(td, func() {
+		PurgeFile(path)
+	})
+}
+
 // 递归删除空目录
 func RemoveEmptyDirectories(dirPath string) error {
 	entries, err := os.ReadDir(dirPath)
@@ -48,24 +55,3 @@ func RemoveEmptyDirectories(dirPath string) error {
 	}
 	return nil
 }
-
-// 在指定时间后删除和清理文件 (定时器)
-func PurgeFileAfter(path string, td time.Duration) {
-	_, err := os.Stat(path)
-	if err != nil {
-		logger.L.Errorf("Failed to create timer for %s: %s", path, err)
-		return
-	}
-	logger.L.Debugf("Purge file after %s: %s", td, path)
-	time.AfterFunc(td, func() {
-		PurgeFile(path)
-	})
-}
-
-func MkCache(path string, data []byte, td time.Duration) {
-	if err := MkFile(path, data); err != nil {
-		logger.L.Errorf("failed to save cache file: %s", err)
-	} else {
-		go PurgeFileAfter(path, td)
-	}
-}

common/req.go

@@ -1,19 +0,0 @@
-package common
-
-import (
-	"path/filepath"
-	"time"
-
-	"github.com/imroc/req/v3"
-	"github.com/krau/SaveAny-Bot/config"
-)
-
-var ReqClient *req.Client
-
-func initClient() {
-	ReqClient = req.NewClient().SetOutputDirectory(config.Cfg.Temp.BasePath).SetTimeout(86400 * time.Second)
-}
-
-func GetDownloadedFilePath(filename string) string {
-	return filepath.Join(config.Cfg.Temp.BasePath, filename)
-}

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

@@ -1,34 +1,93 @@
+#创建文件时，若需要保留中文注释，请务必确保本文件编码为 UTF-8 ，否则会无法读取。
+workers = 4 # 同时下载文件数
+retry = 3   # 下载失败重试次数
+threads = 4 # 单个任务下载最大线程数
+
 [telegram]
-token = ""                       # Bot Token
-admins = [777000]                # 你的 user_id
-api = "https://api.telegram.org"
+# Bot Token
+# 更换 Bot Token 后请删除数据库文件和 session.db
+token = ""
+# Telegram API 配置, 若不配置也可运行, 将使用默认的 API ID 和 API HASH
+# 推荐使用自己的 API ID 和 API HASH (https://my.telegram.org)
+# app_id = 123456
+# app_hash = "0123456789abcdef0123456789abcdef"
 
-[log]
-level = "DEBUG" # 日志等级
+[telegram.proxy]
+# 启用代理连接 telegram, 只支持 socks5
+enable = false
+url = "socks5://127.0.0.1:7890"
 
-[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
-enable = true
-base_path = "/telegram"
-username = "admin"
-password = "password"
-url = "https://alist.com/dav"
+[[storages]]
+name = "MyAlist"
+type = "alist"
+enable = false  #记得启用
+base_path = '/'
+url = 'https://alist.com'
+username = 'admin'
+password = 'password'
+token_exp = 86400  # 86400--1天   604800--7天   1296000--15天   2592000--30天   15552000--180天
+# alist 可直接使用 token 登录, 此时 username, password, token_exp 将被忽略
+# 请自行在 alist 侧配置合理的 token 过期时间
+# token = ""
+
+
+[[storages]]
+name = "MyWebdav"
+type = "webdav"
+enable = false
+base_path = '/path/telegram'
+url = 'https://example.com/dav'
+username = 'username'
+password = 'password'
+
+
+# 用户列表
+[[users]]
+# telegram user id
+id = 114514
+# 开启黑名单，开启后下方留空以使用所有存储，反之则为白名单，白名单请在下方输入允许的存储名
+blacklist = true
+# 将列表留空并开启黑名单模式以允许使用所有存储，此处示例为黑名单模式，用户114514 可使用所有存储
+storages = []
+
+
+[[users]]
+id = 123456
+blacklist = false  #开启白名单模式，此时，用户123456 仅可使用下方列表中的存储
+# 此时该用户只能使用名为 本机1 的存储
+storages = ["本机1"]
+
+
+
+# 其他配置
+
+# [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

@@ -3,27 +3,37 @@ package config
 import (
 	"fmt"
 	"os"
+	"strings"
 
 	"github.com/spf13/viper"
 )
 
 type Config struct {
-	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"`
+	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"`
+	Threads      int  `toml:"threads" mapstructure:"threads" json:"threads"`
+
+	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"`
+	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 {
@@ -31,45 +41,38 @@ type dbConfig struct {
 }
 
 type telegramConfig struct {
-	Token  string  `toml:"token" mapstructure:"token"`
-	API    string  `toml:"api" mapstructure:"api"`
+	Token   string      `toml:"token" mapstructure:"token"`
+	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 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"`
+type proxyConfig struct {
+	Enable bool   `toml:"enable" mapstructure:"enable"`
+	URL    string `toml:"url" mapstructure:"url"`
 }
 
 var Cfg *Config
 
-func Init() {
+func Init() error {
 	viper.SetConfigName("config")
 	viper.AddConfigPath(".")
+	viper.AddConfigPath("/etc/saveany/")
 	viper.SetConfigType("toml")
+	viper.SetEnvPrefix("SAVEANY")
+	viper.AutomaticEnv()
+	replacer := strings.NewReplacer(".", "_")
+	viper.SetEnvKeyReplacer(replacer)
+
+	viper.SetDefault("workers", 3)
+	viper.SetDefault("retry", 3)
+	viper.SetDefault("threads", 4)
+
+	viper.SetDefault("telegram.app_id", 1025907)
+	viper.SetDefault("telegram.app_hash", "452b0359b988148995f22ff0f4229750")
 
 	viper.SetDefault("temp.base_path", "cache/")
 	viper.SetDefault("temp.cache_ttl", 3600)
@@ -80,10 +83,11 @@ func Init() {
 
 	viper.SetDefault("db.path", "data/saveany.db")
 
-	viper.SetDefault("telegram.api", "https://api.telegram.org")
-
-	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)
@@ -91,8 +95,77 @@ func Init() {
 	}
 
 	Cfg = &Config{}
+
 	if err := viper.Unmarshal(Cfg); err != nil {
 		fmt.Println("Error unmarshalling config file, ", err)
 		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) {
+	viper.Set(key, value)
+}
+
+func ReloadConfig() error {
+	if err := viper.WriteConfig(); err != nil {
+		return err
+	}
+	if err := viper.ReadInConfig(); err != nil {
+		return err
+	}
+	if error := viper.Unmarshal(Cfg); error != nil {
+		return error
+	}
+	return nil
 }

core/core.go

@@ -4,92 +4,30 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"time"
 
-	"github.com/imroc/req/v3"
-	"github.com/krau/SaveAny-Bot/bot"
-	"github.com/krau/SaveAny-Bot/common"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/tg"
 	"github.com/krau/SaveAny-Bot/config"
-	"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"
-	"github.com/mymmrac/telego"
-	"github.com/mymmrac/telego/telegoutil"
 )
 
-func processPendingTask(task types.Task) error {
-	fileRecord, err := dao.GetReceivedFileByFileID(task.FileID)
-	if err != nil {
-		return err
-	}
-	fileRecord.Processing = true
-	if err := dao.UpdateReceivedFile(fileRecord); err != nil {
-		return err
-	}
-
-	_, err = common.ReqClient.R().SetOutputFile(task.FileName).SetDownloadCallbackWithInterval(func(info req.DownloadInfo) {
-		if info.Response == nil || info.Response.Response == nil || info.Response.Response.StatusCode != 200 {
-			return
-		}
-		process := fmt.Sprintf("%.2f%%", float64(info.DownloadedSize)/float64(info.Response.ContentLength)*100.0)
-		bot.Bot.EditMessageText(&telego.EditMessageTextParams{
-			ChatID:    telegoutil.ID(task.ChatID),
-			MessageID: task.ReplyMessageID,
-			Text:      "正在下载文件: " + process,
-		})
-	}, time.Second*time.Duration(3*func() int {
-		if queue.Len() > 0 {
-			return queue.Len()
-		}
-		return 1
-	}())).Get(bot.FileDownloadURL(task.FilePath))
-	if err != nil {
-		return err
-	}
-
-	bot.Bot.EditMessageText(&telego.EditMessageTextParams{
-		ChatID:    telegoutil.ID(task.ChatID),
-		MessageID: task.ReplyMessageID,
-		Text:      "下载完成, 正在转存...",
-	})
-
-	defer func() {
-		if config.Cfg.Temp.CacheTTL > 0 {
-			common.PurgeFileAfter(common.GetDownloadedFilePath(task.FileName),
-				time.Duration(config.Cfg.Temp.CacheTTL)*time.Second)
-		} else {
-			if err := common.PurgeFile(common.GetDownloadedFilePath(task.FileName)); err != nil {
-				logger.L.Errorf("Failed to purge file: %s", err)
-			}
-		}
-	}()
-	if task.StoragePath == "" {
-		task.StoragePath = task.FileName
-	}
-
-	if err := storage.Save(task.Storage, task.Ctx, common.GetDownloadedFilePath(task.FileName), task.StoragePath); err != nil {
-		return err
-	}
-	return nil
-}
-
 func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
 	for {
 		semaphore <- struct{}{}
 		task := queue.GetTask()
-		logger.L.Debugf("Got task: %s", task.FileName)
+		logger.L.Debugf("Got task: %s", task.String())
 
 		switch task.Status {
 		case types.Pending:
 			logger.L.Infof("Processing task: %s", task.String())
 			if err := processPendingTask(task); err != nil {
-				logger.L.Errorf("Failed to do task: %s", err)
 				task.Error = err
 				if errors.Is(err, context.Canceled) {
 					task.Status = types.Canceled
 				} else {
+					logger.L.Errorf("Failed to do task: %s", err)
 					task.Status = types.Failed
 				}
 			} else {
@@ -98,38 +36,51 @@ func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
 			queue.AddTask(task)
 		case types.Succeeded:
 			logger.L.Infof("Task succeeded: %s", task.String())
-			bot.Bot.EditMessageText(&telego.EditMessageTextParams{
-				ChatID:    telegoutil.ID(task.ChatID),
-				MessageID: task.ReplyMessageID,
-				Text:      "文件转存完成",
-			})
-			if err := dao.DeleteReceivedFileByFileID(task.FileID); err != nil {
-				logger.L.Errorf("Failed to delete received file: %s", err)
+			extCtx, ok := task.Ctx.(*ext.Context)
+			if !ok {
+				logger.L.Errorf("Context is not *ext.Context: %T", task.Ctx)
+			} else {
+				extCtx.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())
-			bot.Bot.EditMessageText(&telego.EditMessageTextParams{
-				ChatID:    telegoutil.ID(task.ChatID),
-				MessageID: task.ReplyMessageID,
-				Text:      "文件转存失败:" + "\n" + task.Error.Error(),
-			})
-			if err := dao.DeleteReceivedFileByFileID(task.FileID); err != nil {
-				logger.L.Errorf("Failed to delete received file: %s", err)
+			extCtx, ok := task.Ctx.(*ext.Context)
+			if !ok {
+				logger.L.Errorf("Context is not *ext.Context: %T", task.Ctx)
+			} else {
+				extCtx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
+					Message: "文件保存失败\n" + task.Error.Error(),
+					ID:      task.ReplyMessageID,
+				})
 			}
 		case types.Canceled:
 			logger.L.Infof("Task canceled: %s", task.String())
+			extCtx, ok := task.Ctx.(*ext.Context)
+			if !ok {
+				logger.L.Errorf("Context is not *ext.Context: %T", task.Ctx)
+			} else {
+				extCtx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
+					Message: "任务已取消",
+					ID:      task.ReplyMessageID,
+				})
+			}
 		default:
 			logger.L.Errorf("Unknown task status: %s", task.Status)
 		}
 		<-semaphore
-		logger.L.Debugf("Task done: %s", task.FileName)
+		logger.L.Debugf("Task done: %s; status: %s", task.String(), task.Status)
+		queue.DoneTask(task)
 	}
 }
 
 func Run() {
 	logger.L.Info("Start processing tasks...")
-	semaphore := make(chan struct{}, 3)
-	for i := 0; i < 3; i++ {
+	semaphore := make(chan struct{}, config.Cfg.Workers)
+	for i := 0; i < config.Cfg.Workers; i++ {
 		go worker(queue.Queue, semaphore)
 	}
+
 }

core/download.go

@@ -0,0 +1,87 @@
+package core
+
+import (
+	"context"
+	"fmt"
+	"path/filepath"
+	"time"
+
+	"github.com/celestix/gotgproto/ext"
+	"github.com/duke-git/lancet/v2/fileutil"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/bot"
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/logger"
+	"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("处理路径失败: %w", err)
+	}
+	if err := fileutil.CreateDir(filepath.Dir(cacheDestPath)); err != nil {
+		return fmt.Errorf("创建目录失败: %w", err)
+	}
+
+	if task.StoragePath == "" {
+		task.StoragePath = task.File.FileName
+	}
+
+	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, taskStorage, cacheDestPath)
+	}
+
+	ctx, ok := task.Ctx.(*ext.Context)
+	if !ok {
+		return fmt.Errorf("context is not *ext.Context: %T", task.Ctx)
+	}
+
+	cancelCtx, cancel := context.WithCancel(ctx)
+	task.Cancel = cancel
+
+	text, entities := buildProgressMessageEntity(task, 0, task.StartTime, 0)
+	ctx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
+		Message:     text,
+		Entities:    entities,
+		ID:          task.ReplyMessageID,
+		ReplyMarkup: getCancelTaskMarkup(task),
+	})
+	progressCallback := buildProgressCallback(ctx, task, getProgressUpdateCount(task.File.FileSize))
+
+	dest, err := NewTaskLocalFile(cacheDestPath, task.File.FileSize, progressCallback)
+	if err != nil {
+		return fmt.Errorf("创建文件失败: %w", err)
+	}
+	defer dest.Close()
+	task.StartTime = time.Now()
+	downloadBuider := Downloader.Download(bot.Client.API(), task.File.Location).WithThreads(getTaskThreads(task.File.FileSize))
+	_, err = downloadBuider.Parallel(cancelCtx, dest)
+	if err != nil {
+		return fmt.Errorf("下载文件失败: %w", err)
+	}
+
+	defer cleanCacheFile(cacheDestPath)
+
+	fixTaskFileExt(task, cacheDestPath)
+
+	logger.L.Infof("Downloaded file: %s", cacheDestPath)
+	ctx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
+		Message: fmt.Sprintf("下载完成: %s\n正在转存文件...", task.FileName()),
+		ID:      task.ReplyMessageID,
+	})
+
+	return saveFileWithRetry(cancelCtx, task, taskStorage, cacheDestPath)
+}

core/downloader.go

@@ -0,0 +1,9 @@
+package core
+
+import "github.com/gotd/td/telegram/downloader"
+
+var Downloader *downloader.Downloader
+
+func init() {
+	Downloader = downloader.NewDownloader().WithPartSize(1024 * 1024)
+}

core/utils.go

@@ -0,0 +1,235 @@
+package core
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"path"
+	"time"
+
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gabriel-vasile/mimetype"
+	"github.com/gotd/td/telegram/message/entity"
+	"github.com/gotd/td/telegram/message/styling"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/bot"
+	"github.com/krau/SaveAny-Bot/common"
+	"github.com/krau/SaveAny-Bot/config"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/storage"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+func saveFileWithRetry(ctx context.Context, task *types.Task, taskStorage storage.Storage, localFilePath string) error {
+	for i := 0; i <= config.Cfg.Retry; i++ {
+		if err := ctx.Err(); err != nil {
+			return fmt.Errorf("context canceled while saving file: %w", err)
+		}
+		if err := taskStorage.Save(ctx, localFilePath, task.StoragePath); err != nil {
+			if i == config.Cfg.Retry {
+				return fmt.Errorf("failed to save file: %w", err)
+			}
+			logger.L.Errorf("Failed to save file: %s, retrying...", err)
+			select {
+			case <-ctx.Done():
+				return fmt.Errorf("context canceled during retry delay: %w", ctx.Err())
+			case <-time.After(time.Duration(i*500) * time.Millisecond):
+			}
+			continue
+		}
+		return nil
+	}
+	return nil
+}
+
+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,
+		Limit:    1024 * 1024,
+	})
+	if err != nil {
+		return fmt.Errorf("failed to get file: %w", err)
+	}
+
+	result, ok := res.(*tg.UploadFile)
+	if !ok {
+		return fmt.Errorf("unexpected type %T", res)
+	}
+
+	if err := os.WriteFile(cachePath, result.Bytes, os.ModePerm); err != nil {
+		return fmt.Errorf("failed to write file: %w", err)
+	}
+
+	defer cleanCacheFile(cachePath)
+
+	logger.L.Infof("Downloaded file: %s", cachePath)
+
+	return saveFileWithRetry(task.Ctx, task, taskStorage, cachePath)
+}
+
+// func getProgressBar(progress float64, updateCount int) string {
+// 	bar := ""
+// 	barSize := 100 / updateCount
+// 	for i := 0; i < updateCount; i++ {
+// 		if progress >= float64(barSize*(i+1)) {
+// 			bar += "█"
+// 		} else {
+// 			bar += "░"
+// 		}
+// 	}
+// 	return bar
+// }
+
+func cleanCacheFile(destPath string) {
+	if config.Cfg.Temp.CacheTTL > 0 {
+		common.RmFileAfter(destPath, time.Duration(config.Cfg.Temp.CacheTTL)*time.Second)
+	} else {
+		if err := os.Remove(destPath); err != nil {
+			logger.L.Errorf("Failed to purge file: %s", err)
+		}
+	}
+}
+
+// 获取进度需要更新的次数
+func getProgressUpdateCount(fileSize int64) int {
+	updateCount := 5
+	if fileSize > 1024*1024*1000 {
+		updateCount = 50
+	} else if fileSize > 1024*1024*500 {
+		updateCount = 20
+	} else if fileSize > 1024*1024*200 {
+		updateCount = 10
+	}
+	return updateCount
+}
+
+func getSpeed(bytesRead int64, startTime time.Time) string {
+	if startTime.IsZero() {
+		return "0MB/s"
+	}
+	elapsed := time.Since(startTime)
+	speed := float64(bytesRead) / 1024 / 1024 / elapsed.Seconds()
+	return fmt.Sprintf("%.2fMB/s", speed)
+}
+
+func buildProgressMessageEntity(task *types.Task, bytesRead int64, startTime time.Time, progress float64) (string, []tg.MessageEntityClass) {
+	entityBuilder := entity.Builder{}
+	text := fmt.Sprintf("正在处理下载任务\n文件名: %s\n保存路径: %s\n平均速度: %s\n当前进度: %.2f%%",
+		task.FileName(),
+		fmt.Sprintf("[%s]:%s", task.StorageName, task.StoragePath),
+		getSpeed(bytesRead, startTime),
+		progress,
+	)
+	var entities []tg.MessageEntityClass
+	if err := styling.Perform(&entityBuilder,
+		styling.Plain("正在处理下载任务\n文件名: "),
+		styling.Code(task.FileName()),
+		styling.Plain("\n保存路径: "),
+		styling.Code(fmt.Sprintf("[%s]:%s", task.StorageName, task.StoragePath)),
+		styling.Plain("\n平均速度: "),
+		styling.Bold(getSpeed(bytesRead, task.StartTime)),
+		styling.Plain("\n当前进度: "),
+		styling.Bold(fmt.Sprintf("%.2f%%", progress)),
+	); err != nil {
+		logger.L.Errorf("Failed to build entities: %s", err)
+		return text, entities
+	}
+	return entityBuilder.Complete()
+}
+
+func buildProgressCallback(ctx *ext.Context, task *types.Task, updateCount int) func(bytesRead, contentLength int64) {
+	return func(bytesRead, contentLength int64) {
+		progress := float64(bytesRead) / float64(contentLength) * 100
+		logger.L.Tracef("Downloading %s: %.2f%%", task.String(), progress)
+		progressInt := int(progress)
+		if task.File.FileSize < 1024*1024*50 || progressInt == 0 || progressInt%int(100/updateCount) != 0 {
+			return
+		}
+		text, entities := buildProgressMessageEntity(task, bytesRead, task.StartTime, progress)
+		ctx.EditMessage(task.ReplyChatID, &tg.MessagesEditMessageRequest{
+			Message:     text,
+			Entities:    entities,
+			ID:          task.ReplyMessageID,
+			ReplyMarkup: getCancelTaskMarkup(task),
+		})
+	}
+}
+
+func getCancelTaskMarkup(task *types.Task) *tg.ReplyInlineMarkup {
+	return &tg.ReplyInlineMarkup{
+		Rows: []tg.KeyboardButtonRow{{Buttons: []tg.KeyboardButtonClass{&tg.KeyboardButtonCallback{Text: "取消任务", Data: fmt.Appendf(nil, "cancel %s", task.Key())}}}},
+	}
+}
+
+func fixTaskFileExt(task *types.Task, localFilePath string) {
+	if path.Ext(task.FileName()) == "" {
+		mimeType, err := mimetype.DetectFile(localFilePath)
+		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())
+		}
+	}
+}
+
+func getTaskThreads(fileSize int64) int {
+	threads := 1
+	if fileSize > 1024*1024*100 {
+		threads = config.Cfg.Threads
+	} else if fileSize > 1024*1024*50 {
+		threads = config.Cfg.Threads / 2
+	}
+	return threads
+}
+
+type TaskLocalFile struct {
+	file             *os.File
+	size             int64
+	done             int64
+	progressCallback func(bytesRead, contentLength int64)
+	callbackTimes    int64
+	nextCallbackAt   int64
+	callbackInterval int64
+}
+
+func (t *TaskLocalFile) Read(p []byte) (n int, err error) {
+	return t.file.Read(p)
+}
+
+func (t *TaskLocalFile) Close() error {
+	return t.file.Close()
+}
+func (t *TaskLocalFile) WriteAt(p []byte, off int64) (int, error) {
+	n, err := t.file.WriteAt(p, off)
+	if err != nil {
+		return n, err
+	}
+	t.done += int64(n)
+	if t.progressCallback != nil && t.done >= t.nextCallbackAt {
+		t.progressCallback(t.done, t.size)
+		t.nextCallbackAt += t.callbackInterval
+	}
+	return n, nil
+}
+
+func NewTaskLocalFile(filePath string, fileSize int64, progressCallback func(bytesRead, contentLength int64)) (*TaskLocalFile, error) {
+	file, err := os.Create(filePath)
+	if err != nil {
+		return nil, fmt.Errorf("failed to open file: %w", err)
+	}
+	var callbackInterval int64
+	callbackInterval = fileSize / 100
+	if callbackInterval == 0 {
+		callbackInterval = 1
+	}
+	return &TaskLocalFile{
+		file:             file,
+		size:             fileSize,
+		progressCallback: progressCallback,
+		callbackTimes:    100,
+		nextCallbackAt:   callbackInterval,
+		callbackInterval: callbackInterval,
+	}, nil
+}

dao/callback_data.go

@@ -0,0 +1,19 @@
+package dao
+
+func CreateCallbackData(data string) (uint, error) {
+	callbackData := CallbackData{
+		Data: data,
+	}
+	err := db.Create(&callbackData).Error
+	return callbackData.ID, err
+}
+
+func GetCallbackData(id uint) (string, error) {
+	var callbackData CallbackData
+	err := db.First(&callbackData, id).Error
+	return callbackData.Data, err
+}
+
+func DeleteCallbackData(id uint) error {
+	return db.Unscoped().Where("id = ?", id).Delete(&CallbackData{}).Error
+}

dao/db.go

@@ -1,33 +1,83 @@
 package dao
 
 import (
+	"fmt"
 	"os"
 	"path/filepath"
+	"time"
 
 	"github.com/glebarez/sqlite"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
-	"github.com/krau/SaveAny-Bot/model"
 	"gorm.io/gorm"
+	glogger "gorm.io/gorm/logger"
 )
 
 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)
 	}
 	var err error
-	db, err = gorm.Open(sqlite.Open(config.Cfg.DB.Path), &gorm.Config{})
+	db, err = gorm.Open(sqlite.Open(config.Cfg.DB.Path), &gorm.Config{
+		Logger: glogger.New(logger.L, glogger.Config{
+			Colorful:                  true,
+			SlowThreshold:             time.Second * 5,
+			LogLevel:                  glogger.Error,
+			IgnoreRecordNotFoundError: true,
+			ParameterizedQueries:      true,
+		}),
+		PrepareStmt: true,
+	})
 	if err != nil {
 		logger.L.Fatal("Failed to open database: ", err)
 		os.Exit(1)
 	}
 	logger.L.Debug("Database connected")
-	db.AutoMigrate(&model.ReceivedFile{}, &model.User{})
+	if err := db.AutoMigrate(&ReceivedFile{}, &User{}, &Dir{}, &CallbackData{}); 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]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/dir.go

@@ -0,0 +1,43 @@
+package dao
+
+func CreateDirForUser(userID uint, storageName, path string) error {
+	dir := Dir{
+		UserID:      userID,
+		StorageName: storageName,
+		Path:        path,
+	}
+	return db.Create(&dir).Error
+}
+
+func GetDirByID(id uint) (*Dir, error) {
+	dir := &Dir{}
+	err := db.First(dir, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return dir, err
+}
+
+func GetUserDirs(userID uint) ([]Dir, error) {
+	var dirs []Dir
+	err := db.Where("user_id = ?", userID).Find(&dirs).Error
+	return dirs, err
+}
+
+func GetUserDirsByChatID(chatID int64) ([]Dir, error) {
+	user, err := GetUserByChatID(chatID)
+	if err != nil {
+		return nil, err
+	}
+	return GetUserDirs(user.ID)
+}
+
+func GetDirsByUserIDAndStorageName(userID uint, storageName string) ([]Dir, error) {
+	var dirs []Dir
+	err := db.Where("user_id = ? AND storage_name = ?", userID, storageName).Find(&dirs).Error
+	return dirs, err
+}
+
+func DeleteDirForUser(userID uint, storageName, path string) error {
+	return db.Unscoped().Where("user_id = ? AND storage_name = ? AND path = ?", userID, storageName, path).Delete(&Dir{}).Error
+}

dao/file.go

@@ -1,31 +1,15 @@
 package dao
 
-import "github.com/krau/SaveAny-Bot/model"
-
-func AddReceivedFile(receivedFile *model.ReceivedFile) error {
-	return db.Create(receivedFile).Error
-}
-
-func GetReceivedFileByFileID(fileID string) (*model.ReceivedFile, error) {
-	var receivedFile model.ReceivedFile
-	err := db.Where("file_id = ?", fileID).First(&receivedFile).Error
-	if err != nil {
-		return nil, err
+func SaveReceivedFile(receivedFile *ReceivedFile) error {
+	record, err := GetReceivedFileByChatAndMessageID(receivedFile.ChatID, receivedFile.MessageID)
+	if err == nil {
+		receivedFile.ID = record.ID
 	}
-	return &receivedFile, nil
+	return db.Save(receivedFile).Error
 }
 
-func GetReceivedFileByFileUniqueID(fileUniqueID string) (*model.ReceivedFile, error) {
-	var receivedFile model.ReceivedFile
-	err := db.Where("file_unique_id = ?", fileUniqueID).First(&receivedFile).Error
-	if err != nil {
-		return nil, err
-	}
-	return &receivedFile, nil
-}
-
-func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*model.ReceivedFile, error) {
-	var receivedFile model.ReceivedFile
+func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*ReceivedFile, error) {
+	var receivedFile ReceivedFile
 	err := db.Where("chat_id = ? AND message_id = ?", chatID, messageID).First(&receivedFile).Error
 	if err != nil {
 		return nil, err
@@ -33,23 +17,6 @@ func GetReceivedFileByChatAndMessageID(chatID int64, messageID int) (*model.Rece
 	return &receivedFile, nil
 }
 
-func GetReceivedFilesByMediaGroupID(mediaGroupID string) ([]model.ReceivedFile, error) {
-	var receivedFiles []model.ReceivedFile
-	err := db.Where("media_group_id = ?", mediaGroupID).Find(&receivedFiles).Error
-	if err != nil {
-		return nil, err
-	}
-	return receivedFiles, nil
-}
-
-func UpdateReceivedFile(receivedFile *model.ReceivedFile) error {
-	return db.Save(receivedFile).Error
-}
-
-func DeleteReceivedFileByFileID(fileID string) error {
-	return db.Where("file_id = ?", fileID).Delete(&model.ReceivedFile{}).Error
-}
-
-func DeleteReceivedFileByFileUniqueID(fileUniqueID string) error {
-	return db.Where("file_unique_id = ?", fileUniqueID).Delete(&model.ReceivedFile{}).Error
+func DeleteReceivedFile(receivedFile *ReceivedFile) error {
+	return db.Unscoped().Delete(receivedFile).Error
 }

dao/model.go

@@ -0,0 +1,37 @@
+package dao
+
+import (
+	"gorm.io/gorm"
+)
+
+type ReceivedFile struct {
+	gorm.Model
+	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
+	ChatID         int64 `gorm:"uniqueIndex;not null"`
+	Silent         bool
+	DefaultStorage string // Default storage name
+	Dirs           []Dir
+}
+
+type Dir struct {
+	gorm.Model
+	UserID      uint
+	StorageName string
+	Path        string
+}
+
+type CallbackData struct {
+	gorm.Model
+	Data string
+}

dao/user.go

@@ -1,22 +1,30 @@
 package dao
 
-import (
-	"github.com/krau/SaveAny-Bot/model"
-)
-
-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(&model.User{UserID: userID}).Error
+	return db.Create(&User{ChatID: chatID}).Error
 }
 
-func GetUserByUserID(userID int64) (*model.User, error) {
-	var user model.User
-	err := db.Where("user_id = ?", userID).First(&user).Error
+func GetAllUsers() ([]User, error) {
+	var users []User
+	err := db.Preload("Dirs").Find(&users).Error
+	return users, err
+}
+
+func GetUserByChatID(chatID int64) (*User, error) {
+	var user User
+	err := db.
+		Preload("Dirs").
+		Where("chat_id = ?", chatID).First(&user).Error
 	return &user, err
 }
 
-func UpdateUser(user *model.User) error {
+func UpdateUser(user *User) error {
 	return db.Save(user).Error
 }
+
+func DeleteUser(user *User) error {
+	return db.Unscoped().Select("Dirs").Delete(user).Error
+}

docker-compose.yml

@@ -0,0 +1,10 @@
+services:
+  saveany-bot:
+    image: ghcr.io/krau/saveany-bot:latest
+    container_name: saveany-bot
+    restart: unless-stopped
+    volumes:
+      - ./data:/app/data
+      - ./config.toml:/app/config.toml
+      - ./downloads:/app/downloads
+      - ./cache:/app/cache

docs/sabot

@@ -0,0 +1,28 @@
+#!/bin/sh
+
+case "$1" in
+    start)
+        /etc/init.d/saveanybot start
+        ;;
+    stop)
+        /etc/init.d/saveanybot stop
+        ;;
+    restart)
+        /etc/init.d/saveanybot restart
+        ;;
+    status)
+        /etc/init.d/saveanybot status
+        ;;
+    enable)
+        /etc/init.d/saveanybot enable
+        echo "Enable SaveAnyBot auto-start."
+        ;;
+    disable)
+        /etc/init.d/saveanybot disable
+        echo "Disable SaveAnyBot auto-start."
+        ;;
+    *)
+        echo "Usage: $0 {start|stop|restart|status|enable|disable}"
+        exit 1
+        ;;
+esac

docs/saveanybot

@@ -0,0 +1,34 @@
+#!/bin/sh /etc/rc.common
+
+# This is the OpenWRT init.d script for SaveAnyBot
+
+START=99  # 设置启动顺序，数字越大越后启动
+STOP=10   # 设置停止顺序，数字越小越先停止
+
+# 脚本描述
+description="SaveAnyBot"
+
+# 设置工作目录和执行文件路径
+WORKING_DIR="/mnt/mmc1-1/SaveAnyBot"
+EXEC_PATH="$WORKING_DIR/saveany-bot"
+
+# 启动函数
+start() {
+    echo "Starting SaveAnyBot..."
+    # 切换到工作目录并执行程序
+    cd $WORKING_DIR
+    $EXEC_PATH &
+}
+
+# 停止函数
+stop() {
+    echo "Stopping SaveAnyBot..."
+    # 查找并杀死进程
+    killall saveany-bot
+}
+
+# 重启函数
+reload() {
+    stop
+    start
+}

go.mod

@@ -1,97 +1,105 @@
 module github.com/krau/SaveAny-Bot
 
-go 1.23.2
+go 1.23.5
 
 require (
 	github.com/blang/semver v3.5.1+incompatible
-	github.com/gookit/slog v0.5.6
-	github.com/imroc/req/v3 v3.46.1
-	github.com/mymmrac/telego v0.31.3
+	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
 	github.com/rhysd/go-github-selfupdate v1.2.3
 	github.com/spf13/cobra v1.8.1
 	github.com/spf13/viper v1.19.0
-	github.com/studio-b12/gowebdav v0.9.0
+	github.com/studio-b12/gowebdav v0.10.0
+	golang.org/x/net v0.35.0
+	golang.org/x/time v0.10.0
 )
 
 require (
-	github.com/cloudflare/circl v1.4.0 // indirect
+	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
+	github.com/coder/websocket v1.8.12 // indirect
+	github.com/dlclark/regexp2 v1.11.5 // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
-	github.com/glebarez/go-sqlite v1.21.2 // indirect
-	github.com/go-task/slim-sprig/v3 v3.0.0 // indirect
-	github.com/golang/protobuf v1.5.3 // indirect
+	github.com/fatih/color v1.18.0 // indirect
+	github.com/ghodss/yaml v1.0.0 // indirect
+	github.com/glebarez/go-sqlite v1.22.0 // indirect
+	github.com/go-faster/errors v0.7.1 // indirect
+	github.com/go-faster/jx v1.1.0 // indirect
+	github.com/go-faster/xor v1.0.0 // indirect
+	github.com/go-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.0.0 // indirect
-	github.com/google/pprof v0.0.0-20241009165004-a3522334989c // indirect
+	github.com/google/go-querystring v1.1.0 // indirect
+	github.com/google/pprof v0.0.0-20250128161936-077ca0a936bf // indirect
 	github.com/google/uuid v1.6.0 // indirect
-	github.com/hashicorp/errwrap v1.1.0 // indirect
-	github.com/hashicorp/go-multierror v1.1.1 // indirect
+	github.com/gotd/ige v0.2.2 // indirect
+	github.com/gotd/neo v0.1.5 // indirect
 	github.com/inconshreveable/go-update v0.0.0-20160112193335-8152e7eb6ccf // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
-	github.com/mattn/go-isatty v0.0.17 // indirect
-	github.com/onsi/ginkgo/v2 v2.20.2 // indirect
-	github.com/quic-go/qpack v0.5.1 // indirect
-	github.com/quic-go/quic-go v0.47.0 // indirect
-	github.com/refraction-networking/utls v1.6.7 // indirect
+	github.com/mattn/go-colorable v0.1.14 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/ncruces/go-strftime v0.1.9 // indirect
+	github.com/ogen-go/ogen v1.10.0 // indirect
+	github.com/onsi/gomega v1.36.2 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
 	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
+	github.com/segmentio/asm v1.2.0 // indirect
 	github.com/tcnksm/go-gitconfig v0.1.2 // indirect
-	github.com/ulikunitz/xz v0.5.9 // indirect
-	go.uber.org/mock v0.4.0 // indirect
-	golang.org/x/crypto v0.28.0 // indirect
-	golang.org/x/mod v0.21.0 // indirect
-	golang.org/x/net v0.30.0 // indirect
-	golang.org/x/oauth2 v0.18.0 // indirect
-	golang.org/x/tools v0.26.0 // indirect
-	google.golang.org/appengine v1.6.8 // indirect
-	google.golang.org/protobuf v1.34.1 // indirect
-	modernc.org/libc v1.22.5 // indirect
-	modernc.org/mathutil v1.5.0 // indirect
-	modernc.org/memory v1.5.0 // indirect
-	modernc.org/sqlite v1.23.1 // indirect
+	github.com/ulikunitz/xz v0.5.12 // indirect
+	go.opentelemetry.io/otel v1.34.0 // indirect
+	go.opentelemetry.io/otel/metric v1.34.0 // indirect
+	go.opentelemetry.io/otel/trace v1.34.0 // indirect
+	go.uber.org/atomic v1.11.0 // indirect
+	go.uber.org/zap v1.27.0 // indirect
+	golang.org/x/crypto v0.33.0 // indirect
+	golang.org/x/mod v0.23.0 // indirect
+	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
+	modernc.org/sqlite v1.35.0 // indirect
+	rsc.io/qr v0.2.0 // indirect
 )
 
 require (
-	github.com/andybalholm/brotli v1.1.1 // indirect
-	github.com/bytedance/sonic v1.12.2 // indirect
-	github.com/bytedance/sonic/loader v0.2.0 // indirect
-	github.com/cloudwego/base64x v0.1.4 // indirect
-	github.com/cloudwego/iasm v0.2.0 // indirect
-	github.com/duke-git/lancet/v2 v2.3.3
-	github.com/fasthttp/router v1.5.2 // indirect
-	github.com/fsnotify/fsnotify v1.7.0 // indirect
+	github.com/coocood/freecache v1.2.4
+	github.com/duke-git/lancet/v2 v2.3.4
+	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.15
+	github.com/gookit/goutil v0.6.18 // indirect
 	github.com/gookit/gsr v0.1.0 // indirect
-	github.com/grbit/go-json v0.11.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
-	github.com/klauspost/compress v1.17.10 // indirect
-	github.com/klauspost/cpuid/v2 v2.2.6 // indirect
-	github.com/magiconair/properties v1.8.7 // indirect
-	github.com/mitchellh/mapstructure v1.5.0 // indirect
-	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
-	github.com/sagikazarmark/locafero v0.4.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
+	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
-	github.com/savsgio/gotils v0.0.0-20240704082632-aef3928b8a38 // indirect
 	github.com/sourcegraph/conc v0.3.0 // indirect
-	github.com/spf13/afero v1.11.0 // indirect
-	github.com/spf13/cast v1.6.0 // indirect
-	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/spf13/afero v1.12.0 // indirect
+	github.com/spf13/cast v1.7.1 // indirect
+	github.com/spf13/pflag v1.0.6 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
-	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	github.com/valyala/bytebufferpool v1.0.0 // indirect
-	github.com/valyala/fasthttp v1.55.0 // indirect
-	github.com/valyala/fastjson v1.6.4 // indirect
 	github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
-	go.uber.org/atomic v1.9.0 // indirect
-	go.uber.org/multierr v1.9.0 // indirect
-	golang.org/x/arch v0.6.0 // indirect
-	golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c // indirect
-	golang.org/x/sync v0.8.0 // indirect
-	golang.org/x/sys v0.26.0 // indirect
-	golang.org/x/text v0.19.0 // indirect
+	go.uber.org/multierr v1.11.0 // indirect
+	golang.org/x/exp v0.0.0-20250210185358-939b2ce775ac // indirect
+	golang.org/x/sync v0.11.0 // indirect
+	golang.org/x/sys v0.30.0 // indirect
+	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,93 +1,116 @@
-github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
-github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
+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/bytedance/sonic v1.12.2 h1:oaMFuRTpMHYLpCntGca65YWt5ny+wAceDERTkT2L9lg=
-github.com/bytedance/sonic v1.12.2/go.mod h1:B8Gt/XvtZ3Fqj+iSKMypzymZxw/FVwgIGKzMzT9r/rk=
-github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
-github.com/bytedance/sonic/loader v0.2.0 h1:zNprn+lsIP06C/IqCHs3gPQIvnvpKbbxyXQP1iU4kWM=
-github.com/bytedance/sonic/loader v0.2.0/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
-github.com/cloudflare/circl v1.4.0 h1:BV7h5MgrktNzytKmWjpOtdYrf0lkkbF8YMlBGPhJQrY=
-github.com/cloudflare/circl v1.4.0/go.mod h1:PDRU+oXvdD7KCtgKxW95M5Z8BpSCJXQORiZFnBQS5QU=
-github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
-github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
-github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
-github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
+github.com/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=
+github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
+github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
+github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
+github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
+github.com/coder/websocket v1.8.12 h1:5bUXkEPPIbewrnkU8LTCLVaxi4N4J8ahufH2vlo4NAo=
+github.com/coder/websocket v1.8.12/go.mod h1:LNVeNrXQZfe5qhS9ALED3uA+l5pPqvwXg3CKoDBB2gs=
+github.com/coocood/freecache v1.2.4 h1:UdR6Yz/X1HW4fZOuH0Z94KwG851GWOSknua5VUbb/5M=
+github.com/coocood/freecache v1.2.4/go.mod h1:RBUWa/Cy+OHdfTGFEhEuE1pMCMX51Ncizj7rthiQ3vk=
 github.com/cpuguy83/go-md2man/v2 v2.0.4/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
-github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1VwoXQT9A3Wy9MM3WgvqSxFWenqJduM=
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
-github.com/duke-git/lancet/v2 v2.3.3 h1:OhqzNzkbJBS9ZlWLo/C7g+WSAOAAyNj7p9CAiEHurUc=
-github.com/duke-git/lancet/v2 v2.3.3/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
+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=
+github.com/duke-git/lancet/v2 v2.3.4/go.mod h1:zGa2R4xswg6EG9I6WnyubDbFO/+A/RROxIbXcwryTsc=
 github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
 github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
-github.com/fasthttp/router v1.5.2 h1:ckJCCdV7hWkkrMeId3WfEhz+4Gyyf6QPwxi/RHIMZ6I=
-github.com/fasthttp/router v1.5.2/go.mod h1:C8EY53ozOwpONyevc/V7Gr8pqnEjwnkFFqPo1alAGs0=
+github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
+github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
 github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
-github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=
-github.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=
-github.com/glebarez/go-sqlite v1.21.2 h1:3a6LFC4sKahUunAmynQKLZceZCOzUthkRkEAl9gAXWo=
-github.com/glebarez/go-sqlite v1.21.2/go.mod h1:sfxdZyhQjTM2Wry3gVYWaW072Ri1WMdWJi0k6+3382k=
+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=
+github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc=
 github.com/glebarez/sqlite v1.11.0 h1:wSG0irqzP6VurnMEpFGer5Li19RpIRi2qvQz++w0GMw=
 github.com/glebarez/sqlite v1.11.0/go.mod h1:h8/o8j5wiAsqSPoWELDUdJXhjAhsVliSn7bWZjOhrgQ=
+github.com/go-faster/errors v0.7.1 h1:MkJTnDoEdi9pDabt1dpWf7AA8/BaSYZqibYyhZ20AYg=
+github.com/go-faster/errors v0.7.1/go.mod h1:5ySTjWFiphBs07IKuiL69nxdfd5+fzh1u7FPGZP2quo=
+github.com/go-faster/jx v1.1.0 h1:ZsW3wD+snOdmTDy9eIVgQdjUpXRRV4rqW8NS3t+20bg=
+github.com/go-faster/jx v1.1.0/go.mod h1:vKDNikrKoyUmpzaJ0OkIkRQClNHFX/nF3dnTJZb3skg=
+github.com/go-faster/xor v0.3.0/go.mod h1:x5CaDY9UKErKzqfRfFZdfu+OSTfoZny3w5Ak7UxcipQ=
+github.com/go-faster/xor v1.0.0 h1:2o8vTOgErSGHP3/7XwA5ib1FTtUsNtwCoLLBjl31X38=
+github.com/go-faster/xor v1.0.0/go.mod h1:x5CaDY9UKErKzqfRfFZdfu+OSTfoZny3w5Ak7UxcipQ=
+github.com/go-faster/yaml v0.4.6 h1:lOK/EhI04gCpPgPhgt0bChS6bvw7G3WwI8xxVe0sw9I=
+github.com/go-faster/yaml v0.4.6/go.mod h1:390dRIvV4zbnO7qC9FGo6YYutc+wyyUSHBgbXL52eXk=
 github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=
 github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
-github.com/go-task/slim-sprig/v3 v3.0.0 h1:sUs3vkvUymDpBKi3qH1YSqBQk9+9D/8M2mN1vB6EwHI=
-github.com/go-task/slim-sprig/v3 v3.0.0/go.mod h1:W848ghGpv3Qj3dhTPRyJypKRiqCdHZiAzKg9hl15HA8=
+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/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
-github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
-github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
-github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
-github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
 github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
 github.com/google/go-github/v30 v30.1.0 h1:VLDx+UolQICEOKu2m4uAoMti1SxuEBAl7RSEG16L+Oo=
 github.com/google/go-github/v30 v30.1.0/go.mod h1:n8jBpHl45a/rlBUtRJMOG4GhNADUQFEufcolZ95JfU8=
-github.com/google/go-querystring v1.0.0 h1:Xkwi/a1rcvNg1PPYe5vI8GbeBY/jrVuDX5ASuANWTrk=
 github.com/google/go-querystring v1.0.0/go.mod h1:odCYkC5MyYFN7vkCjXpyrEuKhc/BUO6wN/zVPAxq5ck=
-github.com/google/pprof v0.0.0-20241009165004-a3522334989c h1:NDovD0SMpBYXlE1zJmS1q55vWB/fUQBcPAqAboZSccA=
-github.com/google/pprof v0.0.0-20241009165004-a3522334989c/go.mod h1:vavhavw2zAxS5dIdcRluK6cSGGPlZynqzFM8NdvU144=
+github.com/google/go-querystring v1.1.0 h1:AnCroh3fv4ZBgVIf1Iwtovgjaw/GiKJo8M8yD/fhyJ8=
+github.com/google/go-querystring v1.1.0/go.mod h1:Kcdr2DB4koayq7X8pmAG4sNG59So17icRSOU623lUBU=
+github.com/google/pprof v0.0.0-20250128161936-077ca0a936bf h1:BvBLUD2hkvLI3dJTJMiopAq8/wp43AAZKTP7qdpptbU=
+github.com/google/pprof v0.0.0-20250128161936-077ca0a936bf/go.mod h1:vavhavw2zAxS5dIdcRluK6cSGGPlZynqzFM8NdvU144=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/gookit/color v1.5.4 h1:FZmqs7XOyGgCAxmWyPslpiok1k05wmY3SJTytgvYFs0=
 github.com/gookit/color v1.5.4/go.mod h1:pZJOeOS8DM43rXbp4AZo1n9zCU2qjpcRko0b6/QJi9w=
-github.com/gookit/goutil v0.6.15 h1:mMQ0ElojNZoyPD0eVROk5QXJPh2uKR4g06slgPDF5Jo=
-github.com/gookit/goutil v0.6.15/go.mod h1:qdKdYEHQdEtyH+4fNdQNZfJHhI0jUZzHxQVAV3DaMDY=
+github.com/gookit/goutil v0.6.18 h1:MUVj0G16flubWT8zYVicIuisUiHdgirPAkmnfD2kKgw=
+github.com/gookit/goutil v0.6.18/go.mod h1:AY/5sAwKe7Xck+mEbuxj0n/bc3qwrGNe3Oeulln7zBA=
 github.com/gookit/gsr v0.1.0 h1:0gadWaYGU4phMs0bma38t+Do5OZowRMEVlHv31p0Zig=
 github.com/gookit/gsr v0.1.0/go.mod h1:7wv4Y4WCnil8+DlDYHBjidzrEzfHhXEoFjEA0pPPWpI=
-github.com/gookit/slog v0.5.6 h1:fmh+7bfOK8CjidMCwE+M3S8G766oHJpT/1qdmXGALCI=
-github.com/gookit/slog v0.5.6/go.mod h1:RfIwzoaQ8wZbKdcqG7+3EzbkMqcp2TUn3mcaSZAw2EQ=
-github.com/grbit/go-json v0.11.0 h1:bAbyMdYrYl/OjYsSqLH99N2DyQ291mHy726Mx+sYrnc=
-github.com/grbit/go-json v0.11.0/go.mod h1:IYpHsdybQ386+6g3VE6AXQ3uTGa5mquBme5/ZWmtzek=
-github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
-github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I=
-github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
-github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=
-github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=
+github.com/gookit/slog v0.5.7 h1:n3Dhgmr3NP+KppkNg95+vpFcI4YD1csu9VTQwgcEYTs=
+github.com/gookit/slog v0.5.7/go.mod h1:uWCRB4YO+FmgwXEq3s8U7ob1wWP7RStOuY/2a4yC/8o=
+github.com/gotd/contrib v0.21.0 h1:4Fj05jnyBE84toXZl7mVTvt7f732n5uglvztyG6nTr4=
+github.com/gotd/contrib v0.21.0/go.mod h1:ENoUh75IhHGxfz/puVJg8BU4ZF89yrL6Q47TyoNqFYo=
+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.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=
 github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
-github.com/imroc/req/v3 v3.46.1 h1:oahr2hBTb3AaFI4P6jkN0Elj2ZVKJcdQ/IjWqeIKjvc=
-github.com/imroc/req/v3 v3.46.1/go.mod h1:weam9gmyb00QnOtu6HXSnk44dNFkIUQb5QdMx13FeUU=
 github.com/inconshreveable/go-update v0.0.0-20160112193335-8152e7eb6ccf h1:WfD7VjIE6z8dIvMsI4/s+1qr5EL+zoIGev1BQj1eoJ8=
 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=
 github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
-github.com/klauspost/compress v1.17.10 h1:oXAz+Vh0PMUvJczoi+flxpnBEPxoER1IaAnU/NMPtT0=
-github.com/klauspost/compress v1.17.10/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0=
-github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
-github.com/klauspost/cpuid/v2 v2.2.6 h1:ndNyv040zDGIDh8thGkXYjnFtiN02M1PVVF+JE/48xc=
-github.com/klauspost/cpuid/v2 v2.2.6/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
-github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
+github.com/klauspost/compress v1.17.11 h1:In6xLpyWOi1+C7tXUUWv2ot1QvBjxevKAaI6IXrJmUc=
+github.com/klauspost/compress v1.17.11/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
 github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
 github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
@@ -95,186 +118,183 @@ github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
 github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
 github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
-github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY=
-github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
-github.com/mattn/go-isatty v0.0.17 h1:BTarxUcIeDqL27Mc+vyvdWYSL28zpIhv3RoTdsLMPng=
-github.com/mattn/go-isatty v0.0.17/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
+github.com/magiconair/properties v1.8.9 h1:nWcCbLq1N2v/cpNsy5WvQ37Fb+YElfq20WJ/a8RkpQM=
+github.com/magiconair/properties v1.8.9/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
+github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
+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/mymmrac/telego v0.31.3 h1:yZlD+dm+1W6p3OmCG8K+MbS02Y6paUgwPnqfZN3RWQQ=
-github.com/mymmrac/telego v0.31.3/go.mod h1:coOoqXVmjFnwBlzusjfEezbQ7RH9wQnDowJdMm+bnEo=
+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.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=
-github.com/onsi/ginkgo/v2 v2.20.2 h1:7NVCeyIWROIAheY21RLS+3j2bb52W0W82tkberYytp4=
-github.com/onsi/ginkgo/v2 v2.20.2/go.mod h1:K9gyxPIlb+aIvnZ8bd9Ak+YP18w3APlR+5coaZoE2ag=
 github.com/onsi/gomega v1.4.2/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
-github.com/onsi/gomega v1.34.1 h1:EUMJIKUjM8sKjYbtxQI9A4z2o+rruxnzNvpknOXie6k=
-github.com/onsi/gomega v1.34.1/go.mod h1:kU1QgUvBDLXBJq618Xvm2LUX6rSAfRaFRTcdOeDLwwY=
-github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
-github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
-github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/onsi/gomega v1.36.2 h1:koNYke6TVk6ZmnyHrCXba/T/MoLBXFjeC1PtvYgw0A8=
+github.com/onsi/gomega v1.36.2/go.mod h1:DdwyADRjrc825LhMEkD76cHR5+pUnjhUN8GlHlRPHzY=
+github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=
+github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
+github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
+github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 h1:Jamvg5psRIccs7FGNTlIRMkT8wgtp5eCXdBlqhYGL6U=
 github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/quic-go/qpack v0.5.1 h1:giqksBPnT/HDtZ6VhtFKgoLOWmlyo9Ei6u9PqzIMbhI=
-github.com/quic-go/qpack v0.5.1/go.mod h1:+PC4XFrEskIVkcLzpEkbLqq1uCoxPhQuvK5rH1ZgaEg=
-github.com/quic-go/quic-go v0.47.0 h1:yXs3v7r2bm1wmPTYNLKAAJTHMYkPEsfYJmTazXrCZ7Y=
-github.com/quic-go/quic-go v0.47.0/go.mod h1:3bCapYsJvXGZcipOHuu7plYtaV6tnF+z7wIFsU0WK9E=
-github.com/refraction-networking/utls v1.6.7 h1:zVJ7sP1dJx/WtVuITug3qYUq034cDq9B2MR1K67ULZM=
-github.com/refraction-networking/utls v1.6.7/go.mod h1:BC3O4vQzye5hqpmDTWUqi4P5DDhzJfkV1tdqtawQIH0=
-github.com/remyoudompheng/bigfft v0.0.0-20200410134404-eec4a21b6bb0/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
 github.com/rhysd/go-github-selfupdate v1.2.3 h1:iaa+J202f+Nc+A8zi75uccC8Wg3omaM7HDeimXA22Ag=
 github.com/rhysd/go-github-selfupdate v1.2.3/go.mod h1:mp/N8zj6jFfBQy/XMYoWsmfzxazpPAODuqarmPDe2Rg=
-github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
-github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
+github.com/rogpeppe/go-internal v1.13.1 h1:KvO1DLK/DRN07sQ1LQKScxyZJuNnedQ5/wKSR38lUII=
+github.com/rogpeppe/go-internal v1.13.1/go.mod h1:uMEvuHeurkdAXX61udpOXGD/AzZDWNMNyH2VO9fmH0o=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
-github.com/sagikazarmark/locafero v0.4.0 h1:HApY1R9zGo4DBgr7dqsTH/JJxLTTsOt7u6keLGt6kNQ=
-github.com/sagikazarmark/locafero v0.4.0/go.mod h1:Pe1W6UlPYUk/+wc/6KFhbORCfqzgYEpgQ3O5fPuL3H4=
+github.com/sagikazarmark/locafero v0.7.0 h1:5MqpDsTGNDhY8sGp0Aowyf0qKsPrhewaLSsFaodPcyo=
+github.com/sagikazarmark/locafero v0.7.0/go.mod h1:2za3Cg5rMaTMoG/2Ulr9AwtFaIppKXTRYnozin4aB5k=
 github.com/sagikazarmark/slog-shim v0.1.0 h1:diDBnUNK9N/354PgrxMywXnAwEr1QZcOr6gto+ugjYE=
 github.com/sagikazarmark/slog-shim v0.1.0/go.mod h1:SrcSrq8aKtyuqEI1uvTDTK1arOWRIczQRv+GVI1AkeQ=
-github.com/savsgio/gotils v0.0.0-20240704082632-aef3928b8a38 h1:D0vL7YNisV2yqE55+q0lFuGse6U8lxlg7fYTctlT5Gc=
-github.com/savsgio/gotils v0.0.0-20240704082632-aef3928b8a38/go.mod h1:sM7Mt7uEoCeFSCBM+qBrqvEo+/9vdmj19wzp3yzUhmg=
+github.com/segmentio/asm v1.2.0 h1:9BQrFxC+YOHJlTlHGkTrFWf59nbL3XnCoFLTwDCI7ys=
+github.com/segmentio/asm v1.2.0/go.mod h1:BqMnlJP91P8d+4ibuonYZw9mfnzI9HfxselHZr5aAcs=
 github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo=
 github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0=
-github.com/spf13/afero v1.11.0 h1:WJQKhtpdm3v2IzqG8VMqrr6Rf3UYpEF239Jy9wNepM8=
-github.com/spf13/afero v1.11.0/go.mod h1:GH9Y3pIexgf1MTIWtNGyogA5MwRIDXGUr+hbWNoBjkY=
-github.com/spf13/cast v1.6.0 h1:GEiTHELF+vaR5dhz3VqZfFSzZjYbgeKDpBxQVS4GYJ0=
-github.com/spf13/cast v1.6.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=
+github.com/spf13/afero v1.12.0 h1:UcOPyRBYczmFn6yvphxkn9ZEOY65cpwGKb5mL36mrqs=
+github.com/spf13/afero v1.12.0/go.mod h1:ZTlWwG4/ahT8W7T0WQ5uYmjI9duaLQGy3Q2OAl4sk/4=
+github.com/spf13/cast v1.7.1 h1:cuNEagBQEHWN1FnbGEjCXL2szYEXqfJPbP2HNUaca9Y=
+github.com/spf13/cast v1.7.1/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo=
 github.com/spf13/cobra v1.8.1 h1:e5/vxKd/rZsfSJMUX1agtjeTDf+qv1/JdBF8gg5k9ZM=
 github.com/spf13/cobra v1.8.1/go.mod h1:wHxEcudfqmLYa8iTfL+OuZPbBZkmvliBWKIezN3kD9Y=
-github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
 github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
+github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 github.com/spf13/viper v1.19.0 h1:RWq5SEjt8o25SROyN3z2OrDB9l7RPd3lwTWU8EcEdcI=
 github.com/spf13/viper v1.19.0/go.mod h1:GQUN9bilAbhU/jgc1bKs99f/suXKeUMct8Adx5+Ntkg=
-github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
-github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
-github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
-github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
-github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
-github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
-github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
-github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
-github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
-github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
-github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
-github.com/studio-b12/gowebdav v0.9.0 h1:1j1sc9gQnNxbXXM4M/CebPOX4aXYtr7MojAVcN4dHjU=
-github.com/studio-b12/gowebdav v0.9.0/go.mod h1:bHA7t77X/QFExdeAnDzK6vKM34kEZAcE1OX4MfiwjkE=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
+github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/studio-b12/gowebdav v0.10.0 h1:Yewz8FFiadcGEu4hxS/AAJQlHelndqln1bns3hcJIYc=
+github.com/studio-b12/gowebdav v0.10.0/go.mod h1:bHA7t77X/QFExdeAnDzK6vKM34kEZAcE1OX4MfiwjkE=
 github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
 github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
 github.com/tcnksm/go-gitconfig v0.1.2 h1:iiDhRitByXAEyjgBqsKi9QU4o2TNtv9kPP3RgPgXBPw=
 github.com/tcnksm/go-gitconfig v0.1.2/go.mod h1:/8EhP4H7oJZdIPyT+/UIsG87kTzrzM4UsLGSItWYCpE=
-github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
-github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
-github.com/ulikunitz/xz v0.5.9 h1:RsKRIA2MO8x56wkkcd3LbtcE/uMszhb6DpRf+3uwa3I=
 github.com/ulikunitz/xz v0.5.9/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=
+github.com/ulikunitz/xz v0.5.12 h1:37Nm15o69RwBkXM0J6A5OlE67RZTfzUxTj8fB3dfcsc=
+github.com/ulikunitz/xz v0.5.12/go.mod h1:nbz6k7qbPmH4IRqmfOplQw/tblSgqTqBwxkY0oWt/14=
 github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
 github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
-github.com/valyala/fasthttp v1.55.0 h1:Zkefzgt6a7+bVKHnu/YaYSOPfNYNisSVBo/unVCf8k8=
-github.com/valyala/fasthttp v1.55.0/go.mod h1:NkY9JtkrpPKmgwV3HTaS2HWaJss9RSIsRVfcxxoHiOM=
-github.com/valyala/fastjson v1.6.4 h1:uAUNq9Z6ymTgGhcm0UynUAB6tlbakBrz6CQFax3BXVQ=
-github.com/valyala/fastjson v1.6.4/go.mod h1:CLCAqky6SMuOcxStkYQvblddUtoRxhYMGLrsQns1aXY=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
-github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
-github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
-github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
-go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
-go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
-go.uber.org/mock v0.4.0 h1:VcM4ZOtdbR4f6VXfiOpwpVJDL6lCReaZ6mw31wqh7KU=
-go.uber.org/mock v0.4.0/go.mod h1:a6FSlNadKUHUa9IP5Vyt1zh4fC7uAwxMutEAscFbkZc=
-go.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI=
-go.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ=
-golang.org/x/arch v0.6.0 h1:S0JTfE48HbRj80+4tbvZDYsJ3tGv6BUU3XxyZ7CirAc=
-golang.org/x/arch v0.6.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
+go.opentelemetry.io/auto/sdk v1.1.0 h1:cH53jehLUN6UFLY71z+NDOiNJqDdPRaXzTel0sJySYA=
+go.opentelemetry.io/auto/sdk v1.1.0/go.mod h1:3wSPjt5PWp2RhlCcmmOial7AvC4DQqZb7a7wCow3W8A=
+go.opentelemetry.io/otel v1.34.0 h1:zRLXxLCgL1WyKsPVrgbSdMN4c0FMkDAskSTQP+0hdUY=
+go.opentelemetry.io/otel v1.34.0/go.mod h1:OWFPOQ+h4G8xpyjgqo4SxJYdDQ/qmRH+wivy7zzx9oI=
+go.opentelemetry.io/otel/metric v1.34.0 h1:+eTR3U0MyfWjRDhmFMxe2SsW64QrZ84AOhvqS7Y+PoQ=
+go.opentelemetry.io/otel/metric v1.34.0/go.mod h1:CEDrp0fy2D0MvkXE+dPV7cMi8tWZwX3dmaIhwPOaqHE=
+go.opentelemetry.io/otel/trace v1.34.0 h1:+ouXS2V8Rd4hp4580a8q23bg0azF2nI8cqLYnC8mh/k=
+go.opentelemetry.io/otel/trace v1.34.0/go.mod h1:Svm7lSjQD7kG7KJ/MUHPVXSDGz2OX4h0M2jHBhmSfRE=
+go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE=
+go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=
+go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
+go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
+go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
+go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
+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.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
-golang.org/x/crypto v0.28.0 h1:GBDwsMXVQi34v5CCYUm2jkJvu4cbtru2U4TN2PSyQnw=
-golang.org/x/crypto v0.28.0/go.mod h1:rmgy+3RHxRZMyY0jjAJShp2zgEdOqj2AO7U0pYmeQ7U=
-golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c h1:7dEasQXItcW1xKJ2+gg5VOiBnqWrJc+rq0DPKyvvdbY=
-golang.org/x/exp v0.0.0-20241009180824-f66d83c29e7c/go.mod h1:NQtJDoLvd6faHhE7m4T/1IY708gDefGGjR/iUW8yQQ8=
-golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
-golang.org/x/mod v0.21.0 h1:vvrHzRwRfVKSiLrG+d4FMl/Qi4ukBCE6kZlTUkDYRT0=
-golang.org/x/mod v0.21.0/go.mod h1:6SkKJ3Xj0I0BrPOZoBy3bdMptDDU9oJrpohJ3eWZ1fY=
+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-20250210185358-939b2ce775ac h1:l5+whBCLH3iH2ZNHYLbAe58bo7yrN4mVcnkHDYz5vvs=
+golang.org/x/exp v0.0.0-20250210185358-939b2ce775ac/go.mod h1:hH+7mtFmImwwcMvScyxUhjuVHR3HGaDPMn9rMSUUbxo=
+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.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
-golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
-golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
-golang.org/x/net v0.30.0 h1:AcW1SDZMkb8IpzCdQUaIq2sP4sZ4zw+55h6ynffypl4=
-golang.org/x/net v0.30.0/go.mod h1:2wGyMJ5iFasEhkwi13ChkO/t1ECNC4X4eBKkVFyYFlU=
+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.18.0 h1:09qnuIAgzdx1XplqJvW6CQqMCtGZykZWcXzPMPUusvI=
-golang.org/x/oauth2 v0.18.0/go.mod h1:Wf7knwG0MPoWIMMBgFlEaSUDaKskp0dCfrlJRJXbBi8=
+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-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
-golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+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.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
-golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.26.0 h1:KHjCJyddX0LoSTb3J+vWpupP9p0oznkqVk/IfjymZbo=
-golang.org/x/sys v0.26.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+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.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
-golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
-golang.org/x/term v0.25.0 h1:WtHI/ltw4NvSUig5KARz9h521QvRC8RmF/cuYqifU24=
-golang.org/x/term v0.25.0/go.mod h1:RPyXicDX+6vLxogjjRxjgD2TKtmAO6NZBsBRfrOLu7M=
+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.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.5/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
-golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
-golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
-golang.org/x/text v0.19.0 h1:kTxAhCbGbxhK0IwgSKiMO5awPoDQ0RpfiVYBfK860YM=
-golang.org/x/text v0.19.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
-golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
-golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
+golang.org/x/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.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.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
-golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
-golang.org/x/tools v0.26.0 h1:v/60pFQmzmT9ExmjDv2gGIfi3OqfKoEP6I5+umXlbnQ=
-golang.org/x/tools v0.26.0/go.mod h1:TPVVj70c7JJ3WCazhD8OdXcZg/og+b9+tH/KxylGwH0=
-golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+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=
 google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
 google.golang.org/appengine v1.3.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
-google.golang.org/appengine v1.6.8 h1:IhEN5q69dyKagZPYMSdIjS2HqprW324FRQZJcGqPAsM=
-google.golang.org/appengine v1.6.8/go.mod h1:1jJ3jBArFh5pcgW8gCtRJnepW8FzD1V44FJffLiz/Ds=
-google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
-google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
-google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
-google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
-gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
-gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
 gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=
 gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
 gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
-gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.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/libc v1.22.5 h1:91BNch/e5B0uPbJFgqbxXuOnxBQjlS//icfQEGmvyjE=
-modernc.org/libc v1.22.5/go.mod h1:jj+Z7dTNX8fBScMVNRAYZ/jF91K8fdT2hYMThc3YjBY=
-modernc.org/mathutil v1.5.0 h1:rV0Ko/6SfM+8G+yKiyI830l3Wuz1zRutdslNoQ0kfiQ=
-modernc.org/mathutil v1.5.0/go.mod h1:mZW8CKdRPY1v87qxC/wUdX5O1qDzXMP5TH3wjfpga6E=
-modernc.org/memory v1.5.0 h1:N+/8c5rE6EqugZwHii4IFsaJ7MUhoWX07J5tC/iI5Ds=
-modernc.org/memory v1.5.0/go.mod h1:PkUhL0Mugw21sHPeskwZW4D6VscE/GQJOnIpCnW6pSU=
-modernc.org/sqlite v1.23.1 h1:nrSBg4aRQQwq59JpvGEQ15tNxoO5pX/kUjcRNwSAGQM=
-modernc.org/sqlite v1.23.1/go.mod h1:OrDj17Mggn6MhE+iPbBNf7RGKODDE9NFT0f3EwDzJqk=
-nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
+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.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.3 h1:aJVhcqAte49LF+mGveZ5KPlsp4tdGdAOT4sipJXADjw=
+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=
+modernc.org/mathutil v1.7.1/go.mod h1:4p5IwJITfppl0G4sUEDtCr4DthTaT47/N3aT6MhfgJg=
+modernc.org/memory v1.8.2 h1:cL9L4bcoAObu4NkxOlKWBWtNHIsnnACGF/TbqQ6sbcI=
+modernc.org/memory v1.8.2/go.mod h1:ZbjSvMO5NQ1A2i3bWeDiVMxIorXwdClKE/0SZ+BMotU=
+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.35.0 h1:yQps4fegMnZFdphtzlfQTCNBWtS0CZv48pRpW3RFHRw=
+modernc.org/sqlite v1.35.0/go.mod h1:9cr2sicr7jIaWTBKQmAxQLfBv9LL0su4ZTEV+utt3ic=
+modernc.org/strutil v1.2.1 h1:UneZBkQA+DX2Rp35KcM69cSsNES9ly8mQWD71HKlOA0=
+modernc.org/strutil v1.2.1/go.mod h1:EHkiggD70koQxjVdSBM3JKM7k6L0FbGE5eymy9i3B9A=
+modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y=
+modernc.org/token v1.1.0/go.mod h1:UGzOrNV1mAFSEB63lOFHIpNRUVMvYTc6yu1SMY/XTDM=
+nhooyr.io/websocket v1.8.17 h1:KEVeLJkUywCKVsnLIDlD/5gtayKp8VoCkksHCGGfT9Y=
+nhooyr.io/websocket v1.8.17/go.mod h1:rN9OFWIUwuxg4fR5tELlYC04bXYowCP9GX47ivo2l+c=
+rsc.io/qr v0.2.0 h1:6vBLea5/NRMVTz8V66gipeLycZMl/+UlFmk8DvqQ6WY=
+rsc.io/qr v0.2.0/go.mod h1:IF+uZjkb9fqyeF/4tlBoynqmQxUoPfWEKh921coOuXs=

model/model.go

@@ -1,28 +0,0 @@
-package model
-
-import (
-	"gorm.io/gorm"
-)
-
-type ReceivedFile struct {
-	gorm.Model
-	// FileUniqueID 和 FileID 在数据库中均不具有唯一性
-	// 如需确定唯一一行, 使用 ChatID + MessageID
-	FileUniqueID   string `gorm:"index"`
-	FileID         string `gorm:"index"`
-	Processing     bool
-	FileName       string
-	FilePath       string
-	FileSize       int64
-	MediaGroupID   string
-	ChatID         int64
-	MessageID      int
-	ReplyMessageID int
-}
-
-type User struct {
-	gorm.Model
-	UserID         int64 `gorm:"uniqueIndex"`
-	Silent         bool
-	DefaultStorage string
-}

queue/queue.go

@@ -8,30 +8,65 @@ import (
 )
 
 type TaskQueue struct {
-	list  *list.List
-	cond  *sync.Cond
-	mutex *sync.Mutex
+	list      *list.List
+	cond      *sync.Cond
+	mutex     *sync.Mutex
+	activeMap map[string]*types.Task
 }
 
-func (q *TaskQueue) AddTask(task types.Task) {
+func (q *TaskQueue) AddTask(task *types.Task) {
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 	q.list.PushBack(task)
 	q.cond.Signal()
+	if task.Status != types.Pending {
+		delete(q.activeMap, task.Key())
+	}
 }
 
-func (q *TaskQueue) GetTask() types.Task {
+func (q *TaskQueue) GetTask() *types.Task {
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 	for q.list.Len() == 0 {
 		q.cond.Wait()
 	}
 	e := q.list.Front()
-	task := e.Value.(types.Task)
+	task := e.Value.(*types.Task)
 	q.list.Remove(e)
+	if task.Status == types.Pending {
+		q.activeMap[task.Key()] = task
+	}
 	return task
 }
 
+func (q *TaskQueue) DoneTask(task *types.Task) {
+	q.mutex.Lock()
+	defer q.mutex.Unlock()
+	delete(q.activeMap, task.Key())
+}
+
+func (q *TaskQueue) CancelTask(key string) bool {
+	q.mutex.Lock()
+	defer q.mutex.Unlock()
+	if task, ok := q.activeMap[key]; ok {
+		if task.Cancel != nil {
+			task.Cancel()
+			return true
+		}
+	}
+	for e := q.list.Front(); e != nil; e = e.Next() {
+		task := e.Value.(*types.Task)
+		if task.Key() == key {
+			if task.Cancel != nil {
+				task.Cancel()
+			}
+			q.list.Remove(e)
+			return true
+		}
+	}
+	return false
+}
+
 func (q *TaskQueue) Len() int {
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
@@ -47,20 +82,29 @@ func init() {
 func NewQueue() *TaskQueue {
 	m := &sync.Mutex{}
 	return &TaskQueue{
-		list:  list.New(),
-		cond:  sync.NewCond(m),
-		mutex: m,
+		list:      list.New(),
+		cond:      sync.NewCond(m),
+		mutex:     m,
+		activeMap: make(map[string]*types.Task),
 	}
 }
 
-func AddTask(task types.Task) {
+func AddTask(task *types.Task) {
 	Queue.AddTask(task)
 }
 
-func GetTask() types.Task {
+func GetTask() *types.Task {
 	return Queue.GetTask()
 }
 
 func Len() int {
 	return Queue.Len()
 }
+
+func CancelTask(key string) bool {
+	return Queue.CancelTask(key)
+}
+
+func DoneTask(task *types.Task) {
+	Queue.DoneTask(task)
+}

storage/alist/alist.go

@@ -3,107 +3,150 @@ package alist
 import (
 	"context"
 	"encoding/json"
-	"errors"
 	"fmt"
+	"io"
 	"net/http"
 	"net/url"
 	"os"
 	"path"
 	"time"
 
-	"github.com/imroc/req/v3"
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/types"
 )
 
-type Alist struct{}
-
-var (
-	basePath  string
-	baseUrl   string
-	reqClient *req.Client
-	loginReq  *loginRequset
-
-	ErrAlistLoginFailed = errors.New("failed to login to Alist")
-)
-
-type loginRequset struct {
-	Username string `json:"username"`
-	Password string `json:"password"`
+type Alist struct {
+	client    *http.Client
+	token     string
+	baseURL   string
+	loginInfo *loginRequest
+	config    config.AlistStorageConfig
 }
 
-type loginResponse struct {
-	Code    int    `json:"code"`
-	Message string `json:"message"`
-	Data    struct {
-		Token string `json:"token"`
-	} `json:"data"`
-}
+func (a *Alist) Init(cfg config.StorageConfig) error {
+	alistConfig, ok := cfg.(*config.AlistStorageConfig)
+	if !ok {
+		return fmt.Errorf("failed to cast alist config")
+	}
+	if err := alistConfig.Validate(); err != nil {
+		return err
+	}
+	a.config = *alistConfig
 
-func getToken() (string, error) {
-	resp, err := reqClient.R().SetBodyJsonMarshal(loginReq).Post("/api/auth/login")
-	if err != nil {
-		return "", err
-	}
-	var loginResp loginResponse
-	if err := json.Unmarshal(resp.Bytes(), &loginResp); err != nil {
-		return "", err
-	}
-	if loginResp.Code != http.StatusOK {
-		return "", fmt.Errorf("%w: %s", ErrAlistLoginFailed, loginResp.Message)
-	}
-	return loginResp.Data.Token, nil
-}
-
-func refreshToken(client *req.Client) {
-	for {
-		time.Sleep(time.Duration(config.Cfg.Storage.Alist.TokenExp) * time.Second)
-		token, err := getToken()
+	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.Errorf("Failed to refresh jwt token: %v", err)
-			continue
+			logger.L.Fatalf("Failed to create request: %v", err)
+			return err
 		}
-		client.SetCommonHeader("Authorization", token)
-		logger.L.Info("Refreshed Alist jwt token")
-	}
-}
+		req.Header.Set("Authorization", a.token)
 
-func (a *Alist) Init() {
-	basePath = config.Cfg.Storage.Alist.BasePath
-	baseUrl = config.Cfg.Storage.Alist.URL
-	reqClient = req.C().SetTLSHandshakeTimeout(time.Second * 10).SetBaseURL(baseUrl)
-	loginReq = &loginRequset{
-		Username: config.Cfg.Storage.Alist.Username,
-		Password: config.Cfg.Storage.Alist.Password,
+		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
 	}
-	token, err := getToken()
-	if err != nil {
+	a.loginInfo = &loginRequest{
+		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")
-	reqClient.SetCommonHeader("Authorization", token)
-	go refreshToken(reqClient)
+
+	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 {
-	storagePath = path.Join(basePath, storagePath)
+	logger.L.Infof("Saving file %s to %s", filePath, storagePath)
 	file, err := os.Open(filePath)
 	if err != nil {
-		return err
+		return fmt.Errorf("failed to open file: %w", err)
 	}
-	resp, err := reqClient.R().
-		SetContext(ctx).
-		SetBody(file).
-		SetHeaders(map[string]string{
-			"File-Path": url.PathEscape(storagePath),
-			"As-Task":   "true",
-		}).Put("/api/fs/put")
+	defer file.Close()
+
+	filestat, err := file.Stat()
 	if err != nil {
-		return err
+		return fmt.Errorf("failed to get file stats: %w", err)
 	}
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPut, a.baseURL+"/api/fs/put", file)
+	if err != nil {
+		return fmt.Errorf("failed to create request: %w", err)
+	}
+	req.Header.Set("Authorization", a.token)
+	req.Header.Set("File-Path", url.PathEscape(storagePath))
+	req.Header.Set("As-Task", "true")
+	req.Header.Set("Content-Type", "application/octet-stream")
+	req.ContentLength = filestat.Size()
+
+	resp, err := a.client.Do(req)
+	if err != nil {
+		return fmt.Errorf("failed to send request: %w", err)
+	}
+	defer resp.Body.Close()
+
 	if resp.StatusCode != http.StatusOK {
 		return fmt.Errorf("failed to save file to Alist: %s", resp.Status)
 	}
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return fmt.Errorf("failed to read response body: %w", err)
+	}
+
+	var putResp putResponse
+	if err := json.Unmarshal(body, &putResp); err != nil {
+		return fmt.Errorf("failed to unmarshal put response: %w", err)
+	}
+
+	if putResp.Code != http.StatusOK {
+		return fmt.Errorf("failed to save file to Alist: %d, %s", putResp.Code, putResp.Message)
+	}
+
 	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,24 +2,56 @@ 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 {
-	return fileutil.CopyFile(filePath, filepath.Join(config.Cfg.Storage.Local.BasePath, storagePath))
+	logger.L.Infof("Saving file %s to %s", filePath, storagePath)
+	absPath, err := filepath.Abs(storagePath)
+	if err != nil {
+		return err
+	}
+	if err := fileutil.CreateDir(filepath.Dir(absPath)); err != nil {
+		return err
+	}
+	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,8 +2,7 @@ package storage
 
 import (
 	"context"
-	"errors"
-	"sync"
+	"fmt"
 
 	"github.com/krau/SaveAny-Bot/config"
 	"github.com/krau/SaveAny-Bot/logger"
@@ -14,51 +13,103 @@ import (
 )
 
 type Storage interface {
-	Init()
-	Save(cttx context.Context, filePath, storagePath string) error
+	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 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()
+var UserStorages = make(map[int64][]Storage)
+
+// GetStorageByName returns storage by name from cache or creates new one
+func GetStorageByName(name string) (Storage, error) {
+	if name == "" {
+		return nil, ErrStorageNameEmpty
 	}
 
-	logger.L.Debug("Storage initialized")
+	storage, ok := Storages[name]
+	if ok {
+		return storage, nil
+	}
+	cfg := config.Cfg.GetStorageByName(name)
+	if cfg == nil {
+		return nil, fmt.Errorf("未找到存储 %s", name)
+	}
+
+	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 {
-	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)
+	}
+
+	return GetStorageByName(name)
+}
+
+func GetUserStorages(chatID int64) []Storage {
+	if chatID <= 0 {
+		return nil
+	}
+	if storages, ok := UserStorages[chatID]; ok {
+		return storages
+	}
+	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))
+	for user := range config.Cfg.GetUsersID() {
+		UserStorages[int64(user)] = GetUserStorages(int64(user))
 	}
-	errs := make([]error, 0)
-	var wg sync.WaitGroup
-	for _, storage := range Storages {
-		wg.Add(1)
-		go func(storage Storage) {
-			defer wg.Done()
-			if err := storage.Save(ctx, filePath, storagePath); err != nil {
-				errs = append(errs, err)
-			}
-		}(storage)
-	}
-	wg.Wait()
-	if len(errs) > 0 {
-		return errors.Join(errs...)
-	}
-	return nil
 }

storage/webdav/errs.go

@@ -0,0 +1,8 @@
+package webdav
+
+import "errors"
+
+var (
+	ErrFailedToCreateDirectory = errors.New("webdav: failed to create directory")
+	ErrFailedToWriteFile       = errors.New("webdav: failed to write file")
+)

storage/webdav/webdav.go

@@ -2,47 +2,68 @@ package webdav
 
 import (
 	"context"
-	"errors"
+	"fmt"
 	"os"
-	"path/filepath"
-	"strings"
+	"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
-	basePath string
-)
-
-func (w *Webdav) Init() {
-	webdavConfig := config.Cfg.Storage.Webdav
-	basePath = strings.TrimSuffix(webdavConfig.BasePath, "/")
-	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")
 	}
+	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 {
-	storagePath = filepath.Join(basePath, storagePath)
-	if err := Client.MkdirAll(filepath.Dir(storagePath), os.ModePerm); err != nil {
-		logger.L.Errorf("Failed to create directory %s: %v", filepath.Dir(storagePath), err)
-		return errors.New("webdav: failed to create directory")
+	logger.L.Infof("Saving file %s to %s", filePath, storagePath)
+	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
 	}
-	fileBytes, err := os.ReadFile(filePath)
+	file, err := os.Open(filePath)
 	if err != nil {
-		logger.L.Errorf("Failed to read file %s: %v", filePath, err)
+		logger.L.Errorf("Failed to open file %s: %v", filePath, err)
 		return err
 	}
-	if err := Client.Write(storagePath, fileBytes, os.ModePerm); err != nil {
+	defer file.Close()
+
+	if err := w.client.WriteStream(storagePath, file, os.ModePerm); err != nil {
 		logger.L.Errorf("Failed to write file %s: %v", storagePath, err)
-		return errors.New("webdav: failed to write file")
+		return ErrFailedToWriteFile
 	}
 	return nil
 }
+
+func (w *Webdav) JoinStoragePath(task types.Task) string {
+	return path.Join(w.config.BasePath, task.StoragePath)
+}

types/types.go

@@ -1,6 +1,14 @@
 package types
 
-import "context"
+import (
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"fmt"
+	"time"
+
+	"github.com/gotd/td/tg"
+)
 
 type TaskStatus string
 
@@ -14,29 +22,66 @@ 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
-	FileID      string
+	Cancel      context.CancelFunc
 	Error       error
 	Status      TaskStatus
-	FilePath    string // telegram File object's FilePath
-	FileName    string
-	Storage     StorageType
+	File        *File
+	StorageName string
 	StoragePath string
+	StartTime   time.Time
 
-	// For track progress
-	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 t.FileName
+func (t Task) Key() string {
+	return fmt.Sprintf("%d:%d", t.FileChatID, t.FileMessageID)
+}
+
+func (t Task) String() string {
+	return fmt.Sprintf("[%d:%d]:%s", t.FileChatID, t.FileMessageID, t.File.FileName)
+}
+
+func (t Task) FileName() string {
+	return t.File.FileName
+}
+
+type File struct {
+	Location tg.InputFileLocationClass
+	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)
 }