101 lines
3 KiB
Python
101 lines
3 KiB
Python
import logging
|
|
import subprocess
|
|
import os
|
|
from config import config
|
|
from distro import get_base_distros, RepoInfo
|
|
from shlex import quote as shell_quote
|
|
|
|
|
|
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: str,
|
|
arch: str,
|
|
packages: list[str] = ['base'],
|
|
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)
|
|
|
|
logging.info(f'Installing packages to {chroot_name}: {", ".join(packages)}')
|
|
|
|
result = subprocess.run([
|
|
'pacstrap',
|
|
'-C',
|
|
pacman_conf_target,
|
|
'-c',
|
|
'-G',
|
|
chroot_path,
|
|
] + packages + [
|
|
'--needed',
|
|
'--overwrite=*',
|
|
'-yyuu',
|
|
])
|
|
if result.returncode != 0:
|
|
raise Exception(f'Failed to install chroot "{chroot_name}"')
|
|
return chroot_path
|
|
|
|
|
|
def run_chroot_cmd(script: str, chroot_path: str, env: dict[str, str] = {}):
|
|
|
|
env_cmd = ['/usr/bin/env'] + [f'{shell_quote(key)}={shell_quote(value)}' for key, value in env.items()]
|
|
result = subprocess.run(['arch-chroot', chroot_path] + env_cmd + [
|
|
'/bin/bash',
|
|
'-c',
|
|
script,
|
|
])
|
|
return result
|
|
|
|
|
|
def create_chroot_user(
|
|
chroot_path: str,
|
|
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 += 'echo "Set user password:" && passwd'
|
|
result = run_chroot_cmd([install_script], chroot_path=chroot_path)
|
|
if result.returncode != 0:
|
|
raise Exception('Failed to setup user')
|