From 401400463bad2965ca75b8af6062a653fde41c19 Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Tue, 11 Aug 2020 22:24:26 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=89=20dme?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dme.py    | 110 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 list.json |  12 +++++-
 2 files changed, 121 insertions(+), 1 deletion(-)
 create mode 100644 dme.py

diff --git a/dme.py b/dme.py
new file mode 100644
index 0000000..a836fde
--- /dev/null
+++ b/dme.py
@@ -0,0 +1,110 @@
+""" Module to automate message deletion. """
+from asyncio import sleep
+from os import path, remove
+from os.path import exists
+from PIL import Image
+from pagermaid import redis, log, redis_status
+from pagermaid.listener import listener
+
+
+@listener(outgoing=True, command="dme",
+          description="编辑并删除当前对话您发送的特定数量的消息。限制：基于消息 ID 的 1000 条消息，大于 1000 条可能会触发删除消息过快限制。入群消息非管理员无法删除。（倒序）当数字足够大时即可实现删除所有消息。",
+          parameters="<数量> [文本]")
+async def selfprune(context):
+    """ Deletes specific amount of messages you sent. """
+    reply = await context.get_reply_message()
+    if reply and reply.photo:
+        if exists('plugins/dme.jpg'):
+            remove('plugins/dme.jpg')
+        target_file = reply.photo
+        await context.client.download_media(
+            await context.get_reply_message(), file = "plugins/dme.jpg"
+        )
+        await context.edit("替换图片设置完成。")
+    elif reply and reply.sticker:
+        if exists('plugins/dme.jpg'):
+            remove('plugins/dme.jpg')
+        await context.client.download_media(reply.media.document, file = "plugins/dme.webp")
+        im = Image.open("plugins/dme.webp")
+        im.save('plugins/dme.png', "png")
+        remove('plugins/dme.webp')
+        target_file = await context.client.upload_file('plugins/dme.png')
+        await context.edit("替换图片设置完成。")
+    elif path.isfile("plugins/dme.jpg"):
+        target_file = await context.client.upload_file('plugins/dme.jpg')
+    elif path.isfile("plugins/dme.png"):
+        target_file = await context.client.upload_file('plugins/dme.png')
+    else:
+        target_file = False
+        await context.edit("注意：没有图片进行替换。")
+    try:
+        count = int(context.parameter[0]) + 1
+    except ValueError:
+        await context.edit("出错了呜呜呜 ~ 无效的参数。")
+        return
+    except IndexError:
+        await context.edit("出错了呜呜呜 ~ 无效的参数。")
+        return
+    dme_msg = "别搁这防撤回了。。。"
+    if len(context.parameter) == 1:
+        if not redis_status:
+            pass
+        else:
+            try:
+                dme_msg = redis.get("dme_msg").decode()
+            except:
+                pass
+    elif len(context.parameter) == 2:
+        dme_msg = context.parameter[1]
+        if not redis_status():
+            pass
+        elif not dme_msg == str(count):
+            try:
+                redis.set("dme_msg", dme_msg)
+            except:
+                pass
+    count_buffer = 0
+    async for message in context.client.iter_messages(context.chat_id, from_user="me"):
+        if count_buffer == count:
+            break
+        if message.forward or message.via_bot or message.sticker or message.contact or message.poll or message.game or message.geo:
+            pass
+        elif message.text or message.voice:
+            if not message.text == dme_msg:
+                try:
+                    await message.edit(dme_msg)
+                except:
+                    pass
+        elif message.document or message.photo or message.file or message.audio or message.video or message.gif:
+            if target_file:
+                if not message.text == dme_msg:
+                    try:
+                        await message.edit(dme_msg, file=target_file)
+                    except:
+                        pass
+            else:
+                if not message.text == dme_msg:
+                    try:
+                        await message.edit(dme_msg)
+                    except:
+                        pass
+        else:
+            pass
+        await message.delete()
+        count_buffer += 1
+    count -= 1
+    count_buffer -= 1
+    await log(f"批量删除了自行发送的 {str(count_buffer)} / {str(count)} 条消息。")
+    notification = await send_prune_notify(context, count_buffer, count)
+    await sleep(.5)
+    await notification.delete()
+
+
+async def send_prune_notify(context, count_buffer, count):
+    return await context.client.send_message(
+        context.chat_id,
+        "删除了 "
+        + str(count_buffer) + " / " + str(count)
+        + " 条消息。"
+    )
+
diff --git a/list.json b/list.json
index eb957d8..b0e8ac3 100644
--- a/list.json
+++ b/list.json
@@ -5,10 +5,20 @@
             "version": "0.2",
             "section": "chat",
             "maintainer": "xtaodada",
-            "size": "10 kb",
+            "size": "1.9 kb",
             "supported": true,
             "des-short": "自动回复。",
             "des": "此插件用于需要自动回复的场景。"
+        },
+        {
+            "name": "dme",
+            "version": "1.1",
+            "section": "chat",
+            "maintainer": "xtaodada",
+            "size": "4.3 kb",
+            "supported": true,
+            "des-short": "反 TG desktop 防撤回插件",
+            "des": "编辑并删除当前对话您发送的特定数量的消息。支持 Redis 记录上一次指定的文本，方便快速撤回消息。 最近更新：防止编辑消息报错。"
         }
     ]
 }
\ No newline at end of file