Vladless
98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
import asyncio
|
|
import uuid
|
|
from datetime import datetime, timedelta
|
|
|
|
import asyncpg
|
|
from py3xui import AsyncApi
|
|
|
|
from client import add_client
|
|
from config import ADMIN_PASSWORD, ADMIN_USERNAME, DATABASE_URL, PUBLIC_LINK, SERVERS
|
|
from database import store_key
|
|
from handlers.texts import INSTRUCTIONS
|
|
from handlers.utils import generate_random_email
|
|
|
|
|
|
async def create_trial_key(tg_id: int):
|
|
conn = await asyncpg.connect(DATABASE_URL)
|
|
try:
|
|
client_id = str(uuid.uuid4())
|
|
email = generate_random_email()
|
|
|
|
public_link = f"{PUBLIC_LINK}{email}"
|
|
instructions = INSTRUCTIONS
|
|
|
|
result = {"key": public_link, "instructions": instructions}
|
|
|
|
asyncio.create_task(
|
|
generate_and_store_keys(tg_id, client_id, email, public_link)
|
|
)
|
|
|
|
return result
|
|
|
|
finally:
|
|
await conn.close()
|
|
|
|
|
|
async def generate_and_store_keys(
|
|
tg_id: int, client_id: str, email: str, public_link: str
|
|
):
|
|
conn = await asyncpg.connect(DATABASE_URL)
|
|
try:
|
|
current_time = datetime.utcnow()
|
|
expiry_time = current_time + timedelta(days=1, hours=3)
|
|
expiry_timestamp = int(expiry_time.timestamp() * 1000)
|
|
|
|
tasks = []
|
|
for server_id in SERVERS:
|
|
task = create_key_on_server(
|
|
server_id, client_id, email, tg_id, expiry_timestamp
|
|
)
|
|
tasks.append(task)
|
|
|
|
await asyncio.gather(*tasks)
|
|
|
|
await store_key(
|
|
tg_id,
|
|
client_id,
|
|
email,
|
|
expiry_timestamp,
|
|
public_link,
|
|
server_id="all_servers",
|
|
)
|
|
|
|
await conn.execute(
|
|
"""
|
|
INSERT INTO connections (tg_id, trial)
|
|
VALUES ($1, 1)
|
|
ON CONFLICT (tg_id)
|
|
DO UPDATE SET trial = 1
|
|
""",
|
|
tg_id,
|
|
)
|
|
finally:
|
|
await conn.close()
|
|
|
|
|
|
async def create_key_on_server(
|
|
server_id: str, client_id: str, email: str, tg_id: int, expiry_timestamp: int
|
|
):
|
|
"""Создает ключ на сервере и возвращает результат."""
|
|
|
|
xui = AsyncApi(
|
|
SERVERS[server_id]["API_URL"], username=ADMIN_USERNAME, password=ADMIN_PASSWORD
|
|
)
|
|
|
|
response = await add_client(
|
|
xui,
|
|
client_id,
|
|
email,
|
|
tg_id,
|
|
limit_ip=1,
|
|
total_gb=0,
|
|
expiry_time=expiry_timestamp,
|
|
enable=True,
|
|
flow="xtls-rprx-vision",
|
|
)
|
|
|
|
return response
|