kupferbootstrap/chroot.py

import logging
import subprocess
import os
from config import config
from distro import get_base_distros, RepoInfo


def get_chroot_path(chroot_name, override_basepath: str = None) -> str:
    base_path = config.get_path('chroots') if not override_basepath else override_basepath
    return os.path.join(base_path, chroot_name)


def create_chroot(
    chroot_name,
    arch='aarch64',
    packages=['base'],
    pacman_conf='/app/local/etc/pacman.conf',
    extra_repos: dict[str, RepoInfo] = {},
    chroot_base_path: str = None,
):
    base_chroot = f'base_{arch}'
    chroot_path = get_chroot_path(chroot_name, override_basepath=chroot_base_path)
    base_distro = get_base_distros()[arch]
    pacman_conf_target = chroot_path + '/etc/pacman.conf'

    # copy base_chroot instead of creating from scratch every time
    if not (chroot_base_path or chroot_name == base_chroot):
        # only install base package in base_chroot
        base_chroot_path = create_chroot(base_chroot, arch=arch)
        logging.info(f'Copying {base_chroot} chroot to {chroot_name}')
        result = subprocess.run([
            'rsync',
            '-a',
            '--delete',
            '-q',
            '-W',
            '-x',
            f'{base_chroot_path}/',
            f'{chroot_path}/',
        ])
        if result.returncode != 0:
            logging.fatal('Failed to sync chroot copy')
            exit(1)

    os.makedirs(chroot_path + '/etc', exist_ok=True)

    conf_text = base_distro.get_pacman_conf(extra_repos)
    with open(pacman_conf_target, 'w') as file:
        file.write(conf_text)

    result = subprocess.run([
        'pacstrap',
        '-C',
        pacman_conf_target,
        '-c',
        '-G',
        chroot_path,
    ] + packages + [
        '--needed',
        '--overwrite=*',
        '-yyuu',
    ],
                            capture_output=True)
    if result.returncode != 0:
        logging.debug(result.stdout.decode())
        raise Exception(f'Failed to install chroot "{chroot_name}":' + '\n' + result.stderr.decode())
    return chroot_path


def run_chroot_cmd(
    script: str,
    chroot_name,
    chroot_base_path: str = None,
):
    chroot_path = get_chroot_path(chroot_name, override_basepath=chroot_base_path)
    result = subprocess.run([
        'arch-chroot',
        chroot_path,
        '/bin/bash',
        '-c',
        script,
    ])
    return result


def create_chroot_user(
    chroot_name,
    chroot_base_path: str = None,
    user='kupfer',
    password='123456',
    groups=['network', 'video', 'audio', 'optical', 'storage', 'input', 'scanner', 'games', 'lp', 'rfkill', 'wheel'],
):
    install_script = f'''
        set -e
        if ! id -u "{user}" >/dev/null 2>&1; then
          useradd -m {user}
        fi
        usermod -a -G {",".join(groups)} {user}
        chown {user}:{user} /home/{user} -R
    '''
    if password:
        install_script += f'echo "{user}:{password}" | chpasswd'
    else:
        install_script += 'passwd'
    result = run_chroot_cmd(install_script, chroot_name=chroot_name, chroot_base_path=chroot_base_path)
    if result.returncode != 0:
        raise Exception('Failed to setup user')