Add portainer menu

menus/menus.py

@@ -0,0 +1,156 @@
+#!/usr/bin/env python3
+
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
+from telegram.ext import Application, CommandHandler, CallbackQueryHandler, ContextTypes
+
+from uptime_kuma_api import MonitorStatus
+import api.kuma as kuma
+import api.torrent as torrent
+import api.ntfy as ntfy
+
+from menus.portainer import button_handler
+
+
+# --- Menu Definitions ---
+def main_menu_keyboard():
+    return InlineKeyboardMarkup(
+        [
+            [InlineKeyboardButton("Torrents", callback_data="torrent_downloading")],
+            [InlineKeyboardButton("Status", callback_data="menu_status")],
+            [InlineKeyboardButton("Portainer", callback_data="portainer_menu")],
+        ]
+    )
+
+
+def torrents_menu_keyboard():
+    print("Creating meny keyboard")
+
+    return InlineKeyboardMarkup(
+        [
+            # First row: Display the status counts for downloading, paused, seeding
+            [InlineKeyboardButton(f"Downloading", callback_data="torrent_downloading")],
+            [InlineKeyboardButton(f"Active", callback_data="torrent_active")],
+            [InlineKeyboardButton(f"All", callback_data="torrent_all")],
+            # Second row: Back button
+            [InlineKeyboardButton("🔙 Back", callback_data="menu_main")],
+        ]
+    )
+
+
+def status_menu_keyboard():
+    return InlineKeyboardMarkup(
+        [[InlineKeyboardButton("🔙 Back", callback_data="menu_main")]]
+    )
+
+
+def format_torrents(torrents):
+    if len(torrents) == 0:
+        return "No torrents."
+
+    text = ""
+
+    i = 0
+    for torrent in torrents:
+        if i > 5:
+            text += "...\n"
+            return text
+
+        text += f"Name: {torrent['name']}\n"
+        text += f"State: {torrent['state']}\n"
+        text += f"Progress: {torrent['progress']:.2f}%\n"
+        text += f"ETA: {torrent['eta']}\n"
+        text += "-" * 20 + "\n"
+
+        text += f"- {torrent['name']} - {torrent['progress']} ({torrent['eta']})\n"
+        i += 1
+
+    return text
+
+
+# ===================================
+# --- MAIN HANDLER FUNCTION ---
+# ==================================
+
+
+# --- Callback Query Handler ---
+async def handle_menu(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    query = update.callback_query
+    await query.answer()
+
+    if query.data.startswith("portainer_"):
+        await button_handler(update, context)
+
+    else:
+        print(query.data)
+
+        match query.data:
+            case "menu_main":
+                await query.edit_message_text(
+                    "Choose an option:", reply_markup=main_menu_keyboard()
+                )
+
+            case "torrent_downloading":
+
+                t_api = context.bot_data.get("torrent_api", {})
+                torrents = t_api.get_filtered_torrents("downloading")
+
+                if len(torrents) == 0:
+                    text = "No downloading torrents."
+                else:
+                    text = format_torrents(torrents)
+
+                await query.edit_message_text(
+                    text, reply_markup=torrents_menu_keyboard()
+                )
+
+            case "torrent_active":
+                t_api = context.bot_data.get("torrent_api", {})
+                torrents = t_api.get_filtered_torrents("active")
+
+                if len(torrents) == 0:
+                    text = "No active torrents."
+                else:
+                    text = format_torrents(torrents)
+
+                await query.edit_message_text(
+                    text, reply_markup=torrents_menu_keyboard()
+                )
+
+            case "torrent_all":
+                t_api = context.bot_data.get("torrent_api", {})
+                torrents = t_api.get_filtered_torrents("all")
+
+                if len(torrents) == 0:
+                    text = "No torrents."
+                else:
+                    text = format_torrents(torrents)
+
+                await query.edit_message_text(
+                    text, reply_markup=torrents_menu_keyboard()
+                )
+
+            case "menu_status":
+                k_api = context.bot_data.get("kuma_api", {})
+                monitors = k_api.get_status()
+
+                up_text, down_text, paused_text = "", "", ""
+                for _, monitor in monitors.items():
+                    status = monitor["status"]
+
+                    if status == MonitorStatus.UP:
+                        up_text += f"    - {monitor['name']}\n"
+                    elif status == MonitorStatus.DOWN:
+                        down_text += f"    - {monitor['name']}\n"
+                    else:
+                        paused_text += f"    - {monitor['name']}\n"
+
+                status_text = f"📡 *Status:*\n\n 🟢 Up:\n{up_text}\n🔴 Down\n{down_text}\n⏸️ Paused\n{paused_text}"
+
+                await query.edit_message_text(
+                    status_text, reply_markup=status_menu_keyboard()
+                )
+
+            case _:
+                await query.edit_message_text(
+                    "Unknown option selected.", reply_markup=main_menu_keyboard()
+                )