Mini Shell

# coding=utf-8
#
# Copyright CloudLinux Zug GmbH 2010-2018 All Rights Reserved
# Licensed under CLOUD LINUX ZUG GMBH LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENCE.TXT
#

from __future__ import absolute_import
import argparse
import json

from clwizard.modules import get_supported_modules


def validate_json_modules(json_string):
    try:
        data = json.loads(json_string)
    except (TypeError, ValueError):
        raise argparse.ArgumentTypeError('Invalid json passed')

    supported_modules = get_supported_modules()

    # TODO: Validate given json data separately for each module
    for name, instructions in data.items():
        if name not in supported_modules:
            raise argparse.ArgumentTypeError(
                "'%s' module is not supported now, "
                "available keys: %s" % (name, list(supported_modules.keys())))
        if "options" in list(instructions.keys()):
            if not set(instructions["options"].keys()).issubset(supported_modules[name].supported_options()):
                raise argparse.ArgumentTypeError(
                    "'%s' module has incorrect options: %s" % (name, list(instructions["options"].keys())))
        elif instructions:
            raise argparse.ArgumentTypeError(
                "'%s' module has incorrect instructions: %s" % (name, list(instructions.keys())))
    return data


def parse_cloudlinux_wizard_opts(argv):
    """
    Arg parser for cloudlinux-wizard utility
    """
    parser = argparse.ArgumentParser(
        prog='cloudlinux-wizard',
        description='Utility for installing cloudlinux modules'
    )
    # --json parameter is a convenience for UI calls, we do not process it
    parser.add_argument('--json', action='store_true', help=argparse.SUPPRESS)
    subparser = parser.add_subparsers(dest='subparser')
    install_parser = subparser.add_parser(
        'install',
        help='Installation of given modules by default in background'
    )
    install_parser.add_argument(
        '--no-async',
        help='Run install in not background mode',
        action='store_true'
    )
    install_parser.add_argument(
        '--force',
        help='force installer to ignore previous installation',
        action='store_true'
    )
    installer_group = install_parser.add_mutually_exclusive_group(required=True)
    installer_group.add_argument(
        '--json-data',
        help='Json with modules to be installed',
        type=validate_json_modules
    )
    installer_group.add_argument(
        '--resume',
        help="Restart installation process with previously given arguments",
        action='store_true'
    )
    status_parser = subparser.add_parser(
        'status',
        help='Return json with statuses of all modules'
    )
    status_parser.add_argument(
        '--initial', '-i', action='store_true', help='Get current modules status'
    )
    finish_parser = subparser.add_parser(
        'finish',
        help='Finalize installation from UI and close wizard'
    )
    cancel_parser = subparser.add_parser('cancel', help='Remove module from installation queue')
    cancel_parser.add_argument('module', choices=list(get_supported_modules()))
    opts = parser.parse_args(argv)
    return opts