from types import ModuleType
from typing import Dict
from inspect import isclass
from pkgutil import iter_modules
from pathlib import Path
from os.path import dirname
from importlib import import_module
from raven.pluginbase import PluginBase


class PluginRegistry():
    plugins: Dict[str, PluginBase] = {}

    modules: [ModuleType] = []

    @staticmethod
    def register_plugin(name: str, plugin: PluginBase):
        PluginRegistry.plugins[name] = plugin

    @staticmethod
    def get_plugins_by_function(function_name: str) -> [PluginBase]:
        result = []
        for plugin in PluginRegistry.plugins.values():
            fun = getattr(plugin, function_name, None)
            if callable(fun):
                result.append(plugin)
        return result

    @staticmethod
    def load_module(module_name: str) -> ModuleType:
        module_name = f"plugins.{module_name}"
        # import the module and iterate through its attributes
        module = import_module(module_name)
        PluginRegistry.modules.append(module)
        return module

    @staticmethod
    def load_plugin(plugin_name: str):
        module_name = f"plugins.{plugin_name.lower()}"
        module = import_module(module_name)
        if plugin_name in dir(module):
            plugin_class = getattr(module, plugin_name)

            if isclass(plugin_class) and issubclass(plugin_class, PluginBase):
                PluginRegistry.register_plugin(plugin_name, plugin_class)
                print("%s -> %s :: %s in %s" % (plugin_name, plugin_class, module_name, module))
                return plugin_class

    @staticmethod
    def get_modules() -> [ModuleType]:
        return PluginRegistry.modules.copy()

    @staticmethod
    def get_plugins() -> [ModuleType]:
        return PluginRegistry.modules.copy()