feat: complete QEMU power lifecycle

app/handlers/qemu.py

@@ -12,6 +12,7 @@ from app.api.errors import ApiError
 from app.api.registry import HandlerRegistry
 from app.db.pool import AsyncpgDatabase
 from app.db.primitives import ConflictError
+from app.simulation.transitions import InvalidTransitionError, VmState, plan_transition
 from app.tasks.repository import TaskRepository
 from app.tasks.upid import Upid
 
@@ -71,10 +72,10 @@ def register_qemu_handlers(registry: HandlerRegistry) -> None:
         if row is None:
             raise ApiError(404, "virtual machine does not exist")
         current = str(_state(row["state"]).get("status", "stopped"))
-        if (operation == "start" and current != "stopped") or (
-            operation == "stop" and current != "running"
-        ):
-            raise ApiError(409, f"cannot {operation} VM while it is {current}")
+        try:
+            plan_transition(VmState(current), operation)
+        except (InvalidTransitionError, ValueError) as error:
+            raise ApiError(409, f"cannot {operation} VM while it is {current}") from error
         timestamp = int(await database.pool.fetchval("SELECT extract(epoch from now())::bigint"))
         pid = int(await database.pool.fetchval("SELECT pg_backend_pid()"))
         upid = str(
@@ -220,6 +221,21 @@ def register_qemu_handlers(registry: HandlerRegistry) -> None:
     async def stop(request: Request, inputs: dict[str, Any]) -> str:
         return await mutate("stop", request, inputs)
 
+    async def shutdown(request: Request, inputs: dict[str, Any]) -> str:
+        return await mutate("shutdown", request, inputs)
+
+    async def reboot(request: Request, inputs: dict[str, Any]) -> str:
+        return await mutate("reboot", request, inputs)
+
+    async def reset(request: Request, inputs: dict[str, Any]) -> str:
+        return await mutate("reset", request, inputs)
+
+    async def suspend(request: Request, inputs: dict[str, Any]) -> str:
+        return await mutate("suspend", request, inputs)
+
+    async def resume(request: Request, inputs: dict[str, Any]) -> str:
+        return await mutate("resume", request, inputs)
+
     async def task_list(request: Request, inputs: dict[str, Any]) -> list[dict[str, Any]]:
         tasks = await TaskRepository(_database(request).pool).list_for_node(
             str(_values(inputs)["node"])
@@ -262,6 +278,11 @@ def register_qemu_handlers(registry: HandlerRegistry) -> None:
     registry.register("/nodes/{node}/qemu/{vmid}/status/current", "GET", qemu_status)
     registry.register("/nodes/{node}/qemu/{vmid}/status/start", "POST", start)
     registry.register("/nodes/{node}/qemu/{vmid}/status/stop", "POST", stop)
+    registry.register("/nodes/{node}/qemu/{vmid}/status/shutdown", "POST", shutdown)
+    registry.register("/nodes/{node}/qemu/{vmid}/status/reboot", "POST", reboot)
+    registry.register("/nodes/{node}/qemu/{vmid}/status/reset", "POST", reset)
+    registry.register("/nodes/{node}/qemu/{vmid}/status/suspend", "POST", suspend)
+    registry.register("/nodes/{node}/qemu/{vmid}/status/resume", "POST", resume)
     registry.register("/nodes/{node}/tasks", "GET", task_list)
     registry.register("/nodes/{node}/tasks/{upid}/status", "GET", task_status)
     registry.register("/nodes/{node}/tasks/{upid}/log", "GET", task_log)

app/main.py

@@ -46,8 +46,13 @@ def create_app(
                 {
                     "qemu-create": handler,
                     "qemu-delete": handler,
+                    "qemu-reboot": handler,
+                    "qemu-reset": handler,
+                    "qemu-resume": handler,
+                    "qemu-shutdown": handler,
                     "qemu-start": handler,
                     "qemu-stop": handler,
+                    "qemu-suspend": handler,
                     "qemu-update": handler,
                 },
                 concurrency=resolved.task_worker_concurrency,

app/simulation/transitions.py

@@ -29,6 +29,10 @@ class InvalidTransitionError(ValueError):
 TRANSITIONS: dict[tuple[VmState, str], tuple[VmState, VmState]] = {
     (VmState.STOPPED, "start"): (VmState.STARTING, VmState.RUNNING),
     (VmState.RUNNING, "stop"): (VmState.STOPPING, VmState.STOPPED),
+    (VmState.RUNNING, "shutdown"): (VmState.STOPPING, VmState.STOPPED),
+    (VmState.RUNNING, "reboot"): (VmState.STOPPING, VmState.RUNNING),
+    (VmState.RUNNING, "reset"): (VmState.STOPPING, VmState.RUNNING),
+    (VmState.RUNNING, "suspend"): (VmState.PAUSING, VmState.PAUSED),
     (VmState.RUNNING, "pause"): (VmState.PAUSING, VmState.PAUSED),
     (VmState.PAUSED, "resume"): (VmState.RESUMING, VmState.RUNNING),
     (VmState.RUNNING, "migrate"): (VmState.MIGRATING, VmState.RUNNING),