119 lines
4 KiB
Python
119 lines
4 KiB
Python
import json
|
|
import uuid
|
|
import random
|
|
from httpx import AsyncClient
|
|
from fastapi import HTTPException
|
|
|
|
class XUIClient:
|
|
def __init__(self, base_url: str, username: str, password: str):
|
|
self.base_url = base_url.rstrip("/")
|
|
self.username = username
|
|
self.password = password
|
|
self.session = AsyncClient(timeout=20.0)
|
|
self.logged_in = False
|
|
|
|
@classmethod
|
|
async def from_server(cls, server):
|
|
#TODO заменить на https
|
|
base_url = f"http://{server.host}:{server.port}"
|
|
return cls(base_url, server.user, server.password)
|
|
|
|
async def _login(self):
|
|
if self.logged_in:
|
|
return
|
|
try:
|
|
resp = await self.session.post(
|
|
f"{self.base_url}/login",
|
|
json={"username": self.username, "password": self.password}
|
|
)
|
|
resp.raise_for_status()
|
|
self.logged_in = True
|
|
except:
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail="Server error"
|
|
)
|
|
|
|
async def add_client(self, inbound_id: int, client_email: str, display_name: str) -> str:
|
|
await self._login()
|
|
resp = await self.session.post(f"{self.base_url}/panel/inbound/list")
|
|
resp.raise_for_status()
|
|
data = resp.json()
|
|
inbounds = data.get("obj", [])
|
|
inbound = next((i for i in inbounds if i.get("id") == inbound_id), None)
|
|
if not inbound:
|
|
raise HTTPException(
|
|
status_code=404,
|
|
detail=f"Server not found"
|
|
)
|
|
stream_settings = json.loads(inbound.get("streamSettings", "{}"))
|
|
reality = stream_settings.get("realitySettings", {})
|
|
short_ids = reality.get("shortIds", [""])
|
|
server_names = reality.get("serverNames", {})
|
|
suid = random.choice(short_ids)
|
|
sni = random.choice(server_names)
|
|
pbk = reality.get("settings", {}).get("publicKey", "")
|
|
client_uuid = str(uuid.uuid4())
|
|
add_payload = {
|
|
"id": inbound_id,
|
|
"settings": json.dumps({
|
|
"clients": [{
|
|
"id": client_uuid,
|
|
"flow": "xtls-rprx-vision",
|
|
"email": client_email,
|
|
"limitIp": 0,
|
|
"totalGB": 0,
|
|
"expiryTime": 0,
|
|
"enable": True,
|
|
"tgId": "",
|
|
"subId": suid,
|
|
"comment": "",
|
|
"reset": 0
|
|
}]
|
|
})
|
|
}
|
|
resp = await self.session.post(f"{self.base_url}/panel/inbound/addClient", json=add_payload)
|
|
resp.raise_for_status()
|
|
host = inbound.get("remark")
|
|
config_url = (
|
|
f"vless://{client_uuid}@{host}?"
|
|
f"security=reality&"
|
|
f"pbk={pbk}&"
|
|
f"fp=random&"
|
|
f"sni={sni}&"
|
|
f"sid={suid}&"
|
|
f"spx=%2F&"
|
|
f"flow=xtls-rprx-vision#"
|
|
f"{display_name}"
|
|
)
|
|
return config_url
|
|
|
|
async def get_client_traffic(self, client_email: str) -> int:
|
|
await self._login()
|
|
try:
|
|
resp = await self.session.post(
|
|
f"{self.base_url}/panel/inbound/getClientTraffics/{client_email}"
|
|
)
|
|
resp.raise_for_status()
|
|
data = resp.json()
|
|
obj = data.get("obj", {})
|
|
return obj.get("down", 0) + obj.get("up", 0)
|
|
except:
|
|
return 0
|
|
|
|
async def delete_client(self, inbound_id: int, client_email: str):
|
|
await self._login()
|
|
try:
|
|
resp = await self.session.post(
|
|
f"{self.base_url}/panel/inbound/{inbound_id}/delClientByEmail/{client_email}"
|
|
)
|
|
resp.raise_for_status()
|
|
return
|
|
except:
|
|
HTTPException(
|
|
status_code=500,
|
|
detail="Server error"
|
|
)
|
|
|
|
async def close(self):
|
|
await self.session.aclose()
|