arcbot/link_command.py at master · Utanith/arcbot · 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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import random
import re
import urllib.request as request

from ircbot.command import IRCCommand
from ircbot.events import sendmessage

class Link(IRCCommand):
    link_regex = re.compile(  r"^"
        # protocol identifier
        r"(?:(?:https?|ftp)://)"
        # user:pass authentication
        r"(?:\S+(?::\S*)?@)?"
        r"(?:"
        # IP address exclusion
        # private & local networks
        r"(?!(?:10|127)(?:\.\d{1,3}){3})"
        r"(?!(?:169\.254|192\.168)(?:\.\d{1,3}){2})"
        r"(?!172\.(?:1[6-9]|2\d|3[0-1])(?:\.\d{1,3}){2})"
        # IP address dotted notation octets
        # excludes loopback network 0.0.0.0
        # excludes reserved space >= 224.0.0.0
        # excludes network & broacast addresses
        # (first & last IP address of each class)
        r"(?:[1-9]\d?|1\d\d|2[01]\d|22[0-3])"
        r"(?:\.(?:1?\d{1,2}|2[0-4]\d|25[0-5])){2}"
        r"(?:\.(?:[1-9]\d?|1\d\d|2[0-4]\d|25[0-4]))"
        r"|"
        # host name
        r"(?:(?:[a-z\u00a1-\uffff0-9]-*)*[a-z\u00a1-\uffff0-9]+)"
        # domain name
        r"(?:\.(?:[a-z\u00a1-\uffff0-9]-*)*[a-z\u00a1-\uffff0-9]+)*"
        # TLD identifier
        r"(?:\.(?:[a-z\u00a1-\uffff]{2,}))"
        # TLD may end with dot
        r"\.?"
        r")"
        # port number
        r"(?::\d{2,5})?"
        # resource path
        r"(?:[/?#]\S*)?"
        r"$", re.IGNORECASE)

    def __init__(self, link_file):
        super(Link, self).__init__('link', self.link_trigger)

        self.link_file = link_file
        self.load()

    def link_trigger(self, user, chan, args):
        user = user.nick
        if args:
            args = args.split()
            if args[0] in ['delete', 'remove']:
                if user.lower() == 'arctem':
                    if args[1] in self.links:
                        self.links.remove(args[1])
                        self.save()
                        self.fire(sendmessage(chan, '{}: Link {}d.'.format(user, args[0])))
                    else:
                        self.fire(sendmessage(chan, '{}: Could not find {} in links.'.format(user, args[1])))
                else:
                    self.fire(sendmessage(chan, '{}: You do not have permission to {} links.'.format(user, args[0])))
            elif args[0] in self.links:
                self.fire(sendmessage(chan, '{}: Link already in database.'
                    .format(user)))
            else:
                valid, info = Link.valid_link(args[0])
                if valid == True:
                    self.links.append(args[0])
                    self.save()
                    self.fire(sendmessage(chan, '{}: Link added: {}'.format(user, info)))
                else:
                    self.fire(sendmessage(chan, '{}: Could not add link: {}'
                        .format(user, info)))
        else:
            if self.links:
                self.fire(sendmessage(chan, '{}: {}'.format(user,
                    random.choice(self.links))))
            else:
                self.fire(sendmessage(chan, '{}: No links available.'
                    .format(user)))

    def load(self):
        try:
            with open(self.link_file, 'r') as link_file:
                self.links = link_file.read().split('\n')
        except FileNotFoundError:
            self.links = []

    def save(self):
        with open(self.link_file, 'w') as link_file:
            link_file.write('\n'.join(self.links))

    def valid_link(link):
        if not Link.link_regex.match(link):
            return "Did not match regex."
        try:
            req = request.Request(link, headers={ 'User-Agent': "Python's arcbot: The Ultimate Botting Machine!" })
            resp = request.urlopen(req)
            if resp.getcode() // 100 == 2:
                return True, re.search("<title>(?P<title>.*?)</title>", str(resp.read())).group('title')
            else:
                return False, "Return code {}.".format(resp.getcode())
        except Exception as e:
            return False, e