import os
import click
import logging

from config import config
from utils import programs_available
from .docker import DockerWrapper

wrapper_impls = {
    'docker': DockerWrapper,
}


def get_wrapper_type(wrapper_type: str = None):
    return wrapper_type or config.file['wrapper']['type']


def wrap(wrapper_type: str = None):
    wrapper_type = get_wrapper_type(wrapper_type)
    if wrapper_type != 'none':
        wrapper_impls[wrapper_type]().wrap()


def is_wrapped(wrapper_type: str = None):
    wrapper_type = get_wrapper_type(wrapper_type)
    return os.getenv('KUPFERBOOTSTRAP_WRAPPED') == wrapper_type.capitalize()


def enforce_wrap(no_wrapper=False):
    wrapper_type = get_wrapper_type()
    if wrapper_type != 'none' and not is_wrapped(wrapper_type) and not config.runtime['no_wrap'] and not no_wrapper:
        logging.info(f'Wrapping in {wrapper_type}')
        wrap()


def check_programs_wrap(programs):
    if not programs_available(programs):
        enforce_wrap()


nowrapper_option = click.option(
    '--no-wrapper',
    'no_wrapper',
    is_flag=True,
    default=False,
    help='Disable the docker wrapper. Defaults to autodetection.',
)