X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/49a455aa76cae497ffc197f6cbe84682cd0d09d7..d431e54309841dcdb50d440e440234200d0de2fc:/bot.py diff --git a/bot.py b/bot.py index a58b4ea..0f2c7e6 100644 --- a/bot.py +++ b/bot.py @@ -1,65 +1,122 @@ #!/usr/bin/python +# Erebus IRC bot - Author: John Runyon +# "Bot" and "BotConnection" classes (handling a specific "arm") + #TODO: error checking import socket, sys #bots = {'erebus': bot.Bot(nick='Erebus', user='erebus', bind='', server='irc.quakenet.org', port=6667, realname='Erebus')} class Bot(object): - def __init__(self, parent, nick, user, bind, server, port, realname, chans): + def __init__(self, parent, nick, user, bind, server, port, realname): self.parent = parent self.nick = nick - self.chans = chans + self.user = user + self.realname = realname + + self.chans = [] - self.conn = BotConnection(self, nick, user, bind, server, port, realname) + chcurs = self.parent.db.cursor() + chcurs.execute("SELECT chname FROM chans WHERE bot = %s AND active = 1", (self.nick,)) + chans = chcurs.fetchall() + chcurs.close() + + for chrow in chans: + uscurs = self.parent.db.cursor() + uscurs.execute("SELECT user, level FROM chusers WHERE chan = %s", (chrow['chname'],)) + usrow = uscurs.fetchone() + levels = {} + while usrow is not None: + levels[usrow['user']] = -usrow['level'] + usrow = uscurs.fetchone() + uscurs.close() + self.chans.append(self.parent.newchannel(self, chrow['chname'], levels)) + + self.conn = BotConnection(self, bind, server, port) def connect(self): if self.conn.connect(): self.parent.newfd(self, self.conn.socket.fileno()) def getdata(self): - for line in self.conn.read(): - print self.nick, '[I]', line - if not self.conn.registered(): - pieces = line.split() - if pieces[0] == "PING": - self.conn.send("PONG %s" % (pieces[1])) - elif pieces[1] == "001": - self.conn.registered(True) - for c in self.chans: - self.join(c) - else: - self.parse(line) + return self.conn.read() + def parse(self, line): pieces = line.split() - if pieces[1] == "PRIVMSG": + + if not self.conn.registered() and pieces[0] == "NOTICE": + self.conn.register() + + elif pieces[1] == "001": + self.conn.registered(True) + for c in self.chans: + self.join(c.name) + + elif pieces[1] == "PRIVMSG": nick = pieces[0].split('!')[0][1:] user = self.parent.user(nick) - chan = self.parent.channel(pieces[2]) + target = pieces[2] msg = ' '.join(pieces[3:])[1:] - self.parsemsg(user, chan, msg) + self.parsemsg(user, target, msg) + elif pieces[0] == "PING": self.conn.send("PONG %s" % (pieces[1])) + + elif pieces[1] == "354": # WHOX + qt = pieces[3] + nick = pieces[4] + auth = pieces[5] + if auth != '0': + self.parent.user(nick).authed(auth) + elif pieces[1] == "JOIN": nick = pieces[0].split('!')[0][1:] - user = self.parent.user(nick) - chan = self.parent.channel(pieces[2]) #TODO TODO TODO + chan = self.parent.channel(pieces[2]) + + if nick == self.nick: + self.conn.send("WHO %s %%ant,1" % (chan)) + else: + user = self.parent.user(nick, justjoined=True) + chan.userjoin(user) + user.join(chan) - def parsemsg(self, user, chan, msg): - if msg[0] == '!': #TODO check config for trigger - msg = msg[1:] - else: - return + def parsemsg(self, user, target, msg): + chan = None + triggerused = msg[0] == self.parent.trigger + if triggerused: msg = msg[1:] pieces = msg.split() - cmd = pieces[0].upper() - if cmd == "EVAL": - try: ret = eval(' '.join(pieces[1:])) - except: self.msg(chan, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1])) - else: self.msg(chan, "Done: %r" % (ret)) - elif cmd == "EXEC": - try: exec ' '.join(pieces[1:]) - except: self.msg(chan, "Error: %s %s" % (sys.exc_info()[0], sys.exc_info()[1])) - else: self.msg(chan, "Done.") - #TODO + + if target == self.nick: + if len(pieces) > 1: + chanword = pieces[1] + if chanword[0] == '#': + chan = self.parent.channel(chanword) + pieces.pop(1) + + else: # message was sent to a channel + chan = self.parent.channel(target) #TODO check if bot's on channel --- in Erebus.channel() maybe? + if msg[0] == '*': # message may be addressed to bot by "*BOTNICK" trigger? + if pieces[0][1:].lower() == self.nick.lower(): + pieces.pop(0) # command actually starts with next word + msg = ' '.join(pieces) # command actually starts with next word + elif not triggerused: + return # not to bot, don't process! + + cmd = pieces[0].lower() + + if self.parent.hashook(cmd): + callback = self.parent.gethook(cmd) + if chan is None and callback.needchan: + self.msg(user, "You need to specify a channel for that command.") + return + if user.glevel >= callback.reqglevel: + #TODO TODO TODO check reqclevel + cbret = callback(self, user, chan, target, *pieces[1:]) + if cbret is NotImplemented: + self.msg(user, "Command not implemented.") + return + + self.msg(user, "No such command, or you don't have access.") def msg(self, target, msg): if isinstance(target, self.parent.User): self.conn.send("NOTICE %s :%s" % (target.nick, msg)) @@ -68,35 +125,41 @@ class Bot(object): if target[0] == '#': self.conn.send("PRIVMSG %s :%s" % (target, msg)) else: self.conn.send("NOTICE %s :%s" % (target, msg)) else: raise TypeError('Bot.msg() "target" must be Erebus.User, Erebus.Channel, or string') + def join(self, chan): self.conn.send("JOIN %s" % (chan)) + def part(self, chan): self.conn.send("PART %s" % (chan)) + def quit(self, reason="Shutdown"): self.conn.send("QUIT :%s" % (reason)) + def __str__(self): return self.nick + def __repr__(self): return "" % (self.nick) + class BotConnection(object): state = 0 # 0=disconnected, 1=registering, 2=connected - def __init__(self, parent, nick, user, bind, server, port, realname): + def __init__(self, parent, bind, server, port): self.parent = parent self.buffer = '' self.socket = None - self.nick = nick - self.user = user self.bind = bind self.server = server self.port = int(port) - self.realname = realname def connect(self): self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.socket.bind((self.bind, 0)) self.socket.connect((self.server, self.port)) - self.send("NICK %s" % (self.nick)) - self.send("USER %s 0 * :%s" % (self.user, self.realname)) - self.state = 1 + return True + def register(self): + if self.state == 0: + self.send("NICK %s" % (self.parent.nick)) + self.send("USER %s 0 * :%s" % (self.parent.user, self.parent.realname)) + self.state = 1 return True def registered(self, done=False): @@ -105,15 +168,23 @@ class BotConnection(object): #TODO: rewrite send() to queue def send(self, line): - print self.nick, '[O]', line + print self.parent.nick, '[O]', line self.write(line) + def write(self, line): self.socket.sendall(line+"\r\n") + def read(self): self.buffer += self.socket.recv(8192) lines = [] + while '\r\n' in self.buffer: pieces = self.buffer.split('\r\n', 1) + print self.parent.nick, '[I]', pieces[0] lines.append(pieces[0]) self.buffer = pieces[1] + return lines + + def __str__(self): return self.nick + def __repr__(self): return "" % (self.socket.fileno(), self.parent.nick)