diff --git a/blend-inst b/blend-inst index d41c1fa..f054018 100755 --- a/blend-inst +++ b/blend-inst @@ -10,6 +10,7 @@ import os import sys import json +import yaml import signal import subprocess @@ -31,12 +32,19 @@ def preexec(): signal.signal(signal.SIGQUIT, signal.SIG_IGN) -def exec(cmd): +def exec(cmd, input=None): if testing: - print(' '.join(cmd)) + if input != None: + print(' '.join(cmd), '<--', input) + else: + print(' '.join(cmd)) else: - subprocess.call(cmd, shell=False, stdout=sys.stdout, - stderr=sys.stderr, preexec_fn=preexec) + if input != None: + subprocess.run(cmd, shell=False, stdout=sys.stdout, + stderr=sys.stderr, preexec_fn=preexec, input=input.encode()).returncode + else: + subprocess.run(cmd, shell=False, stdout=sys.stdout, + stderr=sys.stderr, preexec_fn=preexec).returncode def exec_chroot(cmd): @@ -107,10 +115,7 @@ def format_and_mount(config, mountpoint, filesystem, blockdevice): if mountpoint == 'System': mountpoint = '/mnt/' elif mountpoint == 'Boot': - if efi: - mountpoint = '/mnt/boot/efi/' - else: - mountpoint = '/mnt/boot/' + mountpoint = '/mnt/boot/' elif mountpoint == 'User': mountpoint = '/mnt/home/' else: @@ -133,60 +138,67 @@ def inst_partition(config): mode = config['partition']['mode'] efi = config['partition']['efi'] partitions = config['partition']['partitions'] - - # Delete partition table - exec(['dd', 'if=/dev/zero', f'of={device}', 'bs=512', 'count=1']) - exec(['sync']) + password = config['partition']['password'] if mode == 'Auto': + # Delete partition table + exec(['wipefs', '-a', device]) + exec(['sync']) + if efi: # Create GPT label exec(['parted', '-s', device, 'mklabel', 'gpt']) # Create EFI partition - exec(['parted', '-s', device, 'mkpart', 'fat32', '0', '500']) + exec(['parted', '-s', device, 'mkpart', 'fat32', '0%', '512MiB']) else: # Create msdos label exec(['parted', '-s', device, 'mklabel', 'msdos']) # Create boot partition exec(['parted', '-s', device, 'mkpart', - 'primary', 'ext4', '1MIB', '600MIB']) + 'primary', 'ext4', '0%', '512MiB']) # Create root partition exec(['parted', '-s', device, 'mkpart', - 'primary', 'ext4', '700MB', '100%']) + 'primary', 'btrfs', '513MiB', '100%']) + + global orig_main_partition if 'nvme' in device or 'mmcblk' in device: # Format EFI/boot partition if efi: exec(['mkfs.vfat', '-F32', f'{device}p1']) else: - exec(['mkfs.ext4', f'{device}p1']) + exec(['mkfs.ext4', '-F', f'{device}p1']) + root_partition = f'{device}p2' + orig_main_partition = root_partition + if password != '': + exec(['cryptsetup', '-q', 'luksFormat', root_partition], input=f'{password}\n') + exec(['cryptsetup', 'open', root_partition, 'new_root', '-'], input=f'{password}\n') + root_partition = '/dev/mapper/new_root' # Format root partition - exec(['mkfs.ext4', f'{device}p2']) + exec(['mkfs.btrfs', '-F', root_partition]) # Mount partitions - mount(f'{device}p2', '/mnt') - if efi: - mkdir('/mnt/boot/efi') - mount(f'{device}p1', '/mnt/boot/efi') - else: - mkdir('/mnt/boot') - mount(f'{device}p1', '/mnt/boot') + mount(root_partition, '/mnt') + mkdir('/mnt/boot') + mount(f'{device}p1', '/mnt/boot') else: # Format EFI/boot partition if efi: exec(['mkfs.vfat', '-F32', f'{device}1']) else: - exec(['mkfs.ext4', f'{device}1']) + exec(['mkfs.ext4', '-F', f'{device}1']) + root_partition = f'{device}2' + orig_main_partition = root_partition + if password != '': + exec(['cryptsetup', '-q', 'luksFormat', root_partition], input=f'{password}\n') + exec(['cryptsetup', 'open', root_partition, 'new_root', '-'], input=f'{password}\n') + root_partition = '/dev/mapper/new_root' # Format root partition - exec(['mkfs.ext4', f'{device}2']) + exec(['mkfs.btrfs', '-F', root_partition]) # Mount partitions - mount(f'{device}2', '/mnt') - if efi: - mkdir('/mnt/boot/efi') - mount(f'{device}1', '/mnt/boot/efi') - else: - mkdir('/mnt/boot') - mount(f'{device}1', '/mnt/boot') + mount(root_partition, '/mnt') + mkdir('/mnt/boot') + mount(f'{device}1', '/mnt/boot') else: partitions.sort(key=lambda x: len(x[1])) for p in partitions: @@ -199,7 +211,7 @@ def inst_partition(config): print( "Invalid manual partitioning configuration. There must be a 'System' partition.") sys.exit(1) - if '/mnt/boot/' not in mountpoints and '/mnt/boot/efi/' not in mountpoints: + if '/mnt/boot/' not in mountpoints: print() print( "Invalid manual partitioning configuration. There must be a 'Boot' partition.") @@ -238,33 +250,41 @@ def inst_setup_base(config): # Refresh package lists, pacman-key --init exec_chroot(['pacman-key', '--init']) exec_chroot(['pacman-key', '--populate', 'archlinux']) + # Add akshara and encrypt hooks + exec_chroot(['bash', '-c', 'echo "MODULES=()" > /etc/mkinitcpio.conf']) + exec_chroot(['bash', '-c', 'echo "BINARIES=()" >> /etc/mkinitcpio.conf']) + exec_chroot(['bash', '-c', 'echo "FILES=()" >> /etc/mkinitcpio.conf']) + if config['partition']['password'] == '': + exec_chroot(['bash', '-c', 'echo "HOOKS=(base udev akshara autodetect keyboard keymap modconf block filesystems fsck)" >> /etc/mkinitcpio.conf']) + else: + exec_chroot(['bash', '-c', 'echo "HOOKS=(base udev akshara autodetect keyboard keymap consolefont modconf block encrypt filesystems fsck)" >> /etc/mkinitcpio.conf']) # Install linux-zen exec_chroot(['pacman', '-Sy', '--noconfirm', 'linux-zen', 'xorriso']) # Remove jade-gui exec_chroot(['pacman', '-Rn', '--noconfirm', 'jade-gui']) - # Install jade-gui-postinst - exec_chroot(['pacman', '-S', '--noconfirm', 'jade-gui-postinst']) def inst_bootloader(config): # Install bootloader if 'NVIDIA' in subprocess.check_output(['lspci']).decode('utf-8'): - exec_chroot(['pacman', '-S', '--noconfirm', 'nvidia-dkms']) exec_chroot( - ['bash', '-c', 'echo -e \'\\nGRUB_CMDLINE_LINUX_DEFAULT="${GRUB_CMDLINE_LINUX_DEFAULT} vt.global_cursor_default=0 nvidia_drm.modeset=1 splash"\' >> /etc/default/grub']) + ['bash', '-c', 'echo -e \'\\nGRUB_CMDLINE_LINUX_DEFAULT="${GRUB_CMDLINE_LINUX_DEFAULT} nvidia_drm.modeset=1 splash"\' >> /etc/default/grub']) mkdir('/mnt/etc/udev/rules.d') exec_chroot(['ln', '-s', '/dev/null', '/etc/udev/rules.d/61-gdm.rules']) else: exec_chroot( - ['bash', '-c', 'echo -e \'\\nGRUB_CMDLINE_LINUX_DEFAULT="${GRUB_CMDLINE_LINUX_DEFAULT} vt.global_cursor_default=0 splash"\' >> /etc/default/grub']) + ['bash', '-c', 'echo -e \'\\nGRUB_CMDLINE_LINUX_DEFAULT="${GRUB_CMDLINE_LINUX_DEFAULT} splash"\' >> /etc/default/grub']) if config['partition']['mode'] == 'Auto': + if config['partition']['password'] != '': + exec_chroot( + ['bash', '-c', 'echo -e \'\\nGRUB_CMDLINE_LINUX_DEFAULT="${GRUB_CMDLINE_LINUX_DEFAULT} cryptdevice=UUID=' + subprocess.check_output(['blkid', '-s', 'UUID', '-o', 'value', orig_main_partition]).decode('UTF-8').strip() + ':root root=/dev/mapper/root"\' >> /etc/default/grub']) exec_chroot( - ['bash', '-c', 'echo -e \'\\nGRUB_TIMEOUT=0\' >> /etc/default/grub']) + ['bash', '-c', 'echo -e \'\\nGRUB_TIMEOUT=3\' >> /etc/default/grub']) if config['bootloader']['type'] == 'grub-efi': - exec_chroot(['grub-install', '--target=x86_64-efi', f'--efi-directory={config["bootloader"]["location"]}', + exec_chroot(['grub-install', '--target=x86_64-efi', f'--efi-directory=/boot', '--bootloader-id=blend', '--removable']) - exec_chroot(['grub-install', '--target=x86_64-efi', f'--efi-directory={config["bootloader"]["location"]}', + exec_chroot(['grub-install', '--target=x86_64-efi', f'--efi-directory=/boot', '--bootloader-id=blend']) else: exec_chroot(['grub-install', '--target=i386-pc', @@ -322,6 +342,14 @@ def inst_users(config): # Akshara def inst_akshara(config): + exec_chroot( + ['rm', '-f', '/etc/locale.gen']) + for locale in ['en_US.UTF-8', 'de_DE.UTF-8', 'fr_FR.UTF-8', 'es_ES.UTF-8', 'zh_CN.UTF-8', 'ja_JP.UTF-8', 'ru_RU.UTF-8', 'ar_EG.UTF-8']: + exec_chroot( + ['bash', '-c', f'echo "{locale} UTF-8" >> /etc/locale.gen']) + exec_chroot( + ['locale-gen']) + system_config = { 'repo': 'https://pkg-repo.blendos.co', 'impl': 'http://github.com/blend-os/tracks/raw/main', @@ -340,7 +368,7 @@ def inst_akshara(config): ] if testing == False: - with open('/system.yaml', 'w') as system_config_file: + with open('/mnt/system.yaml', 'w') as system_config_file: yaml.dump(system_config, system_config_file) else: print(system_config)