Refactor ModuleManager

1 files changed, 83 insertions, 46 deletions

diff --git a/src/ModuleManager.py b/src/ModuleManager.py
index 49d00772..6aafa9b3 100644
--- a/src/ModuleManager.py
+++ b/src/ModuleManager.py
@@ -18,6 +18,11 @@ class ModuleUnloadException(ModuleException):
 class ModuleNotLoadedWarning(ModuleWarning):
     pass
 
+class ModuleDependencyNotFulfilled(ModuleException):
+    def __init__(self, message, dependency):
+        ModuleException.__init__(self, message)
+        self.dependency = dependency
+
 class ModuleType(enum.Enum):
     FILE = 0
     DIRECTORY = 1
@@ -43,6 +48,23 @@ class BaseModule(object):
     def command_line(self, args: str):
         pass
 
+class ModuleDefinition(object):
+    def __init__(self,
+            name: str,
+            filename: str,
+            type: ModuleType,
+            hashflags: typing.List[typing.Tuple[str, str]]):
+        self.name = name
+        self.filename = filename
+        self.type = type
+        self.hashflags = hashflags
+    def get_dependencies(self):
+        dependencies = []
+        for key, value in self.hashflags:
+            if key == "depends-on":
+                dependencies.append(value)
+        return sorted(dependencies)
+
 class LoadedModule(object):
     def __init__(self,
             name: str,
@@ -70,19 +92,39 @@ class ModuleManager(object):
         self.directory = directory
 
         self.modules = {} # type: typing.Dict[str, LoadedModule]
-        self.waiting_requirement = {} # type: typing.Dict[str, typing.Set[str]]
 
-    def list_modules(self) -> typing.List[typing.Tuple[ModuleType, str]]:
+    def list_modules(self) -> typing.List[ModuleDefinition]:
         modules = []
 
         for file_module in glob.glob(os.path.join(self.directory, "*.py")):
-            modules.append((ModuleType.FILE, file_module))
+            modules.append(self.define_module(ModuleType.FILE, file_module))
 
         for directory_module in glob.glob(os.path.join(
                 self.directory, "*", "__init__.py")):
-            directory = os.path.dirname(directory_module)
-            modules.append((ModuleType.DIRECTORY, directory))
-        return sorted(modules, key=lambda module: module[1])
+            modules.append(self.define_module(ModuleType.DIRECTORY,
+                directory_module))
+        return sorted(modules, key=lambda module: module.name)
+
+    def define_module(self, type: ModuleType, filename: str
+            ) -> ModuleDefinition:
+        if type == ModuleType.DIRECTORY:
+            name = os.path.dirname(filename)
+        else:
+            name = filename
+        name = self._module_name(name)
+
+        return ModuleDefinition(name, filename, type,
+            utils.parse.hashflags(filename))
+
+    def find_module(self, name: str) -> ModuleDefinition:
+        type = ModuleType.FILE
+        path = self._module_path(name)
+
+        if os.path.isdir(path):
+            type = ModuleType.DIRECTORY
+            path = os.path.join(path, "__init__.py")
+
+        return self.define_module(type, path)
 
     def _module_name(self, path: str) -> str:
         return os.path.basename(path).rsplit(".py", 1)[0].lower()
@@ -94,7 +136,7 @@ class ModuleManager(object):
     def from_context(self, context: str) -> typing.Optional[LoadedModule]:
         for module in self.modules.values():
             if module.context == context:
-                return  module
+                return module
         return None
     def from_name(self, name: str) -> typing.Optional[LoadedModule]:
         name_lower = name.lower()
@@ -107,15 +149,14 @@ class ModuleManager(object):
             ) -> typing.Any:
         return getattr(obj, magic) if hasattr(obj, magic) else default
 
-    def _load_module(self, bot: "IRCBot.Bot", name: str) -> LoadedModule:
-        path = self._module_path(name)
-        if os.path.isdir(path) and os.path.isfile(os.path.join(
-                path, "__init__.py")):
-            path = os.path.join(path, "__init__.py")
-        else:
-            path = "%s.py" % path
+    def _load_module(self, bot: "IRCBot.Bot", definition: ModuleDefinition
+            ) -> LoadedModule:
+        dependencies = definition.get_dependencies()
+        for dependency in dependencies:
+            if not dependency in self.modules:
+                raise ModuleDependencyNotFulfilled(dependency)
 
-        for hashflag, value in utils.parse.hashflags(path):
+        for hashflag, value in definition.hashflags:
             if hashflag == "ignore":
                # nope, ignore this module.
                raise ModuleNotLoadedWarning("module ignored")
