102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package bot
|
|
|
|
import (
|
|
"fmt"
|
|
"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
|
|
}
|
|
if linkChat == nil {
|
|
logger.L.Errorf("Cannot find chat: %s", chatUsername)
|
|
ctx.Reply(update, ext.ReplyTextString("Cannot find chat"), 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 == "" {
|
|
logger.L.Warnf("Empty file name, use generated name")
|
|
file.FileName = fmt.Sprintf("%d_%d_%s", linkChat.GetID(), messageID, file.Hash())
|
|
}
|
|
|
|
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(),
|
|
})
|
|
}
|