adx

#!/usr/bin/env python3

import argparse
import sys
import util
import os
import logging

parser = argparse.ArgumentParser(
    description='This script provides a variety of tools for managing the'
                ' ADX development environment.'
)
subparsers = parser.add_subparsers(
    title='Actions',
    description='Choose which action to take.',
    metavar='action',
    dest='action'
)
actions = {}
actions['up'] = actions['dc'] = subparsers.add_parser(
    'up',
    description="Start docker containers (docker-compose).",
    help="Start docker containers (docker-compose)."
)
actions['up'].set_defaults(func=util.up)

actions['stop'] = subparsers.add_parser(
    'stop',
    description="Stop docker containers (docker-compose).",
    help="Stop docker containers (docker-compose)."
)
actions['stop'].set_defaults(func=util.run_docker_compose)

actions['down'] = subparsers.add_parser(
    'down',
    description="Stop and destroy docker containers (docker-compose).",
    help="Stop and destroy docker containers (docker-compose)."
)
actions['down'].set_defaults(func=util.run_docker_compose)

actions['restart'] = subparsers.add_parser(
    'restart',
    description="Restart docker containers (docker-compose).",
    help="Restart docker containers (docker-compose)."
)
actions['restart'].set_defaults(func=util.run_docker_compose)

actions['build'] = subparsers.add_parser(
    'build',
    description="Restart docker containers (docker-compose).",
    help="Restart docker containers (docker-compose)."
)
actions['build'].set_defaults(func=util.run_docker_compose)

actions['logs'] = subparsers.add_parser(
    'logs',
    description="Open the logs for a service (docker-compose)",
    help="Open the logs for a service (docker-compose)"
)
actions['logs'].set_defaults(func=util.run_docker_compose)

actions['dc'] = subparsers.add_parser(
    'dc',
    description="Run any docker-compose command.",
    help='Run any docker-compose command'
)
actions['dc'].set_defaults(func=util.run_docker_compose)

actions['exec'] = subparsers.add_parser(
    'exec',
    description="Run any docker-compose exec command.",
    help='Run any docker-compose exec command.'
)
actions['exec'].set_defaults(func=util.run_docker_compose)

# Docker exec bash
actions['bash'] = subparsers.add_parser(
    'bash',
    description="Open a bash prompt in the specified container.",
    help='Open a bash prompt in the specified container'
)
actions['bash'].set_defaults(func=util.run_docker_compose)
actions['bash'].add_argument('container', metavar='container', type=str,
                             help='The service name e.g."frontend."')

# Add the setup command arguments
actions['setup'] = subparsers.add_parser(
    'setup',
    description='Sets up the ADR code base locally',
    help='Sets up the ADR codebase locally.',
)
actions['setup'].set_defaults(func=util.setup)

actions['init'] = subparsers.add_parser(
    'init',
    description="Initializes ADX project",
    help="Configures the already existing ADX codebase with already "
         "checkout repositories."
)
actions['init'].set_defaults(func=util.init)

# Add the setup command arguments
actions['dbsetup'] = subparsers.add_parser(
    'dbsetup',
    description='Runs the initial configuration for CKAN plugins. Should be run'
                'after the initial Ckan startup',
    help='Performs initial CKAN database setup.',
)
actions['dbsetup'].set_defaults(func=util.init_ckan_db)

actions['demodata'] = subparsers.add_parser(
    'demodata',
    description='Loads demo example data to local ckan for easier development'
                'Needs adx dbsetup',
    help="Load demo example data to CKAN"
)
actions['demodata'].set_defaults(func=util.load_demo_data)

# Add the setup command arguments
actions['testsetup'] = subparsers.add_parser(
    'testsetup',
    description='Sets up the db for ckan tests',
    help='Performs initial CKAN test database setup.',
)
actions['testsetup'].set_defaults(func=util.reset_test_db)

actions['test'] = subparsers.add_parser(
    'test',
    description="Test a CKAN extension.",
    help='Test a CKAN extension.'
)
actions['test'].set_defaults(func=util.run_tests)
actions['test'].add_argument('extension', metavar='name', type=str,
                             help='The extension name e.g."validation", "ytp-request"')
actions['test'].add_argument(
    '--no-interaction', action='store_const',
    help='Removes the -it flag from the docker exec command',
    dest="interaction",
    default="-it",
    const=""
)

parser.add_argument(
    "-log",
    "--log",
    default="warning",
    help=(
        "Provide logging level. "
        "Example --log debug', default='warning'"),
)

levels = {
    'critical': logging.CRITICAL,
    'error': logging.ERROR,
    'warn': logging.WARNING,
    'warning': logging.WARNING,
    'info': logging.INFO,
    'debug': logging.DEBUG
}

if os.path.exists((util.ADX_PATH + "/adx_deploy/deploy_master.bash")):
    actions['deploy'] = subparsers.add_parser(
        'deploy',
        description='Opens up master/development comparisons of all repos in github.',
        help='Compare pushed master/development branches and deploy if desired.',
    )
    actions['deploy'].set_defaults(func=util.deploy_master)

if os.path.exists((util.ADX_PATH + "/adx_deploy/deploy_master.bash")):
    actions['deploy'] = subparsers.add_parser(
        'deploy',
        description='Opens up master/development comparisons of all repos in github.',
        help='Compare pushed master/development branches and deploy if desired.',
    )
    actions['deploy'].set_defaults(func=util.deploy_master)


def main(orig_args):
    args, extra_args = parser.parse_known_args(orig_args)

    level = levels.get(args.log.lower())
    if level is None:
        raise ValueError(
            f"log level given: {args.log}"
            f" -- must be one of: {' | '.join(levels.keys())}")
    logging.basicConfig(level=level)
    if hasattr(args, 'func'):
        args.func(args, extra_args)
    else:
        parser.print_help()


if __name__ == '__main__':
    main(sys.argv[1:])