fix: mimetype nil pointer

This commit is contained in:
krau
2025-06-20 22:42:57 +08:00
parent 336309fad0
commit 3b85911e3d
2 changed files with 4 additions and 4 deletions

View File

@@ -82,7 +82,7 @@ func GetFilesFromUpdateLinkMessageWithReplyEdit(ctx *ext.Context, update *ext.Up
files = make([]tfile.TGFileMessage, 0, len(msgLinks)) files = make([]tfile.TGFileMessage, 0, len(msgLinks))
addFile := func(msg *tg.Message) { addFile := func(msg *tg.Message) {
if msg == nil { if msg == nil || msg.Media == nil {
logger.Warn("message is nil, skipping") logger.Warn("message is nil, skipping")
return return
} }

View File

@@ -27,11 +27,11 @@ func GenFileNameFromMessage(message tg.Message) string {
if !ok { if !ok {
return "" return ""
} }
ext := mimetype.Lookup(doc.MimeType).Extension() mmt := mimetype.Lookup(doc.MimeType)
if ext == "" { if mmt == nil || mmt.Extension() == "" {
return "" return ""
} }
return ext return mmt.Extension()
case *tg.MessageMediaPhoto: case *tg.MessageMediaPhoto:
return ".jpg" return ".jpg"
} }