update

server/app/controller/config/config_controller.py

@@ -0,0 +1,121 @@
+from typing import List, Optional
+from fastapi import Depends, HTTPException, Query, Response, APIRouter
+from sqlmodel import Session, select, or_
+from app.component.database import session
+from app.component.auth import Auth, auth_must
+from fastapi_babel import _
+from app.model.config.config import Config, ConfigCreate, ConfigUpdate, ConfigInfo, ConfigOut
+
+router = APIRouter(tags=["Config Management"])
+
+
+@router.get("/configs", name="list configs", response_model=list[ConfigOut])
+async def list_configs(
+    config_group: Optional[str] = None, session: Session = Depends(session), auth: Auth = Depends(auth_must)
+):
+    query = select(Config)
+    user_id = auth.user.id
+    if user_id is not None:
+        query = query.where(Config.user_id == user_id)
+    if config_group is not None:
+        query = query.where(Config.config_group == config_group)
+    configs = session.exec(query).all()
+    return configs
+
+
+@router.get("/configs/{config_id}", name="get config", response_model=ConfigOut)
+async def get_config(
+    config_id: int,
+    session: Session = Depends(session),
+    auth: Auth = Depends(auth_must),
+):
+    query = select(Config).where(Config.user_id == auth.user.id)
+
+    if config_id is not None:
+        query = query.where(Config.id == config_id)
+
+    config = session.exec(query).first()
+
+    if not config:
+        raise HTTPException(status_code=404, detail=_("Configuration not found"))
+    return config
+
+
+@router.post("/configs", name="create config", response_model=ConfigOut)
+async def create_config(config: ConfigCreate, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
+    if not ConfigInfo.is_valid_env_var(config.config_group, config.config_name):
+        raise HTTPException(status_code=400, detail=_("Config Name is valid"))
+
+    # Check if configuration already exists
+    existing_config = session.exec(
+        select(Config).where(Config.user_id == auth.user.id, Config.config_name == config.config_name)
+    ).first()
+
+    if existing_config:
+        raise HTTPException(status_code=400, detail=_("Configuration already exists for this user"))
+
+    db_config = Config(
+        user_id=auth.user.id,
+        config_name=config.config_name,
+        config_value=config.config_value,
+        config_group=config.config_group,
+    )
+    session.add(db_config)
+    session.commit()
+    session.refresh(db_config)
+    return db_config
+
+
+@router.put("/configs/{config_id}", name="update config", response_model=ConfigOut)
+async def update_config(
+    config_id: int, config_update: ConfigUpdate, session: Session = Depends(session), auth: Auth = Depends(auth_must)
+):
+    db_config = session.exec(select(Config).where(Config.id == config_id, Config.user_id == auth.user.id)).first()
+
+    if not db_config:
+        raise HTTPException(status_code=404, detail=_("Configuration not found"))
+
+    # Check if configuration group is valid
+    if not ConfigInfo.is_valid_env_var(config_update.config_group, config_update.config_name):
+        raise HTTPException(status_code=400, detail=_("Invalid configuration group"))
+
+    # Check for conflicts with other configurations
+    existing_config = session.exec(
+        select(Config).where(
+            Config.user_id == auth.user.id,
+            Config.config_name == config_update.config_name,
+            Config.id != config_id,
+        )
+    ).first()
+
+    if existing_config:
+        raise HTTPException(status_code=400, detail=_("Configuration already exists for this user"))
+
+    db_config.config_name = config_update.config_name
+    db_config.config_value = config_update.config_value
+
+    session.add(db_config)
+    session.commit()
+    session.refresh(db_config)
+    return db_config
+
+
+@router.delete("/configs/{config_id}", name="delete config")
+async def delete_config(config_id: int, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
+    db_config = session.exec(select(Config).where(Config.id == config_id, Config.user_id == auth.user.id)).first()
+
+    if not db_config:
+        raise HTTPException(status_code=404, detail=_("Configuration not found"))
+    session.delete(db_config)
+    session.commit()
+    return Response(status_code=204)
+
+
+@router.get("/config/info", name="get config info")
+async def get_config_info(
+    show_all: bool = Query(False, description="Show all config info, including those with empty env_vars"),
+):
+    configs = ConfigInfo.getinfo()
+    if show_all:
+        return configs
+    return {k: v for k, v in configs.items() if v.get("env_vars") and len(v["env_vars"]) > 0}