Mini Shell

#!/opt/cloudlinux/venv/bin/python -bb
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2018 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# https://cloudlinux.com/docs/LICENCE.TXT
#
import argparse
import enum
import logging
import sys

from clcommon.lib import cledition
from clconfigure import setup_logger
from clconfigure.spec import admin, shared, shared_pro, solo
from lve_utils.sentry import init_lve_utils_sentry_client

LOGS_PATH = '/var/log/cloudlinux/clcustomizer.log'


class TargetEdition(enum.Enum):
    ADMIN = 'admin'
    AUTO = 'auto'
    SHARED = 'shared'
    SHARED_PRO = 'shared_pro'
    SOLO = 'solo'


def _resolve_auto_target():
    """
    Resolves edition for 'auto' target.
    """
    if cledition.is_cl_solo_edition():
        return TargetEdition.SOLO.value
    if cledition.is_cl_admin_edition():
        return TargetEdition.ADMIN.value
    if cledition.is_cl_shared_edition():
        return TargetEdition.SHARED.value
    if cledition.is_cl_shared_pro_edition():
        return TargetEdition.SHARED_PRO.value
    raise NotImplementedError()


def main():
    """
    Entry point for this program.
    Parses arguments and calls needed methods.
    """
    setup_logger(None, LOGS_PATH)
    logging.debug('Executing "%s"', ' '.join(sys.argv))

    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest='action')
    reconfigure = subparsers.add_parser('reconfigure')
    reconfigure.add_argument('-t', '--target',
                             choices=[e.value for e in TargetEdition],
                             required=True)

    preconfigure = subparsers.add_parser('preconfigure')
    preconfigure.add_argument('-t', '--target',
                              choices=[e.value for e in TargetEdition],
                              required=True)

    args = parser.parse_args()
    if args.action is None:
        parser.print_help()
        return

    target = _resolve_auto_target() if args.target == 'auto' else args.target

    handlers = {
        TargetEdition.SOLO.value: solo,
        TargetEdition.ADMIN.value: admin,
        TargetEdition.SHARED.value: shared,
        TargetEdition.SHARED_PRO.value: shared_pro,
    }

    handler = handlers.get(target)
    if handler is None:
        raise ValueError(f'Unknown target {target}')

    actions = {
        'reconfigure': 'provision',
        'preconfigure': 'preprovision'
    }
    action = actions.get(args.action)
    if action is None:
        raise ValueError(f"Unknown action {args.action}")

    getattr(handler, action)(edition=target)


if __name__ == '__main__':
    logging.basicConfig(level=logging.INFO)

    init_lve_utils_sentry_client('clconfigure')
    try:
        main()
    finally:
        logging.info('You can find complete logs of the operation here: %s', LOGS_PATH)