#!/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()