X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/b25d436812b01503103b6eb8a6ea806b5b4d5bdf..839d2b358368b12c685ad30e6e6a534179dd0478:/bot.py diff --git a/bot.py b/bot.py index f82f143..a57b0d4 100644 --- a/bot.py +++ b/bot.py @@ -1,76 +1,172 @@ #!/usr/bin/python +# Erebus IRC bot - Author: John Runyon +# "Bot" and "BotConnection" classes (handling a specific "arm") + #TODO: error checking -import socket +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): self.parent = parent self.nick = nick + self.user = user + self.realname = realname self.chans = chans - self.conn = BotConnection(self, nick, user, bind, server, port, realname) + self.conn = BotConnection(self, bind, server, port) def connect(self): - self.conn.connect() + if self.conn.connect(): + self.parent.newfd(self, self.conn.socket.fileno()) + def getdata(self): for line in self.conn.read(): - self.parse(line) + 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) def parse(self, line): pieces = line.split() - if not self.conn.registered(): - if pieces[0] == "PING": - self.conn.send("PONG %s" % (pieces[1])) - elif pieces[1] == "001": - self.conn.registered(True) - print "!!!REGISTERED!!!" + + if pieces[1] == "PRIVMSG": + nick = pieces[0].split('!')[0][1:] + user = self.parent.user(nick) + target = pieces[2] + msg = ' '.join(pieces[3:])[1:] + 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]) + + if nick == self.nick: + self.conn.send("WHO %s %%ant,1" % (chan)) + else: + pass #TODO TODO TODO add to common chans! + + def parsemsg(self, user, target, msg): + chan = None + triggerused = msg[0] == self.parent.trigger + if triggerused: msg = msg[1:] + pieces = msg.split() + + if target == self.nick: + 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() + + print "%r %r %r %r" % (cmd, user, target, msg) + + 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 + callback(self, user, chan, *pieces[1:]) + 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)) + elif isinstance(target, self.parent.Channel): self.conn.send("PRIVMSG %s :%s" % (target.name, msg)) + elif isinstance(target, basestring): + 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.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): if done: self.state = 2 return self.state == 2 - def fileno(self): - if self.socket is not None: - return self.socket.fileno() - else: - return None - #TODO: rewrite send() to queue def send(self, line): + print self.parent.nick, '[O]', line self.write(line) + def write(self, line): - print self.nick, '[O]', 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.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)