WIP: Add rootfs building and flashing and other misc commands

boot.py

@@ -0,0 +1,54 @@
+import os
+import urllib.request
+from image import get_device_and_flavour, get_image_name
+from logger import *
+import click
+import subprocess
+
+
+FASTBOOT = 'fastboot'
+
+JUMPDRIVE = 'jumpdrive'
+jumpdrive_version = '0.8'
+
+boot_strategies = {
+    'oneplus-enchilada': FASTBOOT,
+    'xiaomi-beryllium-ebbg': FASTBOOT,
+    'xiaomi-beryllium-tianma': FASTBOOT,
+}
+
+
+def dump_bootimg(image_name: str) -> str:
+    path = '/tmp/boot.img'
+    result = subprocess.run(['debugfs',
+                             image_name,
+                             '-R', f'dump /boot/boot.img {path}'])
+    if result.returncode != 0:
+        logging.fatal(f'Faild to dump boot.img')
+        exit(1)
+    return path
+
+
+@click.command(name='boot', help=f'Leave TYPE empty or choose \'{JUMPDRIVE}\'')
+@verbose_option
+@click.argument('type', required=False)
+def cmd_boot(verbose, type):
+    setup_logging(verbose)
+
+    device, flavour = get_device_and_flavour()
+    image_name = get_image_name(device, flavour)
+    strategy = boot_strategies[device]
+
+    if strategy == FASTBOOT:
+        if type == JUMPDRIVE:
+            file = f'boot-{device}.img'
+            path = os.path.join('/var/cache/jumpdrive', file)
+            urllib.request.urlretrieve(
+                f'https://github.com/dreemurrs-embedded/Jumpdrive/releases/download/{jumpdrive_version}/{file}', path)
+        else:
+            path = dump_bootimg(image_name)
+
+        result = subprocess.run(['fastboot', 'boot', path])
+        if result.returncode != 0:
+            logging.fatal(f'Failed to boot {path} using fastboot')
+            exit(1)