from aiogram.filters.callback_data import CallbackData
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
from aiogram.utils.keyboard import InlineKeyboardBuilder

from handlers.buttons import BACK


class AdminServerCallback(CallbackData, prefix="admin_server"):
    action: str
    data: str


def build_manage_server_kb(
    server_name: str, cluster_name: str, enabled: bool
) -> InlineKeyboardMarkup:

    builder = InlineKeyboardBuilder()

    toggle_text = "🔴 Отключить" if enabled else "🟢 Включить"
    toggle_action = "disable" if enabled else "enable"

    builder.button(
        text=toggle_text,
        callback_data=AdminServerCallback(
            action=toggle_action, data=server_name
        ).pack(),
    )

    builder.button(
        text="📈 Задать лимит",
        callback_data=AdminServerCallback(action="set_limit", data=server_name).pack(),
    )

    builder.button(
        text="🗑️ Удалить",
        callback_data=AdminServerCallback(action="delete", data=server_name).pack(),
    )

    builder.button(
        text="✏️ Редактировать",
        callback_data=f"edit_server|{server_name}",
    )

    builder.button(
        text="🔙 Назад",
        callback_data=f"cluster_servers|{cluster_name}",
    )

    builder.adjust(1)
    return builder.as_markup()


def build_edit_server_fields_kb(server_name: str, server_data: dict) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()

    builder.row(
        InlineKeyboardButton(
            text="📝 Имя сервера", 
            callback_data=f"edit_server_field|{server_name}|server_name"
        )
    )
    
    builder.row(
        InlineKeyboardButton(
            text="🗂 Кластер", 
            callback_data=f"edit_server_field|{server_name}|cluster_name"
        )
    )
    
    builder.row(
        InlineKeyboardButton(
            text="🌐 API URL", 
            callback_data=f"edit_server_field|{server_name}|api_url"
        )
    )

    if server_data.get("subscription_url"):
        builder.row(
            InlineKeyboardButton(
                text="📡 Subscription URL", 
                callback_data=f"edit_server_field|{server_name}|subscription_url"
            )
        )
    
    builder.row(
        InlineKeyboardButton(
            text="🔑 Inbound ID", 
            callback_data=f"edit_server_field|{server_name}|inbound_id"
        )
    )
    
    builder.row(
        InlineKeyboardButton(
            text="⚙️ Тип панели", 
            callback_data=f"select_panel_type|{server_name}"
        )
    )
    
    builder.row(
        InlineKeyboardButton(
            text="⬅️ Назад", 
            callback_data=AdminServerCallback(action="manage", data=server_name).pack()
        )
    )
    
    return builder.as_markup()


def build_panel_type_selection_kb(server_name: str) -> InlineKeyboardMarkup:
    return InlineKeyboardMarkup(
        inline_keyboard=[
            [InlineKeyboardButton(text="🌐 3x-ui", callback_data=f"set_panel_type|{server_name}|3x-ui")],
            [InlineKeyboardButton(text="🌀 remnawave", callback_data=f"set_panel_type|{server_name}|remnawave")],
            [InlineKeyboardButton(
                text="⬅️ Назад", 
                callback_data=f"edit_server|{server_name}"
            )]
        ]
    )


def build_cluster_selection_kb(server_name: str, clusters: list[str]) -> InlineKeyboardMarkup:
    builder = InlineKeyboardBuilder()
    
    for cluster in clusters:
        builder.row(
            InlineKeyboardButton(
                text=cluster, 
                callback_data=f"set_cluster|{server_name}|{cluster}"
            )
        )
    
    builder.row(
        InlineKeyboardButton(
            text="⬅️ Назад", 
            callback_data=f"edit_server|{server_name}"
        )
    )
    
    return builder.as_markup()


def build_cancel_edit_kb(server_name: str) -> InlineKeyboardMarkup:
    return InlineKeyboardMarkup(
        inline_keyboard=[
            [
                InlineKeyboardButton(
                    text="❌ Отменить", 
                    callback_data=f"edit_server|{server_name}"
                )
            ]
        ]
    )