"""Wartungs-Endpoints: Update-Badge, OS-/Engine-Update, Reboot, Restart, Logs."""

from fastapi import APIRouter, HTTPException, Header
from pydantic import BaseModel

from services import maintenance

router = APIRouter(prefix="/api")


class SudoReq(BaseModel):
    sudo_password: str | None = None


class RestartReq(BaseModel):
    service: str
    sudo_password: str | None = None


@router.get("/maintenance/updates")
def updates() -> dict:
    return maintenance.updates()


@router.get("/maintenance/update-details")
def update_details(kind: str) -> dict:
    if kind not in ("os", "engine", "hermes"):
        raise HTTPException(400, "Unbekannte Update-Art.")
    return maintenance.update_details(kind)

@router.post("/maintenance/check-updates")
def check_updates(body: SudoReq) -> dict:
    res = maintenance.check_updates_job(body.sudo_password)
    if isinstance(res, dict) and not res.get("ok", True):
        return res
    return res


@router.post("/maintenance/os-update")
def os_update(body: SudoReq) -> dict:
    res = maintenance.os_update_job(body.sudo_password)
    if isinstance(res, dict) and not res.get("ok", True):
        return res
    return res


@router.post("/maintenance/engine-update")
def engine_update(body: SudoReq) -> dict:
    res = maintenance.engine_update_job(body.sudo_password)
    if not res:
        raise HTTPException(400, "Kein Engine-Update-Befehl gesetzt (MC_ENGINE_UPDATE_CMD).")
    return res


@router.post("/maintenance/hermes-update")
def hermes_update() -> dict:
    return maintenance.hermes_update_job()


@router.post("/maintenance/reboot")
def reboot(body: SudoReq) -> dict:
    return maintenance.reboot(body.sudo_password)


@router.post("/maintenance/restart")
def restart(body: RestartReq) -> dict:
    return maintenance.restart_service(body.service, body.sudo_password)


@router.get("/maintenance/logs")
def logs(service: str, lines: int = 200, x_sudo_password: str | None = Header(None)) -> dict:
    return maintenance.logs(service, lines, x_sudo_password)