summaryrefslogtreecommitdiff
path: root/plugins/display_corrections.py
blob: 9937387fdbf79213e4efa32e875bdb615013b3ac (plain)
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
# A plugin that adds the /display_corrections command, to view the previous
# versions of a corrected message.

from plugin import BasePlugin
from common import shell_split
import tabs

class Plugin(BasePlugin):
    def init(self):
        usage = 'Usage: /display_corrections <number>\nDisplay_corrections: display all the corrections of the number-th last corrected message.'
        for tab_type in (tabs.MucTab, tabs.PrivateTab, tabs.ConversationTab):
            self.add_tab_command(tab_type, 'display_corrections', self.command_display_corrections, usage)

    def find_corrected(self, nb):
        messages = self.core.get_conversation_messages()
        if not messages:
            return None
        for message in messages[::-1]:
            if message.old_message:
                if nb == 1:
                    return message
                else:
                    nb -= 1
        return None

    def command_display_corrections(self, args):
        args = shell_split(args)
        if len(args) == 1:
            try:
                nb = int(args[0])
            except:
                return self.core.command_help('display_corrections')
        else:
            nb = 1
        message = self.find_corrected(nb)
        if message:
            display = []
            while message:
                display.append('%s %s%s%s %s' % (message.str_time, '* ' if message.me else '', message.nickname, '' if message.me else '>', message.txt))
                message = message.old_message
            self.core.information('Older versions:\n' + '\n'.join(display[::-1]), 'Info')
        else:
            self.core.information('No corrected message found.', 'Warning')

    def cleanup(self):
        del self.config