63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
import logging
|
|
import subprocess
|
|
import os
|
|
import shutil
|
|
from config import config
|
|
|
|
|
|
def get_chroot_path(chroot_name, override_basepath: str = None) -> str:
|
|
base_path = config.file['paths']['chroots'] if not override_basepath else override_basepath
|
|
return os.path.join(base_path, chroot_name)
|
|
|
|
|
|
def create_chroot(chroot_name, packages=['base'], pacman_conf='/app/local/etc/pacman.conf', extra_repos={}, chroot_base_path: str = None):
|
|
chroot_path = get_chroot_path(chroot_name, override_basepath=chroot_base_path)
|
|
pacman_conf_target = chroot_path + '/etc/pacman.conf'
|
|
|
|
os.makedirs(chroot_path + '/etc', exist_ok=True)
|
|
shutil.copyfile(pacman_conf, pacman_conf_target)
|
|
|
|
extra_conf = ''
|
|
for repo_name, repo_options in extra_repos.items():
|
|
extra_conf += f'\n\n[{repo_name}]\n'
|
|
extra_conf += '\n'.join(['%s = %s' % (name, value) for name, value in repo_options.items()])
|
|
with open(pacman_conf_target, 'a') as file:
|
|
file.write(extra_conf)
|
|
|
|
result = subprocess.run(['pacstrap', '-C', pacman_conf_target, '-c', '-G', chroot_path] + packages + [
|
|
'--needed',
|
|
'--overwrite=*',
|
|
'-yyuu',
|
|
])
|
|
if result.returncode != 0:
|
|
raise Exception('Failed to install chroot')
|
|
return chroot_path
|
|
|
|
|
|
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'],
|
|
):
|
|
chroot_path = get_chroot_path(chroot_name, override_basepath=chroot_base_path)
|
|
|
|
install_script = '\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',
|
|
chroot_path,
|
|
'/bin/bash',
|
|
'-c',
|
|
install_script,
|
|
])
|
|
if result.returncode != 0:
|
|
logging.fatal('Failed to setup user')
|
|
exit(1)
|