From 1128ee20817269f8ba2f0c890cad0cc663a565dc Mon Sep 17 00:00:00 2001
From: krau <71133316+krau@users.noreply.github.com>
Date: Sat, 9 Nov 2024 11:38:29 +0800
Subject: [PATCH] fix: update reply message id in callback query

---
 bot/handlers.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/bot/handlers.go b/bot/handlers.go
index 2d0489e..245fb3c 100644
--- a/bot/handlers.go
+++ b/bot/handlers.go
@@ -339,6 +339,13 @@ func AddToQueue(ctx *ext.Context, update *ext.Update) error {
 		})
 		return dispatcher.EndGroups
 	}
+	if update.CallbackQuery.MsgID != record.ReplyMessageID {
+		record.ReplyMessageID = update.CallbackQuery.MsgID
+		if err := dao.UpdateReceivedFile(record); err != nil {
+			logger.L.Errorf("Failed to update received file: %s", err)
+		}
+	}
+
 	file, err := FileFromMessage(ctx, Client, record.ChatID, record.MessageID)
 	if err != nil {
 		logger.L.Errorf("Failed to get file from message: %s", err)