import logging
import os

from glob import glob
from shutil import rmtree

from constants import Arch
from exec.cmd import run_root_cmd
from exec.file import makedir, root_makedir
from config import config

from .abstract import Chroot, get_chroot
from .helpers import base_chroot_name


class BaseChroot(Chroot):

    copy_base: bool = False

    def create_rootfs(self, reset, pacman_conf_target, active_previously):
        if reset:
            logging.info(f'Resetting {self.name}')
            for dir in glob(os.path.join(self.path, '*')):
                rmtree(dir)
        makedir(config.get_path('chroots'))
        root_makedir(self.get_path())

        self.write_pacman_conf()
        self.mount_pacman_cache()

        logging.info(f'Pacstrapping chroot {self.name}: {", ".join(self.base_packages)}')

        result = run_root_cmd([
            'pacstrap',
            '-C',
            pacman_conf_target,
            '-G',
            self.path,
        ] + self.base_packages + [
            '--needed',
            '--overwrite=*',
            '-yyuu',
        ])
        if result.returncode != 0:
            raise Exception(f'Failed to initialize chroot "{self.name}"')
        self.initialized = True


def get_base_chroot(arch: Arch) -> BaseChroot:
    name = base_chroot_name(arch)
    default = BaseChroot(name, arch, copy_base=False, initialize=False)
    chroot = get_chroot(name, initialize=False, default=default)
    assert isinstance(chroot, BaseChroot)
    return chroot