1 files changed, 19 insertions, 3 deletions

diff --git a/bot.py b/bot.py
index aea62af..1d59f9e 100644
--- a/bot.py
+++ b/bot.py
@@ -42,6 +42,7 @@ from command import (
     help_command,
     tos_command,
     toggle_visibility_command,
+    callback_toggle_visibility,
     logout_command,
     list_command
 )
@@ -55,7 +56,8 @@ from config import (
     ADD_COMMENT,
     BOT_TOKEN,
     BOT_SCOPE,
-    MAIN_MENU
+    MAIN_MENU,
+    WAIT_VISIBILITY
 )
 
 from prompt.string import PROMPT_CHOOSE_ACTION
@@ -161,13 +163,27 @@ async def main() -> None:
 
     # register handlers
     application.add_handler(CommandHandler("tos", tos_command))
-    application.add_handler(CommandHandler("vis", toggle_visibility_command))
+    visibility_conversation_handler = ConversationHandler(
+        entry_points=[
+            CommandHandler("vis", toggle_visibility_command)
+        ],
+        states={
+            WAIT_VISIBILITY: [
+                CallbackQueryHandler(callback_toggle_visibility)
+        ]},
+        fallbacks=[CommandHandler("cancel", cancel_command)],
+        per_message=False,
+        allow_reentry=True,
+    )
+
     application.add_handler(CommandHandler("logout", logout_command))
     application.add_handler(CommandHandler("list", list_command))
     application.add_handler(CommandHandler("Help", help_command))
-    application.add_handler(checkin_handler)
     application.add_handler(TypeHandler(type=FediLoginCallbackUpdate, callback=process_oauth_login_callback))
 
+    application.add_handler(visibility_conversation_handler, 2)
+    application.add_handler(checkin_handler, 1)
+
     # Pass webhook settings to telegram
     await application.bot.set_webhook(url=f"{BOT_DOMAIN}{TELEGRAM_WEBHOOK_URL}")