import os

from scale_build.bootstrap.bootstrapdir import PackageBootstrapDir
from scale_build.utils.run import run


class BootstrapMixin:
    def setup_chroot_basedir(self):
        self.logger.debug('Restoring CHROOT_BASEDIR for runs...')
        os.makedirs(self.tmpfs_path, exist_ok=True)
        if self.tmpfs:
            run(['mount', '-t', 'tmpfs', '-o', f'size={self.tmpfs_size}G', 'tmpfs', self.tmpfs_path])
        PackageBootstrapDir().restore_cache(self.chroot_base_directory)