345 lines
13 KiB
Python
345 lines
13 KiB
Python
import enum, gc, glob, importlib, io, inspect, os, sys, typing, uuid
|
|
from src import Config, EventManager, Exports, IRCBot, Logging, Timers, utils
|
|
|
|
class ModuleException(Exception):
|
|
pass
|
|
class ModuleWarning(Exception):
|
|
pass
|
|
|
|
class ModuleNotFoundException(ModuleException):
|
|
pass
|
|
class ModuleNameCollisionException(ModuleException):
|
|
pass
|
|
class ModuleLoadException(ModuleException):
|
|
pass
|
|
class ModuleUnloadException(ModuleException):
|
|
pass
|
|
|
|
class ModuleNotLoadedWarning(ModuleWarning):
|
|
pass
|
|
|
|
class ModuleDependencyNotFulfilled(ModuleException):
|
|
def __init__(self, module, dependency):
|
|
ModuleException.__init__(self, "Dependency for %s not fulfilled: %s"
|
|
% (module, dependency))
|
|
self.module = module
|
|
self.dependency = dependency
|
|
class ModuleCircularDependency(ModuleException):
|
|
pass
|
|
|
|
class ModuleType(enum.Enum):
|
|
FILE = 0
|
|
DIRECTORY = 1
|
|
|
|
class BaseModule(object):
|
|
def __init__(self,
|
|
bot: "IRCBot.Bot",
|
|
events: EventManager.EventHook,
|
|
exports: Exports.Exports,
|
|
timers: Timers.Timers,
|
|
log: Logging.Log):
|
|
self.bot = bot
|
|
self.events = events
|
|
self.exports = exports
|
|
self.timers = timers
|
|
self.log = log
|
|
self.on_load()
|
|
def on_load(self):
|
|
pass
|
|
def unload(self):
|
|
pass
|
|
|
|
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, typing.Optional[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,
|
|
module: BaseModule,
|
|
context: str,
|
|
import_name: str):
|
|
self.name = name
|
|
self.module = module
|
|
self.context = context
|
|
self.import_name = import_name
|
|
|
|
class ModuleManager(object):
|
|
def __init__(self,
|
|
events: EventManager.EventHook,
|
|
exports: Exports.Exports,
|
|
timers: Timers.Timers,
|
|
config: Config.Config,
|
|
log: Logging.Log,
|
|
directory: str):
|
|
self.events = events
|
|
self.exports = exports
|
|
self.config = config
|
|
self.timers = timers
|
|
self.log = log
|
|
self.directory = directory
|
|
|
|
self.modules = {} # type: typing.Dict[str, LoadedModule]
|
|
|
|
def list_modules(self) -> typing.List[ModuleDefinition]:
|
|
modules = []
|
|
|
|
for file_module in glob.glob(os.path.join(self.directory, "*.py")):
|
|
modules.append(self.define_module(ModuleType.FILE, file_module))
|
|
|
|
for directory_module in glob.glob(os.path.join(
|
|
self.directory, "*", "__init__.py")):
|
|
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")
|
|
else:
|
|
path = "%s.py" % path
|
|
|
|
return self.define_module(type, path)
|
|
|
|
def _module_name(self, path: str) -> str:
|
|
return os.path.basename(path).rsplit(".py", 1)[0].lower()
|
|
def _module_path(self, name: str) -> str:
|
|
return os.path.join(self.directory, name)
|
|
def _import_name(self, name: str) -> str:
|
|
return "bitbot_%s" % name
|
|
|
|
def from_context(self, context: str) -> typing.Optional[LoadedModule]:
|
|
for module in self.modules.values():
|
|
if module.context == context:
|
|
return module
|
|
return None
|
|
def from_name(self, name: str) -> typing.Optional[LoadedModule]:
|
|
name_lower = name.lower()
|
|
for module in self.modules.values():
|
|
if module.name.lower() == name_lower:
|
|
return module
|
|
return None
|
|
|
|
def _get_magic(self, obj: typing.Any, magic: str, default: typing.Any
|
|
) -> typing.Any:
|
|
return getattr(obj, magic) if hasattr(obj, magic) else default
|
|
|
|
def _load_module(self, bot: "IRCBot.Bot", definition: ModuleDefinition,
|
|
check_dependency: bool=True) -> LoadedModule:
|
|
if check_dependency:
|
|
dependencies = definition.get_dependencies()
|
|
for dependency in dependencies:
|
|
if not dependency in self.modules:
|
|
raise ModuleDependencyNotFulfilled(definition.name,
|
|
dependency)
|
|
|
|
for hashflag, value in definition.hashflags:
|
|
if hashflag == "ignore":
|
|
# nope, ignore this module.
|
|
raise ModuleNotLoadedWarning("module ignored")
|
|
|
|
elif hashflag == "require-config" and value:
|
|
if not self.config.get(value.lower(), None):
|
|
# nope, required config option not present.
|
|
raise ModuleNotLoadedWarning("required config not present")
|
|
|
|
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)
|
|
loader.exec_module(module)
|
|
|
|
module_object_pointer = getattr(module, "Module", None)
|
|
if not module_object_pointer:
|
|
raise ModuleLoadException("module '%s' doesn't have a "
|
|
"'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." % definition.name)
|
|
|
|
context = str(uuid.uuid4())
|
|
context_events = self.events.new_context(context)
|
|
context_exports = self.exports.new_context(context)
|
|
context_timers = self.timers.new_context(context)
|
|
module_object = module_object_pointer(bot, context_events,
|
|
context_exports, context_timers, self.log)
|
|
|
|
if not hasattr(module_object, "_name"):
|
|
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,
|
|
utils.consts.BITBOT_HOOKS_MAGIC, []):
|
|
context_events.on(hook["event"]).hook(attribute,
|
|
**hook["kwargs"])
|
|
for export in self._get_magic(module_object,
|
|
utils.consts.BITBOT_EXPORTS_MAGIC, []):
|
|
context_exports.add(export["setting"], export["value"])
|
|
|
|
if definition.name in self.modules:
|
|
raise ModuleNameCollisionException("Module name '%s' "
|
|
"attempted to be used twice" % definition.name)
|
|
|
|
return LoadedModule(definition.name, module_object, context,
|
|
import_name)
|
|
|
|
def load_module(self, bot: "IRCBot.Bot", definition: ModuleDefinition
|
|
) -> LoadedModule:
|
|
try:
|
|
loaded_module = self._load_module(bot, definition,
|
|
check_dependency=False)
|
|
except ModuleWarning as warning:
|
|
self.log.warn("Module '%s' not loaded", [definition.name])
|
|
raise
|
|
except Exception as e:
|
|
self.log.error("Failed to load module \"%s\": %s",
|
|
[definition.name, str(e)])
|
|
raise
|
|
|
|
self.modules[loaded_module.name] = loaded_module
|
|
self.log.debug("Module '%s' loaded", [loaded_module.name])
|
|
return loaded_module
|
|
|
|
def _dependency_sort(self, definitions: typing.List[ModuleDefinition]):
|
|
definitions_ordered = []
|
|
|
|
definition_names = {d.name: d for d in definitions}
|
|
definition_dependencies = {
|
|
d.name: d.get_dependencies() for d in definitions}
|
|
|
|
for name, deps in definition_dependencies.items():
|
|
for dep in deps:
|
|
if not dep in definition_dependencies:
|
|
# unknown dependency!
|
|
raise ModuleDependencyNotFulfilled(name, dep)
|
|
|
|
while definition_dependencies:
|
|
changed = False
|
|
|
|
to_remove = []
|
|
for name, dependencies in definition_dependencies.items():
|
|
if not dependencies:
|
|
changed = True
|
|
# pop things with no unfufilled dependencies
|
|
to_remove.append(name)
|
|
for name in to_remove:
|
|
definitions_ordered.append(name)
|
|
del definition_dependencies[name]
|
|
for deps in definition_dependencies.values():
|
|
if name in deps:
|
|
changed = True
|
|
# fulfill dependencies for things we just popped
|
|
deps.remove(name)
|
|
|
|
if not changed:
|
|
for name, deps in definition_dependencies.items():
|
|
for dep_name in deps:
|
|
if name in definition_dependencies[dep_name]:
|
|
self.log.warn(
|
|
"Direct circular dependency detected: %s<->%s",
|
|
[name, dep_name])
|
|
changed = True
|
|
# snap a circular dependence
|
|
deps.remove(dep_name)
|
|
definition_dependencies[dep_name].remove(name)
|
|
if not changed:
|
|
raise ModuleCircularDependency()
|
|
|
|
return [definition_names[name] for name in definitions_ordered]
|
|
|
|
def load_modules(self, bot: "IRCBot.Bot", whitelist: typing.List[str]=[],
|
|
blacklist: typing.List[str]=[], safe: bool=False
|
|
) -> typing.Tuple[typing.List[str], typing.List[str]]:
|
|
fail = []
|
|
success = []
|
|
|
|
module_definitions = self._dependency_sort(self.list_modules())
|
|
|
|
for definition in module_definitions:
|
|
if definition.name in whitelist or (
|
|
not whitelist and not definition.name in blacklist):
|
|
try:
|
|
self.load_module(bot, definition)
|
|
except ModuleWarning:
|
|
fail.append(definition.name)
|
|
continue
|
|
except Exception as e:
|
|
if safe:
|
|
fail.append(definition.name)
|
|
continue
|
|
else:
|
|
raise
|
|
success.append(definition.name)
|
|
return success, fail
|
|
|
|
def unload_module(self, name: str):
|
|
if not name in self.modules:
|
|
raise ModuleNotFoundException(name)
|
|
loaded_module = self.modules[name]
|
|
if hasattr(loaded_module.module, "unload"):
|
|
try:
|
|
loaded_module.module.unload()
|
|
except:
|
|
pass
|
|
del self.modules[loaded_module.name]
|
|
|
|
context = loaded_module.context
|
|
self.events.purge_context(context)
|
|
self.exports.purge_context(context)
|
|
self.timers.purge_context(context)
|
|
|
|
module = loaded_module.module
|
|
del loaded_module.module
|
|
|
|
del sys.modules[loaded_module.import_name]
|
|
namespace = "%s." % loaded_module.import_name
|
|
for import_name in list(sys.modules.keys()):
|
|
if import_name.startswith(namespace):
|
|
del sys.modules[import_name]
|
|
|
|
references = sys.getrefcount(module)
|
|
referrers = gc.get_referrers(module)
|
|
del module
|
|
references -= 1 # 'del module' removes one reference
|
|
references -= 1 # one of the refs is from getrefcount
|
|
|
|
self.log.debug("Module '%s' unloaded (%d reference%s)",
|
|
[loaded_module.name, references,
|
|
"" if references == 1 else "s"])
|
|
if references > 0:
|
|
self.log.debug("References left for '%s': %s",
|
|
[loaded_module.name,
|
|
", ".join([str(referrer) for referrer in referrers])])
|