Hello, there!

pmb/flasher/frontend.py

@@ -0,0 +1,88 @@
+"""
+Copyright 2017 Oliver Smith
+
+This file is part of pmbootstrap.
+
+pmbootstrap is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+pmbootstrap is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with pmbootstrap.  If not, see <http://www.gnu.org/licenses/>.
+"""
+import logging
+import os
+
+import pmb.flasher
+import pmb.install
+import pmb.chroot.other
+
+
+def kernel(args):
+    # Parse the kernel flavor
+    suffix = "rootfs_" + args.device
+    flavor = args.flavor
+    flavors = pmb.chroot.other.installed_kernel_flavors(args, suffix)
+    if flavor:
+        if flavor not in flavors:
+            raise RuntimeError("No kernel installed with flavor " + flavor + "!" +
+                               " Run 'pmbootstrap flasher list_flavors' to get a list.")
+    elif not len(flavors):
+        raise RuntimeError(
+            "No kernel flavors installed in chroot " + suffix + "!")
+    else:
+        flavor = flavors[0]
+
+    # Generate the paths and run the flasher
+    pmb.flasher.init(args)
+    mnt = "/mnt/rootfs_" + args.device
+    kernel = mnt + "/boot/vmlinuz-" + flavor
+    ramdisk = mnt + "/boot/initramfs-" + flavor
+    if args.action_flasher == "boot":
+        logging.info("(native) boot " + flavor + " kernel")
+        pmb.flasher.run(args, "boot", kernel, ramdisk)
+    else:
+        logging.info("(native) flash kernel '" + flavor + "'")
+        pmb.flasher.run(args, "flash_kernel", kernel, ramdisk)
+
+
+def list_flavors(args):
+    suffix = "rootfs_" + args.device
+    logging.info("(" + suffix + ") installed kernel flavors:")
+    for flavor in pmb.chroot.other.installed_kernel_flavors(args, suffix):
+        logging.info("* " + flavor)
+
+
+def system(args):
+    # Generate system image, install flasher
+    img_path = "/home/user/rootfs/" + args.device + ".img"
+    if not os.path.exists(args.work + "/chroot_native" + img_path):
+        setattr(args, "sdcard", None)
+        pmb.install.install(args, False)
+    pmb.flasher.init(args)
+
+    # Run the flasher
+    logging.info("(native) flash system image")
+    pmb.flasher.run(args, "flash_system", image=img_path)
+
+
+def list_devices(args):
+    pmb.flasher.run(args, "list_devices")
+
+
+def frontend(args):
+    action = args.action_flasher
+    if action in ["boot", "flash_kernel"]:
+        kernel(args)
+    if action == "flash_system":
+        system(args)
+    if action == "list_flavors":
+        list_flavors(args)
+    if action == "list_devices":
+        list_devices(args)