@@ -125,16 +166,9 @@ class ModuleManager(object):
                     # nope, required config option not present.
                     raise ModuleNotLoadedWarning("required config not present")
 
-            elif hashflag == "require-module" and value:
-                requirement = value.lower()
-                if not requirement in self.modules:
-                    if not requirement in self.waiting_requirement:
-                        self.waiting_requirement[requirement] = set([])
-                        self.waiting_requirement[requirement].add(path)
-                    raise ModuleNotLoadedWarning("waiting for requirement")
-
-        import_name = self._import_name(name)
-        import_spec = importlib.util.spec_from_file_location(import_name, path)
+        import_name = self._import_name(definition.name)
+        import_spec = importlib.util.spec_from_file_location(import_name,
+            definition.filename)
         module = importlib.util.module_from_spec(import_spec)
         sys.modules[import_name] = module
         loader = typing.cast(importlib.abc.Loader, import_spec.loader)
@@ -143,10 +177,10 @@ class ModuleManager(object):
         module_object_pointer = getattr(module, "Module", None)
         if not module_object_pointer:
             raise ModuleLoadException("module '%s' doesn't have a "
-                "'Module' class." % name)
+                "'Module' class." % definition.name)
         if not inspect.isclass(module_object_pointer):
             raise ModuleLoadException("module '%s' has a 'Module' attribute "
-                "but it is not a class." % name)
+                "but it is not a class." % definition.name)
 
         context = str(uuid.uuid4())
         context_events = self.events.new_context(context)
@@ -156,7 +190,7 @@ class ModuleManager(object):
             context_exports, context_timers, self.log)
 
         if not hasattr(module_object, "_name"):
-            module_object._name = name.title()
+            module_object._name = definition.name.title()
         for attribute_name in dir(module_object):
             attribute = getattr(module_object, attribute_name)
             for hook in self._get_magic(attribute,
@@ -167,28 +201,26 @@ class ModuleManager(object):
                 utils.consts.BITBOT_EXPORTS_MAGIC, []):
             context_exports.add(export["setting"], export["value"])
 
-        if name in self.modules:
+        if definition.name in self.modules:
             raise ModuleNameCollisionException("Module name '%s' "
-                "attempted to be used twice")
+                "attempted to be used twice" % definition.name)
 
-        return LoadedModule(name, module_object, context, import_name)
+        return LoadedModule(definition.name, module_object, context,
+            import_name)
 
-    def load_module(self, bot: "IRCBot.Bot", name: str) -> LoadedModule:
+    def load_module(self, bot: "IRCBot.Bot", definition: ModuleDefinition
+            ) -> LoadedModule:
         try:
-            loaded_module = self._load_module(bot, name)
+            loaded_module = self._load_module(bot, definition)
         except ModuleWarning as warning:
-            self.log.warn("Module '%s' not loaded", [name])
+            self.log.warn("Module '%s' not loaded", [definition.name])
             raise
         except Exception as e:
             self.log.error("Failed to load module \"%s\": %s",
-                [name, str(e)])
+                [definition.name, str(e)])
             raise
 
         self.modules[loaded_module.name] = loaded_module
-        if loaded_module.name in self.waiting_requirement:
-            for requirement_name in self.waiting_requirement[
-                    loaded_module.name]:
-                self.load_module(bot, requirement_name)
         self.log.debug("Module '%s' loaded", [loaded_module.name])
         return loaded_module
 
@@ -197,21 +229,26 @@ class ModuleManager(object):
             ) -> typing.Tuple[typing.List[str], typing.List[str]]:
         fail = []
         success = []
-        for type, path in self.list_modules():
-            name = self._module_name(path)
-            if name in whitelist or (not whitelist and not name in blacklist):
+
+        module_definitions = self.list_modules()
+
+        #TODO figure out dependency tree
+
+        for definition in module_definitions:
+            if definition.name in whitelist or (
+                    not whitelist and not definition.name in blacklist):
                 try:
-                    self.load_module(bot, name)
+                    self.load_module(bot, definition)
                 except ModuleWarning:
-                    fail.append(name)
+                    fail.append(definition.name)
                     continue
                 except Exception as e:
                     if safe:
-                        fail.append(name)
+                        fail.append(definition.name)
                         continue
                     else:
                         raise
-                success.append(name)
+                success.append(definition.name)
         return success, fail
 
     def unload_module(self, name: str):