Добавление логики переноса

- Логика переноса ключей при удалении сервера/кластера

handlers/admin/servers/servers_handler.py

@@ -1,7 +1,10 @@
 from typing import Any
 
 from aiogram import F, Router, types
-from aiogram.types import CallbackQuery
+from aiogram.fsm.context import FSMContext
+from aiogram.types import CallbackQuery, InlineKeyboardButton
+from aiogram.utils.keyboard import InlineKeyboardBuilder
+from handlers.buttons import BACK
 
 from database import delete_server, get_servers
 from filters.admin import IsAdminFilter
@@ -9,7 +12,6 @@ from filters.admin import IsAdminFilter
 from ..panel.keyboard import build_admin_back_kb
 from .keyboard import (
     AdminServerCallback,
-    build_delete_server_kb,
     build_manage_server_kb,
 )
 
@@ -47,23 +49,139 @@ async def handle_server_manage(callback_query: CallbackQuery, callback_data: Adm
 
 
 @router.callback_query(AdminServerCallback.filter(F.action == "delete"), IsAdminFilter())
-async def handle_server_delete(callback_query: CallbackQuery, callback_data: AdminServerCallback):
-    server_name = callback_data.data
-
-    await callback_query.message.edit_text(
-        text=f"🗑️ Вы уверены, что хотите удалить сервер {server_name}?",
-        reply_markup=build_delete_server_kb(server_name),
-    )
-
-
-@router.callback_query(AdminServerCallback.filter(F.action == "delete_confirm"), IsAdminFilter())
-async def handle_server_delete_confirm(
-    callback_query: types.CallbackQuery, callback_data: AdminServerCallback, session: Any
+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
 
-    await delete_server(server_name, session)
+    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
 
-    await callback_query.message.edit_text(
-        text=f"🗑️ Сервер {server_name} успешно удален.", reply_markup=build_admin_back_kb("clusters")
+    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"),
+        )