pick kernel image based on architecture

Sponsored-by: author

1 files changed, 11 insertions, 2 deletions

diff --git a/check-one b/check-one
index c49ec55..4e72640 100755
--- a/check-one
+++ b/check-one
@@ -8,6 +8,15 @@ import tempfile
 import yaml
 
 
+KERNELS = {
+    "amd64": "linux-image-amd64",
+    "arm64": "linux-image-arm64",
+    "armhf": "linux-image-armmp-lpae",
+    "i386": "linux-image-686",
+    "ppc64el": "linux-image-powerpc64le",
+}
+
+
 class Config:
     def __init__(self):
         p = argparse.ArgumentParser()
@@ -15,7 +24,7 @@ class Config:
         p.add_argument("--vmdb", action="store")
         p.add_argument("--tarball-directory", action="store", default=".")
         p.add_argument("--debian-release", action="store")
-        p.add_argument("--grub", action="store", choices=["bios", "uefi"])
+        p.add_argument("--grub", action="store", choices=["bios", "uefi", "ieee1275"])
         p.add_argument("--mklabel", action="store", choices=["msdos", "gpt"])
         p.add_argument("--arch", action="store")
         args = p.parse_args()
@@ -34,7 +43,7 @@ class Config:
         if args.arch is not None:
             debootstrap["arch"] = args.arch
             debootstrap["include"] = debootstrap.get("include", []) + [
-                f"linux-image-{args.arch}"
+                KERNELS[args.arch]
             ]
         if args.mklabel is not None:
             mklabel["mklabel"] = args.mklabel