mirror of
https://github.com/krau/SaveAny-Bot.git
synced 2026-05-11 23:39:46 +08:00
114 lines
3.2 KiB
Go
114 lines
3.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/celestix/gotgproto/dispatcher"
|
|
"github.com/celestix/gotgproto/ext"
|
|
"github.com/charmbracelet/log"
|
|
"github.com/gotd/td/telegram/message/styling"
|
|
"github.com/krau/SaveAny-Bot/core"
|
|
)
|
|
|
|
func handleTaskCmd(ctx *ext.Context, update *ext.Update) error {
|
|
logger := log.FromContext(ctx)
|
|
args := strings.Fields(update.EffectiveMessage.Text)
|
|
if len(args) == 1 {
|
|
showRunningTasks(ctx, update)
|
|
return dispatcher.EndGroups
|
|
}
|
|
|
|
switch args[1] {
|
|
case "running", "run", "r":
|
|
showRunningTasks(ctx, update)
|
|
case "queued", "queue", "q", "waiting":
|
|
showQueuedTasks(ctx, update)
|
|
case "cancel", "c":
|
|
if len(args) < 3 {
|
|
ctx.Reply(update, ext.ReplyTextString("用法: /tasks cancel <task_id>"), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
taskID := args[2]
|
|
if err := core.CancelTask(ctx, taskID); err != nil {
|
|
logger.Errorf("取消任务 %s 失败: %v", taskID, err)
|
|
ctx.Reply(update, ext.ReplyTextString("取消任务失败: "+err.Error()), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
ctx.Reply(update, ext.ReplyTextStyledTextArray([]styling.StyledTextOption{
|
|
styling.Plain("已请求取消任务: "),
|
|
styling.Code(taskID),
|
|
}), nil)
|
|
default:
|
|
ctx.Reply(update, ext.ReplyTextString("用法: /tasks [running|queued|cancel <task_id>]"), nil)
|
|
}
|
|
return dispatcher.EndGroups
|
|
}
|
|
|
|
func showRunningTasks(ctx *ext.Context, update *ext.Update) {
|
|
tasks := core.GetRunningTasks(ctx)
|
|
if len(tasks) == 0 {
|
|
ctx.Reply(update, ext.ReplyTextString("当前没有正在运行的任务"), nil)
|
|
return
|
|
}
|
|
opts := make([]styling.StyledTextOption, 0, 2+len(tasks)*4)
|
|
opts = append(opts,
|
|
styling.Bold("当前正在运行的任务:"),
|
|
styling.Plain(fmt.Sprintf("\n总数: %d\n", len(tasks))),
|
|
)
|
|
for _, t := range tasks {
|
|
created := t.Created.In(time.Local).Format("2006-01-02 15:04:05")
|
|
status := "运行中"
|
|
if t.Cancelled {
|
|
status = "已请求取消"
|
|
}
|
|
opts = append(opts,
|
|
styling.Plain("\nID: "),
|
|
styling.Code(t.ID),
|
|
styling.Plain("\n名称: "),
|
|
styling.Code(t.Title),
|
|
styling.Plain("\n创建时间: "),
|
|
styling.Code(created),
|
|
styling.Plain("\n状态: "),
|
|
styling.Code(status),
|
|
)
|
|
}
|
|
ctx.Reply(update, ext.ReplyTextStyledTextArray(opts), nil)
|
|
}
|
|
|
|
func showQueuedTasks(ctx *ext.Context, update *ext.Update) {
|
|
tasks := core.GetQueuedTasks(ctx)
|
|
if len(tasks) == 0 {
|
|
ctx.Reply(update, ext.ReplyTextString("当前没有排队中的任务"), nil)
|
|
return
|
|
}
|
|
opts := make([]styling.StyledTextOption, 0, 2+len(tasks)*3)
|
|
opts = append(opts,
|
|
styling.Bold("当前排队中的任务:"),
|
|
styling.Plain(fmt.Sprintf("\n总数: %d\n", len(tasks))),
|
|
)
|
|
for _, t := range tasks {
|
|
created := t.Created.In(time.Local).Format("2006-01-02 15:04:05")
|
|
status := "排队中"
|
|
if t.Cancelled {
|
|
status = "已请求取消"
|
|
}
|
|
opts = append(opts,
|
|
styling.Plain("\nID: "),
|
|
styling.Code(t.ID),
|
|
styling.Plain("\n名称: "),
|
|
styling.Code(t.Title),
|
|
styling.Plain("\n创建时间: "),
|
|
styling.Code(created),
|
|
styling.Plain("\n状态: "),
|
|
styling.Code(status),
|
|
)
|
|
if len(tasks) > 10 {
|
|
opts = append(opts, styling.Plain("\n...\n只显示前 10 个任务, 共 "+fmt.Sprintf("%d", len(tasks))+" 个任务"))
|
|
break
|
|
}
|
|
}
|
|
ctx.Reply(update, ext.ReplyTextStyledTextArray(opts), nil)
|
|
}
|