erebus.py

   1 #!/usr/bin/python
   2
   3 #TODO: tons
   4
   5 import os, sys, select, MySQLdb, MySQLdb.cursors
   6 import bot, config, ctlmod
   7
   8 class Erebus(object):
   9         bots = {}
  10         fds = {}
  11         mods = {}
  12         msghandlers = {}
  13
  14         class User(object):
  15                 chans = []
  16
  17                 def __init__(self, nick, auth=None):
  18                         self.nick = nick
  19                         self.auth = auth
  20
  21                         if auth is not None:
  22                                 self.checklevel()
  23
  24                 def authed(self, auth):
  25                         self.auth = auth
  26                         self.checklevel()
  27
  28                 def checklevel(self): self.level = 9999 #TODO get level from db
  29
  30                 def __str__(self): return self.nick
  31                 def __repr__(self): return "<User %r>" % (self.nick)
  32
  33         class Channel(object):
  34                 users = []
  35                 voices = []
  36                 ops = []
  37
  38                 def __init__(self, name):
  39                         self.name = name
  40
  41                 def userjoin(self, user, level=None):
  42                         if user not in self.users: self.users.append(user)
  43                         if level == 'op' and user not in self.ops: self.ops.append(user)
  44                         if level == 'voice' and user not in self.voices: self.voices.append(user)
  45                 def userpart(self, user):
  46                         if user in self.ops: self.ops.remove(user)
  47                         if user in self.voices: self.voices.remove(user)
  48                         if user in self.users: self.users.remove(user)
  49
  50                 def userop(self, user):
  51                         if user in self.users and user not in self.ops: self.ops.append(user)
  52                 def uservoice(self, user):
  53                         if user in self.users and user not in self.voices: self.voices.append(user)
  54                 def userdeop(self, user):
  55                         if user in self.ops: self.ops.remove(user)
  56                 def userdevoice(self, user):
  57                         if user in self.voices: self.voices.remove(user)
  58
  59                 def __str__(self): return self.name
  60                 def __repr__(self): return "<Channel %r>" % (self.name)
  61
  62         def __init__(self):
  63                 if os.name == "posix":
  64                         self.potype = "poll"
  65                         self.po = select.poll()
  66                 else: # f.e. os.name == "nt" (Windows)
  67                         self.potype = "select"
  68                         self.fdlist = []
  69
  70         def newbot(self, nick, user, bind, server, port, realname, chans):
  71                 if bind is None: bind = ''
  72                 obj = bot.Bot(self, nick, user, bind, server, port, realname, chans)
  73                 self.bots[nick.lower()] = obj
  74
  75         def newfd(self, obj, fileno):
  76                 self.fds[fileno] = obj
  77                 if self.potype == "poll":
  78                         self.po.register(fileno, select.POLLIN)
  79                 elif self.potype == "select":
  80                         self.fdlist.append(fileno)
  81
  82         def bot(self, name): #get Bot() by name (nick)
  83                 return self.bots[name.lower()]
  84         def fd(self, fileno): #get Bot() by fd/fileno
  85                 return self.fds[fileno]
  86
  87         def user(self, nick): #TODO #get User() by nick
  88                 return self.User(nick.lower())
  89         def channel(self, name): #TODO #get Channel() by name
  90                 return self.Channel(name.lower())
  91
  92         def poll(self):
  93                 if self.potype == "poll":
  94                         return [fd for (fd, ev) in self.po.poll()]
  95                 elif self.potype == "select":
  96                         return select.select(self.fdlist, [], [])[0]
  97
  98         def connectall(self):
  99                 for bot in self.bots.itervalues():
 100                         if bot.conn.state == 0:
 101                                 bot.connect()
 102
 103         #module functions
 104         def modlist(self): pass
 105         def hasmod(self, name): pass
 106         def loadmod(self, name): pass
 107         def unloadmod(self, name): pass
 108         def reloadmod(self, name): pass
 109
 110         #bind functions
 111         def hook(self, word, handler):
 112                 print "hooked %r to %r" % (word, handler)
 113                 self.msghandlers[word] = handler
 114         def unhook(self, word):
 115                 del self.msghandlers[word]
 116         def hashook(self, word):
 117                 return word in self.msghandlers
 118         def gethook(self, word):
 119                 return self.msghandlers[word]
 120
 121 def setup():
 122         global cfg, main
 123
 124         cfg = config.Config('bot.config')
 125         main = Erebus()
 126
 127         autoloads = [mod for mod, yes in cfg.items('autoloads') if int(yes) == 1]
 128         for mod in autoloads:
 129                 ctlmod.load(main, mod)
 130
 131         main.db = MySQLdb.connect(host=cfg.dbhost, user=cfg.dbuser, passwd=cfg.dbpass, db=cfg.dbname, cursorclass=MySQLdb.cursors.DictCursor)
 132         c = main.db.cursor()
 133         c.execute("SELECT nick, user, bind FROM bots WHERE active = 1")
 134         rows = c.fetchall()
 135         c.close()
 136         for row in rows:
 137                 c2 = main.db.cursor()
 138                 c2.execute("SELECT chname FROM chans WHERE bot = %s AND active = 1", (row['nick'],))
 139                 chans = [chdic['chname'] for chdic in c2.fetchall()]
 140                 c2.close()
 141                 main.newbot(row['nick'], row['user'], row['bind'], cfg.host, cfg.port, cfg.realname, chans)
 142         main.connectall()
 143
 144 def loop():
 145         poready = main.poll()
 146         for fileno in poready:
 147                 main.fd(fileno).getdata()
 148
 149 if __name__ == '__main__':
 150         setup()
 151         while True: loop()