from __future__ import absolute_import

import importlib as _importlib
import pkgutil as _pkgutil

def _load_all_modules():
    """Dynamically loads all modules in the 'mongo' package so that any
    commands declared within them are registered.
    """

    for (_, module, _) in _pkgutil.walk_packages(path=__path__):
        _importlib.import_module("." + module, package=__name__)

_load_all_modules()

from __future__ import absolute_import

# TODO: Combine the logic of this function with register_mongo_command() to support the
# "mongodb-help" command.
def register(name, command_class, **kwargs):
    """Registers a gdb command.

    @commandutil.register("hello-world", gdb.COMMAND_USER)
    class HelloWorld(gdb.Command):
      '''Greet the whole world.'''

      def invoke(self, arg, from_tty):
        print("Hello, World!")
    """

    def wrapper(cls):
        # A GDB command is registered by calling gdb.Command.__init__(). We construct an instance of
        # 'cls' so the command is automatically registered as a result of decorating a class with
        # the register() function.
        cls(name, command_class, **kwargs)
        return cls
    return wrapper