Update config_controller.py

config_controller.py

@@ -1,36 +1,33 @@
-from fastapi import APIRouter, Request, HTTPException
-from service_config import ServiceConfig
+from fastapi import APIRouter, HTTPException, Request, Depends
+from app import get_config, ServiceConfig
 import json
 
 router = APIRouter()
-service_config = ServiceConfig()
-service_config.load()
 
 @router.get("/get")
-def get_config():
+def get_config_values(config: ServiceConfig = Depends(get_config)):
     return {
-        "work_mode": service_config.work_mode,
-        "cloud_token": service_config.cloud_token,
-        "spark_endpoint": service_config.spark_endpoint,
-        "last_version_number": service_config.last_version_number
+        "work_mode": config.work_mode,
+        "cloud_token": config.cloud_token
     }
 
 @router.post("/update")
-async def update_config(request: Request):
+async def update_config(request: Request, config: ServiceConfig = Depends(get_config)):
     data = await request.json()
     work_mode = data.get("work_mode")
     cloud_token = data.get("cloud_token")
-    spark_endpoint = data.get("spark_endpoint")
 
-    if not work_mode or not spark_endpoint:
-        raise HTTPException(status_code=400, detail="work_mode and spark_endpoint cannot be empty")
+    if work_mode not in ["hfcloud", "cloud", "on-premise"]:
+        raise HTTPException(status_code=400, detail="Invalid work mode")
 
-    service_config.work_mode = work_mode
-    service_config.cloud_token = cloud_token
-    service_config.spark_endpoint = spark_endpoint
+    if (work_mode in ["hfcloud", "cloud"]) and not cloud_token:
+        raise HTTPException(status_code=400, detail="Cloud token required for selected mode")
 
-    with open(service_config.config_path, "w", encoding="utf-8") as f:
-        json.dump(service_config, f, indent=2)
+    config.work_mode = work_mode
+    config.cloud_token = cloud_token
 
-    return {"message": "Configuration updated successfully"}
+    with open(config.config_path, "w", encoding="utf-8") as f:
+        json.dump(config, f, indent=2)
+
+    return {"message": "Configuration updated"}