GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-05-12 00:09:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: task progress

Browse Source
This commit is contained in:
krau
2024-10-12 13:03:11 +08:00
parent c8376f78a2
commit f58e28412f
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/core.go
View File

@@ -21,6 +21,7 @@ func processPendingTask(task types.Task) error {
return err
}
logger.L.Debugf("Start downloading file: %s", task.FileName)
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "正在下载文件...")
dest, err := message.Download(&telegram.DownloadOptions{
FileName: common.GetCacheFilePath(task.FileName),
// ProgressCallback: func(totalBytes, downloadedBytes int64) {},
@@ -42,9 +43,11 @@ func processPendingTask(task types.Task) error {
task.StoragePath = task.FileName
}
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "下载完成, 正在转存文件...")
if err := storage.Save(task.Storage, task.Ctx, dest, task.StoragePath); err != nil {
return err
}
return nil
}
@@ -71,8 +74,10 @@ func worker(queue *queue.TaskQueue, semaphore chan struct{}) {
queue.AddTask(task)
case types.Succeeded:
logger.L.Infof("Task succeeded: %s", task.String())
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "文件保存成功")
case types.Failed:
logger.L.Errorf("Task failed: %s", task.String())
bot.Client.EditMessage(task.ChatID, task.ReplyMessageID, "文件保存失败")
case types.Canceled:
logger.L.Infof("Task canceled: %s", task.String())
default: