fix: handle empty file name in HandleFileMessage

krau · Oct 10, 2024 · 7a288e3 · 7a288e3
1 parent f9d51a6
commit 7a288e3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bot/handlers.go b/bot/handlers.go
@@ -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,