implement qemu resize command

Author: evakhoni

packages/jumpstarter-driver-qemu/jumpstarter_driver_qemu/client.py

@@ -1,8 +1,11 @@
 from contextlib import contextmanager
 
+import click
 from jumpstarter_driver_composite.client import CompositeClient
 from jumpstarter_driver_network.adapters import FabricAdapter, NovncAdapter
 
+from jumpstarter.client.decorators import driver_click_group
+
 
 class QemuClient(CompositeClient):
     @property
@@ -17,6 +20,14 @@ def username(self) -> str:
     def password(self) -> str:
         return self.call("get_password")
 
+    def set_disk_size(self, size: str) -> None:
+        """Set the disk size for resizing before boot."""
+        self.call("set_disk_size", size)
+
+    def set_memory_size(self, size: str) -> None:
+        """Set the memory size for next boot."""
+        self.call("set_memory_size", size)
+
     @contextmanager
     def novnc(self):
         with NovncAdapter(client=self.vnc) as url:
@@ -30,3 +41,30 @@ def shell(self):
             connect_kwargs={"password": self.password},
         ) as conn:
             yield conn
+
+    def cli(self):
+        @driver_click_group(self)
+        def base():
+            """QEMU virtual machine operations"""
+            pass
+
+        @base.group()
+        def resize():
+            """Resize QEMU resources"""
+            pass
+
+        @resize.command(name="disk")
+        @click.argument("size")
+        def resize_disk(size):
+            """Resize the root disk (e.g., 20G). Run before power on."""
+            self.set_disk_size(size)
+            click.echo(f"Disk will be resized to {size} on next power on")
+
+        @resize.command(name="memory")
+        @click.argument("size")
+        def resize_memory(size):
+            """Set memory size (e.g., 2G, 4G). Takes effect on next boot."""
+            self.set_memory_size(size)
+            click.echo(f"Memory will be set to {size} on next power on")
+
+        return base

packages/jumpstarter-driver-qemu/jumpstarter_driver_qemu/driver.py

@@ -410,3 +410,17 @@ def _parse_size(self, size: str) -> int:
             return int(TypeAdapter(ByteSize).validate_python(size + "iB" if size[-1] in "kmgtKMGT" else size))
         except (ValidationError, IndexError):
             raise ValueError(f"Invalid size: '{size}'. Use e.g. '20G', '512M', '2T'") from None
+
+    @export
+    @validate_call(validate_return=True)
+    def set_disk_size(self, size: str) -> None:
+        """Set the disk size for resizing before boot."""
+        self._parse_size(size)  # Validate
+        self.disk_size = size
+
+    @export
+    @validate_call(validate_return=True)
+    def set_memory_size(self, size: str) -> None:
+        """Set the memory size for next boot."""
+        self._parse_size(size)  # Validate
+        self.mem = size