kupferbootstrap/chroot/__init__.py
2022-02-17 21:49:24 +01:00

58 lines
1.9 KiB
Python

import click
import logging
import os
from config import config
from wrapper import enforce_wrap
from .base import get_base_chroot, Chroot
from .build import get_build_chroot
from .device import get_device_chroot
# export Chroot class
Chroot = Chroot
@click.command('chroot')
@click.argument('type', required=False, default='build')
@click.argument('arch', required=False, default=None)
def cmd_chroot(type: str = 'build', arch: str = None, enable_crossdirect=True):
"""Open a shell in a chroot"""
chroot_path = ''
if type not in ['base', 'build', 'rootfs']:
raise Exception('Unknown chroot type: ' + type)
enforce_wrap()
if type == 'rootfs':
if arch:
name = 'rootfs_' + arch
else:
raise Exception('"rootfs" without args not yet implemented, sorry!')
# TODO: name = config.get_profile()[...]
chroot_path = os.path.join(config.get_path('chroots'), name)
if not os.path.exists(chroot_path):
raise Exception(f"rootfs {name} doesn't exist")
else:
if not arch:
#TODO: arch = config.get_profile()[...]
arch = 'aarch64'
if type == 'base':
chroot = get_base_chroot(arch)
if not os.path.exists(os.path.join(chroot.path, 'bin')):
chroot.initialize()
chroot.initialized = True
elif type == 'build':
chroot = get_build_chroot(arch, activate=True)
if not os.path.exists(os.path.join(chroot.path, 'bin')):
chroot.initialize()
chroot.initialized = True
chroot.mount_pkgbuilds()
if config.file['build']['crossdirect'] and enable_crossdirect:
chroot.mount_crossdirect()
else:
raise Exception('Really weird bug')
chroot.activate()
logging.debug(f'Starting shell in {chroot.name}:')
chroot.run_cmd('bash', attach_tty=True)
chroot.run_cmd('bash', attach_tty=True)