From c98c6af14cb3888fe18e664803c179db8510776c Mon Sep 17 00:00:00 2001 From: Vlad Date: Tue, 17 Sep 2024 00:39:06 +0300 Subject: [PATCH] =?UTF-8?q?=D1=80=D0=B0=D0=B1=D0=BE=D1=87=D0=B8=D0=B9=20?= =?UTF-8?q?=D0=B2=D0=B0=D1=80=D0=B8=D0=B0=D0=BD=D1=82=20=D0=B2=D1=8B=D0=B4?= =?UTF-8?q?=D0=B0=D1=87=D0=B8=20=D0=BA=D0=BB=D1=8E=D1=87=D0=B5=D0=B9=20?= =?UTF-8?q?=D1=87=D0=B5=D1=80=D0=B5=D0=B7=20=D0=BF=D0=BE=D0=B4=D1=82=D0=B2?= =?UTF-8?q?=D0=B5=D1=80=D0=B6=D0=B4=D0=B5=D0=BD=D0=B8=D0=B5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- auth.py | 4 +- bot.py | 210 ++++++++++++++++++++++++------------------------------ client.py | 69 +++++++++++------- 3 files changed, 138 insertions(+), 145 deletions(-) diff --git a/auth.py b/auth.py index 48ea4624..c1d0880e 100644 --- a/auth.py +++ b/auth.py @@ -26,7 +26,7 @@ def get_clients(session): else: raise Exception(f"Ошибка при получении клиентов: {response.status_code}, {response.text}") -def link(session, client_id: str): +def link(session, client_id: str, email: str): """ Получение ссылки для подключения по ID клиента. :param session: requests.Session - авторизованная сессия @@ -48,6 +48,6 @@ def link(session, client_id: str): flow = stream_settings.get('flow', 'xtls-rprx-vision') # Создание ссылки для подключения VLESS - val = f"vless://{client_id}@solonet.pocomacho.ru:443?type={tcp}&security={reality}&fp=chrome&pbk=Ik3cfHdM5RXYCuzTxhaFfo58Fzwjvw11AtKsz5-izxA&fp=chrome&sni=discord.com&sid=6fc9&spx=%2F&flow={flow}#VPN_F-{client_id}" + val = f"vless://{client_id}@solonet.pocomacho.ru:443?type={tcp}&security={reality}&pbk=Ik3cfHdM5RXYCuzTxhaFfo58Fzwjvw11AtKsz5-izxA&fp=chrome&sni=discord.com&sid=6fc9&spx=%2F&flow={flow}#SoloNet_DE-{email}" return val diff --git a/bot.py b/bot.py index e0037853..7524c868 100644 --- a/bot.py +++ b/bot.py @@ -1,21 +1,23 @@ from aiogram import Bot, Dispatcher, Router, F, types from aiogram.filters import Command -from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message +from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup, CallbackQuery, Message from aiogram.fsm.storage.memory import MemoryStorage from aiogram.fsm.context import FSMContext from aiogram.fsm.state import State, StatesGroup -from auth import login_with_credentials, link, get_clients +from auth import login_with_credentials, link from client import add_client from datetime import datetime, timedelta from config import API_TOKEN, ADMIN_PASSWORD, ADMIN_USERNAME, ADMIN_CHAT_ID -from database import add_connection, has_active_key, get_active_key_email, get_key_expiry_time +from database import add_connection, has_active_key, get_active_key_email, DATABASE_PATH import uuid import aiosqlite +import re class Form(StatesGroup): waiting_for_key_name = State() waiting_for_admin_confirmation = State() - key_to_create = State() # Состояние для хранения ключа, который будет создан + waiting_for_statistics = State() + waiting_for_expiry_date = State() bot = Bot(token=API_TOKEN) storage = MemoryStorage() @@ -36,27 +38,10 @@ async def start_command(message: Message): await message.reply(welcome_text, reply_markup=keyboard) -@router.callback_query(F.data == "view_expiry") -async def process_callback_view_expiry_date(callback_query: types.CallbackQuery): - tg_id = callback_query.from_user.id - - # Получаем дату окончания ключа из базы данных - expiry_time = await get_key_expiry_time(tg_id) - - if expiry_time: - expiry_date = expiry_time.strftime('%Y-%m-%d %H:%M:%S') - response_message = f"Дата окончания вашего ключа: {expiry_date}" - else: - response_message = "У вас нет активного ключа." - - await callback_query.message.reply(response_message) - await callback_query.answer() - @dp.callback_query(F.data == 'create_key') async def process_callback_create_key(callback_query: types.CallbackQuery, state: FSMContext): await callback_query.message.reply("Введите имя вашего профиля VPN:") await state.set_state(Form.waiting_for_key_name) - await state.update_data(tg_id=callback_query.from_user.id) # Сохраняем tg_id для дальнейшего использования await callback_query.answer() @dp.message() @@ -65,61 +50,47 @@ async def handle_text(message: types.Message, state: FSMContext): if current_state == Form.waiting_for_key_name.state: tg_id = message.from_user.id - + key_name = message.text + + if not key_name: + await message.reply("Имя клиента не указано.") + await state.clear() + return + if await has_active_key(tg_id): await message.reply("У вас уже есть активный ключ. Вы не можете создать новый.") await state.clear() return - - # Сохраняем имя ключа и tg_id для подтверждения - await state.set_state(Form.waiting_for_admin_confirmation) - await state.update_data(key_name=message.text, tg_id=tg_id) - + + # Сохраняем информацию о запросе в контексте состояния + await state.update_data(key_name=key_name, tg_id=tg_id) + # Отправляем запрос на подтверждение админу - admin_message = f"Запрос на создание ключа от пользователя {message.from_user.username}. Имя ключа: {message.text}. Подтвердите создание ключа." - button_yes = InlineKeyboardButton(text='Подтвердить', callback_data='confirm_key') - button_no = InlineKeyboardButton(text='Отклонить', callback_data='reject_key') - keyboard = InlineKeyboardMarkup(inline_keyboard=[[button_yes, button_no]]) + admin_message = ( + f"Новый запрос на создание ключа:\n" + f"Имя клиента: {key_name}\n" + f"Введите 'yes' для подтверждения или 'no' для отклонения." + ) + keyboard = InlineKeyboardMarkup(inline_keyboard=[ + [InlineKeyboardButton(text='Да', callback_data=f'confirm_key_{tg_id}_{key_name}')], + [InlineKeyboardButton(text='Нет', callback_data=f'reject_key_{tg_id}')] + ]) await bot.send_message(ADMIN_CHAT_ID, admin_message, reply_markup=keyboard) - await message.reply("Ваш запрос на создание ключа отправлен администратору. Ожидайте подтверждения.") - await state.clear() -@dp.message() -async def handle_text(message: types.Message, state: FSMContext): - current_state = await state.get_state() - - if current_state == Form.waiting_for_key_name.state: - tg_id = message.from_user.id - - if await has_active_key(tg_id): - await message.reply("У вас уже есть активный ключ. Вы не можете создать новый.") - await state.clear() - return - await state.set_state(Form.waiting_for_admin_confirmation) - await state.update_data(key_name=message.text, tg_id=tg_id) # Сохраняем имя ключа и tg_id для подтверждения - - # Отправляем запрос на подтверждение админу - admin_message = f"Запрос на создание ключа от пользователя {message.from_user.username}. Имя ключа: {message.text}. Подтвердите создание ключа." - button_yes = InlineKeyboardButton(text='Подтвердить', callback_data='confirm_key') - button_no = InlineKeyboardButton(text='Отклонить', callback_data='reject_key') - keyboard = InlineKeyboardMarkup(inline_keyboard=[[button_yes, button_no]]) - await bot.send_message(476217106, admin_message, reply_markup=keyboard) - await message.reply("Ваш запрос на создание ключа отправлен администратору. Ожидайте подтверждения.") + await message.reply("Ожидайте подтверждения администратора.") await state.clear() -@dp.callback_query(F.data == 'confirm_key') -async def process_admin_confirmation(callback_query: types.CallbackQuery, state: FSMContext): - user_data = await state.get_data() - key_name = user_data.get('key_name') - tg_id = user_data.get('tg_id') - - if not key_name or not tg_id: - await callback_query.answer("Ошибка: не удалось получить данные для создания ключа.") +@dp.callback_query(F.data.startswith('confirm_key_')) +async def handle_admin_confirmation(callback_query: CallbackQuery, state: FSMContext): + if callback_query.message.chat.id != int(ADMIN_CHAT_ID): return - + + data = callback_query.data.split('_', 3) + tg_id = int(data[2]) + key_name = data[3] + try: - # Генерация уникального client_id и создание ключа client_id = str(uuid.uuid4()) email = key_name limit_ip = 1 @@ -131,72 +102,75 @@ async def process_admin_confirmation(callback_query: types.CallbackQuery, state: add_client(session, client_id, email, tg_id, limit_ip, total_gb, expiry_time, enable, flow) - # Генерация ссылки для подключения - connection_link = link(session, email) + connection_link = link(session, client_id, email) - # Сохранение данных о подключении в базе данных await add_connection(tg_id, client_id, email, expiry_time) - # Отправка ключа клиенту await bot.send_message(tg_id, f"Ключ создан:\n
{connection_link}
", parse_mode="HTML") - await bot.send_message(callback_query.from_user.id, "Ключ успешно создан и отправлен клиенту.") - - await callback_query.answer("Ключ создан и отправлен клиенту.") + await callback_query.message.reply("Ключ создан и отправлен клиенту.") except Exception as e: - await bot.send_message(callback_query.from_user.id, f"Ошибка: {e}") - - await state.clear() + await callback_query.message.reply(f"Ошибка при создании ключа: {e}") + + await callback_query.answer() + +@dp.callback_query(F.data.startswith('reject_key_')) +async def handle_rejection(callback_query: CallbackQuery, state: FSMContext): + if callback_query.message.chat.id != int(ADMIN_CHAT_ID): + return + + tg_id = int(callback_query.data.split('_', 3)[2]) + await bot.send_message(tg_id, "Создание ключа отклонено.") + await callback_query.message.reply("Запрос отклонен.") + await callback_query.answer() @dp.callback_query(F.data == 'view_stats') async def process_callback_view_stats(callback_query: types.CallbackQuery): tg_id = callback_query.from_user.id - # Проверяем, есть ли активный ключ для этого пользователя - email = await get_active_key_email(tg_id) - - if not email: - await callback_query.message.reply("У вас нет активного ключа для просмотра статистики.") - return - - # Получаем статистику через API сервера try: - clients = get_clients(session) # Предположим, что get_clients возвращает список клиентов с их данными - client_data = next((client for client in clients if client['email'] == email), None) - - if not client_data: - await callback_query.message.reply("Не удалось получить статистику для этого пользователя.") - return - - # Подготавливаем статистику для отображения - used_gb = client_data.get('usedGB', 0) # Использованные данные - limit_gb = client_data.get('totalGB', 0) # Лимит данных - expiry_time = client_data.get('expiryTime', 0) # Время окончания подписки - - expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime('%Y-%m-%d %H:%M:%S') - - stats_message = (f"Статистика использования:\n" - f"Использовано: {used_gb} ГБ\n" - f"Лимит: {limit_gb} ГБ\n" - f"Дата окончания ключа: {expiry_date}") - - await callback_query.message.reply(stats_message) - + email = await get_active_key_email(tg_id) + if email: + connection_link = link(session, email) + + # Извлечение данных о загрузке и выгрузке из ссылки + up_match = re.search(r'up=(\d+)', connection_link) + down_match = re.search(r'down=(\d+)', connection_link) + + up = up_match.group(1) if up_match else "Неизвестно" + down = down_match.group(1) if down_match else "Неизвестно" + + statistics = f"Статистика вашего ключа:\nЗагрузка: {up} MB\nВыгрузка: {down} MB" + else: + statistics = "У вас нет активных ключей." + except Exception as e: - await callback_query.message.reply(f"Ошибка при получении статистики: {e}") - + statistics = f"Ошибка при получении статистики: {e}" + + await callback_query.message.reply(f"Ваша статистика:\n{statistics}") await callback_query.answer() - -@dp.callback_query(F.data == 'reject_key') -async def process_admin_rejection(callback_query: types.CallbackQuery, state: FSMContext): - await bot.send_message(callback_query.from_user.id, "Создание ключа отклонено.") +@dp.callback_query(F.data == 'view_expiry') +async def process_callback_view_expiry(callback_query: types.CallbackQuery): + tg_id = callback_query.from_user.id + + try: + email = await get_active_key_email(tg_id) + if email: + async with aiosqlite.connect(DATABASE_PATH) as db: + async with db.execute("SELECT expiry_time FROM connections WHERE tg_id = ? AND expiry_time > ?", + (tg_id, int(datetime.utcnow().timestamp() * 1000))) as cursor: + record = await cursor.fetchone() + if record: + expiry_time = record[0] + expiry_date = datetime.utcfromtimestamp(expiry_time / 1000).strftime("%Y-%m-%d %H:%M:%S") + message_text = f"Дата окончания вашего ключа: {expiry_date}" + else: + message_text = "У вас нет активных ключей." + else: + message_text = "У вас нет активных ключей." + + except Exception as e: + message_text = f"Ошибка при получении даты окончания ключа: {e}" + + await callback_query.message.reply(message_text) await callback_query.answer() - await state.clear() - -async def main(): - dp.include_router(router) # Подключение роутера - await dp.start_polling(bot) # Запуск поллинга - -if __name__ == '__main__': - import asyncio - asyncio.run(main()) diff --git a/client.py b/client.py index d0eec05f..925427ee 100644 --- a/client.py +++ b/client.py @@ -1,41 +1,60 @@ - +import requests import json +from config import ADMIN_USERNAME, ADMIN_PASSWORD, GET_INBOUNDS_URL import uuid -from config import ADD_CLIENT_URL -def add_client(session, client_id, email, tg_id, limit_ip, total_gb, expiry_time, enable, flow): - settings = { - "clients": [ - { - "id": client_id, - "alterId": 0, - "email": email, - "limitIp": limit_ip, - "totalGB": total_gb, - "expiryTime": expiry_time, - "enable": enable, - "tgId": tg_id, - "subId": "", - "flow": flow - } - ] +import requests +from config import ADMIN_USERNAME, ADMIN_PASSWORD + +import requests +from config import ADMIN_USERNAME, ADMIN_PASSWORD + +def add_client(session, client_id: str, email: str, tg_id: str, limit_ip: int, total_gb: int, expiry_time: int, enable: bool, flow: str): + url = 'https://solonet.pocomacho.ru:62553/solonet/panel/api/inbounds/addClient' + + # Преобразуем email в нижний регистр + email = email.lower() + + # Формируем данные клиента + client_data = { + "id": client_id, + "alterId": 0, + "email": email, + "limitIp": limit_ip, + "totalGB": total_gb, + "expiryTime": expiry_time, + "enable": enable, + "tgId": tg_id, + "subId": "", + "flow": flow, } + # Преобразуем данные клиента в строку JSON + settings = json.dumps({"clients": [client_data]}) + + # Формируем тело запроса data = { - "id": 1, # Можно изменить в зависимости от ваших требований - "settings": json.dumps(settings) + "id": 1, # Если id динамически изменяется, замените это значение + "settings": settings } - + headers = { - "Accept": "application/json" + 'Content-Type': 'application/json', # Если требуется токен авторизации } - response = session.post(ADD_CLIENT_URL, headers=headers, json=data) + response = session.post(url, json=data, headers=headers) + # Выводим информацию о запросе и ответе + print(f"Запрос на добавление клиента: {data}") + print(f"Статус ответа: {response.status_code}") + print(f"Ответ от сервера: {response.text}") + if response.status_code == 200: - return response.json() + print(f"Клиент добавлен: email={email}") else: - raise Exception(f"Ошибка при добавлении клиента: {response.status_code}, {response.text}") + print(f"Ошибка при добавлении клиента: {response.status_code}, {response.text}") + + def generate_client_id(): return str(uuid.uuid4())