add memory resize

Author: evakhoni

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

@@ -24,6 +24,10 @@ 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:
@@ -56,4 +60,11 @@ def resize_disk(size):
             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

@@ -385,3 +385,9 @@ def get_password(self) -> str:
     def set_disk_size(self, size: str) -> None:
         """Set the disk size for resizing before boot."""
         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.mem = size