115 lines
3.2 KiB
Python
115 lines
3.2 KiB
Python
"""Editor helper endpoints — merge wizard fragments into compose YAML."""
|
|
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
from auth import get_current_user
|
|
from models.user import User
|
|
from services import compose_edit_service as edit
|
|
from services import compose_service
|
|
|
|
router = APIRouter(prefix="/api/editor", tags=["editor"])
|
|
|
|
|
|
class ValidateBody(BaseModel):
|
|
yaml: str
|
|
env: str = ""
|
|
|
|
|
|
class AddVolumeBody(BaseModel):
|
|
yaml: str
|
|
service: str
|
|
spec: dict
|
|
|
|
|
|
class SetGpuBody(BaseModel):
|
|
yaml: str
|
|
service: str
|
|
config: dict
|
|
|
|
|
|
class DeviceBody(BaseModel):
|
|
yaml: str
|
|
service: str
|
|
host_path: str
|
|
target: str | None = None
|
|
|
|
|
|
class PrivilegedBody(BaseModel):
|
|
yaml: str
|
|
service: str
|
|
value: bool
|
|
|
|
|
|
class ResourcesBody(BaseModel):
|
|
yaml: str
|
|
service: str
|
|
cpus: float | None = None
|
|
memory: str | None = None
|
|
cpus_reserve: float | None = None
|
|
memory_reserve: str | None = None
|
|
|
|
|
|
def _run(fn, *args) -> dict:
|
|
try:
|
|
return {"yaml": fn(*args)}
|
|
except edit.EditError as exc:
|
|
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
|
|
|
|
|
@router.post("/validate")
|
|
async def validate(body: ValidateBody, _user: User = Depends(get_current_user)) -> dict:
|
|
"""Run `docker compose config -q` on the supplied YAML (+ optional .env)."""
|
|
return await compose_service.validate_yaml(body.yaml, body.env)
|
|
|
|
|
|
@router.post("/services")
|
|
def services(body: dict, _user: User = Depends(get_current_user)) -> dict:
|
|
try:
|
|
return {"services": edit.list_services(body.get("yaml", ""))}
|
|
except edit.EditError as exc:
|
|
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
|
|
|
|
|
@router.post("/add-volume")
|
|
def add_volume(body: AddVolumeBody, _user: User = Depends(get_current_user)) -> dict:
|
|
return _run(edit.add_volume, body.yaml, body.service, body.spec)
|
|
|
|
|
|
@router.post("/set-gpu")
|
|
def set_gpu(body: SetGpuBody, _user: User = Depends(get_current_user)) -> dict:
|
|
return _run(edit.set_gpu, body.yaml, body.service, body.config)
|
|
|
|
|
|
@router.post("/add-device")
|
|
def add_device(body: DeviceBody, _user: User = Depends(get_current_user)) -> dict:
|
|
return _run(edit.add_device, body.yaml, body.service, body.host_path, body.target)
|
|
|
|
|
|
@router.post("/remove-device")
|
|
def remove_device(body: DeviceBody, _user: User = Depends(get_current_user)) -> dict:
|
|
return _run(edit.remove_device, body.yaml, body.service, body.host_path)
|
|
|
|
|
|
@router.post("/set-privileged")
|
|
def set_privileged(body: PrivilegedBody, _user: User = Depends(get_current_user)) -> dict:
|
|
return _run(edit.set_privileged, body.yaml, body.service, body.value)
|
|
|
|
|
|
@router.post("/set-resources")
|
|
def set_resources(body: ResourcesBody, _user: User = Depends(get_current_user)) -> dict:
|
|
try:
|
|
return {
|
|
"yaml": edit.set_resources(
|
|
body.yaml,
|
|
body.service,
|
|
cpus=body.cpus,
|
|
memory=body.memory,
|
|
cpus_reserve=body.cpus_reserve,
|
|
memory_reserve=body.memory_reserve,
|
|
)
|
|
}
|
|
except edit.EditError as exc:
|
|
raise HTTPException(status_code=400, detail=str(exc)) from exc
|