From 5fb5f33ff477a255830a36f7394fbbec7cbf933d Mon Sep 17 00:00:00 2001
From: deexno <50229919+deexno@users.noreply.github.com>
Date: Thu, 22 Jun 2023 10:17:13 +0200
Subject: [PATCH] Bug Fix

Bug fix regarding: https://github.com/deexno/checkmk-telegram-plus/issues/3
---
 resources/telegram_bot.py | 149 +++++++++++++++++++++++++++++---------
 1 file changed, 113 insertions(+), 36 deletions(-)

diff --git a/resources/telegram_bot.py b/resources/telegram_bot.py
index 70e935a..724a89f 100644
--- a/resources/telegram_bot.py
+++ b/resources/telegram_bot.py
@@ -1454,7 +1454,6 @@ def main() -> None:
     bot_handler.add_handler(CommandHandler("start", start))
     bot_handler.add_handler(CommandHandler("menu", start))
     bot_handler.add_handler(CommandHandler("help", help_command))
-    bot_handler.add_handler(CommandHandler("cancel", cancel))
 
     # Add conversation handlers for various commands
     # "⭕ GET HOST STATUS" command
@@ -1466,10 +1465,18 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_host_name)],
-                HOSTNAME: [MessageHandler(filters.TEXT, print_host_status)],
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_host_name
+                    )
+                ],
+                HOSTNAME: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), print_host_status
+                    )
+                ],
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1482,10 +1489,18 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_host_name)],
-                HOSTNAME: [MessageHandler(filters.TEXT, get_services)],
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_host_name
+                    )
+                ],
+                HOSTNAME: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_services
+                    )
+                ],
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1498,11 +1513,24 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_host_name)],
-                HOSTNAME: [MessageHandler(filters.TEXT, get_service_name)],
-                SERVICE: [MessageHandler(filters.TEXT, print_service_details)],
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_host_name
+                    )
+                ],
+                HOSTNAME: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_service_name
+                    )
+                ],
+                SERVICE: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND),
+                        print_service_details,
+                    )
+                ],
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1516,9 +1544,13 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_host_problems)]
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_host_problems
+                    )
+                ]
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1532,9 +1564,13 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_service_problems)]
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_service_problems
+                    )
+                ]
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1542,8 +1578,14 @@ def main() -> None:
     bot_handler.add_handler(
         ConversationHandler(
             entry_points=[CommandHandler("authenticate", get_pw_for_auth)],
-            states={PW: [MessageHandler(filters.TEXT, try_to_authenticate)]},
-            fallbacks=[],
+            states={
+                PW: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), try_to_authenticate
+                    )
+                ]
+            },
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1558,10 +1600,13 @@ def main() -> None:
             ],
             states={
                 NOTIFICATION_SETTING: [
-                    MessageHandler(filters.TEXT, change_notifications_setting)
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND),
+                        change_notifications_setting,
+                    )
                 ]
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1574,11 +1619,23 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_host_name)],
-                HOSTNAME: [MessageHandler(filters.TEXT, get_service_name)],
-                SERVICE: [MessageHandler(filters.TEXT, print_service_graphs)],
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_host_name
+                    )
+                ],
+                HOSTNAME: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_service_name
+                    )
+                ],
+                SERVICE: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), print_service_graphs
+                    )
+                ],
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1591,10 +1648,18 @@ def main() -> None:
                 )
             ],
             states={
-                HOSTGROUP: [MessageHandler(filters.TEXT, get_host_name)],
-                HOSTNAME: [MessageHandler(filters.TEXT, reschedule_check)],
+                HOSTGROUP: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), get_host_name
+                    )
+                ],
+                HOSTNAME: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), reschedule_check
+                    )
+                ],
             },
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1607,7 +1672,7 @@ def main() -> None:
                 )
             ],
             states={},
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1618,7 +1683,7 @@ def main() -> None:
                 MessageHandler(filters.Regex("^(📖 GET LOGS)$"), get_logs)
             ],
             states={},
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1631,7 +1696,7 @@ def main() -> None:
                 )
             ],
             states={},
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1641,8 +1706,14 @@ def main() -> None:
             entry_points=[
                 MessageHandler(filters.Regex("^(🔒 CHANGE PASSWORD)$"), get_pw)
             ],
-            states={PW: [MessageHandler(filters.TEXT, change_password)]},
-            fallbacks=[],
+            states={
+                PW: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), change_password
+                    )
+                ]
+            },
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1653,7 +1724,7 @@ def main() -> None:
                 MessageHandler(filters.Regex("^(👥 LIST USERS)$"), list_users)
             ],
             states={},
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1663,8 +1734,14 @@ def main() -> None:
             entry_points=[
                 MessageHandler(filters.Regex("^(🗑️ DELETE USERS)$"), get_user)
             ],
-            states={OPTION: [MessageHandler(filters.TEXT, delete_user)]},
-            fallbacks=[],
+            states={
+                OPTION: [
+                    MessageHandler(
+                        filters.TEXT & (~filters.COMMAND), delete_user
+                    )
+                ]
+            },
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1678,7 +1755,7 @@ def main() -> None:
                 )
             ],
             states={},
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )
 
@@ -1691,7 +1768,7 @@ def main() -> None:
                 )
             ],
             states={},
-            fallbacks=[],
+            fallbacks=[CommandHandler("cancel", cancel)],
         )
     )