feat: support save content protect channel message by handle link

2026-07-09 06:21:52 +08:00 · 2025-02-16 11:38:26 +08:00
parent ec09289d5f
commit db69688722
6 changed files with 198 additions and 117 deletions
--- a/bot/handle_link.go
+++ b/bot/handle_link.go
@@ -0,0 +1,97 @@
+package bot
+
+import (
+	"regexp"
+	"strconv"
+	"strings"
+
+	"github.com/celestix/gotgproto/dispatcher"
+	"github.com/celestix/gotgproto/ext"
+	"github.com/gotd/td/tg"
+	"github.com/krau/SaveAny-Bot/dao"
+	"github.com/krau/SaveAny-Bot/logger"
+	"github.com/krau/SaveAny-Bot/types"
+)
+
+var (
+	linkRegexString = `t.me/.*/\d+`
+	linkRegex       = regexp.MustCompile(linkRegexString)
+)
+
+func handleLinkMessage(ctx *ext.Context, update *ext.Update) error {
+	logger.L.Trace("Got link message")
+	link := linkRegex.FindString(update.EffectiveMessage.Text)
+	if link == "" {
+		return dispatcher.ContinueGroups
+	}
+	strSlice := strings.Split(link, "/")
+	if len(strSlice) < 3 {
+		return dispatcher.ContinueGroups
+	}
+	messageID, err := strconv.Atoi(strSlice[2])
+	if err != nil {
+		logger.L.Errorf("Failed to parse message ID: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("Failed to parse message ID"), nil)
+		return dispatcher.EndGroups
+	}
+	chatUsername := strSlice[1]
+	linkChat, err := ctx.ResolveUsername(chatUsername)
+	if err != nil {
+		logger.L.Errorf("Failed to resolve chat ID: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("Failed to resolve chat ID"), nil)
+		return dispatcher.EndGroups
+	}
+	user, err := dao.GetUserByUserID(update.GetUserChat().GetID())
+	if err != nil {
+		logger.L.Errorf("Failed to get user: %s", err)
+		return dispatcher.EndGroups
+	}
+	replied, err := ctx.Reply(update, ext.ReplyTextString("正在获取文件..."), nil)
+	if err != nil {
+		logger.L.Errorf("Failed to reply: %s", err)
+		return dispatcher.EndGroups
+	}
+
+	file, err := FileFromMessage(ctx, linkChat.GetID(), messageID, "")
+	if err != nil {
+		logger.L.Errorf("Failed to get file from message: %s", err)
+		ctx.Reply(update, ext.ReplyTextString("获取文件失败: "+err.Error()), nil)
+		return dispatcher.EndGroups
+	}
+	if file.FileName == "" {
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "无法获取文件名",
+			ID:      replied.ID,
+		})
+		return dispatcher.EndGroups
+	}
+	receivedFile := &types.ReceivedFile{
+		Processing:     false,
+		FileName:       file.FileName,
+		ChatID:         linkChat.GetID(),
+		MessageID:      messageID,
+		ReplyMessageID: replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+	}
+	if err := dao.SaveReceivedFile(receivedFile); err != nil {
+		logger.L.Errorf("Failed to save received file: %s", err)
+		ctx.EditMessage(update.EffectiveChat().GetID(), &tg.MessagesEditMessageRequest{
+			Message: "无法保存文件: " + err.Error(),
+			ID:      replied.ID,
+		})
+		return dispatcher.EndGroups
+	}
+	if !user.Silent {
+		return ProvideSelectMessage(ctx, update, file, int(linkChat.GetID()), messageID, replied.ID)
+	}
+	return HandleSilentAddTask(ctx, update, user, &types.Task{
+		Ctx:            ctx,
+		Status:         types.Pending,
+		File:           file,
+		Storage:        types.StorageType(user.DefaultStorage),
+		FileChatID:     linkChat.GetID(),
+		FileMessageID:  messageID,
+		ReplyMessageID: replied.ID,
+		ReplyChatID:    update.GetUserChat().GetID(),
+	})
+}