Vlad
141 lines
5.1 KiB
Python
141 lines
5.1 KiB
Python
from aiogram import Bot, Dispatcher, Router, F, types
|
||
from aiogram.filters import Command
|
||
from aiogram.types import Message
|
||
from aiogram.filters import Command
|
||
import requests
|
||
from database import add_user, get_user, update_subscription
|
||
from aiogram.types import InlineKeyboardButton, InlineKeyboardMarkup
|
||
from aiogram.fsm.storage.memory import MemoryStorage
|
||
from aiogram.fsm.context import FSMContext
|
||
from aiogram.fsm.state import State, StatesGroup
|
||
|
||
import asyncio
|
||
import json
|
||
from datetime import datetime, timedelta
|
||
import uuid
|
||
|
||
|
||
API_TOKEN = '7436953421:AAH9iUg86cLC8oxO_XZeqYEfSBhDGBodV5o'
|
||
API_URL = "https://vpn.pocomacho.ru:34268/solonet"
|
||
ADMIN_USERNAME = "paneladmin"
|
||
ADMIN_PASSWORD = "L7CMDASN9x7g"
|
||
ADD_CLIENT_URL = f"{API_URL}/panel/api/inbounds/addClient"
|
||
|
||
bot = Bot(token=API_TOKEN)
|
||
storage = MemoryStorage()
|
||
dp = Dispatcher(storage=storage)
|
||
router = Router()
|
||
|
||
class Form(StatesGroup):
|
||
waiting_for_key_name = State()
|
||
|
||
def login_with_credentials(username, password):
|
||
auth_url = "https://vpn.pocomacho.ru:34268/solonet/login/"
|
||
data = {
|
||
"username": ADMIN_USERNAME,
|
||
"password": ADMIN_PASSWORD
|
||
}
|
||
response = requests.post(auth_url, json=data)
|
||
if response.status_code == 200:
|
||
# Если API возвращает токен, используйте его в заголовке
|
||
# return {"Authorization": f"Bearer {response.json().get('token')}"}
|
||
|
||
# Если API использует сессии (например, с cookies)
|
||
session = requests.Session()
|
||
session.cookies.update(response.cookies)
|
||
return session
|
||
|
||
else:
|
||
raise Exception(f"Ошибка авторизации: {response.status_code}, {response.text}")
|
||
|
||
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
|
||
}
|
||
]
|
||
}
|
||
|
||
data = {
|
||
"id": 1, # Можно изменить в зависимости от ваших требований
|
||
"settings": json.dumps(settings)
|
||
}
|
||
|
||
headers = {
|
||
"Accept": "application/json"
|
||
}
|
||
|
||
response = session.post(ADD_CLIENT_URL, headers=headers, json=data)
|
||
|
||
if response.status_code == 200:
|
||
return response.json()
|
||
else:
|
||
raise Exception(f"Ошибка при добавлении клиента: {response.status_code}, {response.text}")
|
||
|
||
# Обработчик команды /start
|
||
@dp.message(Command("start"))
|
||
async def start_command(message: Message):
|
||
# Создаем кнопки
|
||
button_create_key = InlineKeyboardButton(text='Создать ключ', callback_data='create_key')
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=[[button_create_key]])
|
||
|
||
await message.reply("Выберите действие:", reply_markup=keyboard)
|
||
|
||
# Обработчик нажатия кнопки
|
||
@dp.callback_query(F.data == 'create_key')
|
||
async def process_callback_create_key(callback_query: types.CallbackQuery, state: FSMContext):
|
||
await callback_query.message.reply("Введите имя для нового ключа:")
|
||
await state.set_state(Form.waiting_for_key_name) # Устанавливаем состояние
|
||
|
||
await callback_query.answer()
|
||
|
||
# Обработчик текстовых сообщений
|
||
@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:
|
||
try:
|
||
# Авторизация
|
||
session = login_with_credentials(ADMIN_USERNAME, ADMIN_PASSWORD)
|
||
|
||
# Параметры клиента
|
||
client_id = str(uuid.uuid4()) # Пример UUID
|
||
email = message.text
|
||
limit_ip = 1
|
||
tg_id = message.from_user.id
|
||
total_gb = 0 # В байтах
|
||
current_time = datetime.utcnow()
|
||
expiry_time = int((current_time + timedelta(days=30)).timestamp() * 1000) # Временная метка в миллисекундах
|
||
enable = True
|
||
flow = "xtls-rprx-vision"
|
||
|
||
# Добавление клиента
|
||
result = add_client(session, client_id, email, tg_id, limit_ip, total_gb, expiry_time, enable, flow)
|
||
|
||
await message.reply(f"Ключ создан: {result}")
|
||
|
||
# Сброс состояния
|
||
await state.clear()
|
||
except Exception as e:
|
||
await message.reply(f"Произошла ошибка: {e}")
|
||
|
||
# Запуск бота
|
||
async def main():
|
||
dp.include_router(router) # Подключение роутера
|
||
await dp.start_polling(bot) # Запуск поллинга
|
||
|
||
if __name__ == '__main__':
|
||
asyncio.run(main())
|