Add portainer menu

menus/portainer.py

@@ -0,0 +1,138 @@
+#!/usr/bin/env python3
+
+import requests
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
+from telegram.ext import (
+    ApplicationBuilder,
+    CommandHandler,
+    CallbackQueryHandler,
+    ContextTypes,
+)
+
+from config import BOT_TOKEN, PORTAINER_API_KEY
+
+PORTAINER_URL = "https://192.168.1.17:9443/api"
+STACKS_PER_PAGE = 5
+user_page = 0  # Tracks current page per user
+
+
+def get_stacks():
+    headers = {"X-API-Key": f"{PORTAINER_API_KEY}"}
+    response = requests.get(
+        f"{PORTAINER_URL}/stacks?filters=" + '{"EndpointID":2}',
+        headers=headers,
+        verify=False,
+    )
+
+    return response.json() if response.ok else []
+
+
+def get_stack_status(stack_id):
+    """Fetch the stack status based on its ID."""
+    headers = {"X-API-Key": f"{PORTAINER_API_KEY}"}
+    response = requests.get(
+        f"{PORTAINER_URL}/stacks/{stack_id}?endpointId=2", headers=headers, verify=False
+    )
+    if response.ok:
+        stack_info = response.json()
+        return stack_info.get(
+            "Status", ""
+        )  # Assuming the status is returned in lowercase
+    return None
+
+
+async def start_portainer(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    global user_page
+
+    user_page = 0
+    await send_stack_list(update, context)
+
+
+def build_stack_markup(stacks, page):
+    start_idx = page * STACKS_PER_PAGE
+    end_idx = start_idx + STACKS_PER_PAGE
+    sliced = stacks[start_idx:end_idx]
+
+    keyboard = []
+    for stack in sliced:
+        stack_name = stack["Name"]
+        stack_id = stack["Id"]
+        status = get_stack_status(stack_id)
+
+        # Add smiley based on stack status
+        if status == 1:
+            smiley = "🟢"  # Stack is up
+        else:
+            smiley = "🔴"  # Stack is down
+
+        # Button labels: Stack name with status smiley
+        row = [
+            InlineKeyboardButton(
+                f"{smiley} {stack_name}",
+                callback_data=f"portainer_select_{stack_id}",
+            ),
+            InlineKeyboardButton(
+                f"▶️ Start", callback_data=f"portainer_start_{stack_id}"
+            ),
+            InlineKeyboardButton(f"⏹ Stop", callback_data=f"portainer_stop_{stack_id}"),
+        ]
+        keyboard.append(row)
+
+    nav_row = []
+    if page > 0:
+        nav_row.append(InlineKeyboardButton("⬅️ Prev", callback_data="portainer_prev"))
+    if end_idx < len(stacks):
+        nav_row.append(InlineKeyboardButton("➡️ Next", callback_data="portainer_next"))
+    keyboard.append(nav_row)
+
+    keyboard.append([InlineKeyboardButton("🔙 Main Menu", callback_data="menu_main")])
+    return InlineKeyboardMarkup(keyboard)
+
+
+async def send_stack_list(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    global user_page
+
+    page = user_page
+    stacks = get_stacks()
+
+    if not stacks:
+        await update.message.reply_text(
+            "No stacks found or unable to connect to Portainer."
+        )
+        return
+
+    if update.callback_query:
+        await update.callback_query.answer()
+        await update.callback_query.edit_message_text(
+            text=f"Select a stack (page {page + 1}):",
+            reply_markup=build_stack_markup(stacks, page),
+        )
+    else:
+        await update.message.reply_text(
+            text=f"Select a stack (page {page + 1}):",
+            reply_markup=build_stack_markup(stacks, page),
+        )
+
+
+async def button_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    global user_page
+
+    query = update.callback_query
+    data = query.data
+
+    if data == "portainer_next":
+        user_page += 1
+    elif data == "portainer_prev":
+        user_page = max(0, user_page - 1)
+    elif data.startswith("portainer_start_") or data.startswith("portainer_stop_"):
+        _, action, stack_id = data.split("_")
+        await handle_stack_action(query, action, stack_id)
+    await send_stack_list(update, context)
+
+
+async def handle_stack_action(query, action, stack_id):
+    endpoint = f"{PORTAINER_URL}/stacks/{stack_id}/{action}?endpointId=2"
+    headers = {"X-API-Key": f"{PORTAINER_API_KEY}"}
+    response = requests.post(endpoint, headers=headers, verify=False)
+    msg = "✅ Success!" if response.ok else f"❌ Failed: {response.text}"
+    await query.answer(text=msg, show_alert=True)