set custom afk & pm permit (#105)

* set custom afk message

* set custom antipm message

modules/afk.py

@@ -23,6 +23,7 @@
 
 from utils.misc import modules_help, prefix
 from utils.scripts import ReplyCheck
+from utils.db import db
 
 # Variables
 AFK = False
@@ -59,13 +60,18 @@ async def collect_afk_messages(bot: Client, message: Message):
         CHAT_TYPE = GROUPS if is_group else USERS
 
         if GetChatID(message) not in CHAT_TYPE:
-            text = (
-                f"<b>Beep boop. This is an automated message.\n"
-                f"I am not available right now.\n"
-                f"Last seen: {last_seen}\n"
-                f"Reason: <code>{AFK_REASON.upper()}</code>\n"
-                f"See you after I'm done doing whatever I'm doing.</b>"
-            )
+            text = db.get("core.afk", "afk_msg", None)
+            if text is None:
+                text = (
+                    f"<b>Beep boop. This is an automated message.\n"
+                    f"I am not available right now.\n"
+                    f"Last seen: {last_seen}\n"
+                    f"Reason: <code>{AFK_REASON.upper()}</code>\n"
+                    f"See you after I'm done doing whatever I'm doing.</b>"
+                )
+            else:
+                last_seen = last_seen.replace("ago", "").strip()
+                text = text.format(last_seen=last_seen, reason=AFK_REASON)
             await bot.send_message(
                 chat_id=GetChatID(message),
                 text=text,
@@ -144,6 +150,39 @@ async def afk_unset(_, message: Message):
     await message.delete()
 
 
+@Client.on_message(filters.command("setafkmsg", prefix) & filters.me, group=3)
+async def set_afk_msg(_, message: Message):
+    if not message.reply_to_message:
+        return await message.edit("Reply to a message to set it as your AFK message.")
+
+    msg = message.reply_to_message
+    afk_msg = msg.text or msg.caption
+
+    if not afk_msg:
+        return await message.edit(
+            "Reply to a text or caption message to set it as your AFK message."
+        )
+
+    if len(afk_msg) > 200:
+        return await message.edit(
+            "AFK message is too long. It should be less than 200 characters."
+        )
+    if "{reason}" not in afk_msg:
+        return await message.edit(
+            "AFK message should contain <code>{reason}</code> to indicate where the reason will be placed."
+        )
+    if "{last_seen}" not in afk_msg:
+        return await message.edit(
+            "AFK message should contain <code>{last_seen}</code> to indicate where the last seen time will be placed."
+        )
+
+    old_afk_msg = db.get("core.afk", "afk_msg", None)
+    if old_afk_msg:
+        db.remove("core.afk", "afk_msg")
+    db.set("core.afk", "afk_msg", afk_msg)
+    await message.edit(f"AFK message set to:\n\n{afk_msg}")
+
+
 @Client.on_message(filters.me, group=3)
 async def auto_afk_unset(_, message: Message):
     global AFK, AFK_TIME, AFK_REASON, USERS, GROUPS
@@ -166,4 +205,5 @@ async def auto_afk_unset(_, message: Message):
 modules_help["afk"] = {
     "afk [reason]": "Go to AFK mode with reason as anything after .afk\nUsage: <code>.afk <reason></code>",
     "unafk": "Get out of AFK",
+    "setafkmsg [reply to message]*": "Set your AFK message. Use <code>{reason}</code> and <code>{last_seen}</code> to indicate where the reason and last seen time will be placed.",
 }

modules/antipm.py

@@ -52,14 +52,18 @@ async def anti_pm_handler(client: Client, message: Message):
     user = await client.get_users(ids)
     u_f = user.first_name
     user_info = await client.resolve_peer(ids)
-    default_text = f"""<b>Hello, {u_f}!
+    default_text = db.get("core.antipm", "antipm_msg", None)
+    if default_text is None:
+        default_text = f"""<b>Hello, {u_f}!
 This is the Assistant Of {u_n}.</b>
 <i>My Boss is away or busy as of now, You can wait for him to respond.
 Do not spam further messages else I may have to block you!</i>
 
 <b>This is an automated message by the assistant.</b>
 <b><u>Currently You Have <code>{warns}</code> Warnings.</u></b>
-"""
+    """
+    else:
+        default_text = default_text.format(user=u_f, my_name=u_n, warns=warns)
     if db.get("core.antipm", "spamrep", False):
         await client.invoke(functions.messages.ReportSpam(peer=user_info))
     if db.get("core.antipm", "block", False):
@@ -179,10 +183,52 @@ async def del_contact(_, message: Message):
     await message.edit("User DisApproved!")
 
 
+@Client.on_message(filters.command(["setantipmmsg", "sam"], prefix) & filters.me)
+async def set_antipm_msg(_, message: Message):
+    if not message.reply_to_message:
+        return await message.edit(
+            "Reply to a message to set it as your antipm message."
+        )
+
+    msg = message.reply_to_message
+    afk_msg = msg.text or msg.caption
+
+    if not afk_msg:
+        return await message.edit(
+            "Reply to a text or caption message to set it as your antipm message."
+        )
+
+    if len(afk_msg) > 200:
+        return await message.edit(
+            "antipm message is too long. It should be less than 200 characters."
+        )
+
+    if "{user}" not in afk_msg:
+        return await message.edit(
+            "antipm message must contain <code>{user}</code> to mention the user."
+        )
+    if "{my_name}" not in afk_msg:
+        return await message.edit(
+            "antipm message must contain <code>{my_name}</code> to mention your name."
+        )
+    if "{warns}" not in afk_msg:
+        return await message.edit(
+            "antipm message must contain <code>{warns}</code> to mention the warns count."
+        )
+
+    old_afk_msg = db.get("core.antipm", "antipm_msg", None)
+    if old_afk_msg:
+        db.remove("core.antipm", "antipm_msg")
+    db.set("core.antipm", "antipm_msg", afk_msg)
+    await message.edit(f"antipm message set to:\n\n{afk_msg}")
+
+
 modules_help["antipm"] = {
     "antipm [enable|disable]*": "Enable Pm permit",
     "antipm_report [enable|disable]*": "Enable spam reporting",
     "antipm_block [enable|disable]*": "Enable user blocking",
+    "setantipmmsg [reply to message]*": "Set antipm message. Use {user} to mention the user and {my_name} to mention your name and {warns} to mention the warns count.",
+    "sam [reply to message]*": "Set antipm message. Use {user} to mention the user and {my_name} to mention your name and {warns} to mention the warns count.",
     "a": "Approve User",
     "d": "DisApprove User",
 }