fix: handle empty file name in HandleFileMessage

This commit is contained in:
krau
2024-10-11 01:14:00 +08:00
parent f9d51a688c
commit 7a288e3155

View File

@@ -3,6 +3,7 @@ package bot
import (
"context"
"fmt"
"path/filepath"
"strconv"
"strings"
@@ -151,7 +152,8 @@ func HandleFileMessage(ctx context.Context, bot *telego.Bot, message telego.Mess
fileName = message.Audio.FileName
}
if fileID == "" || fileName == "" {
if fileID == "" {
logger.L.Error("File ID is empty")
ReplyMessage(message, "文件信息获取失败")
return
}
@@ -182,6 +184,9 @@ func HandleFileMessage(ctx context.Context, bot *telego.Bot, message telego.Mess
return
}
}
if fileName == "" {
fileName = filepath.Base(file.FilePath)
}
err = dao.AddReceivedFile(&model.ReceivedFile{
FileUniqueID: file.FileUniqueID,