Create prefixprem.py

Akeno/utils/prefixprem.py

@@ -0,0 +1,68 @@
+import re
+from pyrogram import Client, Message
+from Akeno.utils.base_sqlite import *
+
+def command(commands: Union[str, List[str]], case_sensitive: bool = False):
+    command_re = re.compile(
+        r"([\"'])(.*?)(?<!\\)\1|(\S+)",
+        flags=re.UNICODE,
+    )
+
+    async def func(flt, client: Client, message: Message):
+        username = client.me.username or ""
+        user_id = message.from_user.id
+        text = message.text or message.caption
+        message.command = None
+
+        if not text:
+            return False
+
+        stored_prefix = get_prefix(user_id)
+        if message.entities and stored_prefix:
+            for entity in message.entities:
+                if entity.type == "custom_emoji" and entity.custom_emoji_id == stored_prefix:
+                    without_prefix = text[entity.length:]
+                    for cmd in flt.commands:
+                        if re.match(
+                            f"^(?:{cmd}(?:@?{username})?)(?:\s|$)",
+                            without_prefix,
+                            flags=0 if flt.case_sensitive else re.IGNORECASE,
+                        ):
+                            without_command = re.sub(
+                                f"{cmd}(?:@?{username})?\s?",
+                                "",
+                                without_prefix,
+                                count=1,
+                                flags=0 if flt.case_sensitive else re.IGNORECASE,
+                            )
+                            message.command = [cmd] + [
+                                re.sub(r"\\([\"'])", r"\1", m.group(2) or m.group(3) or "")
+                                for m in command_re.finditer(without_command)
+                            ]
+                            return True
+
+        if stored_prefix and text.startswith(stored_prefix):
+            without_prefix = text[len(stored_prefix):]
+
+            for cmd in flt.commands:
+                if re.match(
+                    f"^(?:{cmd}(?:@?{username})?)(?:\s|$)",
+                    without_prefix,
+                    flags=0 if flt.case_sensitive else re.IGNORECASE,
+                ):
+                    without_command = re.sub(
+                        f"{cmd}(?:@?{username})?\s?",
+                        "",
+                        without_prefix,
+                        count=1,
+                        flags=0 if flt.case_sensitive else re.IGNORECASE,
+                    )
+                    message.command = [cmd] + [
+                        re.sub(r"\\([\"'])", r"\1", m.group(2) or m.group(3) or "")
+                        for m in command_re.finditer(without_command)
+                    ]
+                    return True
+        return False
+    commands = commands if isinstance(commands, list) else [commands]
+    commands = {c if case_sensitive else c.lower() for c in commands}
+    return create(func, "CommandFilter", commands=commands, case_sensitive=case_sensitive)