X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/a4eacae2beaf344a5fd5b1d231f772e9fcbe4e10..5477b368b6f982ae96db41d62bbd3ce7abbc9d5d:/erebus.py diff --git a/erebus.py b/erebus.py index bf8305f..4e1934b 100644 --- a/erebus.py +++ b/erebus.py @@ -1,46 +1,72 @@ #!/usr/bin/python +# Erebus IRC bot - Author: John Runyon +# main startup code + #TODO: tons -import sys, select -import bot +import os, sys, select, MySQLdb, MySQLdb.cursors +import bot, config, ctlmod class Erebus(object): bots = {} fds = {} mods = {} msghandlers = {} + users = {} + chans = {} class User(object): - chans = [] - def __init__(self, nick, auth=None): self.nick = nick self.auth = auth + self.checklevel() - if auth is not None: - self.checklevel() + self.chans = [] + + def isauthed(self): + return self.auth is not None def authed(self, auth): + if auth == '0': auth = None self.auth = auth self.checklevel() - def checklevel(self): self.level = 9999 #TODO get level from db + def checklevel(self): + if self.auth is None: + self.glevel = -1 + else: + c = main.db.cursor() + c.execute("SELECT level FROM users WHERE auth = %s", (self.auth,)) + row = c.fetchone() + if row is not None: + self.glevel = row['level'] + else: + self.glevel = 0 + return self.glevel + + def join(self, chan): + self.chans.append(chan) + def part(self, chan): + self.chans.remove(chan) + def __str__(self): return self.nick - def __repr__(self): return "" % (self.nick) - class Channel(object): - users = [] - voices = [] - ops = [] + def __repr__(self): return "" % (self.nick,self.glevel) - def __init__(self, name): + class Channel(object): + def __init__(self, name, bot, levels={}): self.name = name + self.bot = bot + self.levels = levels + + self.users = [] + self.voices = [] + self.ops = [] def userjoin(self, user, level=None): if user not in self.users: self.users.append(user) if level == 'op' and user not in self.ops: self.ops.append(user) if level == 'voice' and user not in self.voices: self.voices.append(user) - def userpart(self, user): if user in self.ops: self.ops.remove(user) if user in self.voices: self.voices.remove(user) @@ -48,77 +74,106 @@ class Erebus(object): def userop(self, user): if user in self.users and user not in self.ops: self.ops.append(user) - def uservoice(self, user): if user in self.users and user not in self.voices: self.voices.append(user) - def userdeop(self, user): if user in self.ops: self.ops.remove(user) - def userdevoice(self, user): if user in self.voices: self.voices.remove(user) def __str__(self): return self.name def __repr__(self): return "" % (self.name) - def __init__(self): - self.po = select.poll() + def __init__(self, trigger): + self.trigger = trigger + if os.name == "posix": + self.potype = "poll" + self.po = select.poll() + else: # f.e. os.name == "nt" (Windows) + self.potype = "select" + self.fdlist = [] - def newbot(self, nick, user, bind, server, port, realname, chans): + def newbot(self, nick, user, bind, server, port, realname): if bind is None: bind = '' - obj = bot.Bot(self, nick, user, bind, server, port, realname, chans) + obj = bot.Bot(self, nick, user, bind, server, port, realname) self.bots[nick.lower()] = obj def newfd(self, obj, fileno): - print "newfd(Erebus(), %r, %r)" % (obj, fileno) self.fds[fileno] = obj - self.po.register(fileno, select.POLLIN) + if self.potype == "poll": + self.po.register(fileno, select.POLLIN) + elif self.potype == "select": + self.fdlist.append(fileno) - def bot(self, name): + def bot(self, name): #get Bot() by name (nick) return self.bots[name.lower()] - - def fd(self, fileno): + def fd(self, fileno): #get Bot() by fd/fileno return self.fds[fileno] - def user(self, nick): #TODO - return self.User(nick.lower()) - - def channel(self, name): #TODO - return self.Channel(name.lower()) + def user(self, nick): + nick = nick.lower() + if nick in self.users: + return self.users[nick] + else: + user = self.User(nick) + self.users[nick] = user + return user + def channel(self, name): #get Channel() by name + if name.lower() in self.chans: + return self.chans[name.lower()] + else: + return None + + def newchannel(self, bot, name, levels={}): + chan = self.Channel(name.lower(), bot, levels) + self.chans[name.lower()] = chan + return chan def poll(self): - return self.po.poll(60000) + if self.potype == "poll": + return [fd for (fd, ev) in self.po.poll()] + elif self.potype == "select": + return select.select(self.fdlist, [], [])[0] def connectall(self): for bot in self.bots.itervalues(): if bot.conn.state == 0: bot.connect() - #module functions - def modlist(self): pass - def hasmod(self, name): pass - def loadmod(self, name): pass - def unloadmod(self, name): pass - def reloadmod(self, name): pass - #bind functions - def bind(self, word, handler): pass - def addbind(self, word, handler): pass - def rmbind(self, word, handler): pass - def getbind(self, word, handler): pass + def hook(self, word, handler): + self.msghandlers[word] = handler + def unhook(self, word): + del self.msghandlers[word] + def hashook(self, word): + return word in self.msghandlers + def gethook(self, word): + return self.msghandlers[word] +def setup(): + global cfg, main -main = Erebus() + cfg = config.Config('bot.config') + main = Erebus(cfg.trigger) -def setup(): - main.newbot('Erebus', 'erebus', None, 'irc.quakenet.org', 6667, 'Erebus', ['#dimetest']) - main.bot('erebus').connect() + autoloads = [mod for mod, yes in cfg.items('autoloads') if int(yes) == 1] + for mod in autoloads: + ctlmod.load(main, mod) + + main.db = MySQLdb.connect(host=cfg.dbhost, user=cfg.dbuser, passwd=cfg.dbpass, db=cfg.dbname, cursorclass=MySQLdb.cursors.DictCursor) + c = main.db.cursor() + c.execute("SELECT nick, user, bind FROM bots WHERE active = 1") + rows = c.fetchall() + c.close() + for row in rows: + main.newbot(row['nick'], row['user'], row['bind'], cfg.host, cfg.port, cfg.realname) + main.connectall() def loop(): poready = main.poll() - - for (fileno,mask) in poready: - main.fd(fileno).getdata() + for fileno in poready: + for line in main.fd(fileno).getdata(): + main.fd(fileno).parse(line) if __name__ == '__main__': setup()