stages(kickstart): support autopart

This commit implements the `autopart` kickstart option and adds
matching tests.

stages/org.osbuild.kickstart

@@ -170,6 +170,65 @@ SCHEMA = r"""
         "type": "boolean"
       }
     }
+  },
+  "autopart": {
+    "description": "Automatically creates partitions",
+    "type": "object",
+    "properties": {
+      "type": {
+        "description": "Selects one of the predefined automatic partitioning schemes you want to use",
+        "type": "string",
+         "enum": ["lvm", "btrfs", "plain", "thinp"]
+      },
+      "fstype": {
+        "description": "Specify a supported file system (such as ext4 or xfs) to replace the default when doing automatic partitioning",
+        "type": "string"
+      },
+      "nolvm": {
+        "description": "Do not use LVM or Btrfs for automatic partitioning. This option is equal to --type=plain",
+        "type": "boolean"
+      },
+      "encrypted": {
+        "description": "Encrypts all partitions",
+        "type": "boolean"
+      },
+      "passphrase": {
+        "description": "Provides a default system-wide passphrase for all encrypted devices",
+        "type": "string"
+      },
+      "escrowcert": {
+        "description": "Stores data encryption keys of all encrypted volumes as files in /root, encrypted using the X.509 certificate from the URL specified",
+        "type": "string"
+      },
+      "backuppassphrase": {
+        "description": "Adds a randomly-generated passphrase to each encrypted volume",
+        "type": "string"
+      },
+      "cipher": {
+        "description": "Specifies which type of encryption will be used if the Anaconda default aes-xts-plain64 is not satisfactory",
+        "type": "string"
+      },
+      "luks-version": {
+        "description": "Specifies which version of LUKS should be used to encrypt the system",
+        "type": "string"
+      },
+      "pbkdf": {
+        "description": "Sets Password-Based Key Derivation Function (PBKDF) algorithm for the LUKS keyslot",
+        "type": "string"
+      },
+      "pbkdf-memory": {
+        "description": "Sets the memory cost for PBKDF",
+        "type": "integer"
+      },
+      "pbkdf-time": {
+        "description": "Sets the number of milliseconds to spend with PBKDF passphrase processing",
+        "type": "integer"
+      },
+      "pbkdf-iterations": {
+        "description": "Sets the number of iterations for passphrase processing directly",
+        "type": "integer"
+      }
+    }
   }
 }
 """
@@ -259,6 +318,26 @@ def make_clearpart(options: Dict) -> str:
     return cmd
 
 
+def make_autopart(options: Dict) -> str:
+    autopart = options.get("autopart")
+    if autopart is None:
+        return ""
+    cmd = "autopart"
+    # TODO: make pkdf-{time,iterations} mutally exclusive at the
+    # schema level
+    for key in ["type", "fstype", "nolvm", "encrypted", "passphrase",
+                 "escrowcert", "backuppassphrase", "cipher", "luks-version",
+                 "pbkdf", "pbkdf-memory", "pbkdf-time", "pbkdf-iterations"]:
+        if not key in autopart:
+            continue
+        val = autopart[key]
+        if type(val) is bool:
+            cmd += f" --{key}"
+        else:
+            cmd += f" --{key}={val}"
+    return cmd
+
+
 def main(tree, options):
     path = options["path"].lstrip("/")
     ostree = options.get("ostree")
@@ -301,6 +380,9 @@ def main(tree, options):
     clearpart = make_clearpart(options)
     if clearpart:
         config += [clearpart]
+    autopart = make_autopart(options)
+    if autopart:
+        config += [autopart]
 
     target = os.path.join(tree, path)
     base = os.path.dirname(target)

stages/test/test_kickstart.py

@@ -16,6 +16,10 @@
       "timezone": "UTC",
       },
      "lang en_US.UTF-8\nkeyboard us\ntimezone UTC"),
+    ({"autopart": {}},"autopart"),
+    ({"autopart": {"type": "plain"}},"autopart --type=plain"),
+    ({"autopart": {"type": "lvm"}},"autopart --type=lvm"),
+    ({"autopart": {"nolvm": True}},"autopart --nolvm"),
 ])
 def test_kickstart(tmp_path, test_input, expected):
     ks_stage_path = os.path.join(os.path.dirname(__file__), "../org.osbuild.kickstart")