From eaf12d85dddf362c2fb8345b14a156eabc53e346 Mon Sep 17 00:00:00 2001
From: Iam54r1n4 <iam54r1n40@proton.me>
Date: Sun, 26 Jan 2025 11:18:08 +0000
Subject: [PATCH] Add api/v1/config/warp/... endpoint

---
 .../webpanel/routers/api/v1/config/warp.py    | 89 +++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 core/scripts/webpanel/routers/api/v1/config/warp.py

diff --git a/core/scripts/webpanel/routers/api/v1/config/warp.py b/core/scripts/webpanel/routers/api/v1/config/warp.py
new file mode 100644
index 0000000..6dcaa7b
--- /dev/null
+++ b/core/scripts/webpanel/routers/api/v1/config/warp.py
@@ -0,0 +1,89 @@
+import re
+from fastapi import APIRouter, HTTPException
+from ..schema.response import DetailResponse
+from ..schema.config.warp import ConfigureInputBody, StatusResponse
+
+import cli_api
+
+router = APIRouter()
+
+
+@router.get('/install', response_model=DetailResponse)
+async def install():
+    try:
+        cli_api.install_warp()
+        return DetailResponse(detail='WARP installed successfully.')
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f'Error: {str(e)}')
+
+
+@router.get('/uninstall', response_model=DetailResponse)
+async def uninstall():
+    try:
+        cli_api.uninstall_warp()
+        return DetailResponse(detail='WARP uninstalled successfully.')
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f'Error: {str(e)}')
+
+
+@router.get('/configure', response_model=DetailResponse)
+async def configure(body: ConfigureInputBody):
+    try:
+        cli_api.configure_warp(body.all, body.popular_sites, body.domestic_sites,
+                               body.block_adult_sites, body.warp_option, body.warp_key)
+        return DetailResponse(detail='WARP configured successfully.')
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f'Error: {str(e)}')
+
+
+@router.get('/status', response_model=StatusResponse)
+async def status():
+    try:
+        if res := cli_api.warp_status():
+            return __parse_status(res)
+        raise HTTPException(status_code=404, detail='WARP status not available.')
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f'Error: {str(e)}')
+
+
+def __parse_status(status: str) -> StatusResponse:
+    data = {}
+    # Example output(status) from cli_api.warp_status():
+    # --------------------------------
+    # Current WARP Configuration:
+    # All traffic: Inactive
+    # Popular sites (Google, Netflix, etc.): Inactive
+    # Domestic sites (geosite:ir, geoip:ir): Inactive
+    # Block adult content: Inactive
+    # --------------------------------
+
+    # Remove ANSI escape sequences(colors) (e.g., \x1b[1;35m)
+    clean_status = re.sub(r'\x1b\[[0-9;]*m', '', status)
+
+    for line in clean_status.split('\n'):
+        if ':' not in line:
+            continue
+        if 'Current WARP Configuration:' in line:
+            continue
+        key, _, value = line.partition(':')
+        key = key.strip().lower()
+        value = value.strip()
+
+        if not key or not value:
+            continue
+
+        if 'all traffic' in key:
+            data['all_traffic'] = value == 'active'
+        elif 'popular sites' in key:
+            data['popular_sites'] = value == 'active'
+        elif 'domestic sites' in key:
+            data['domestic_sites'] = value == 'active'
+        elif 'block adult content' in key:
+            data['block_adult_sites'] = value == 'active'
+
+    if not data:
+        raise ValueError('Invalid WARP status')
+    try:
+        return StatusResponse(**data)
+    except Exception as e:
+        raise ValueError(f'Invalid or incomplete WARP status: {e}')