#! /usr/bin/env python3
"""
WROLPi is a self-contained collection of software to help you survive the world Without Rule of Law.

WROLPi is intended to be run on a Raspberry Pi with an optional external drive attached.  It serves up it's own wifi
network so that any person with a laptop/tablet/phone can connect and use the data previously collected by the user.
"""
import argparse
import logging
import sys

from wrolpi import web
from wrolpi.cmd import update_db, import_settings_configs, save_settings_configs
from wrolpi.common import logger
from wrolpi.user_plugins import PLUGINS


def update_choices_to_mains(sub_commands, choices_to_mains, main):
    """Associate a sub-command with the provided main, but only if that sub-command hasn't already been claimed
    by another main."""
    for choice in sub_commands.choices:
        if choice not in choices_to_mains:
            choices_to_mains[choice] = main


def main():
    from wrolpi.common import CONFIG_PATH

    parser = argparse.ArgumentParser()
    parser.add_argument('-v', '--verbose', action='count')
    parser.add_argument('-c', '--config', default=CONFIG_PATH,
                        help='The config file used to start WROLPi.')
    parser.add_argument('-d', '--db', action='store_true', default=False,
                        help='Initialize the database (all data will be lost).')
    parser.add_argument('-s', '--save-config', action='store_const', const=CONFIG_PATH,
                        help=f'Save the settings to this config file.  DEFAULT: {CONFIG_PATH}')
    parser.add_argument('-i', '--import-config', action='store_const', const=CONFIG_PATH,
                        help=f'Import this config, updating existing settings.  DEFAULT: {CONFIG_PATH}')

    sub_commands = parser.add_subparsers(title='sub-commands', dest='sub_commands')

    # Web parser is always present
    web_parser = sub_commands.add_parser('web')
    # This dict keeps track of which main will be called for each sub-command
    choices_to_mains = {'web': web.main}
    web.init_parser(web_parser)

    # Setup the plugins' sub-commands
    for plugin_name, plugin in PLUGINS.items():
        plugin.init_parser(sub_commands)
        update_choices_to_mains(sub_commands, choices_to_mains, plugin.main.main)

    args = parser.parse_args()

    if args.verbose == 1:
        logger.warning('Setting verbosity to INFO')
        logger.setLevel(logging.INFO)
    elif args.verbose == 2:
        logger.warning('Setting verbosity to DEBUG')
        logger.setLevel(logging.DEBUG)

    if args.sub_commands:
        sub_main = choices_to_mains[args.sub_commands]
        return sub_main(args)
    elif args.db:
        return_code = update_db(PLUGINS)
        logger.info('DB updated.')
    elif args.save_config:
        return_code = save_settings_configs(PLUGINS)
        logger.info('Config written.')
    elif args.import_config:
        return_code = import_settings_configs(PLUGINS)
        logger.info('Config imported.')
    else:
        parser.print_help()
        return_code = 1

    return return_code


if __name__ == '__main__':
    sys.exit(main())