900823cdb9
* 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
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
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))
|
|
}
|
|
}
|