1 files changed, 58 insertions, 0 deletions
diff --git a/labbot/commands.py b/labbot/commands.py
new file mode 100644
index 0000000..10b6325
--- /dev/null
+++ b/labbot/commands.py
@@ -0,0 +1,58 @@
+"""
+Command Framework for Lab-Bot
+Allows the creation of functions that run when specific comments are made
+e.g.
+!close -> closes issue
+"""
+
+import logging
+
+log = logging.getLogger(__name__)
+
+class Commands:
+
+    def __init__(self, **kwargs):
+        self.prefix = kwargs.get("prefix", "!")
+        self.hooks = {}
+
+    def command(self, *args, name=None, aliases=None):
+        def decorator(func):
+            nonlocal name
+            nonlocal aliases
+
+            if name is None:
+                name = func.__name__
+
+            if aliases is None:
+                aliases = []
+
+            aliases.append(name)
+
+            for alias in aliases:
+                self.hooks[alias] = func
+
+        return decorator
+
+    async def process_note(self, event, gl, *args, **kwargs):
+        note = event.object_attributes.get("note", "").strip()
+        if not note.startswith(self.prefix):
+            return
+
+        note = note[len(self.prefix):]
+
+        note = note.split()
+        command = note[0]
+        arguments = note[1:]
+
+        hook = None
+        try:
+            hook = self.hooks[command]
+        except KeyError:
+            log.warn(f"Attempted to invoke nonexistant command `{command}`")
+
+        if hook:
+            await hook(event, gl, *arguments)
+    def setup_hook(self, bot):
+        bot.register_note_hook(self.process_note)
+
+commands = Commands()