47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
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.',
|
|
)
|