Merge pull request #45 from Alpha4615/master

[irc/evilnet/x3.git] / src / plugins / __init__.py

diff --git a/src/plugins/__init__.py b/src/plugins/__init__.py
index 7b1dc9c7dbb2971a642eb1859922f9c983c86ac1..abf4a5af081476d6ca81a44a20c6c0b251326d07 100644 (file)
--- a/src/plugins/__init__.py
+++ b/src/plugins/__init__.py
@@ -1 +1,58 @@
-__all__ = [ "annoy" ]
+import imp
+import os.path
+
+class Plugin(object):
+    def server_link(self, server):
+        pass
+
+    def new_user(self, user):
+        pass
+
+    def nick_change(self, user, oldnick):
+        pass
+
+    def del_user(self, user, killer, why):
+        pass
+
+    def topic(self, who, chan, old_topic):
+        pass
+
+def load_path(path, prefix):
+    mods = []
+
+    for entry in os.listdir(path):
+        if os.path.isfile(os.path.join(path, entry)):
+            if os.path.splitext(entry)[1] != '.py':
+                continue
+
+        if entry.startswith('.') or entry.startswith('__'):
+            continue
+
+        try:
+            args = imp.find_module(os.path.splitext(entry)[0], [path])
+        except ImportError:
+            continue
+
+        if args:
+            mod = imp.load_module(prefix + os.path.splitext(entry)[0], *args)
+            if args[0]:
+                args[0].close()
+            if mod:
+                mods.append(mod)
+
+        if os.path.isdir(os.path.join(path, entry)):
+            mods.extend(load_path(os.path.join(path, entry), prefix + entry + '.'))
+
+    return mods
+
+def load():
+    mods = load_path(os.path.dirname(__file__), 'plugins.')
+    plugins = {}
+
+    # for some reason this returns multiple instances of the same plugin types
+    candidates = Plugin.__subclasses__()
+    for plg in candidates:
+        if plg.__name__ not in plugins:
+            plugins[plg.__name__] = plg()
+
+    return plugins.values()