init: Set/get single disk

pkgs/clan-cli/clan_cli/api/disk.py

@@ -0,0 +1,65 @@
+from clan_cli.api import API
+from clan_cli.inventory import (
+    ServiceMeta,
+    ServiceSingleDisk,
+    ServiceSingleDiskRole,
+    ServiceSingleDiskRoleDefault,
+    SingleDiskConfig,
+    load_inventory_eval,
+    load_inventory_json,
+    save_inventory,
+)
+
+
+def get_instance_name(machine_name: str) -> str:
+    return f"{machine_name}-single-disk"
+
+
+@API.register
+def set_single_disk_uuid(
+    base_path: str,
+    machine_name: str,
+    disk_uuid: str,
+) -> None:
+    """
+    Set the disk UUID of single disk machine
+    """
+    inventory = load_inventory_json(base_path)
+
+    instance_name = get_instance_name(machine_name)
+
+    single_disk_config: ServiceSingleDisk = ServiceSingleDisk(
+        meta=ServiceMeta(name=instance_name),
+        roles=ServiceSingleDiskRole(
+            default=ServiceSingleDiskRoleDefault(
+                config=SingleDiskConfig(device=disk_uuid)
+            )
+        ),
+    )
+
+    inventory.services.single_disk[instance_name] = single_disk_config
+
+    save_inventory(
+        inventory,
+        base_path,
+        f"Set disk UUID: '{disk_uuid}' on machine: '{machine_name}'",
+    )
+
+
+@API.register
+def get_single_disk_uuid(
+    base_path: str,
+    machine_name: str,
+) -> str | None:
+    """
+    Get the disk UUID of single disk machine
+    """
+    inventory = load_inventory_eval(base_path)
+
+    instance_name = get_instance_name(machine_name)
+
+    single_disk_config: ServiceSingleDisk = inventory.services.single_disk[
+        instance_name
+    ]
+
+    return single_disk_config.roles.default.config.device