feat: support save content protect channel message by handle link

bot/utils.go

@@ -1,16 +1,18 @@
 package bot
 
 import (
-	"context"
 	"errors"
 	"fmt"
 	"time"
 
-	"github.com/celestix/gotgproto"
 	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/telegram/message/entity"
+	"github.com/gotd/td/telegram/message/styling"
 	"github.com/gotd/td/tg"
 	"github.com/krau/SaveAny-Bot/common"
 	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/queue"
 	"github.com/krau/SaveAny-Bot/storage"
 	"github.com/krau/SaveAny-Bot/types"
 )
@@ -44,12 +46,12 @@ var StorageDisplayNames = map[string]string{
 	"webdav": "WebDAV",
 }
 
-func getAddTaskMarkup(messageID int) *tg.ReplyInlineMarkup {
+func getAddTaskMarkup(chatID, messageID int) *tg.ReplyInlineMarkup {
 	storageButtons := make([]tg.KeyboardButtonClass, 0)
 	for _, name := range storage.StorageKeys {
 		storageButtons = append(storageButtons, &tg.KeyboardButtonCallback{
 			Text: StorageDisplayNames[string(name)],
-			Data: []byte(fmt.Sprintf("add %d %s", messageID, name)),
+			Data: []byte(fmt.Sprintf("add %d %d %s", chatID, messageID, name)),
 		})
 	}
 
@@ -74,7 +76,7 @@ func getAddTaskMarkup(messageID int) *tg.ReplyInlineMarkup {
 				Buttons: []tg.KeyboardButtonClass{
 					&tg.KeyboardButtonCallback{
 						Text: "全部",
-						Data: []byte(fmt.Sprintf("add %d all", messageID)),
+						Data: []byte(fmt.Sprintf("add %d %d all", chatID, messageID)),
 					},
 				},
 			},
@@ -144,7 +146,7 @@ func FileFromMedia(media tg.MessageMediaClass, customFileName string) (*types.Fi
 	return nil, fmt.Errorf("unexpected type %T", media)
 }
 
-func FileFromMessage(ctx context.Context, client *gotgproto.Client, chatID int64, messageID int, customFileName string) (*types.File, error) {
+func FileFromMessage(ctx *ext.Context, chatID int64, messageID int, customFileName string) (*types.File, error) {
 	key := fmt.Sprintf("file:%d:%d", chatID, messageID)
 	logger.L.Debugf("Getting file: %s", key)
 	var cachedFile types.File
@@ -152,8 +154,7 @@ func FileFromMessage(ctx context.Context, client *gotgproto.Client, chatID int64
 	if err == nil {
 		return &cachedFile, nil
 	}
-
-	message, err := GetTGMessage(ctx, client, messageID)
+	message, err := GetTGMessage(ctx, chatID, messageID)
 	if err != nil {
 		return nil, err
 	}
@@ -167,20 +168,60 @@ func FileFromMessage(ctx context.Context, client *gotgproto.Client, chatID int64
 	return file, nil
 }
 
-func GetTGMessage(ctx context.Context, client *gotgproto.Client, messageID int) (*tg.Message, error) {
+func GetTGMessage(ctx *ext.Context, chatId int64, messageID int) (*tg.Message, error) {
 	logger.L.Debugf("Fetching message: %d", messageID)
-	res, err := client.API().MessagesGetMessages(ctx, []tg.InputMessageClass{
-		&tg.InputMessageID{
-			ID: messageID,
-		},
-	})
+	messages, err := ctx.GetMessages(chatId, []tg.InputMessageClass{&tg.InputMessageID{ID: messageID}})
 	if err != nil {
 		return nil, err
 	}
-	messages := res.(*tg.MessagesMessages)
-	msg := messages.Messages[0]
-	if _, ok := msg.(*tg.Message); !ok {
-		return nil, fmt.Errorf("unexpected type %T, this file may be deleted", msg)
+	if len(messages) == 0 {
+		return nil, errors.New("no messages found")
 	}
-	return msg.(*tg.Message), nil
+	msg := messages[0]
+	tgMessage, ok := msg.(*tg.Message)
+	if !ok {
+		return nil, fmt.Errorf("unexpected message type: %T", msg)
+	}
+	return tgMessage, nil
+}
+
+func ProvideSelectMessage(ctx *ext.Context, update *ext.Update, file *types.File, chatID int, fileMsgID, toEditMsgID int) error {
+	entityBuilder := entity.Builder{}
+	var entities []tg.MessageEntityClass
+	text := fmt.Sprintf("文件名: %s\n请选择存储位置", file.FileName)
+	if err := styling.Perform(&entityBuilder,
+		styling.Plain("文件名: "),
+		styling.Code(file.FileName),
+		styling.Plain("\n请选择存储位置"),
+	); err != nil {
+		logger.L.Errorf("Failed to build entity: %s", err)
+	} else {
+		text, entities = entityBuilder.Complete()
+	}
+	_, err := ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message:     text,
+		Entities:    entities,
+		ReplyMarkup: getAddTaskMarkup(chatID, fileMsgID),
+		ID:          toEditMsgID,
+	})
+	if err != nil {
+		logger.L.Errorf("Failed to reply: %s", err)
+	}
+	return dispatcher.EndGroups
+}
+
+func HandleSilentAddTask(ctx *ext.Context, update *ext.Update, user *types.User, task *types.Task) error {
+	if user.DefaultStorage == "" {
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "请先使用 /storage 设置默认存储位置",
+			ID:      task.ReplyMessageID,
+		})
+		return dispatcher.EndGroups
+	}
+	queue.AddTask(*task)
+	ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+		Message: fmt.Sprintf("已添加到队列: %s\n当前排队任务数: %d", task.FileName(), queue.Len()),
+		ID:      task.ReplyMessageID,
+	})
+	return dispatcher.EndGroups
 }