__package__ = 'archivebox.cli'

import os
from importlib import import_module

CLI_DIR = os.path.dirname(os.path.abspath(__file__))

required_attrs = ('__package__', '__command__', '__description__', 'main')


def list_subcommands():
    COMMANDS = {}
    for filename in os.listdir(CLI_DIR):
        if filename.startswith('archivebox_') and filename.endswith('.py'):
            subcommand = filename.replace('archivebox_', '').replace('.py', '')
            module = import_module('.archivebox_{}'.format(subcommand), __package__)

            assert all(hasattr(module, attr) for attr in required_attrs)
            assert module.__command__.split(' ')[-1] == subcommand
            COMMANDS[subcommand] = module.__description__

    return COMMANDS


def run_subcommand(subcommand: str, args=None):
    module = import_module('.archivebox_{}'.format(subcommand), __package__)
    return module.main(args)    # type: ignore