Init

main.py

@@ -0,0 +1,131 @@
+#!/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
+
+TOKEN = "7396669954:AAH8_I0Y-qg3j_LfbUdRTOLPDKh80NdijMo"
+
+# --- Menu Definitions ---
+def main_menu_keyboard():
+    return InlineKeyboardMarkup([
+        [InlineKeyboardButton("Torrents", callback_data='status_downloading')],
+        [InlineKeyboardButton("Status", callback_data='menu_status')]
+    ])
+
+def torrents_menu_keyboard():
+    return InlineKeyboardMarkup([
+        # First row: Display the status counts for downloading, paused, seeding
+        [InlineKeyboardButton(f"Downloading", callback_data='status_downloading')],
+        [InlineKeyboardButton(f"Active", callback_data='status_active')],
+        [InlineKeyboardButton(f"All", callback_data='status_all')],
+        # Second row: Back button
+        [InlineKeyboardButton("🔙 Back", callback_data='menu_main')]
+    ])
+
+def status_menu_keyboard():
+    return InlineKeyboardMarkup([
+        [InlineKeyboardButton("🔙 Back", callback_data='menu_main')]
+    ])
+
+# --- Command Handlers ---
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    await update.message.reply_text("Choose an option:", reply_markup=main_menu_keyboard())
+
+
+def format_torrents(torrents):
+    if len(torrents) == 0:
+        return "No torrents."
+        
+    text = ""
+
+    i = 0
+    for torrent in torrents:
+        if i > 10:
+            text += "...\n"
+            return text
+            
+        text += f"- {torrent['name']} - {torrent['progress']} ({torrent['eta']})\n"
+        i += 1
+
+    return text
+
+# --- Callback Query Handler ---
+async def handle_menu(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    query = update.callback_query
+    await query.answer()
+
+    match query.data:
+        case 'menu_main':
+            await query.edit_message_text("Choose an option:", reply_markup=main_menu_keyboard())
+
+        case 'status_downloading':
+            torrent_api = context.bot_data.get("torrent_api", {})
+            torrents = torrent_api.get_filtered_torrents("downloading")
+
+            text = format_torrents(torrents)
+            
+            await query.edit_message_text(text, reply_markup=torrents_menu_keyboard())
+
+        case 'status_active':
+            torrent_api = context.bot_data.get("torrent_api", {})
+            torrents = torrent_api.get_filtered_torrents("active")
+
+            text = format_torrents(torrents)
+
+            await query.edit_message_text(text, reply_markup=torrents_menu_keyboard())
+
+
+        case 'status_all':
+            torrent_api = context.bot_data.get("torrent_api", {})
+            torrents = torrent_api.get_filtered_torrents("all")
+
+            text = format_torrents(torrents)
+
+            await query.edit_message_text(text, reply_markup=torrents_menu_keyboard())
+            
+        case 'menu_status':
+            kuma_api = context.bot_data.get("kuma_api", {})
+            monitors = kuma_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())
+
+# --- Main Function ---
+def main():
+    # Initiate api
+    kuma_api = kuma.KumaAPI("http://192.168.1.2:36667", "k!PTfyvoIJho9o*gX6F1")
+    torrent_api = torrent.TorrentApi("http://192.168.1.17:8112", "tMHNjrJr7nhjyhJrYsahi4anq2h6LJ")
+
+    app = Application.builder().token(TOKEN).build()
+    app.bot_data["kuma_api"] = kuma_api
+    app.bot_data["torrent_api"] = torrent_api
+
+    app.add_handler(CommandHandler("start", start))
+    app.add_handler(CallbackQueryHandler(handle_menu))
+
+    print("Bot is running... Press Ctrl+C to stop.")
+    app.run_polling()
+
+if __name__ == "__main__":
+    main()