refactor: refactor task logic for better scalability (#76)

* refactor: a big refactor. wip

* refactor: port handle file

* refactor: place all handlers

* fix: task info nil pointer

* feat: enhance task progress tracking and context management

* feat: cancel task

* feat: stream mode

* feat: silent mode

* feat: dir cmd

* refactor: remove unused old file

* feat: rule cmd

* feat: handle silent mode

* feat: batch task

* fix: batch task progress and temp file cleanup

* refactor: update file creation and cleanup methods for better resource management

* feat: add save command with silent mode handling

* feat: message link

* feat: update message prompts to include file count in storage selection

* feat: slient save links

* refactor: reduce dup code

* feat: rule type

* feat: chose dir

* feat: refactor file handling and storage rules, improve error handling and logging

* feat: rule mode

* feat: telegraph pics

* fix: tphpics nil pointer and inaccurate dirpath

* feat: silent save telegraph

* feat: add suffix to avoid file overwrite

* feat: new storage telegram

* chore: tidy go mod

storage/load.go

@@ -0,0 +1,80 @@
+package storage
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/charmbracelet/log"
+	"github.com/krau/SaveAny-Bot/config"
+)
+
+var UserStorages = make(map[int64][]Storage)
+
+// GetStorageByName returns storage by name from cache or creates new one
+func getStorageByName(ctx context.Context, name string) (Storage, error) {
+	if name == "" {
+		return nil, ErrStorageNameEmpty
+	}
+
+	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(ctx, cfg)
+	if err != nil {
+		return nil, err
+	}
+	Storages[name] = storage
+	return storage, nil
+}
+
+// 检查 user 是否可用指定的 storage, 若不可用则返回未找到错误
+func GetStorageByUserIDAndName(ctx context.Context, chatID int64, name string) (Storage, error) {
+	if name == "" {
+		return nil, ErrStorageNameEmpty
+	}
+
+	if !config.Cfg.HasStorage(chatID, name) {
+		return nil, fmt.Errorf("没有找到用户 %d 的存储 %s", chatID, name)
+	}
+
+	return getStorageByName(ctx, name)
+}
+
+func GetUserStorages(ctx context.Context, 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(ctx, name)
+		if err != nil {
+			continue
+		}
+		storages = append(storages, storage)
+	}
+	return storages
+}
+
+func LoadStorages(ctx context.Context) {
+	logger := log.FromContext(ctx)
+	logger.Info("加载存储...")
+	for _, storage := range config.Cfg.Storages {
+		_, err := getStorageByName(ctx, storage.GetName())
+		if err != nil {
+			logger.Errorf("加载存储 %s 失败: %v", storage.GetName(), err)
+		}
+	}
+	logger.Infof("成功加载 %d 个存储", len(Storages))
+	for user := range config.Cfg.GetUsersID() {
+		UserStorages[int64(user)] = GetUserStorages(ctx, int64(user))
+	}
+}