added todos as well as regex based command triggers.HEADmaster

1 files changed, 48 insertions, 0 deletions

diff --git a/lib/commands.py b/lib/commands.py
new file mode 100644
index 0000000..638ab9b
--- /dev/null
+++ b/lib/commands.py
@@ -0,0 +1,48 @@
+import typing
+
+import pyrogram
+from pyrogram.api import types as tgtypes
+
+from lib.common import CommonContext
+from util import property_decorator
+
+commands = {}
+
+
+def get_all_commands():
+    return commands.values()
+
+
+class CommandContext(CommonContext):
+    def __init__(self, client: pyrogram.Client, channel, args: typing.List[str], message: tgtypes.Message):
+        super().__init__(client, channel, message)
+        import re
+        self.args = args
+        self.rest_content = re.sub('^.*? ', '', message.message)
+
+
+
+def register_command(func):
+    if not is_command(func):
+        return
+    commands[get_command_name(func)] = func
+
+
+def is_command(func):
+    return hasattr(func, 'name') and func.name is not None
+
+
+def get_command_name(func):
+    return func.name
+
+
+def get_command_description(func):
+    return func.description
+
+
+def get_command_by_name(command_name: str):
+    return commands[command_name.lower()]
+
+
+name = property_decorator('name')
+description = property_decorator('description')