Mini Shell

# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENCE.TXT
#
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) as json_error:
        raise argparse.ArgumentTypeError("Invalid JSON data passed") from json_error

    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(
                f"'{name}' module is not supported at the moment, "
                f"available keys: {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(
                    f"'{name}' module has incorrect options: {list(instructions['options'].keys())}"
                )
        elif instructions:
            raise argparse.ArgumentTypeError(
                f"'{name}' module has incorrect instructions: {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="Install a module. The installation is performed in the background by default",
    )
    install_parser.add_argument(
        "--no-async", help="Install in the foreground", action="store_true"
    )
    install_parser.add_argument(
        "--force",
        help="Force the installer to ignore existing installations",
        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 the 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 the current modules' status"
    )

    finish_parser = subparser.add_parser(
        "finish", help="Finalize the installation from UI and close the wizard"
    )

    cancel_parser = subparser.add_parser(
        "cancel", help="Remove a module from the installation queue"
    )
    cancel_parser.add_argument("module", choices=list(get_supported_modules()))

    opts = parser.parse_args(argv)
    return opts