Update controllers/config_controller.py

controllers/config_controller.py

@@ -1,18 +1,88 @@
-from fastapi import APIRouter
-import threading
-from service_config import ServiceConfig
-from log import log
-
-router = APIRouter()
-service_config = ServiceConfig()
-
-@router.post("/reload")
-def reload_config():
-    def background_reload():
-        try:
-            service_config.load()
-            log("✅ Service config reloaded successfully.")
-        except Exception as e:
-            log(f"❌ Error reloading config: {e}")
-    threading.Thread(target=background_reload, daemon=True).start()
-    return {"status": "accepted", "message": "Config reload started in background."}
+from fastapi import APIRouter, Request
+import threading
+import json
+from service_config import ServiceConfig
+from log import log
+
+router = APIRouter()
+service_config = ServiceConfig()
+service_config.load()
+
+def save_config():
+    with open(service_config.config_path, "w", encoding="utf-8") as f:
+        json.dump({
+            "config": {
+                "work_mode": service_config.work_mode,
+                "cloud_token": service_config.cloud_token,
+                "system_prompt": service_config.system_prompt,
+                "llm_inference_service_url": service_config.llm_inference_service_url
+            },
+            "projects": service_config.projects,
+            "apis": service_config.apis
+        }, f, indent=2)
+
+@router.post("/reload")
+def reload_config():
+    def background_reload():
+        try:
+            service_config.load()
+            log("✅ Service config reloaded successfully.")
+        except Exception as e:
+            log(f"❌ Error reloading config: {e}")
+    threading.Thread(target=background_reload, daemon=True).start()
+    return {"status": "accepted", "message": "Config reload started in background."}
+
+# === API CRUD ===
+
+@router.get("/list_apis")
+def list_apis():
+    api_names = list(service_config.apis.keys())
+    log("📋 Listing APIs.")
+    return {"apis": api_names}
+
+@router.post("/add_api")
+async def add_api(request: Request):
+    data = await request.json()
+    api_name = data.get("api_name")
+    api_config = data.get("api_config")
+    if not api_name or not api_config:
+        return {"error": "API name and config required."}
+
+    if api_name in service_config.apis:
+        return {"error": "API already exists."}
+
+    service_config.apis[api_name] = api_config
+    save_config()
+    log(f"🆕 Added API: {api_name}")
+    return {"message": f"API {api_name} added."}
+
+@router.post("/delete_api")
+async def delete_api(request: Request):
+    data = await request.json()
+    api_name = data.get("api_name")
+    if not api_name:
+        return {"error": "API name required."}
+
+    if api_name not in service_config.apis:
+        return {"error": "API not found."}
+
+    del service_config.apis[api_name]
+    save_config()
+    log(f"🗑️ Deleted API: {api_name}")
+    return {"message": f"API {api_name} deleted."}
+
+@router.post("/update_api")
+async def update_api(request: Request):
+    data = await request.json()
+    api_name = data.get("api_name")
+    api_config = data.get("api_config")
+    if not api_name or not api_config:
+        return {"error": "API name and updated config required."}
+
+    if api_name not in service_config.apis:
+        return {"error": "API not found."}
+
+    service_config.apis[api_name] = api_config
+    save_config()
+    log(f"✏️ Updated API: {api_name}")
+    return {"message": f"API {api_name} updated."}