Refactor API

API/utils/xui.py

@@ -0,0 +1,119 @@
+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()