commit to not forget code

plugins/plugin_manager.py

@@ -0,0 +1,109 @@
+#!/usr/bin/env python3
+
+import importlib
+import inspect
+from collections import OrderedDict
+from plugins.plugin import Plugin
+
+
+class PluginManager:
+    def __init__(self, bot, config):
+        self.bot = bot
+        self.config = config
+        self.plugins = OrderedDict()
+        self.commands = OrderedDict()
+        self.command_names = []
+
+    def load_plugin(self, name):
+        mod = importlib.import_module('plugins.{0}'.format(name))
+
+        classes = inspect.getmembers(mod, inspect.isclass)
+
+        for class_name, c in classes:
+            if c != Plugin and issubclass(c, Plugin):
+                self.plugins[name] = c(self)
+                self.plugins[name]._module = mod
+                for method_name, method in inspect.getmembers(self.plugins[name], inspect.ismethod):
+                    if getattr(method, '_plugin_command', False):
+                        options = method._plugin_command['options']
+                        command = options.pop('command', None)
+                        if command is None:
+                            command = ' '.join(method.__name__.split('_'))
+                        self.add_command(command, method, **options)
+                self.plugins[name].config = self.config
+
+    # Only for debug
+    def reload_plugins(self):
+        plugin_names = self.plugins.keys()
+        self.deactivate_plugins()
+        importlib.invalidate_caches()
+
+        # Force rebuilding modules
+        for plugin in self.plugins.values():
+            importlib.reload(plugin._module)
+
+        self.plugins = OrderedDict()
+        self.commands = OrderedDict()
+        self.command_names = []
+
+        for plugin_name in plugin_names:
+            self.load_plugin(plugin_name)
+
+        self.activate_plugins()
+
+    def activate_plugins(self):
+        for k, plugin in self.plugins.items():
+            plugin.activate()
+
+    def deactivate_plugins(self):
+        for k, plugin in self.plugins.items():
+            plugin.deactivate()
+
+    def notify_message(self, room, event, private=False):
+        # for k, plugin in self.plugins.items():
+        #     reply = plugin.notify_message(event, private)
+        #     if reply:
+        #         self.bot.send_message(mto=message['from'].full if private else message['from'].bare, mbody=reply, mtype='chat' if private else 'groupchat')
+        body = event['content']['body']
+        if body.startswith('!'):
+            command_found = False
+            for command in self.command_names:
+                command_chain = command.split(' ')
+                body_chain = body.split(' ')
+                if len(body_chain) >= len(command_chain) and body_chain[:len(command_chain)] == command_chain:
+                    admin_needed = self.commands[command]['options'].get('admin', False)
+                    if admin_needed and not self.is_sender_admin(event):
+                        reply = "Unauthorized"
+                    else:
+                        args = [a for a in body[len(command):].strip().split(' ') if a]
+                        f = self.commands[command]['function']
+                        reply = f(room, event, args)
+                    command_found = True
+                    break
+            if not command_found:
+                reply = 'Unknown command'
+            if reply and isinstance(reply, str):
+                room.send_text(reply)
+
+    def is_sender_admin(self, event):
+        return False
+        """Checks if author is admin"""
+        if message['type'] == 'groupchat':
+            # We don't know the jid. Either there's a problem or it's an anonymous room
+            if message['from'].full not in self.bot.presences:
+                return False
+            jid = self.bot.presences[message['from'].full]
+        else:
+            jid = message['from'].full.split('/')[0]
+        return jid in self.config.ADMINS
+
+    def is_admin_command(self, command):
+        return command in self.commands and self.commands[command]['options'].get('admin', False)
+
+    def add_command(self, command, function, **options):
+        # TODO: check the command doesn't exist. If not, try another name
+        command_name = '!{0}'.format(command)
+        class_name = inspect.getmro(function.__self__.__class__)[0].__name__
+        self.commands[command_name] = {'function': function, 'class': class_name, 'options': options}
+        self.command_names.append(command_name)
+        self.command_names.sort(key=len, reverse=True)