From 770cd0ec5eb556d912bd9d200b2da76c7e3bd7c8 Mon Sep 17 00:00:00 2001
From: clarkzjw <i@jinwei.me>
Date: Thu, 23 Feb 2023 00:13:47 -0800
Subject: bot: fix login flow

---
 callback.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'callback.py')

diff --git a/callback.py b/callback.py
index 23b814b..d968523 100644
--- a/callback.py
+++ b/callback.py
@@ -97,10 +97,11 @@ async def callback_generate_fedi_login_url(update: Update, context: ContextTypes
                                                  scopes=BOT_SCOPE,
                                                  state=state)
 
-    await update.message.reply_text(PROMPT_FEDI_LOGIN,
+    msg = await update.message.reply_text(PROMPT_FEDI_LOGIN,
                                     reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("Login", url=oauth_url)]]),
                                     parse_mode=ParseMode.MARKDOWN)
 
+    context.user_data[PROMPT_FEDI_LOGIN] = msg.message_id
     return FEDI_LOGIN
 
 
-- 
cgit v1.2.3