GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-06-03 06:31:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* 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
This commit is contained in:
Krau
2025-06-15 23:57:49 +08:00
committed by GitHub
parent 280745cae3
commit 900823cdb9
150 changed files with 5730 additions and 3923 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
storage/local/local.go
View File

@@ -6,18 +6,20 @@ import (
"io"
"os"
"path/filepath"
"strings"
"github.com/charmbracelet/log"
"github.com/duke-git/lancet/v2/fileutil"
"github.com/krau/SaveAny-Bot/common"
config "github.com/krau/SaveAny-Bot/config/storage"
"github.com/krau/SaveAny-Bot/types"
storenum "github.com/krau/SaveAny-Bot/pkg/enums/storage"
)
type Local struct {
config config.LocalStorageConfig
logger *log.Logger
}
func (l *Local) Init(cfg config.StorageConfig) error {
func (l *Local) Init(ctx context.Context, cfg config.StorageConfig) error {
localConfig, ok := cfg.(*config.LocalStorageConfig)
if !ok {
return fmt.Errorf("failed to cast local config")
@@ -30,25 +32,33 @@ func (l *Local) Init(cfg config.StorageConfig) error {
if err != nil {
return fmt.Errorf("failed to create local storage directory: %w", err)
}
l.logger = log.FromContext(ctx).WithPrefix(fmt.Sprintf("local[%s]", l.config.Name))
return nil
}
func (l *Local) Type() types.StorageType {
return types.StorageTypeLocal
func (l *Local) Type() storenum.StorageType {
return storenum.Local
}
func (l *Local) Name() string {
return l.config.Name
}
func (l *Local) JoinStoragePath(task types.Task) string {
return filepath.Join(l.config.BasePath, task.StoragePath)
func (l *Local) JoinStoragePath(path string) string {
return filepath.Join(l.config.BasePath, path)
}
func (l *Local) Save(ctx context.Context, r io.Reader, storagePath string) error {
common.Log.Infof("Saving file to %s", storagePath)
l.logger.Infof("Saving file to %s", storagePath)
absPath, err := filepath.Abs(storagePath)
ext := filepath.Ext(storagePath)
base := strings.TrimSuffix(storagePath, ext)
candidate := storagePath
for i := 1; l.Exists(ctx, candidate); i++ {
candidate = fmt.Sprintf("%s_%d%s", base, i, ext)
}
absPath, err := filepath.Abs(candidate)
if err != nil {
return err
}