From 3b42ba86b3e55a751ec615fb0b491279c20c0f14 Mon Sep 17 00:00:00 2001
From: krau <71133316+krau@users.noreply.github.com>
Date: Sat, 4 Jan 2025 20:56:24 +0800
Subject: [PATCH] fix: generate file name from hash if empty in FileFromMedia
 function

---
 bot/utils.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/bot/utils.go b/bot/utils.go
index 31a8778..b283f1e 100644
--- a/bot/utils.go
+++ b/bot/utils.go
@@ -2,6 +2,7 @@ package bot
 
 import (
 	"context"
+	"crypto/md5"
 	"fmt"
 
 	"github.com/celestix/gotgproto"
@@ -84,10 +85,14 @@ func FileFromMedia(media tg.MessageMediaClass) (*types.File, error) {
 		var fileName string
 		for _, attribute := range document.Attributes {
 			if name, ok := attribute.(*tg.DocumentAttributeFilename); ok {
-				fileName = name.FileName
+				fileName = name.GetFileName()
 				break
 			}
 		}
+		if fileName == "" {
+			fileName = fmt.Sprintf("%x", md5.Sum(document.GetFileReference()))
+			logger.L.Warnf("File name is empty, using hash: %s", fileName)
+		}
 		return &types.File{
 			Location: document.AsInputDocumentFileLocation(),
 			FileSize: document.Size,