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]
    settings = json.loads(inbounds['settings'])
    
    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.")