UdoChudo
c2f7d2a88e
feat: /create command now check profile existed in vless and ss inbounds and if yes just send sub link
102 lines
4.0 KiB
Python
102 lines
4.0 KiB
Python
import json
|
||
from aiogram import types
|
||
from aiogram.enums import ParseMode
|
||
from pyxui.errors import BadLogin
|
||
|
||
from bot.services.client import ClientService
|
||
from bot.utils.logging import logger
|
||
|
||
|
||
class ClientHandlers:
|
||
"""Обработчики команд для работы с клиентами."""
|
||
|
||
def __init__(self, client_service: ClientService):
|
||
self.client_service = client_service
|
||
|
||
async def cmd_create(self, message: types.Message):
|
||
"""Обработчик команды /create."""
|
||
args = (message.text or "").strip().split(maxsplit=1)
|
||
if len(args) < 2:
|
||
await message.answer(
|
||
"❌ Укажи Telegram ID или username после команды.\n"
|
||
"Пример:\n<code>/create udochudo</code>"
|
||
)
|
||
return
|
||
|
||
telegram_id = args[1].lstrip("@").strip()
|
||
|
||
try:
|
||
success, result_message = await self.client_service.create_client_profile(telegram_id)
|
||
|
||
if success:
|
||
await message.answer(result_message, parse_mode=ParseMode.HTML)
|
||
else:
|
||
await message.answer(f"❌ {result_message}")
|
||
|
||
except BadLogin:
|
||
await message.answer("❌ Ошибка: неверный логин или пароль XUI.")
|
||
except Exception as e:
|
||
logger.error(f"Неожиданная ошибка при создании профиля: {e}")
|
||
await message.answer(f"❌ Произошла ошибка при создании профиля. Попробуйте позже.\n {e}")
|
||
|
||
async def cmd_info(self, message: types.Message):
|
||
"""Обработчик команды /info с выводом конфига и статистики."""
|
||
args = (message.text or "").strip().split(maxsplit=1)
|
||
if len(args) < 2:
|
||
await message.answer(
|
||
"❌ Укажи Telegram ID или username после команды.\n"
|
||
"Пример:\n<code>/info udochudo</code>"
|
||
)
|
||
return
|
||
|
||
telegram_id = args[1].lstrip("@").strip()
|
||
|
||
try:
|
||
(vless_client, vless_stats), (ss_client, ss_stats) = \
|
||
self.client_service.get_client_info_with_stats(telegram_id)
|
||
|
||
def format_info(client, stats, name):
|
||
if not client:
|
||
return f"❌ Клиент <b>{name}</b> не найден.\n"
|
||
|
||
# Конфиг клиента
|
||
json_info = json.dumps(client, ensure_ascii=False, indent=2)
|
||
text = f"🔹 <b>{name}</b>:\n<pre>{json_info}</pre>"
|
||
|
||
# Статистика
|
||
if stats:
|
||
up = stats.get("up", 0)
|
||
down = stats.get("down", 0)
|
||
total = up + down
|
||
|
||
def human_size(num):
|
||
for unit in ["B", "KB", "MB", "GB", "TB"]:
|
||
if num < 1024:
|
||
return f"{num:.2f} {unit}"
|
||
num /= 1024
|
||
return f"{num:.2f} PB"
|
||
|
||
text += (
|
||
f"\n📊 Трафик: ↑ {human_size(up)} / ↓ {human_size(down)}"
|
||
f" (Σ {human_size(total)})"
|
||
)
|
||
else:
|
||
text += "\n📊 Статистика не найдена."
|
||
|
||
return text
|
||
|
||
response = (
|
||
format_info(vless_client, vless_stats, "VLESS")
|
||
+ "\n\n"
|
||
+ format_info(ss_client, ss_stats, "Shadowsocks")
|
||
)
|
||
|
||
await message.answer(response, parse_mode=ParseMode.HTML)
|
||
|
||
except BadLogin:
|
||
await message.answer("❌ Ошибка входа в панель XUI.")
|
||
except Exception as e:
|
||
logger.error(f"Ошибка получения информации: {e}")
|
||
await message.answer("❌ Ошибка при получении информации. Проверь лог.")
|
||
|