feat: Add plain text support for afk/brb commands without prefixes

DragMusic/plugins/tools/afk.py

@@ -9,7 +9,7 @@ from DragMusic.Mongo.afkdb import add_afk, is_afk, remove_afk
 
 
 
-@app.on_message(filters.command(["afk", "brb"], prefixes=["/", "!"]))
+@app.on_message(filters.command(["afk", "brb"], prefixes=["/", "!"]) | filters.regex(r"^(afk|brb)(?:\s+(.+))?$"))
 async def active_afk(_, message: Message):
     if message.sender_chat:
         return
@@ -61,103 +61,101 @@ async def active_afk(_, message: Message):
                 disable_web_page_preview=True,
             )
 
-    if len(message.command) == 1 and not message.reply_to_message:
-        details = {
-            "type": "text",
-            "time": time.time(),
-            "data": None,
-            "reason": None,
-        }
-    elif len(message.command) > 1 and not message.reply_to_message:
-        _reason = (message.text.split(None, 1)[1].strip())[:100]
-        details = {
-            "type": "text_reason",
-            "time": time.time(),
-            "data": None,
-            "reason": _reason,
-        }
-    elif len(message.command) == 1 and message.reply_to_message.animation:
-        _data = message.reply_to_message.animation.file_id
-        details = {
-            "type": "animation",
-            "time": time.time(),
-            "data": _data,
-            "reason": None,
-        }
-    elif len(message.command) > 1 and message.reply_to_message.animation:
-        _data = message.reply_to_message.animation.file_id
-        _reason = (message.text.split(None, 1)[1].strip())[:100]
-        details = {
-            "type": "animation",
-            "time": time.time(),
-            "data": _data,
-            "reason": _reason,
-        }
-    elif len(message.command) == 1 and message.reply_to_message.photo:
-        await app.download_media(
-            message.reply_to_message, file_name=f"/tmp/downloads/{user_id}.jpg"
-        )
-        details = {
-            "type": "photo",
-            "time": time.time(),
-            "data": None,
-            "reason": None,
-        }
-    elif len(message.command) > 1 and message.reply_to_message.photo:
-        await app.download_media(
-            message.reply_to_message, file_name=f"/tmp/downloads/{user_id}.jpg"
-        )
-        _reason = message.text.split(None, 1)[1].strip()
-        details = {
-            "type": "photo",
-            "time": time.time(),
-            "data": None,
-            "reason": _reason,
-        }
-    elif len(message.command) == 1 and message.reply_to_message.sticker:
-        if message.reply_to_message.sticker.is_animated:
+    # Handle both command format and plain text format
+    if message.command:
+        # Command format: /afk or /afk reason
+        if len(message.command) == 1 and not message.reply_to_message:
             details = {
                 "type": "text",
                 "time": time.time(),
                 "data": None,
                 "reason": None,
             }
-        else:
-            await app.download_media(
-                message.reply_to_message, file_name=f"/tmp/downloads/{user_id}.jpg"
-            )
+        elif len(message.command) > 1 and not message.reply_to_message:
+            _reason = (message.text.split(None, 1)[1].strip())[:100]
             details = {
-                "type": "photo",
+                "type": "text_reason",
                 "time": time.time(),
                 "data": None,
-                "reason": None,
+                "reason": _reason,
             }
-    elif len(message.command) > 1 and message.reply_to_message.sticker:
-        _reason = (message.text.split(None, 1)[1].strip())[:100]
-        if message.reply_to_message.sticker.is_animated:
+    else:
+        # Plain text format: afk or afk reason
+        text_parts = message.text.strip().split(None, 1)
+        if len(text_parts) == 1:
             details = {
-                "type": "text_reason",
+                "type": "text",
                 "time": time.time(),
                 "data": None,
-                "reason": _reason,
+                "reason": None,
             }
         else:
-            await app.download_media(
-                message.reply_to_message, file_name=f"/tmp/downloads/{user_id}.jpg"
-            )
+            _reason = text_parts[1].strip()[:100]
             details = {
-                "type": "photo",
+                "type": "text_reason",
                 "time": time.time(),
                 "data": None,
                 "reason": _reason,
             }
-    else:
-        details = {
-            "type": "text",
-            "time": time.time(),
-            "data": None,
-            "reason": None,
-        }
+
+    # Handle reply to media (works for both command and plain text)
+    if message.reply_to_message:
+        if message.reply_to_message.animation:
+            _data = message.reply_to_message.animation.file_id
+            if details.get("reason"):
+                details.update({
+                    "type": "animation",
+                    "data": _data,
+                })
+            else:
+                details.update({
+                    "type": "animation",
+                    "data": _data,
+                    "reason": None,
+                })
+        elif message.reply_to_message.photo:
+            await app.download_media(
+                message.reply_to_message, file_name=f"/tmp/downloads/{user_id}.jpg"
+            )
+            if details.get("reason"):
+                details.update({
+                    "type": "photo",
+                    "data": None,
+                })
+            else:
+                details.update({
+                    "type": "photo",
+                    "data": None,
+                    "reason": None,
+                })
+        elif message.reply_to_message.sticker:
+            if message.reply_to_message.sticker.is_animated:
+                if details.get("reason"):
+                    details.update({
+                        "type": "text_reason",
+                        "data": None,
+                    })
+                else:
+                    details.update({
+                        "type": "text",
+                        "data": None,
+                        "reason": None,
+                    })
+            else:
+                await app.download_media(
+                    message.reply_to_message, file_name=f"/tmp/downloads/{user_id}.jpg"
+                )
+                if details.get("reason"):
+                    details.update({
+                        "type": "photo",
+                        "data": None,
+                    })
+                else:
+                    details.update({
+                        "type": "photo",
+                        "data": None,
+                        "reason": None,
+                    })
 
     await add_afk(user_id, details)    
     await message.reply_text(f"{message.from_user.first_name} ɪs ɴᴏᴡ ᴀғᴋ!")