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())