summaryrefslogtreecommitdiff
path: root/plugins/link.py
blob: 0cd1a0f1ffbc2a55c8ce377ede80c56993ce8d72 (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
47
48
# A plugin that adds the /link command, letting you open links that are pasted
# in the conversation, without having to click them.

import os
import re

from plugin import BasePlugin, PluginConfig
from xhtml import clean_text
import common

url_pattern = re.compile(r'\b(http[s]?://(?:\S+))\b', re.I|re.U)

class Plugin(BasePlugin):
    def init(self):
        self.add_command('link', self.command_link, "Usage: /link\nLink: opens the last link from the conversation into a browser.")

    def find_link(self, nb):
        messages = self.core.get_conversation_messages()
        if not messages:
            return None
        for message in messages[::-1]:
            match = url_pattern.search(clean_text(message.txt))
            if match:
                self.core.information('[%s]' % (match.groups(),))
                for url in list(match.groups())[::-1]:
                    if nb == 1:
                        return url
                    else:
                        nb -= 1
        return None

    def command_link(self, args):
        args = common.shell_split(args)
        if len(args) == 1:
            try:
                nb = int(args[0])
            except:
                return self.core.command_help('link')
        else:
            nb = 1
        link = self.find_link(nb)
        if link:
            self.core.exec_command('%s %s' % (self.config.get('browser', 'firefox'), link))
        else:
            self.core.information('No URL found.', 'Warning')

    def cleanup(self):
        del self.config