import asyncpg
import uuid
from config import DATABASE_URL, SERVERS, ADMIN_USERNAME, ADMIN_PASSWORD
from auth import login_with_credentials, link_subscription
from client import add_client
from database import store_key, add_connection
from handlers.texts import INSTRUCTIONS
from datetime import datetime, timedelta
from handlers.utils import generate_random_email, get_least_loaded_server


async def create_trial_key(tg_id: int):
    conn = await asyncpg.connect(DATABASE_URL)
    try:
        server_id = await get_least_loaded_server(conn)
        session = await login_with_credentials(server_id, ADMIN_USERNAME, ADMIN_PASSWORD)
        current_time = datetime.utcnow()

        expiry_time = current_time + timedelta(days=1, hours=3)
        expiry_timestamp = int(expiry_time.timestamp() * 1000)
        
        client_id = str(uuid.uuid4())
        email = generate_random_email()
        response = await add_client(
            session, server_id, client_id, email, tg_id,
            limit_ip=1, total_gb=0, expiry_time=expiry_timestamp,
            enable=True, flow="xtls-rprx-vision"
        )

        if response.get("success"):
            # Генерация ссылки подписки
            connection_link = await link_subscription(email, server_id)

            existing_connection = await conn.fetchrow('SELECT * FROM connections WHERE tg_id = $1', tg_id)

            if existing_connection:
                await conn.execute('UPDATE connections SET trial = 1 WHERE tg_id = $1', tg_id)
            else:
                await add_connection(tg_id, 0, 1)

            await store_key(tg_id, client_id, email, expiry_timestamp, connection_link, server_id)

            instructions = INSTRUCTIONS
            return {
                'key': connection_link,
                'instructions': instructions
            }
        else:
            return {'error': 'Не удалось добавить клиента на панель'}
    finally:
        await conn.close()