1 files changed, 56 insertions, 0 deletions

diff --git a/wmd/actions/modules.py b/wmd/actions/modules.py
new file mode 100644
index 0000000..ed55609
--- /dev/null
+++ b/wmd/actions/modules.py
@@ -0,0 +1,56 @@
+from wmd.actions import Action
+
+import settings
+
+class ReloadModule(Action):
+    def recv_msg(self, irc, obj_data):
+        username = obj_data.get_username()
+
+        if username in settings.ADMINS:
+            args = obj_data.params.split(" ")
+            if "PRIVMSG" in obj_data.command and "RELOAD" in args[1].upper():
+                module = args[2]
+                if not module in irc.actions:
+                    irc.privmsg(username, "Invalid Module: %s")
+                    return
+                elif module == self.__class__.__name__:
+                    irc.privmsg(username, "Unable to reload self. Try restarting")
+                    return
+                module_class = irc.actions[module].__module__
+                module_path = module_class + '.' + module
+
+                del(irc.actions[module])
+                irc.load_action(module_path)
+
+                msg = "Reloaded %s" % (module_path,)
+                self.log(msg)
+                irc.privmsg(username, msg)
+
+class LoadModule(Action):
+    def recv_msg(self, irc, obj_data):
+        username = obj_data.get_username()
+
+        if username in settings.ADMINS:
+            args = obj_data.params.split(" ")
+            if "PRIVMSG" in obj_data.command and "LOAD" in args[1].upper():
+                module_path = args[2]
+                if module_path in irc.actions:
+                    irc.privmsg(username, "Module %s already loaded" % (module_path,))
+                    return
+                #irc.load_action(module_path)
+
+                msg = "Loading %s" % (module_path,)
+                self.log(msg)
+                irc.privmsg(username, msg)
+                return {'load': module_path}
+
+class ListModules(Action):
+    def recv_msg(self, irc, obj_data):
+        username = obj_data.get_username()
+
+        if username in settings.ADMINS:
+            args = obj_data.params.split(" ")
+            if "PRIVMSG" in obj_data.command and "LIST" in args[1].upper():
+                for module in irc.actions:
+                    msg = module
+                    irc.privmsg(username, msg)