46 lines
1.5 KiB
Go
46 lines
1.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/celestix/gotgproto/dispatcher"
|
|
"github.com/celestix/gotgproto/ext"
|
|
"github.com/charmbracelet/log"
|
|
"github.com/gotd/td/tg"
|
|
"github.com/krau/SaveAny-Bot/client/bot/handlers/utils/msgelem"
|
|
"github.com/krau/SaveAny-Bot/core"
|
|
)
|
|
|
|
func handleCancelCallback(ctx *ext.Context, update *ext.Update) error {
|
|
taskid := strings.Split(string(update.CallbackQuery.Data), " ")[1]
|
|
if err := core.CancelTask(ctx, taskid); err != nil {
|
|
log.FromContext(ctx).Errorf("error cancelling task %s: %v", taskid, err)
|
|
ctx.AnswerCallback(msgelem.AlertCallbackAnswer(update.CallbackQuery.GetQueryID(), "取消任务失败: "+err.Error()))
|
|
return dispatcher.EndGroups
|
|
}
|
|
|
|
ctx.EditMessage(update.CallbackQuery.GetUserID(), &tg.MessagesEditMessageRequest{
|
|
ID: update.CallbackQuery.GetMsgID(),
|
|
Message: "正在取消任务...",
|
|
})
|
|
|
|
return dispatcher.EndGroups
|
|
}
|
|
|
|
func handleCancelCmd(ctx *ext.Context, update *ext.Update) error {
|
|
logger := log.FromContext(ctx)
|
|
args := strings.Fields(update.EffectiveMessage.Text)
|
|
if len(args) < 2 {
|
|
ctx.Reply(update, ext.ReplyTextString("用法: /cancel <task_id>"), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
taskID := args[1]
|
|
if err := core.CancelTask(ctx, taskID); err != nil {
|
|
logger.Errorf("failed to cancel task %s: %v", taskID, err)
|
|
ctx.Reply(update, ext.ReplyTextString("取消任务失败: "+err.Error()), nil)
|
|
return dispatcher.EndGroups
|
|
}
|
|
ctx.Reply(update, ext.ReplyTextString("已请求取消任务: "+taskID), nil)
|
|
return dispatcher.EndGroups
|
|
}
|