Files
Solo_bot/auth.py
T
Vladless 223559e44c bug fix
2024-11-11 07:42:22 +03:00

73 lines
2.6 KiB
Python

import json
import aiohttp
from config import SERVERS
session = None
async def login_with_credentials(server_id: str, username: str, password: str):
global session
session = aiohttp.ClientSession()
api_url = SERVERS[server_id]["API_URL"]
auth_url = f"{api_url}/login/"
data = {"username": username, "password": password}
async with session.post(auth_url, json=data) as response:
if response.status == 200:
session.cookie_jar.update_cookies(response.cookies)
return session
else:
raise Exception(
f"Ошибка авторизации: {response.status}, {await response.text()}"
)
async def get_clients(session, server_id):
api_url = SERVERS[server_id]["API_URL"]
async with session.get(f"{api_url}/panel/api/inbounds/list/") as response:
if response.status == 200:
return await response.json()
else:
raise Exception(
f"Ошибка при получении клиентов: {response.status}, {await response.text()}"
)
async def link(session, server_id: str, client_id: str, email: str):
"""
Получение ссылки для подключения по ID клиента.
:param server_id: str - идентификатор сервера
:param client_id: str - идентификатор клиента
:param email: str - электронная почта клиента
:return: str - ссылка для подключения
"""
response = await get_clients(session, server_id)
if "obj" not in response or len(response["obj"]) == 0:
raise Exception("Не удалось получить данные клиентов.")
inbounds = response["obj"][0]
stream_settings = json.loads(inbounds["streamSettings"])
tcp = stream_settings.get("network", "tcp")
reality = stream_settings.get("security", "reality")
flow = stream_settings.get("flow", "xtls-rprx-vision")
val = (
f"vless://{client_id}@{SERVERS[server_id]['DOMEN']}?type={tcp}&security={reality}&pbk={SERVERS[server_id]['PBK']}"
f"&fp=chrome&sni={SERVERS[server_id]['SNI']}&sid={SERVERS[server_id]['SID']}=%2F&flow={flow}#{SERVERS[server_id]['PREFIX']}-{email}"
)
return val
async def link_subscription(email, server_id):
server = SERVERS.get(server_id)
if server:
subscription_url = f"{server['SUBSCRIPTION']}/{email}"
return subscription_url
else:
raise ValueError(f"Server '{server_id}' not found in configuration.")