WIP: Add rootfs building and flashing and other misc commands

image.py

@@ -0,0 +1,198 @@
+import atexit
+from logging import root
+import os
+import shutil
+import signal
+import subprocess
+import time
+import click
+from logger import *
+
+devices = {
+    'oneplus-enchilada': ['sdm845-oneplus-enchilada'],
+    'xiaomi-beryllium-ebbg': ['sdm845-xiaomi-beryllium-ebbg'],
+    'xiaomi-beryllium-tianma': ['sdm845-xiaomi-beryllium-tianma'],
+}
+
+flavours = {
+    'barebone': [],
+    'phosh': [],
+    'plasma-mobile': [],
+}
+
+
+def get_device_and_flavour() -> tuple[str, str]:
+    if not os.path.exists('.device'):
+        logging.fatal(
+            f'Please set the device using \'kupferbootstrap image device ...\'')
+        exit(1)
+    if not os.path.exists('.flavour'):
+        logging.fatal(
+            f'Please set the flavour using \'kupferbootstrap image flavour ...\'')
+        exit(1)
+
+    with open('.device', 'r') as file:
+        device = file.read()
+    with open('.flavour', 'r') as file:
+        flavour = file.read()
+
+    return (device, flavour)
+
+
+def get_image_name(device, flavour):
+    return f'{device}-{flavour}-rootfs.img'
+
+
+def mount_rootfs_image(path):
+    rootfs_mount = '/mnt/kupfer/rootfs'
+    if not os.path.exists(rootfs_mount):
+        os.makedirs(rootfs_mount)
+
+    def umount():
+        subprocess.run(['umount', '-lc', rootfs_mount],
+                       stderr=subprocess.DEVNULL)
+    atexit.register(umount)
+
+    result = subprocess.run(['mount',
+                             '-o', 'loop',
+                             path,
+                             rootfs_mount])
+    if result.returncode != 0:
+        logging.fatal(f'Failed to loop mount {path} to {rootfs_mount}')
+        exit(1)
+
+    return rootfs_mount
+
+
+@click.group(name='image')
+def cmd_image():
+    pass
+
+
+@click.command(name='device')
+@verbose_option
+@click.argument('device')
+def cmd_device(verbose, device):
+    setup_logging(verbose)
+
+    for key in devices.keys():
+        if '-'.join(key.split('-')[1:]) == device:
+            device = key
+            break
+
+    if device not in devices:
+        logging.fatal(
+            f'Unknown device {device}. Pick one from:\n{", ".join(devices.keys())}')
+        exit(1)
+
+    logging.info(f'Setting device to {device}')
+
+    with open('.device', 'w') as file:
+        file.write(device)
+
+
+@click.command(name='flavour')
+@verbose_option
+@click.argument('flavour')
+def cmd_flavour(verbose, flavour):
+    setup_logging(verbose)
+
+    if flavour not in flavours:
+        logging.fatal(
+            f'Unknown flavour {flavour}. Pick one from:\n{", ".join(flavours.keys())}')
+        exit(1)
+
+    logging.info(f'Setting flavour to {flavour}')
+
+    with open('.flavour', 'w') as file:
+        file.write(flavour)
+
+
+@click.command(name='build')
+@verbose_option
+def cmd_build(verbose):
+    setup_logging(verbose)
+
+    device, flavour = get_device_and_flavour()
+    image_name = get_image_name(device, flavour)
+
+    shutil.copyfile('/app/src/pacman.conf', '/app/src/pacman_copy.conf')
+    with open('/app/src/pacman_copy.conf', 'a') as file:
+        file.write(
+            '\n\n[main]\nServer = https://gitlab.com/kupfer/packages/prebuilts/-/raw/main/$repo')
+        file.write(
+            '\n\n[device]\nServer = https://gitlab.com/kupfer/packages/prebuilts/-/raw/main/$repo')
+
+    if not os.path.exists(image_name):
+        result = subprocess.run(['fallocate',
+                                 '-l', '4G',
+                                 image_name])
+        if result.returncode != 0:
+            logging.fatal(f'Failed to allocate {image_name}')
+            exit(1)
+
+        result = subprocess.run(['mkfs.ext4',
+                                 '-L', 'kupfer',
+                                 image_name])
+        if result.returncode != 0:
+            logging.fatal(f'Failed to create ext4 filesystem on {image_name}')
+            exit(1)
+
+    rootfs_mount = mount_rootfs_image(image_name)
+
+    result = subprocess.run(['pacstrap',
+                             '-C', '/app/src/pacman_copy.conf',
+                             '-c',
+                             '-G',
+                             rootfs_mount,
+                             'base', 'base-kupfer']
+                            + devices[device]
+                            + flavours[flavour]
+                            + ['--needed', '--overwrite=*', '-yyuu'])
+
+    with open(os.path.join(rootfs_mount, 'install'), 'w') as file:
+        user = 'kupfer'
+        password = '123456'
+        groups = ['network', 'video', 'audio', 'optical', 'storage',
+                  'input', 'scanner', 'games', 'lp', 'rfkill', 'wheel']
+        file.write('\n'.join([
+            f'if ! id -u "{user}" >/dev/null 2>&1; then',
+            f'  useradd -m {user}',
+            f'fi',
+            f'usermod -a -G {",".join(groups)} {user}',
+            f'echo "{user}:{password}" | chpasswd',
+            f'chown {user}:{user} /home/{user} -R',
+        ]))
+    result = subprocess.run(['arch-chroot',
+                             rootfs_mount,
+                             '/bin/bash', '/install'])
+    os.unlink(os.path.join(rootfs_mount, 'install'))
+    if result.returncode != 0:
+        logging.fatal('Failed to setup user')
+        exit(1)
+
+
+"""
+This doesn't work, because the mount isn't passed through to the real host
+"""
+
+"""
+@click.command(name='inspect')
+@verbose_option
+def cmd_inspect(verbose):
+    setup_logging(verbose)
+
+    device, flavour = get_device_and_flavour()
+    image_name = get_image_name(device, flavour)
+
+    rootfs_mount = mount_rootfs_image(image_name)
+
+    logging.info(f'Inspect the rootfs image at {rootfs_mount}')
+
+    signal.pause()
+"""
+
+cmd_image.add_command(cmd_device)
+cmd_image.add_command(cmd_flavour)
+cmd_image.add_command(cmd_build)
+# cmd_image.add_command(cmd_inspect)