stages(kickstart): implement reboot option

stages/org.osbuild.kickstart

@@ -170,6 +170,19 @@ SCHEMA = r"""
         "type": "boolean"
       }
     }
+  },
+  "reboot": {
+    "description": "Reboot after the installation is successfully completed",
+    "type": "object",
+    "properties": {
+      "eject": {
+        "description": "Attempt to eject the installation media before rebooting",
+        "type": "boolean"
+      }
+      "kexec": {
+        "description": "Use this option to reboot into the new system using the kexec",
+        "type": "boolean"
+      }
   }
 }
 """
@@ -302,6 +315,15 @@ def main(tree, options):
     if clearpart:
         config += [clearpart]
 
+    reboot = options.get("reboot", None)
+    if reboot is not None:
+        cmd = "reboot"
+        if reboot.get("eject"):
+            cmd += " --eject"
+        if reboot.get("kexec"):
+            cmd += " --kexec"
+        config += [cmd]
+
     target = os.path.join(tree, path)
     base = os.path.dirname(target)
     os.makedirs(base, exist_ok=True)

stages/test/test_kickstart.py

@@ -87,6 +87,9 @@
       },
         "lang en_US.UTF-8\nkeyboard us\ntimezone UTC\nzerombr\nclearpart --all --drives=sd*|hd*|vda,/dev/vdc",
      ),
+    ({"reboot": {}}, "reboot"),
+    ({"reboot": {"eject": True}}, "reboot --eject"),
+    ({"reboot": {"kexec": True}}, "reboot --kexec"),
 ])
 def test_kickstart(tmp_path, test_input, expected):
     ks_stage_path = os.path.join(os.path.dirname(__file__), "../org.osbuild.kickstart")