Solo_bot/handlers/admin/servers/servers_handler.py

from typing import Any

from aiogram import F, Router, types
from aiogram.fsm.context import FSMContext
from aiogram.types import CallbackQuery, InlineKeyboardButton
from aiogram.utils.keyboard import InlineKeyboardBuilder

from database import get_servers
from filters.admin import IsAdminFilter
from handlers.buttons import BACK

from ..panel.keyboard import build_admin_back_kb
from .keyboard import (
    AdminServerCallback,
    build_manage_server_kb,
)


router = Router()


@router.callback_query(AdminServerCallback.filter(F.action == "manage"), IsAdminFilter())
async def handle_server_manage(callback_query: CallbackQuery, callback_data: AdminServerCallback):
    server_name = callback_data.data
    servers = await get_servers(include_enabled=True)

    cluster_name, server = next(
        ((c, s) for c, cs in servers.items() for s in cs if s["server_name"] == server_name), (None, None)
    )

    if server:
        api_url = server["api_url"]
        subscription_url = server["subscription_url"]
        inbound_id = server["inbound_id"]

        text = (
            f"<b>🔧 Информация о сервере {server_name}:</b>\n\n"
            f"<b>📡 API URL:</b> {api_url}\n"
            f"<b>🌐 Subscription URL:</b> {subscription_url}\n"
            f"<b>🔑 Inbound ID:</b> {inbound_id}"
        )

        await callback_query.message.edit_text(
            text=text,
            reply_markup=build_manage_server_kb(server_name, cluster_name, enabled=server.get("enabled", True)),
        )
    else:
        await callback_query.message.edit_text(text="❌ Сервер не найден.")


@router.callback_query(AdminServerCallback.filter(F.action == "delete"), IsAdminFilter())
async def process_callback_delete_server(
    callback_query: CallbackQuery, callback_data: AdminServerCallback, state: FSMContext, session: Any
):
    from ..clusters.clusters_handler import AdminClusterStates

    server_name = callback_data.data

    servers = await get_servers(session)
    cluster_name = None
    for c_name, server_list in servers.items():
        for server in server_list:
            if server["server_name"] == server_name:
                cluster_name = c_name
                break
        if cluster_name:
            break

    if not cluster_name:
        await callback_query.message.edit_text(
            text=f"❌ Не удалось найти кластер для сервера '{server_name}'.",
            reply_markup=build_admin_back_kb("clusters"),
        )
        return

    keys_count = await session.fetchval("SELECT COUNT(*) FROM keys WHERE server_id = $1", server_name)

    if keys_count > 0:
        await state.update_data(server_name=server_name, cluster_name=cluster_name)

        all_servers = await session.fetch(
            """
            SELECT server_name, (SELECT COUNT(*) FROM keys WHERE server_id = servers.server_name) as key_count
            FROM servers
            WHERE server_name != $1
            """,
            server_name,
        )

        if all_servers:
            builder = InlineKeyboardBuilder()
            for server in all_servers:
                builder.row(
                    InlineKeyboardButton(
                        text=f"{server['server_name']} ({server['key_count']})",
                        callback_data=f"transfer_to_server|{server['server_name']}|{server_name}",
                    )
                )
            builder.row(
                InlineKeyboardButton(
                    text=BACK, callback_data=AdminServerCallback(action="manage", data=server_name).pack()
                )
            )

            await callback_query.message.edit_text(
                text=f"⚠️ На сервере '{server_name}' есть {keys_count} ключей. Выберите сервер для переноса ключей:",
                reply_markup=builder.as_markup(),
            )
            await state.set_state(AdminClusterStates.waiting_for_server_transfer)
            return

    remaining_servers = await session.fetchval(
        "SELECT COUNT(*) FROM servers WHERE cluster_name = $1 AND server_name != $2", cluster_name, server_name
    )

    if remaining_servers == 0:
        other_clusters = await session.fetch(
            "SELECT DISTINCT cluster_name FROM servers WHERE cluster_name != $1", cluster_name
        )

        if other_clusters:
            cluster_keys_count = await session.fetchval("SELECT COUNT(*) FROM keys WHERE server_id = $1", cluster_name)

            if cluster_keys_count > 0:
                await state.update_data(server_name=server_name, cluster_name=cluster_name)

                all_clusters = await session.fetch(
                    """
                    SELECT cluster_name, (SELECT COUNT(*) FROM keys WHERE server_id = servers.cluster_name) as key_count
                    FROM servers
                    WHERE cluster_name != $1
                    GROUP BY cluster_name
                    """,
                    cluster_name,
                )

                builder = InlineKeyboardBuilder()
                for cluster in all_clusters:
                    builder.row(
                        InlineKeyboardButton(
                            text=f"{cluster['cluster_name']} ({cluster['key_count']})",
                            callback_data=f"transfer_to_cluster|{cluster['cluster_name']}|{cluster_name}|{server_name}",
                        )
                    )
                builder.row(
                    InlineKeyboardButton(
                        text=BACK, callback_data=AdminServerCallback(action="manage", data=server_name).pack()
                    )
                )

                await callback_query.message.edit_text(
                    text=f"⚠️ Это последний сервер в кластере '{cluster_name}'. На кластере есть {cluster_keys_count} ключей. Выберите кластер для переноса ключей:",
                    reply_markup=builder.as_markup(),
                )
                await state.set_state(AdminClusterStates.waiting_for_cluster_transfer)
                return

        await session.execute(
            "DELETE FROM servers WHERE cluster_name = $1 AND server_name = $2", cluster_name, server_name
        )
        await callback_query.message.edit_text(
            text=f"✅ Сервер '{server_name}' удален. Кластер '{cluster_name}' также удален, так как в нем не осталось серверов.",
            reply_markup=build_admin_back_kb("clusters"),
        )
    else:
        await session.execute(
            "DELETE FROM servers WHERE cluster_name = $1 AND server_name = $2", cluster_name, server_name
        )
        await callback_query.message.edit_text(
            text=f"✅ Сервер '{server_name}' удален.",
            reply_markup=build_admin_back_kb("clusters"),
        )


@router.callback_query(
    AdminServerCallback.filter(F.action.in_(["enable", "disable"])), IsAdminFilter()
)
async def toggle_server_enabled(callback_query: CallbackQuery, callback_data: AdminServerCallback, session: Any):
    server_name = callback_data.data
    action = callback_data.action

    new_status = action == "enable"

    await session.execute(
        "UPDATE servers SET enabled = $1 WHERE server_name = $2",
        new_status,
        server_name
    )

    servers = await get_servers(include_enabled=True)

    cluster_name, server = next(
        ((c, s) for c, cs in servers.items() for s in cs if s["server_name"] == server_name), (None, None)
    )

    if not server:
        await callback_query.message.edit_text("❌ Сервер не найден.")
        return

    text = (
        f"<b>🔧 Информация о сервере {server_name}:</b>\n\n"
        f"<b>📡 API URL:</b> {server['api_url']}\n"
        f"<b>🌐 Subscription URL:</b> {server['subscription_url']}\n"
        f"<b>🔑 Inbound ID:</b> {server['inbound_id']}"
    )

    await callback_query.message.edit_text(
        text=text,
        reply_markup=build_manage_server_kb(server_name, cluster_name, enabled=new_status)
    )