summaryrefslogtreecommitdiff
path: root/autoexec.py
blob: 79532671a894bfc6c3b5d8a3998d907a24088853 (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
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
#!/usr/bin/python
import re


class Autoexec(object):
    def __init__(self):
        self.networks = {}

    def onModuleAdd(self, IRC, label, onconnect=None, onregister=None, autojoin=None, usermodes=None, wallet=None, opername=None, opermodes=None, snomasks=None, operexec=None, operjoin=None):
        labels = [v[0] for v in self.networks.values()]
        if label in labels:
            raise BaseException("Label already exists")
        if IRC in self.networks.keys():
            raise BaseException("Network already exists")
        self.networks[IRC] = (label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin)

    def onModuleRem(self, IRC):
        del self.networks[IRC]

    def onConnect(self, IRC):
        (label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin) = self.networks[IRC]
        if onconnect:
            for line in onconnect:
                IRC.raw(line, origin=self)

    def onRegistered(self, IRC):
        (label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin) = self.networks[IRC]
        if onregister:
            for line in onregister:
                IRC.raw(line, origin=self)
        if usermodes:
            IRC.raw("MODE %s %s"%(IRC.identity.nick, usermodes), origin=self)
        if opername and wallet and "%s/opers/%s"%(label, opername) in wallet.keys():
            IRC.raw("OPER %s %s"%(opername, wallet[
                "%s/opers/%s"%(label, opername)]), origin=self)
        if autojoin:
            IRC.raw("JOIN %s"%(",".join(autojoin)), origin=self)

    def onRecv(self, IRC, line, data):
        if data is None:
            return
        (label, onconnect, onregister, autojoin, usermodes, wallet, opername, opermodes, snomasks, operexec, operjoin) = self.networks[IRC]
        (origin, ident, host, cmd, target, params, extinfo) = data
        if cmd == "381" and opermodes:
            if operexec:
                for line in operexec:
                    IRC.raw(line, origin=self)
            if opermodes:
                IRC.raw("MODE %s %s"%(IRC.identity.nick,
                                      opermodes), origin=self)
            if snomasks:
                IRC.raw("MODE %s +s %s"%(
                    IRC.identity.nick, snomasks), origin=self)
            if operjoin:
                IRC.raw("JOIN %s"%(",".join(operjoin)), origin=self)


class NickServ(object):
    def __init__(self):
        self.networks = {}

    def onModuleAdd(self, IRC, label, wallet=None, autojoin=None):
        labels = [v[0] for v in self.networks.values()]
        #print labels
        if label in labels:
            raise BaseException("Label already exists")
        if IRC in self.networks.keys():
            raise BaseException("Network already exists")
        self.networks[IRC] = (label, wallet, autojoin)

    def onModuleRem(self, IRC):
        del self.networks[IRC]

    def onRecv(self, IRC, line, data):
        if data is None:
            return
        (origin, ident, host, cmd, target, params, extinfo) = data
        label, wallet, autojoin = self.networks[IRC]
        if target == IRC.identity.nick and origin == "NickServ" and re.match("This nickname is registered and protected.", extinfo) and wallet and "%s/NickServ/%s"%(label, target.lower()) in wallet.keys():
            IRC.user("NickServ").msg("identify %s" % wallet[
                "%s/NickServ/%s"%(label, target.lower())])
        if cmd == "900" and autojoin:
            IRC.raw("JOIN %s"%(",".join(autojoin)), origin=self)