FTG-Modules/recent_actions.py at main · GeekTG/FTG-Modules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# -*- coding: utf-8 -*-

import telethon

from .. import loader, utils


@loader.tds
class RecentActionsMod(loader.Module):
    """Reads recent actions"""

    strings = {
        "name": "Recent Actions",
        "reply_start": "<b>Reply to a message to specify where to start</b>",
        "invalid_chat": "<b>This isn't a supergroup or channel</b>",
        "needs_admin": "<b>Admin rights are required to read deleted messages</b>",
        "recovered": "Deleted message {} recovered. Originally sent at {} by {}, deleted at {} by {}",
    }

    @loader.group_admin
    @loader.ratelimit
    async def recoverdeletedcmd(self, message):
        """Restores deleted messages sent after replied message (optionally specify how many to recover)"""
        msgs = message.client.iter_admin_log(message.to_id, delete=True)
        if not message.is_reply:
            await utils.answer(message, self.strings("reply_start", message))
            return
        if not isinstance(message.to_id, telethon.tl.types.PeerChannel):
            await utils.answer(message, self.strings("invalid_chat", message))
            return
        target = (await message.get_reply_message()).date
        ret = []
        try:
            async for msg in msgs:
                if msg.original.date < target:
                    break
                if msg.original.action.message.date < target:
                    continue
                ret += [msg]
        except telethon.errors.rpcerrorlist.ChatAdminRequiredError:
            await utils.answer(message, self.strings("needs_admin", message))
        args = utils.get_args(message)
        if len(args) > 0:
            try:
                count = int(args[0])
                ret = ret[-count:]
            except ValueError:
                pass
        for msg in reversed(ret):
            orig = msg.original.action.message
            deldate = msg.original.date.isoformat()
            origdate = orig.date.isoformat()
            await message.respond(
                self.strings("recovered", message).format(
                    msg.id, origdate, orig.sender_id, deldate, msg.user_id
                )
            )
            if isinstance(orig, telethon.tl.types.MessageService):
                await message.respond(
                    "<b>" + utils.escape_html(orig.stringify()) + "</b>"
                )
            else:
                await message.respond(orig)