| 1 | #!/usr/bin/env python
|
|---|
| 2 | # -*- coding: utf-8 -*-
|
|---|
| 3 | #
|
|---|
| 4 | # Copyright (C) 2007 Daniele Tricoli aka Eriol <eriol@mornie.org>
|
|---|
| 5 | # Copyright (C) 2010 Andrea Milazzo aka mancausoft <me@mancausoft.org>
|
|---|
| 6 | #
|
|---|
| 7 | # Daniele Tricoli, 2010 Reliceinsed to WTFPL
|
|---|
| 8 | #
|
|---|
| 9 | # This Program is free software. It comes without any warranty, to
|
|---|
| 10 | # the extent permitted by applicable law. You can redistribute it
|
|---|
| 11 | # and/or modify it under the terms of the Do What The Fuck You Want
|
|---|
| 12 | # To Public License, Version 2, as published by Sam Hocevar. See
|
|---|
| 13 | # http://sam.zoy.org/wtfpl/COPYING for more details.
|
|---|
| 14 |
|
|---|
| 15 |
|
|---|
| 16 | import os
|
|---|
| 17 | import re
|
|---|
| 18 |
|
|---|
| 19 | from twisted.words.protocols import irc
|
|---|
| 20 | from twisted.internet import reactor, protocol, ssl
|
|---|
| 21 |
|
|---|
| 22 | from dataslavebot import *
|
|---|
| 23 |
|
|---|
| 24 | import logging as log
|
|---|
| 25 | log.basicConfig(level=log.DEBUG,
|
|---|
| 26 | format='[%(asctime)s] %(levelname)s %(message)s',
|
|---|
| 27 | datefmt='%d/%m/%Y %H:%M:%S %Z',
|
|---|
| 28 | filename='slavebot.log',
|
|---|
| 29 | filemode='a')
|
|---|
| 30 | console = log.StreamHandler()
|
|---|
| 31 | console.setLevel(log.INFO)
|
|---|
| 32 | log.getLogger().addHandler(console)
|
|---|
| 33 |
|
|---|
| 34 |
|
|---|
| 35 | def save():
|
|---|
| 36 | f = open('dataslavebot.py', 'w')
|
|---|
| 37 | f.write('BOT_NICKNAME = "' + BOT_NICKNAME +'"\n')
|
|---|
| 38 | f.write('MASTER_USER = ' + str(MASTER_USER))
|
|---|
| 39 | f.write('\n\nRE_CONV = ' + str(RE_CONV))
|
|---|
| 40 | f.write('\nRE_CONV_ME = ' + str(RE_CONV_ME))
|
|---|
| 41 | f.write("\nOP_BOT = '"+ OP_BOT + "'\n")
|
|---|
| 42 | f.write('SERVER = '+ str(SERVER))
|
|---|
| 43 |
|
|---|
| 44 | class SlaveBot(irc.IRCClient):
|
|---|
| 45 |
|
|---|
| 46 | nickname = BOT_NICKNAME
|
|---|
| 47 |
|
|---|
| 48 | def __init__(self):
|
|---|
| 49 | self.op = False
|
|---|
| 50 | self.nicks = {}
|
|---|
| 51 |
|
|---|
| 52 | def opOnlyMaster (self, nick, channel):
|
|---|
| 53 | if ((nick in MASTER_USER) or (nick[:-1] in MASTER_USER) or (nick[:-2] in MASTER_USER)) and self.op:
|
|---|
| 54 | if nick == "tosky":
|
|---|
| 55 | self.msg(channel, "Ďekuji vám, pane " + nick + ", protože mi jste ctíl jako operátor.")
|
|---|
| 56 | elif nick == "geenna":
|
|---|
| 57 | self.msg(channel, "Grazie mio padrone " + nick + " mettiti a 90 che cominciamo!.")
|
|---|
| 58 | else:
|
|---|
| 59 | self.msg(channel, "Grazie mio padrone " + nick + " per avermi dato l'onore di poterti oppare.")
|
|---|
| 60 | self.sendLine('MODE ' + channel + ' +o ' + nick)
|
|---|
| 61 | self.nicks[nick] = 1
|
|---|
| 62 |
|
|---|
| 63 | def signedOn(self):
|
|---|
| 64 | log.info('Signed On:')
|
|---|
| 65 | for chan in self.factory.channel:
|
|---|
| 66 | self.join(chan)
|
|---|
| 67 |
|
|---|
| 68 | def joined(self, channel):
|
|---|
| 69 | log.info('Joined: %s', channel)
|
|---|
| 70 | self.nicks = {}
|
|---|
| 71 | self.msg(channel, "Grazie di avermi creato, datemi l'onore del'op e saro' il vostro schiavo per il resto della mia vita")
|
|---|
| 72 |
|
|---|
| 73 | def action (self, username, channel, emote):
|
|---|
| 74 | if (emote.find("si inchina") != -1) or (emote.find("s'inchina") != -1):
|
|---|
| 75 | self.msg(channel, username.split('!')[0] + ": Stai attento in questo canale e' pericoloso fare cose del genere!")
|
|---|
| 76 |
|
|---|
| 77 | def userLeft (self, user, channel):
|
|---|
| 78 | del self.nicks[user]
|
|---|
| 79 |
|
|---|
| 80 | def userRenamed(self, oldname, newname):
|
|---|
| 81 | self.nicks[newname] = self.nicks[oldname]
|
|---|
| 82 | del self.nicks[oldname]
|
|---|
| 83 |
|
|---|
| 84 | def userJoined (self, user, channel):
|
|---|
| 85 | self.me(channel, "si prostra davanti a " + user)
|
|---|
| 86 | self.nicks[user] = 0
|
|---|
| 87 | self.opOnlyMaster(user, channel)
|
|---|
| 88 |
|
|---|
| 89 | def irc_RPL_NAMREPLY(self, prefix, params):
|
|---|
| 90 | log.info('irc_RPL_NAMREPLY')
|
|---|
| 91 | nicks = params[3].split()
|
|---|
| 92 | for nick in nicks:
|
|---|
| 93 | if nick[0] == '@':
|
|---|
| 94 | self.nicks[nick[1:]] = 1
|
|---|
| 95 | else:
|
|---|
| 96 | self.nicks[nick] = 0
|
|---|
| 97 | if OP_BOT in self.nicks.keys():
|
|---|
| 98 | self.msg(params[2], "ciao " + OP_BOT)
|
|---|
| 99 |
|
|---|
| 100 | def irc_ERR_NICKNAMEINUSE (self, prefix, params):
|
|---|
| 101 | self.nickname = self.nickname + "_"
|
|---|
| 102 | self.setNick(self.nickname)
|
|---|
| 103 |
|
|---|
| 104 | def privmsg(self, user, channel, message):
|
|---|
| 105 | user = user.split('!')[0]
|
|---|
| 106 | if channel != self.nickname:
|
|---|
| 107 | for regexp in RE_CONV:
|
|---|
| 108 | rexp = re.sub('(?i)\$bot', self.nickname, regexp)
|
|---|
| 109 | try:
|
|---|
| 110 | if re.search(rexp, message, re.IGNORECASE):
|
|---|
| 111 | msg = re.sub('(?i)\$nick', user, RE_CONV[regexp])
|
|---|
| 112 | self.msg(channel, re.sub('(?i)\$bot', self.nickname, msg))
|
|---|
| 113 | except:
|
|---|
| 114 | print "ERRORE PORC!"
|
|---|
| 115 | for regexp in RE_CONV_ME:
|
|---|
| 116 | rexp = re.sub('(?i)\$bot', self.nickname, regexp)
|
|---|
| 117 | if re.search(rexp, message, re.IGNORECASE):
|
|---|
| 118 | msg =re.sub('(?i)\$nick', user, RE_CONV_ME[regexp])
|
|---|
| 119 | self.me(channel, re.sub('(?i)\$bot', self.nickname, msg))
|
|---|
| 120 | if re.search(self.nickname, message, re.IGNORECASE):
|
|---|
| 121 | if re.search('aiutami|help', message, re.IGNORECASE):
|
|---|
| 122 | self.msg(channel, self.polygen('if')[:440].rsplit('.',1)[0] + '.')
|
|---|
| 123 | elif re.search('spot', message, re.IGNORECASE):
|
|---|
| 124 | self.msg(channel, self.polygen('poster').replace('\n',''))
|
|---|
| 125 | elif re.search('consigli', message, re.IGNORECASE):
|
|---|
| 126 | msg = self.polygen('kamasutra')
|
|---|
| 127 | m = re.search('(?i)\<strong\> (.*) \</strong\>', msg)
|
|---|
| 128 | message = m.groups()[0]
|
|---|
| 129 | m = re.search('\</strong\>.*\n.*\n+(.*)', msg)
|
|---|
| 130 | message += ': ' + m.groups()[0]
|
|---|
| 131 | self.msg(channel, message)
|
|---|
| 132 |
|
|---|
| 133 | else:
|
|---|
| 134 | if re.match('(?i) *LIST( |$)', message):
|
|---|
| 135 | self.msg(user, "MESSAGE: ")
|
|---|
| 136 | for i in RE_CONV:
|
|---|
| 137 | self.msg(user, i + " -> " + RE_CONV[i])
|
|---|
| 138 | self.msg(user, "/ME: ")
|
|---|
| 139 | for i in RE_CONV_ME:
|
|---|
| 140 | self.msg(user, i + " -> " + RE_CONV_ME[i])
|
|---|
| 141 |
|
|---|
| 142 | m = re.match('(?i) *ADD +"(.*)".*,.*"(.*)"', message)
|
|---|
| 143 | if m:
|
|---|
| 144 | RE_CONV[m.groups()[0]] = m.groups()[1]
|
|---|
| 145 | for i in RE_CONV:
|
|---|
| 146 | self.msg(user, i + " -> " + RE_CONV[i])
|
|---|
| 147 | log.info(user + ": "+ message)
|
|---|
| 148 | save()
|
|---|
| 149 |
|
|---|
| 150 | m = re.match('(?i) *DEL +"(.*)".*', message)
|
|---|
| 151 | if m:
|
|---|
| 152 | del RE_CONV[m.groups()[0]]
|
|---|
| 153 | for i in RE_CONV:
|
|---|
| 154 | self.msg(user, i + " -> " + RE_CONV[i])
|
|---|
| 155 | log.info(user + ": "+ message)
|
|---|
| 156 | save()
|
|---|
| 157 |
|
|---|
| 158 | m = re.match('(?i) *ADDME +"(.*)".*,.*"(.*)"', message)
|
|---|
| 159 | if m:
|
|---|
| 160 | RE_CONV_ME[m.groups()[0]] = m.groups()[1]
|
|---|
| 161 | for i in RE_CONV_ME:
|
|---|
| 162 | self.msg(user, i + " -> " + RE_CONV_ME[i])
|
|---|
| 163 | log.info(user + ": "+ message)
|
|---|
| 164 | save()
|
|---|
| 165 |
|
|---|
| 166 | m = re.match('(?i) *DELME +"(.*)".*', message)
|
|---|
| 167 | if m:
|
|---|
| 168 | del RE_CONV_ME[m.groups()[0]]
|
|---|
| 169 | for i in RE_CONV_ME:
|
|---|
| 170 | self.msg(user, i + " -> " + RE_CONV_ME[i])
|
|---|
| 171 | log.info(user + ": "+ message)
|
|---|
| 172 | save()
|
|---|
| 173 |
|
|---|
| 174 | m = re.match('(?i) *NICK +"(.*)".*', message)
|
|---|
| 175 | if m:
|
|---|
| 176 | BOT_NICKNAME = m.groups()[0]
|
|---|
| 177 | self.nickname = BOT_NICKNAME
|
|---|
| 178 | self.setNick(self.nickname)
|
|---|
| 179 | save()
|
|---|
| 180 |
|
|---|
| 181 | m = re.match('(?i) *TALK +(.*)', message)
|
|---|
| 182 | if m:
|
|---|
| 183 | msg = m.groups()[0].split(' ',1)
|
|---|
| 184 | self.msg(msg[0], msg[1])
|
|---|
| 185 |
|
|---|
| 186 | m = re.match('(?i) *ADDOP +(.*)', message)
|
|---|
| 187 | if m:
|
|---|
| 188 | for nick in m.groups()[0].split(' '):
|
|---|
| 189 | MASTER_USER.append(nick)
|
|---|
| 190 | self.msg(user, str(MASTER_USER))
|
|---|
| 191 | save()
|
|---|
| 192 |
|
|---|
| 193 | if re.match('(?i) *HELP *.*', message):
|
|---|
| 194 | self.msg(user.split('!')[0], 'TALK #channel message | LIST | ADD "reg-exp", "message" | DEL "regexp" | ADDME "reg-exp", "message" | DELME "regexp" | NICK "NEW NICK"')
|
|---|
| 195 | self.msg(user.split('!')[0], 'In message you can use $nick. $nick will replace with the name to sender.')
|
|---|
| 196 | self.msg(user.split('!')[0], 'In message and regexp you can use $bot. $bot will replace with the name to bot.')
|
|---|
| 197 |
|
|---|
| 198 | def modeChanged (self, user, channel, set, modes, args):
|
|---|
| 199 | toOp = False
|
|---|
| 200 | for i in range(len(args)):
|
|---|
| 201 | if modes[i] == 'o':
|
|---|
| 202 | user = user.split('!')[0]
|
|---|
| 203 | self.op = set
|
|---|
| 204 | if set:
|
|---|
| 205 | if args[i] == self.nickname:
|
|---|
| 206 | self.msg(channel, "Grazie " + user + " saro' il vostro schiavo per il resto della mia vita.")
|
|---|
| 207 | toOp = True
|
|---|
| 208 | self.nicks[args[i]] = 1
|
|---|
| 209 | else:
|
|---|
| 210 | if args[i] == self.nickname:
|
|---|
| 211 | self.msg(channel, "Grazie " + user + " non sono degno di essere il vostro schiavo. La mia vita e' nelle vostre mani!")
|
|---|
| 212 | self.nicks[args[i]] = 0
|
|---|
| 213 | if toOp:
|
|---|
| 214 | for nick in self.nicks:
|
|---|
| 215 | if self.nicks[nick] == 0 and nick != BOT_NICKNAME:
|
|---|
| 216 | self.opOnlyMaster(nick, channel)
|
|---|
| 217 |
|
|---|
| 218 | def polygen (self, name):
|
|---|
| 219 | filename = '/usr/share/polygen/ita/' + name + '.grm'
|
|---|
| 220 | prog = os.popen("polygen " + filename)
|
|---|
| 221 | return prog.read()
|
|---|
| 222 |
|
|---|
| 223 | def ctcpUnknownQuery (self, user, channel, tag, data):
|
|---|
| 224 | pass
|
|---|
| 225 |
|
|---|
| 226 | class SlaveBotFactory(protocol.ClientFactory):
|
|---|
| 227 |
|
|---|
| 228 | protocol = SlaveBot
|
|---|
| 229 |
|
|---|
| 230 | def __init__(self, channel):
|
|---|
| 231 | self.channel = channel
|
|---|
| 232 |
|
|---|
| 233 | def clientConnectionLost(self, connector, reason):
|
|---|
| 234 | log.critical('Connection lost: %s', reason)
|
|---|
| 235 | connector.connect()
|
|---|
| 236 |
|
|---|
| 237 | def clientConnectionFailed(self, connector, reason):
|
|---|
| 238 | log.critical('Connection failed: %s', reason)
|
|---|
| 239 | connector.connect()
|
|---|
| 240 |
|
|---|
| 241 | if __name__ == '__main__':
|
|---|
| 242 |
|
|---|
| 243 | for server in SERVER:
|
|---|
| 244 | f = SlaveBotFactory(server['channel'])
|
|---|
| 245 | if server['ssl'] == 1:
|
|---|
| 246 | contextFactory = ssl.ClientContextFactory()
|
|---|
| 247 | reactor.connectSSL(server['name'], server['port'], f, contextFactory)
|
|---|
| 248 | else:
|
|---|
| 249 | reactor.connectTCP(server['name'], server['port'], f)
|
|---|
| 250 | reactor.run()
|
|---|
