feat: POST /<vm_id>/rename endpoint

Author: skl1017

backend/app/core/policies.py

@@ -69,6 +69,10 @@
         "policy": "vms:stop",
         "description": "Allows the user to stop virtual machines.",
     },
+    {
+        "policy": "vms:rename",
+        "description": "Allows the user rename a vm.",
+    },
     {
         "policy": "vms:connect",
         "description": "Allows the user to connect to a virtual machine via the dashboard tunnel.",

backend/app/models/vm.py

@@ -33,6 +33,10 @@ class VmCreateXML(VmBase):
     id: UUID
 
 
+class VmRename(BaseModel):
+    name: str
+
+
 class VmCredentialCreateRequest(BaseModel):
     name: str = Field(min_length=1)
     password: Optional[str] = None

backend/app/routes/vm.py

@@ -4,7 +4,7 @@
 from fastapi.responses import Response
 from sqlmodel import Session
 from app.models.user_management import MissingPoliciesResponse
-from app.models.vm import VmCreate, VmRead, VmCredentialCreateRequest, VmCredentialRead, RecoverableVm, RecoverableVmCreate
+from app.models.vm import VmCreate, VmRead, VmCredentialCreateRequest, VmCredentialRead, RecoverableVm, RecoverableVmCreate, VmRename
 from app.services.vm_service import VmService
 from app.services.vm_screenshot import capture_screenshot
 from app.utils.auth import require_policy, decode_access_token, user_has_policy
@@ -137,6 +137,16 @@ def restart_vm(vm_id: str):
     return vm
 
 
+@router.post("/{vm_id}/rename",
+             status_code=status.HTTP_200_OK,
+             response_model=VmRead,
+             dependencies=[Depends(require_policy("vms:rename"))],
+             responses={403: {"model": MissingPoliciesResponse}})
+def rename_vm(vm_id: str, vm_rename: VmRename):
+    vm = VmService.rename_vm(vm_id, vm_rename)
+    return vm
+
+
 @router.post("/{vm_id}/start",
              status_code=status.HTTP_200_OK,
              response_model=VmRead,

backend/app/services/vm_service.py

@@ -6,7 +6,7 @@
 from app.utils.vm import wait_for_state
 from typing import Optional
 from app.core.constants import VMS_DIR, IMAGES_DIR, VM_STATE_NAMES
-from app.models.vm import VmCreate, VmRead, VmCredentialCreateRequest, RecoverableVm, RecoverableVmCreate, VmCreateXML
+from app.models.vm import VmCreate, VmRead, VmCredentialCreateRequest, RecoverableVm, RecoverableVmCreate, VmCreateXML, VmRename
 from app.models.image import ImageRead
 from app.core.xml_builder import build_xml
 from app.core.config import QEMUConfig, engine
@@ -422,7 +422,10 @@ def create_vm(vm_create: VmCreate):
         return vm
 
     @staticmethod
-    def _auto_pick_node(required_mem: int, required_vcpus: int, required_disk: int):
+    def _auto_pick_node(
+            required_mem: int,
+            required_vcpus: int,
+            required_disk: int):
         """Pick the best node, prioritizing master. Returns slave UUID or None for master."""
         from app.services.host_service import HostService
         from app.services.slave_service import SlaveService
@@ -788,3 +791,12 @@ def duplicate_vm(vm_id: str):
                 )
 
             return Vm.get(str(duplicate_vm.id))
+
+    @staticmethod
+    def rename_vm(vm_id: str, vm_rename: VmRename):
+        with Session(engine) as session:
+            vm = VmService._get_vm_or_404(session, vm_id)
+            vm.name = vm_rename.name
+            session.add(vm)
+            session.commit()
+            return Vm.get(str(vm.id))