feat: implement transfer command for file transfers between storages

This commit is contained in:
krau
2026-01-19 21:01:50 +08:00
parent e6d8cc775a
commit 3d20fbd0fe
8 changed files with 166 additions and 159 deletions
+1 -1
View File
@@ -31,7 +31,7 @@ var CommandHandlers = []DescCommandHandler{
{"dl", i18nk.BotMsgCmdDl, handleDlCmd}, {"dl", i18nk.BotMsgCmdDl, handleDlCmd},
{"aria2dl", i18nk.BotMsgCmdAria2dl, handleAria2DlCmd}, {"aria2dl", i18nk.BotMsgCmdAria2dl, handleAria2DlCmd},
{"ytdlp", i18nk.BotMsgCmdYtdlp, handleYtdlpCmd}, {"ytdlp", i18nk.BotMsgCmdYtdlp, handleYtdlpCmd},
{"import", i18nk.BotMsgCmdImport, handleImportCmd}, {"transfer", i18nk.BotMsgCmdTransfer, handleTransferCmd},
{"task", i18nk.BotMsgCmdTask, handleTaskCmd}, {"task", i18nk.BotMsgCmdTask, handleTaskCmd},
{"cancel", i18nk.BotMsgCmdCancel, handleCancelCmd}, {"cancel", i18nk.BotMsgCmdCancel, handleCancelCmd},
{"config", i18nk.BotMsgCmdConfig, handleConfigCmd}, {"config", i18nk.BotMsgCmdConfig, handleConfigCmd},
@@ -3,6 +3,7 @@ package handlers
import ( import (
"fmt" "fmt"
"regexp" "regexp"
"strings"
"github.com/celestix/gotgproto/dispatcher" "github.com/celestix/gotgproto/dispatcher"
"github.com/celestix/gotgproto/ext" "github.com/celestix/gotgproto/ext"
@@ -12,8 +13,6 @@ import (
"github.com/krau/SaveAny-Bot/common/i18n/i18nk" "github.com/krau/SaveAny-Bot/common/i18n/i18nk"
"github.com/krau/SaveAny-Bot/common/utils/strutil" "github.com/krau/SaveAny-Bot/common/utils/strutil"
"github.com/krau/SaveAny-Bot/common/utils/tgutil" "github.com/krau/SaveAny-Bot/common/utils/tgutil"
"github.com/krau/SaveAny-Bot/config"
storconfig "github.com/krau/SaveAny-Bot/config/storage"
"github.com/krau/SaveAny-Bot/core" "github.com/krau/SaveAny-Bot/core"
"github.com/krau/SaveAny-Bot/core/tasks/batchimport" "github.com/krau/SaveAny-Bot/core/tasks/batchimport"
"github.com/krau/SaveAny-Bot/pkg/storagetypes" "github.com/krau/SaveAny-Bot/pkg/storagetypes"
@@ -21,81 +20,105 @@ import (
"github.com/rs/xid" "github.com/rs/xid"
) )
func handleImportCmd(ctx *ext.Context, update *ext.Update) error { func handleTransferCmd(ctx *ext.Context, update *ext.Update) error {
logger := log.FromContext(ctx) logger := log.FromContext(ctx)
args := strutil.ParseArgsRespectQuotes(update.EffectiveMessage.Text) args := strutil.ParseArgsRespectQuotes(update.EffectiveMessage.Text)
if len(args) < 3 { if len(args) < 3 {
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgImportUsage, nil)), nil) ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferUsage, nil)), nil)
return dispatcher.EndGroups return dispatcher.EndGroups
} }
storageName := args[1] // Parse source: storage_name:/path
dirPath := args[2] sourceParts := strings.SplitN(args[1], ":", 2)
if len(sourceParts) != 2 {
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferErrorInvalidSource, nil)), nil)
return dispatcher.EndGroups
}
sourceStorageName := sourceParts[0]
sourcePath := sourceParts[1]
// Parse target: storage_name:/path
targetParts := strings.SplitN(args[2], ":", 2)
if len(targetParts) != 2 {
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferErrorInvalidTarget, nil)), nil)
return dispatcher.EndGroups
}
targetStorageName := targetParts[0]
targetPath := targetParts[1]
userID := update.GetUserChat().GetID() userID := update.GetUserChat().GetID()
stor, err := storage.GetStorageByUserIDAndName(ctx, userID, storageName) // Get source storage
sourceStorage, err := storage.GetStorageByUserIDAndName(ctx, userID, sourceStorageName)
if err != nil { if err != nil {
logger.Errorf("Failed to get storage by user ID and name: %s", err) logger.Errorf("Failed to get source storage by user ID and name: %s", err)
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgImportErrorStorageNotFound, map[string]any{ ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferErrorStorageNotFound, map[string]any{
"StorageName": storageName, "StorageName": sourceStorageName,
"Error": err, "Error": err,
})), nil) })), nil)
return dispatcher.EndGroups return dispatcher.EndGroups
} }
listable, ok := stor.(storage.StorageListable) // Check if source storage supports listing
listable, ok := sourceStorage.(storage.StorageListable)
if !ok { if !ok {
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgImportErrorStorageNotListable, map[string]any{ ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferErrorStorageNotListable, map[string]any{
"StorageName": storageName, "StorageName": sourceStorageName,
})), nil) })), nil)
return dispatcher.EndGroups return dispatcher.EndGroups
} }
_, ok = stor.(storage.StorageReadable) // Check if source storage supports reading
_, ok = sourceStorage.(storage.StorageReadable)
if !ok { if !ok {
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgImportErrorStorageNotReadable, map[string]any{ ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferErrorStorageNotReadable, map[string]any{
"StorageName": storageName, "StorageName": sourceStorageName,
})), nil) })), nil)
return dispatcher.EndGroups return dispatcher.EndGroups
} }
telegramStorage, err := storage.GetTelegramStorageByUserID(ctx, userID) // Get target storage
targetStorage, err := storage.GetStorageByUserIDAndName(ctx, userID, targetStorageName)
if err != nil { if err != nil {
ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgImportErrorNoTelegramStorage, map[string]any{ logger.Errorf("Failed to get target storage by user ID and name: %s", err)
"Error": err, ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferErrorTargetNotFound, map[string]any{
"StorageName": targetStorageName,
"Error": err,
})), nil) })), nil)
return dispatcher.EndGroups return dispatcher.EndGroups
} }
replied, err := ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgImportInfoFetchingFiles, nil)), nil) // Fetch file list
replied, err := ctx.Reply(update, ext.ReplyTextString(i18n.T(i18nk.BotMsgTransferInfoFetchingFiles, nil)), nil)
if err != nil { if err != nil {
logger.Errorf("Failed to reply: %s", err) logger.Errorf("Failed to reply: %s", err)
return dispatcher.EndGroups return dispatcher.EndGroups
} }
files, err := listable.ListFiles(ctx, dirPath) files, err := listable.ListFiles(ctx, sourcePath)
if err != nil { if err != nil {
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{ ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID, ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportErrorListFilesFailed, map[string]any{"Error": err}), Message: i18n.T(i18nk.BotMsgTransferErrorListFilesFailed, map[string]any{"Error": err}),
}) })
return dispatcher.EndGroups return dispatcher.EndGroups
} }
// Optional filter
var filter *regexp.Regexp var filter *regexp.Regexp
if len(args) >= 5 { if len(args) >= 4 {
filter, err = regexp.Compile(args[4]) filter, err = regexp.Compile(args[3])
if err != nil { if err != nil {
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{ ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID, ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportErrorInvalidRegex, map[string]any{"Error": err}), Message: i18n.T(i18nk.BotMsgTransferErrorInvalidRegex, map[string]any{"Error": err}),
}) })
return dispatcher.EndGroups return dispatcher.EndGroups
} }
} }
// Filter files
filteredFiles := make([]storagetypes.FileInfo, 0) filteredFiles := make([]storagetypes.FileInfo, 0)
for _, file := range files { for _, file := range files {
if file.IsDir { if file.IsDir {
@@ -110,47 +133,21 @@ func handleImportCmd(ctx *ext.Context, update *ext.Update) error {
if len(filteredFiles) == 0 { if len(filteredFiles) == 0 {
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{ ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID, ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportErrorNoFilesToImport, nil), Message: i18n.T(i18nk.BotMsgTransferErrorNoFilesToTransfer, nil),
})
return dispatcher.EndGroups
}
// Get default chat_id from Telegram storage config
targetChatID := int64(0)
if telegramCfg := config.C().GetStorageByName(telegramStorage.Name()); telegramCfg != nil {
if tgCfg, ok := telegramCfg.(*storconfig.TelegramStorageConfig); ok {
targetChatID = tgCfg.ChatID
}
}
if len(args) >= 4 {
parsedChatID, err := tgutil.ParseChatID(ctx, args[3])
if err != nil {
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportErrorInvalidChatId, map[string]any{"Error": err}),
})
return dispatcher.EndGroups
}
targetChatID = parsedChatID
}
if targetChatID == 0 {
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportErrorNoTargetChatId, nil),
}) })
return dispatcher.EndGroups return dispatcher.EndGroups
} }
// Create task elements
elems := make([]batchimport.TaskElement, 0, len(filteredFiles)) elems := make([]batchimport.TaskElement, 0, len(filteredFiles))
var totalSize int64 var totalSize int64
for _, file := range filteredFiles { for _, file := range filteredFiles {
elem := batchimport.NewTaskElement(stor, file, telegramStorage, targetChatID) elem := batchimport.NewTaskElement(sourceStorage, file, targetStorage, targetPath)
elems = append(elems, *elem) elems = append(elems, *elem)
totalSize += file.Size totalSize += file.Size
} }
// Create and add task
taskID := xid.New().String() taskID := xid.New().String()
injectCtx := tgutil.ExtWithContext(ctx.Context, ctx) injectCtx := tgutil.ExtWithContext(ctx.Context, ctx)
task := batchimport.NewBatchImportTask( task := batchimport.NewBatchImportTask(
@@ -164,14 +161,14 @@ func handleImportCmd(ctx *ext.Context, update *ext.Update) error {
if err := core.AddTask(injectCtx, task); err != nil { if err := core.AddTask(injectCtx, task); err != nil {
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{ ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID, ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportErrorAddTaskFailed, map[string]any{"Error": err}), Message: i18n.T(i18nk.BotMsgTransferErrorAddTaskFailed, map[string]any{"Error": err}),
}) })
return dispatcher.EndGroups return dispatcher.EndGroups
} }
ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{ ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
ID: replied.ID, ID: replied.ID,
Message: i18n.T(i18nk.BotMsgImportInfoTaskAdded, map[string]any{ Message: i18n.T(i18nk.BotMsgTransferInfoTaskAdded, map[string]any{
"Count": len(elems), "Count": len(elems),
"SizeMB": fmt.Sprintf("%.2f", float64(totalSize)/(1024*1024)), "SizeMB": fmt.Sprintf("%.2f", float64(totalSize)/(1024*1024)),
"TaskID": taskID, "TaskID": taskID,
+29 -28
View File
@@ -31,6 +31,7 @@ const (
BotMsgCmdStorage Key = "bot.msg.cmd.storage" BotMsgCmdStorage Key = "bot.msg.cmd.storage"
BotMsgCmdSyncpeers Key = "bot.msg.cmd.syncpeers" BotMsgCmdSyncpeers Key = "bot.msg.cmd.syncpeers"
BotMsgCmdTask Key = "bot.msg.cmd.task" BotMsgCmdTask Key = "bot.msg.cmd.task"
BotMsgCmdTransfer Key = "bot.msg.cmd.transfer"
BotMsgCmdUnwatch Key = "bot.msg.cmd.unwatch" BotMsgCmdUnwatch Key = "bot.msg.cmd.unwatch"
BotMsgCmdUpdate Key = "bot.msg.cmd.update" BotMsgCmdUpdate Key = "bot.msg.cmd.update"
BotMsgCmdWatch Key = "bot.msg.cmd.watch" BotMsgCmdWatch Key = "bot.msg.cmd.watch"
@@ -106,20 +107,6 @@ const (
BotMsgDlInfoFilesSelectStorage Key = "bot.msg.dl.info_files_select_storage" BotMsgDlInfoFilesSelectStorage Key = "bot.msg.dl.info_files_select_storage"
BotMsgDlUsage Key = "bot.msg.dl.usage" BotMsgDlUsage Key = "bot.msg.dl.usage"
BotMsgHelpTextFmt Key = "bot.msg.help_text_fmt" BotMsgHelpTextFmt Key = "bot.msg.help_text_fmt"
BotMsgImportErrorAddTaskFailed Key = "bot.msg.import.error_add_task_failed"
BotMsgImportErrorInvalidChatId Key = "bot.msg.import.error_invalid_chat_id"
BotMsgImportErrorInvalidRegex Key = "bot.msg.import.error_invalid_regex"
BotMsgImportErrorListFilesFailed Key = "bot.msg.import.error_list_files_failed"
BotMsgImportErrorNoFilesToImport Key = "bot.msg.import.error_no_files_to_import"
BotMsgImportErrorNoTargetChatId Key = "bot.msg.import.error_no_target_chat_id"
BotMsgImportErrorNoTelegramStorage Key = "bot.msg.import.error_no_telegram_storage"
BotMsgImportErrorStorageNotFound Key = "bot.msg.import.error_storage_not_found"
BotMsgImportErrorStorageNotListable Key = "bot.msg.import.error_storage_not_listable"
BotMsgImportErrorStorageNotReadable Key = "bot.msg.import.error_storage_not_readable"
BotMsgImportInfoFetchingFiles Key = "bot.msg.import.info_fetching_files"
BotMsgImportInfoTaskAdded Key = "bot.msg.import.info_task_added"
BotMsgImportStartStats Key = "bot.msg.import.start_stats"
BotMsgImportUsage Key = "bot.msg.import.usage"
BotMsgMediaGroupErrorBuildStorageSelectKeyboardFailed Key = "bot.msg.media_group.error_build_storage_select_keyboard_failed" BotMsgMediaGroupErrorBuildStorageSelectKeyboardFailed Key = "bot.msg.media_group.error_build_storage_select_keyboard_failed"
BotMsgMediaGroupInfoGroupFoundFilesSelectStorage Key = "bot.msg.media_group.info_group_found_files_select_storage" BotMsgMediaGroupInfoGroupFoundFilesSelectStorage Key = "bot.msg.media_group.info_group_found_files_select_storage"
BotMsgMediaGroupInfoSavingFiles Key = "bot.msg.media_group.info_saving_files" BotMsgMediaGroupInfoSavingFiles Key = "bot.msg.media_group.info_saving_files"
@@ -164,20 +151,6 @@ const (
BotMsgProgressFileProcessingPrefix Key = "bot.msg.progress.file_processing_prefix" BotMsgProgressFileProcessingPrefix Key = "bot.msg.progress.file_processing_prefix"
BotMsgProgressFileSizePrefix Key = "bot.msg.progress.file_size_prefix" BotMsgProgressFileSizePrefix Key = "bot.msg.progress.file_size_prefix"
BotMsgProgressFileStartPrefix Key = "bot.msg.progress.file_start_prefix" BotMsgProgressFileStartPrefix Key = "bot.msg.progress.file_start_prefix"
BotMsgProgressImportAvgSpeedPrefix Key = "bot.msg.progress.import_avg_speed_prefix"
BotMsgProgressImportElapsedTimePrefix Key = "bot.msg.progress.import_elapsed_time_prefix"
BotMsgProgressImportFailedFilesPrefix Key = "bot.msg.progress.import_failed_files_prefix"
BotMsgProgressImportFailedPrefix Key = "bot.msg.progress.import_failed_prefix"
BotMsgProgressImportProcessingMore Key = "bot.msg.progress.import_processing_more"
BotMsgProgressImportProcessingPrefix Key = "bot.msg.progress.import_processing_prefix"
BotMsgProgressImportProgressPrefix Key = "bot.msg.progress.import_progress_prefix"
BotMsgProgressImportRemainingTimePrefix Key = "bot.msg.progress.import_remaining_time_prefix"
BotMsgProgressImportSpeedPrefix Key = "bot.msg.progress.import_speed_prefix"
BotMsgProgressImportStartPrefix Key = "bot.msg.progress.import_start_prefix"
BotMsgProgressImportSuccessPrefix Key = "bot.msg.progress.import_success_prefix"
BotMsgProgressImportTotalFilesPrefix Key = "bot.msg.progress.import_total_files_prefix"
BotMsgProgressImportTotalSizePrefix Key = "bot.msg.progress.import_total_size_prefix"
BotMsgProgressImportUploadedPrefix Key = "bot.msg.progress.import_uploaded_prefix"
BotMsgProgressParsedDonePrefix Key = "bot.msg.progress.parsed_done_prefix" BotMsgProgressParsedDonePrefix Key = "bot.msg.progress.parsed_done_prefix"
BotMsgProgressParsedStartPrefix Key = "bot.msg.progress.parsed_start_prefix" BotMsgProgressParsedStartPrefix Key = "bot.msg.progress.parsed_start_prefix"
BotMsgProgressProcessingListPrefix Key = "bot.msg.progress.processing_list_prefix" BotMsgProgressProcessingListPrefix Key = "bot.msg.progress.processing_list_prefix"
@@ -190,6 +163,20 @@ const (
BotMsgProgressTelegraphProgressPrefix Key = "bot.msg.progress.telegraph_progress_prefix" BotMsgProgressTelegraphProgressPrefix Key = "bot.msg.progress.telegraph_progress_prefix"
BotMsgProgressTelegraphStartPrefix Key = "bot.msg.progress.telegraph_start_prefix" BotMsgProgressTelegraphStartPrefix Key = "bot.msg.progress.telegraph_start_prefix"
BotMsgProgressTotalSizePrefix Key = "bot.msg.progress.total_size_prefix" BotMsgProgressTotalSizePrefix Key = "bot.msg.progress.total_size_prefix"
BotMsgProgressTransferAvgSpeedPrefix Key = "bot.msg.progress.transfer_avg_speed_prefix"
BotMsgProgressTransferElapsedTimePrefix Key = "bot.msg.progress.transfer_elapsed_time_prefix"
BotMsgProgressTransferFailedFilesPrefix Key = "bot.msg.progress.transfer_failed_files_prefix"
BotMsgProgressTransferFailedPrefix Key = "bot.msg.progress.transfer_failed_prefix"
BotMsgProgressTransferProcessingMore Key = "bot.msg.progress.transfer_processing_more"
BotMsgProgressTransferProcessingPrefix Key = "bot.msg.progress.transfer_processing_prefix"
BotMsgProgressTransferProgressPrefix Key = "bot.msg.progress.transfer_progress_prefix"
BotMsgProgressTransferRemainingTimePrefix Key = "bot.msg.progress.transfer_remaining_time_prefix"
BotMsgProgressTransferSpeedPrefix Key = "bot.msg.progress.transfer_speed_prefix"
BotMsgProgressTransferStartPrefix Key = "bot.msg.progress.transfer_start_prefix"
BotMsgProgressTransferSuccessPrefix Key = "bot.msg.progress.transfer_success_prefix"
BotMsgProgressTransferTotalFilesPrefix Key = "bot.msg.progress.transfer_total_files_prefix"
BotMsgProgressTransferTotalSizePrefix Key = "bot.msg.progress.transfer_total_size_prefix"
BotMsgProgressTransferUploadedPrefix Key = "bot.msg.progress.transfer_uploaded_prefix"
BotMsgProgressYtdlpDone Key = "bot.msg.progress.ytdlp_done" BotMsgProgressYtdlpDone Key = "bot.msg.progress.ytdlp_done"
BotMsgProgressYtdlpDownloading Key = "bot.msg.progress.ytdlp_downloading" BotMsgProgressYtdlpDownloading Key = "bot.msg.progress.ytdlp_downloading"
BotMsgProgressYtdlpStart Key = "bot.msg.progress.ytdlp_start" BotMsgProgressYtdlpStart Key = "bot.msg.progress.ytdlp_start"
@@ -245,6 +232,20 @@ const (
BotMsgTelegraphInfoPicCountPrefix Key = "bot.msg.telegraph.info_pic_count_prefix" BotMsgTelegraphInfoPicCountPrefix Key = "bot.msg.telegraph.info_pic_count_prefix"
BotMsgTelegraphInfoPromptSelectStorage Key = "bot.msg.telegraph.info_prompt_select_storage" BotMsgTelegraphInfoPromptSelectStorage Key = "bot.msg.telegraph.info_prompt_select_storage"
BotMsgTelegraphInfoTitlePrefix Key = "bot.msg.telegraph.info_title_prefix" BotMsgTelegraphInfoTitlePrefix Key = "bot.msg.telegraph.info_title_prefix"
BotMsgTransferErrorAddTaskFailed Key = "bot.msg.transfer.error_add_task_failed"
BotMsgTransferErrorInvalidRegex Key = "bot.msg.transfer.error_invalid_regex"
BotMsgTransferErrorInvalidSource Key = "bot.msg.transfer.error_invalid_source"
BotMsgTransferErrorInvalidTarget Key = "bot.msg.transfer.error_invalid_target"
BotMsgTransferErrorListFilesFailed Key = "bot.msg.transfer.error_list_files_failed"
BotMsgTransferErrorNoFilesToTransfer Key = "bot.msg.transfer.error_no_files_to_transfer"
BotMsgTransferErrorStorageNotFound Key = "bot.msg.transfer.error_storage_not_found"
BotMsgTransferErrorStorageNotListable Key = "bot.msg.transfer.error_storage_not_listable"
BotMsgTransferErrorStorageNotReadable Key = "bot.msg.transfer.error_storage_not_readable"
BotMsgTransferErrorTargetNotFound Key = "bot.msg.transfer.error_target_not_found"
BotMsgTransferInfoFetchingFiles Key = "bot.msg.transfer.info_fetching_files"
BotMsgTransferInfoTaskAdded Key = "bot.msg.transfer.info_task_added"
BotMsgTransferStartStats Key = "bot.msg.transfer.start_stats"
BotMsgTransferUsage Key = "bot.msg.transfer.usage"
BotMsgUpdateButtonUpgrade Key = "bot.msg.update.button_upgrade" BotMsgUpdateButtonUpgrade Key = "bot.msg.update.button_upgrade"
BotMsgUpdateErrorCheckLatestFailed Key = "bot.msg.update.error_check_latest_failed" BotMsgUpdateErrorCheckLatestFailed Key = "bot.msg.update.error_check_latest_failed"
BotMsgUpdateErrorNoReleaseFound Key = "bot.msg.update.error_no_release_found" BotMsgUpdateErrorNoReleaseFound Key = "bot.msg.update.error_no_release_found"
+28 -21
View File
@@ -54,6 +54,7 @@ bot:
aria2dl: "Download files using Aria2" aria2dl: "Download files using Aria2"
ytdlp: "Download video/audio using yt-dlp" ytdlp: "Download video/audio using yt-dlp"
import: "Import files from storage to Telegram" import: "Import files from storage to Telegram"
transfer: "Transfer files between storages"
task: "Manage task queue" task: "Manage task queue"
cancel: "Cancel task" cancel: "Cancel task"
watch: "Watch chats (UserBot)" watch: "Watch chats (UserBot)"
@@ -296,20 +297,26 @@ bot:
info_urls_select_storage: "Found {{.Count}} links, please select storage" info_urls_select_storage: "Found {{.Count}} links, please select storage"
info_downloading: "Downloading via yt-dlp..." info_downloading: "Downloading via yt-dlp..."
error_download_failed: "yt-dlp download failed: {{.Error}}" error_download_failed: "yt-dlp download failed: {{.Error}}"
import: transfer:
usage: "Usage: /import <storage_name> <dir_path> [target_chat_id] [filter]\n\nExamples:\n/import local1 /downloads\n/import MyAlist /media/photos -1001234567890\n/import MyLocal /backup \".*[.]mp4$\"" usage: |
Usage: /transfer <source_storage>:/<source_path> <target_storage>:/<target_path> [filter]
Examples:
/transfer local1:/downloads tg1:/backup
/transfer alist1:/media/photos local1:/photos
/transfer webdav1:/files tg1:/archive ".*\.mp4$"
error_invalid_source: "Invalid source path format, should be: storage_name:/path"
error_invalid_target: "Invalid target path format, should be: storage_name:/path"
error_storage_not_found: "Storage '{{.StorageName}}' not found or access denied: {{.Error}}" error_storage_not_found: "Storage '{{.StorageName}}' not found or access denied: {{.Error}}"
error_storage_not_listable: "Storage '{{.StorageName}}' does not support listing files" error_storage_not_listable: "Storage '{{.StorageName}}' does not support listing files"
error_storage_not_readable: "Storage '{{.StorageName}}' does not support reading files" error_storage_not_readable: "Storage '{{.StorageName}}' does not support reading files"
error_no_telegram_storage: "No Telegram storage found: {{.Error}}" error_target_not_found: "Target storage '{{.StorageName}}' not found or access denied: {{.Error}}"
info_fetching_files: "Fetching file list..." info_fetching_files: "Fetching file list..."
error_list_files_failed: "Failed to list files: {{.Error}}" error_list_files_failed: "Failed to list files: {{.Error}}"
error_invalid_regex: "Invalid regular expression: {{.Error}}" error_invalid_regex: "Invalid regular expression: {{.Error}}"
error_no_files_to_import: "No files to import in directory" error_no_files_to_transfer: "No files to transfer in directory"
error_invalid_chat_id: "Invalid Chat ID: {{.Error}}"
error_no_target_chat_id: "No target channel ID specified and Telegram storage has no default chat_id configured"
error_add_task_failed: "Failed to add task: {{.Error}}" error_add_task_failed: "Failed to add task: {{.Error}}"
info_task_added: "Added {{.Count}} files to import queue\nTotal size: {{.SizeMB}} MB\nTask ID: {{.TaskID}}" info_task_added: "Added {{.Count}} files to transfer queue\nTotal size: {{.SizeMB}} MB\nTask ID: {{.TaskID}}"
start_stats: "Total files: {{.Count}}\nTotal size: {{.SizeMB}} MB"
cancel: cancel:
usage: "Usage: /cancel <task_id>" usage: "Usage: /cancel <task_id>"
error_cancel_failed: "Failed to cancel task: {{.Error}}" error_cancel_failed: "Failed to cancel task: {{.Error}}"
@@ -358,20 +365,20 @@ bot:
ytdlp_done: "yt-dlp download completed and transferred ({{.Count}} files)\n" ytdlp_done: "yt-dlp download completed and transferred ({{.Count}} files)\n"
downloaded_prefix: "\nDownloaded: " downloaded_prefix: "\nDownloaded: "
current_speed_prefix: "\nCurrent speed: " current_speed_prefix: "\nCurrent speed: "
import_start_prefix: "Importing: " transfer_start_prefix: "Importing: "
import_progress_prefix: "Import progress: " transfer_progress_prefix: "Import progress: "
import_uploaded_prefix: "\nUploaded: " transfer_uploaded_prefix: "\nUploaded: "
import_speed_prefix: "\nSpeed: " transfer_speed_prefix: "\nSpeed: "
import_remaining_time_prefix: "\nRemaining time: " transfer_remaining_time_prefix: "\nRemaining time: "
import_processing_prefix: "\nProcessing:\n" transfer_processing_prefix: "\nProcessing:\n"
import_processing_more: "...and {{.Count}} more files\n" transfer_processing_more: "...and {{.Count}} more files\n"
import_failed_prefix: "Import failed\n" transfer_failed_prefix: "Import failed\n"
import_success_prefix: "Import completed\n" transfer_success_prefix: "Import completed\n"
import_total_files_prefix: "\nTotal files: " transfer_total_files_prefix: "\nTotal files: "
import_total_size_prefix: "\nTotal size: " transfer_total_size_prefix: "\nTotal size: "
import_elapsed_time_prefix: "\nElapsed time: " transfer_elapsed_time_prefix: "\nElapsed time: "
import_avg_speed_prefix: "\nAverage speed: " transfer_avg_speed_prefix: "\nAverage speed: "
import_failed_files_prefix: "\nFailed files: " transfer_failed_files_prefix: "\nFailed files: "
syncpeers: syncpeers:
start: "Starting to sync peers..." start: "Starting to sync peers..."
done: "Peer sync completed, total {{.Count}} chats synced" done: "Peer sync completed, total {{.Count}} chats synced"
+25 -24
View File
@@ -55,6 +55,7 @@ bot:
aria2dl: "使用 Aria2 下载给定链接的文件" aria2dl: "使用 Aria2 下载给定链接的文件"
ytdlp: "使用 yt-dlp 下载视频/音频" ytdlp: "使用 yt-dlp 下载视频/音频"
import: "从存储端导入文件到 Telegram" import: "从存储端导入文件到 Telegram"
transfer: "在存储端之间传输文件"
task: "管理任务队列" task: "管理任务队列"
cancel: "取消任务" cancel: "取消任务"
watch: "监听聊天(UserBot)" watch: "监听聊天(UserBot)"
@@ -297,25 +298,25 @@ bot:
info_urls_select_storage: "共 {{.Count}} 个链接, 请选择存储位置" info_urls_select_storage: "共 {{.Count}} 个链接, 请选择存储位置"
info_downloading: "正在通过 yt-dlp 下载..." info_downloading: "正在通过 yt-dlp 下载..."
error_download_failed: "yt-dlp 下载失败: {{.Error}}" error_download_failed: "yt-dlp 下载失败: {{.Error}}"
import: transfer:
usage: | usage: |
用法: /import <storage_name> <dir_path> [target_chat_id] [filter] 用法: /transfer <source_storage>:/<source_path> <target_storage>:/<target_path> [filter]
示例: 示例:
/import 本机1 /downloads /transfer local1:/downloads tg1:/backup
/import MyAlist /media/photos -1001234567890 /transfer alist1:/media/photos local1:/photos
/import MyLocal /backup ".*\.mp4$" /transfer webdav1:/files tg1:/archive ".*\.mp4$"
error_invalid_source: "源路径格式无效,应为: storage_name:/path"
error_invalid_target: "目标路径格式无效,应为: storage_name:/path"
error_storage_not_found: "存储端 '{{.StorageName}}' 不存在或您无权访问: {{.Error}}" error_storage_not_found: "存储端 '{{.StorageName}}' 不存在或您无权访问: {{.Error}}"
error_storage_not_listable: "存储端 '{{.StorageName}}' 不支持列举文件功能" error_storage_not_listable: "存储端 '{{.StorageName}}' 不支持列举文件功能"
error_storage_not_readable: "存储端 '{{.StorageName}}' 不支持读取文件功能" error_storage_not_readable: "存储端 '{{.StorageName}}' 不支持读取文件功能"
error_no_telegram_storage: "未找到可用的 Telegram 存储: {{.Error}}" error_target_not_found: "目标存储端 '{{.StorageName}}' 不存在或您无权访问: {{.Error}}"
info_fetching_files: "正在获取文件列表..." info_fetching_files: "正在获取文件列表..."
error_list_files_failed: "获取文件列表失败: {{.Error}}" error_list_files_failed: "获取文件列表失败: {{.Error}}"
error_invalid_regex: "正则表达式无效: {{.Error}}" error_invalid_regex: "正则表达式无效: {{.Error}}"
error_no_files_to_import: "目录中没有可导入的文件" error_no_files_to_transfer: "目录中没有可传输的文件"
error_invalid_chat_id: "无效的 Chat ID: {{.Error}}"
error_no_target_chat_id: "未指定目标频道 ID,且 Telegram 存储未配置默认 chat_id"
error_add_task_failed: "添加任务失败: {{.Error}}" error_add_task_failed: "添加任务失败: {{.Error}}"
info_task_added: "已添加 {{.Count}} 个文件到导入队列\n总大小: {{.SizeMB}} MB\n任务 ID: {{.TaskID}}" info_task_added: "已添加 {{.Count}} 个文件到传输队列\n总大小: {{.SizeMB}} MB\n任务 ID: {{.TaskID}}"
start_stats: "总文件数: {{.Count}}\n总大小: {{.SizeMB}} MB" start_stats: "总文件数: {{.Count}}\n总大小: {{.SizeMB}} MB"
cancel: cancel:
usage: "用法: /cancel <task_id>" usage: "用法: /cancel <task_id>"
@@ -365,20 +366,20 @@ bot:
ytdlp_done: "yt-dlp 下载完成并已转存 ({{.Count}} 个文件)\n" ytdlp_done: "yt-dlp 下载完成并已转存 ({{.Count}} 个文件)\n"
downloaded_prefix: "\n已下载: " downloaded_prefix: "\n已下载: "
current_speed_prefix: "\n当前速度: " current_speed_prefix: "\n当前速度: "
import_start_prefix: "正在导入: " transfer_start_prefix: "正在导入: "
import_progress_prefix: "导入进度: " transfer_progress_prefix: "导入进度: "
import_uploaded_prefix: "\n已上传: " transfer_uploaded_prefix: "\n已上传: "
import_speed_prefix: "\n速度: " transfer_speed_prefix: "\n速度: "
import_remaining_time_prefix: "\n剩余时间: " transfer_remaining_time_prefix: "\n剩余时间: "
import_processing_prefix: "\n正在处理:\n" transfer_processing_prefix: "\n正在处理:\n"
import_processing_more: "...和其他 {{.Count}} 个文件\n" transfer_processing_more: "...和其他 {{.Count}} 个文件\n"
import_failed_prefix: "导入失败\n" transfer_failed_prefix: "导入失败\n"
import_success_prefix: "导入完成\n" transfer_success_prefix: "导入完成\n"
import_total_files_prefix: "\n总文件数: " transfer_total_files_prefix: "\n总文件数: "
import_total_size_prefix: "\n总大小: " transfer_total_size_prefix: "\n总大小: "
import_elapsed_time_prefix: "\n耗时: " transfer_elapsed_time_prefix: "\n耗时: "
import_avg_speed_prefix: "\n平均速度: " transfer_avg_speed_prefix: "\n平均速度: "
import_failed_files_prefix: "\n失败文件数: " transfer_failed_files_prefix: "\n失败文件数: "
syncpeers: syncpeers:
start: "正在同步对话列表..." start: "正在同步对话列表..."
success: "对话列表同步完成, 共同步 {{.Count}} 个对话" success: "对话列表同步完成, 共同步 {{.Count}} 个对话"
+7 -6
View File
@@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"io" "io"
"os" "os"
"path"
"path/filepath" "path/filepath"
"github.com/charmbracelet/log" "github.com/charmbracelet/log"
@@ -84,15 +85,15 @@ func (t *Task) processElement(ctx context.Context, elem TaskElement) error {
} }
defer reader.Close() defer reader.Close()
// Build Telegram storage path: /<chat_id>/<filename> // Build target storage path: /target_path/filename
storagePath := fmt.Sprintf("/%d/%s", elem.TargetChatID, elem.FileInfo.Name) storagePath := path.Join(elem.TargetPath, elem.FileInfo.Name)
// 注入文件大小到 context // Inject file size into context
ctx = context.WithValue(ctx, ctxkey.ContentLength, size) ctx = context.WithValue(ctx, ctxkey.ContentLength, size)
if config.C().Stream { if config.C().Stream {
if err := elem.TargetStorage.Save(ctx, reader, storagePath); err != nil { if err := elem.TargetStorage.Save(ctx, reader, storagePath); err != nil {
return fmt.Errorf("failed to upload file to telegram: %w", err) return fmt.Errorf("failed to upload file to storage: %w", err)
} }
} else { } else {
logger.Info("Downloading to temporary file for ReadSeeker support") logger.Info("Downloading to temporary file for ReadSeeker support")
@@ -107,9 +108,9 @@ func (t *Task) processElement(ctx context.Context, elem TaskElement) error {
return fmt.Errorf("failed to seek temp file: %w", err) return fmt.Errorf("failed to seek temp file: %w", err)
} }
logger.Infof("Uploading file to Telegram storage (size: %d bytes)", size) logger.Infof("Uploading file to storage (size: %d bytes)", size)
if err := elem.TargetStorage.Save(ctx, tempFile, storagePath); err != nil { if err := elem.TargetStorage.Save(ctx, tempFile, storagePath); err != nil {
return fmt.Errorf("failed to upload file to telegram: %w", err) return fmt.Errorf("failed to upload file to storage: %w", err)
} }
} }
+19 -19
View File
@@ -43,14 +43,14 @@ func (p *Progress) OnStart(ctx context.Context, info TaskInfo) {
log.FromContext(ctx).Debugf("Batch import task progress tracking started for message %d in chat %d", p.MessageID, p.ChatID) log.FromContext(ctx).Debugf("Batch import task progress tracking started for message %d in chat %d", p.MessageID, p.ChatID)
sizeMB := float64(info.TotalSize()) / (1024 * 1024) sizeMB := float64(info.TotalSize()) / (1024 * 1024)
statsText := i18n.T(i18nk.BotMsgImportStartStats, map[string]any{ statsText := i18n.T(i18nk.BotMsgTransferStartStats, map[string]any{
"SizeMB": fmt.Sprintf("%.2f", sizeMB), "SizeMB": fmt.Sprintf("%.2f", sizeMB),
"Count": info.Count(), "Count": info.Count(),
}) })
entityBuilder := entity.Builder{} entityBuilder := entity.Builder{}
if err := styling.Perform(&entityBuilder, if err := styling.Perform(&entityBuilder,
styling.Plain(i18n.T(i18nk.BotMsgProgressImportStartPrefix, nil)), styling.Plain(i18n.T(i18nk.BotMsgProgressTransferStartPrefix, nil)),
styling.Code(statsText), styling.Code(statsText),
); err != nil { ); err != nil {
log.FromContext(ctx).Errorf("Failed to build entities: %s", err) log.FromContext(ctx).Errorf("Failed to build entities: %s", err)
@@ -94,9 +94,9 @@ func (p *Progress) OnProgress(ctx context.Context, info TaskInfo) {
entityBuilder := entity.Builder{} entityBuilder := entity.Builder{}
var progressText strings.Builder var progressText strings.Builder
progressText.WriteString(i18n.T(i18nk.BotMsgProgressImportProgressPrefix, nil)) progressText.WriteString(i18n.T(i18nk.BotMsgProgressTransferProgressPrefix, nil))
progressText.WriteString(fmt.Sprintf("%d%%", percent)) progressText.WriteString(fmt.Sprintf("%d%%", percent))
progressText.WriteString(i18n.T(i18nk.BotMsgProgressImportUploadedPrefix, nil)) progressText.WriteString(i18n.T(i18nk.BotMsgProgressTransferUploadedPrefix, nil))
progressText.WriteString(fmt.Sprintf("%.2f MB / %.2f MB", progressText.WriteString(fmt.Sprintf("%.2f MB / %.2f MB",
float64(info.Uploaded())/(1024*1024), float64(info.Uploaded())/(1024*1024),
float64(info.TotalSize())/(1024*1024))) float64(info.TotalSize())/(1024*1024)))
@@ -104,22 +104,22 @@ func (p *Progress) OnProgress(ctx context.Context, info TaskInfo) {
if p.start.Unix() > 0 { if p.start.Unix() > 0 {
elapsed := time.Since(p.start) elapsed := time.Since(p.start)
speed := float64(info.Uploaded()) / elapsed.Seconds() speed := float64(info.Uploaded()) / elapsed.Seconds()
progressText.WriteString(i18n.T(i18nk.BotMsgProgressImportSpeedPrefix, nil)) progressText.WriteString(i18n.T(i18nk.BotMsgProgressTransferSpeedPrefix, nil))
progressText.WriteString(dlutil.FormatSize(int64(speed)) + "/s") progressText.WriteString(dlutil.FormatSize(int64(speed)) + "/s")
if info.Uploaded() > 0 { if info.Uploaded() > 0 {
remaining := time.Duration(float64(info.TotalSize()-info.Uploaded()) / speed * float64(time.Second)) remaining := time.Duration(float64(info.TotalSize()-info.Uploaded()) / speed * float64(time.Second))
progressText.WriteString(i18n.T(i18nk.BotMsgProgressImportRemainingTimePrefix, nil)) progressText.WriteString(i18n.T(i18nk.BotMsgProgressTransferRemainingTimePrefix, nil))
progressText.WriteString(formatDuration(remaining)) progressText.WriteString(formatDuration(remaining))
} }
} }
processing := info.Processing() processing := info.Processing()
if len(processing) > 0 { if len(processing) > 0 {
progressText.WriteString(i18n.T(i18nk.BotMsgProgressImportProcessingPrefix, nil)) progressText.WriteString(i18n.T(i18nk.BotMsgProgressTransferProcessingPrefix, nil))
for i, elem := range processing { for i, elem := range processing {
if i >= 3 { if i >= 3 {
progressText.WriteString(i18n.T(i18nk.BotMsgProgressImportProcessingMore, map[string]any{"Count": len(processing) - 3})) progressText.WriteString(i18n.T(i18nk.BotMsgProgressTransferProcessingMore, map[string]any{"Count": len(processing) - 3}))
break break
} }
fmt.Fprintf(&progressText, "- %s\n", elem.FileName()) fmt.Fprintf(&progressText, "- %s\n", elem.FileName())
@@ -162,36 +162,36 @@ func (p *Progress) OnDone(ctx context.Context, info TaskInfo, err error) {
var resultText strings.Builder var resultText strings.Builder
if err != nil { if err != nil {
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportFailedPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferFailedPrefix, nil))
resultText.WriteString(i18n.T(i18nk.BotMsgProgressErrorPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressErrorPrefix, nil))
fmt.Fprintf(&resultText, "%v\n", err) fmt.Fprintf(&resultText, "%v\n", err)
} else { } else {
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportSuccessPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferSuccessPrefix, nil))
} }
elapsed := time.Since(p.start) elapsed := time.Since(p.start)
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportTotalFilesPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferTotalFilesPrefix, nil))
fmt.Fprintf(&resultText, "%d\n", info.Count()) fmt.Fprintf(&resultText, "%d\n", info.Count())
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportTotalSizePrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferTotalSizePrefix, nil))
fmt.Fprintf(&resultText, "%.2f MB\n", float64(info.TotalSize())/(1024*1024)) fmt.Fprintf(&resultText, "%.2f MB\n", float64(info.TotalSize())/(1024*1024))
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportUploadedPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferUploadedPrefix, nil))
fmt.Fprintf(&resultText, "%.2f MB\n", float64(info.Uploaded())/(1024*1024)) fmt.Fprintf(&resultText, "%.2f MB\n", float64(info.Uploaded())/(1024*1024))
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportElapsedTimePrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferElapsedTimePrefix, nil))
fmt.Fprintf(&resultText, "%s\n", formatDuration(elapsed)) fmt.Fprintf(&resultText, "%s\n", formatDuration(elapsed))
if elapsed.Seconds() > 0 { if elapsed.Seconds() > 0 {
avgSpeed := float64(info.Uploaded()) / elapsed.Seconds() avgSpeed := float64(info.Uploaded()) / elapsed.Seconds()
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportAvgSpeedPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferAvgSpeedPrefix, nil))
fmt.Fprintf(&resultText, "%s/s\n", dlutil.FormatSize(int64(avgSpeed))) fmt.Fprintf(&resultText, "%s/s\n", dlutil.FormatSize(int64(avgSpeed)))
} }
failedFiles := info.FailedFiles() failedFiles := info.FailedFiles()
if len(failedFiles) > 0 { if len(failedFiles) > 0 {
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportFailedFilesPrefix, nil)) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferFailedFilesPrefix, nil))
fmt.Fprintf(&resultText, "%d\n", len(failedFiles)) fmt.Fprintf(&resultText, "%d\n", len(failedFiles))
for i, name := range failedFiles { for i, name := range failedFiles {
if i >= 5 { if i >= 5 {
resultText.WriteString(i18n.T(i18nk.BotMsgProgressImportProcessingMore, map[string]any{"Count": len(failedFiles) - 5})) resultText.WriteString(i18n.T(i18nk.BotMsgProgressTransferProcessingMore, map[string]any{"Count": len(failedFiles) - 5}))
break break
} }
fmt.Fprintf(&resultText, "- %s\n", name) fmt.Fprintf(&resultText, "- %s\n", name)
+3 -3
View File
@@ -21,7 +21,7 @@ type TaskElement struct {
SourcePath string SourcePath string
FileInfo storagetypes.FileInfo FileInfo storagetypes.FileInfo
TargetStorage storage.Storage TargetStorage storage.Storage
TargetChatID int64 TargetPath string
} }
type Task struct { type Task struct {
@@ -56,7 +56,7 @@ func NewTaskElement(
sourceStorage storage.Storage, sourceStorage storage.Storage,
fileInfo storagetypes.FileInfo, fileInfo storagetypes.FileInfo,
targetStorage storage.Storage, targetStorage storage.Storage,
targetChatID int64, targetPath string,
) *TaskElement { ) *TaskElement {
id := xid.New().String() id := xid.New().String()
return &TaskElement{ return &TaskElement{
@@ -65,7 +65,7 @@ func NewTaskElement(
SourcePath: fileInfo.Path, SourcePath: fileInfo.Path,
FileInfo: fileInfo, FileInfo: fileInfo,
TargetStorage: targetStorage, TargetStorage: targetStorage,
TargetChatID: targetChatID, TargetPath: targetPath,
} }
} }