From 82845e7978f4ec4afe9b1d2c2abb575fbe4fbc5e Mon Sep 17 00:00:00 2001
From: stevefoxru <53931856+stevefoxru@users.noreply.github.com>
Date: Tue, 22 Apr 2025 14:16:37 +0300
Subject: [PATCH] Update add_client.py

---
 handlers/add_client.py | 43 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/handlers/add_client.py b/handlers/add_client.py
index 8b13789..d3a94ea 100644
--- a/handlers/add_client.py
+++ b/handlers/add_client.py
@@ -1 +1,44 @@
+from aiogram import types
+from aiogram.dispatcher import FSMContext
+from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
+from aiogram.dispatcher.filters.state import State, StatesGroup
+import subprocess
+from loader import dp, bot  # Убедись, что loader.py существует
 
+class AddClient(StatesGroup):
+    protocol = State()
+    username = State()
+
+@dp.message_handler(commands=["add_client"])
+async def cmd_add_client(message: types.Message):
+    keyboard = InlineKeyboardMarkup()
+    keyboard.add(
+        InlineKeyboardButton("WireGuard", callback_data="proto_wg"),
+        InlineKeyboardButton("XRay", callback_data="proto_xray")
+    )
+    await message.answer("Выберите протокол для нового клиента:", reply_markup=keyboard)
+    await AddClient.protocol.set()
+
+@dp.callback_query_handler(lambda c: c.data.startswith("proto_"), state=AddClient.protocol)
+async def process_protocol_choice(callback_query: types.CallbackQuery, state: FSMContext):
+    protocol = callback_query.data.replace("proto_", "")
+    await state.update_data(protocol=protocol)
+    await bot.send_message(callback_query.from_user.id, f"Выбран протокол: {protocol.upper()}\nВведите имя клиента:")
+    await AddClient.username.set()
+
+@dp.message_handler(state=AddClient.username)
+async def process_username(message: types.Message, state: FSMContext):
+    data = await state.get_data()
+    username = message.text.strip()
+    protocol = data.get("protocol", "wg")
+
+    try:
+        subprocess.run(
+            ["amnezia", "client", "add", "--proto", protocol, "--name", username],
+            check=True
+        )
+        await message.answer(f"✅ Клиент `{username}` с протоколом `{protocol.upper()}` успешно создан.", parse_mode="Markdown")
+    except subprocess.CalledProcessError as e:
+        await message.answer(f"❌ Ошибка при создании клиента:\n```\n{str(e)}\n```", parse_mode="Markdown")
+
+    await state.finish()