erebus.py

   1 #!/usr/bin/python
   2
   3 #TODO: tons
   4
   5 import sys, select
   6 import bot
   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                 self.po = select.poll()
  64
  65         def newbot(self, nick, user, bind, server, port, realname, chans):
  66                 if bind is None: bind = ''
  67                 obj = bot.Bot(self, nick, user, bind, server, port, realname, chans)
  68                 self.bots[nick.lower()] = obj
  69
  70         def newfd(self, obj, fileno):
  71                 print "newfd(Erebus(), %r, %r)" % (obj, fileno)
  72                 self.fds[fileno] = obj
  73                 self.po.register(fileno, select.POLLIN)
  74
  75         def bot(self, name): #get Bot() by name (nick)
  76                 return self.bots[name.lower()]
  77         def fd(self, fileno): #get Bot() by fd/fileno
  78                 return self.fds[fileno]
  79
  80         def user(self, nick): #TODO #get User() by nick
  81                 return self.User(nick.lower())
  82         def channel(self, name): #TODO #get Channel() by name
  83                 return self.Channel(name.lower())
  84
  85         def poll(self):
  86                 return self.po.poll(60000)
  87
  88         def connectall(self):
  89                 for bot in self.bots.itervalues():
  90                         if bot.conn.state == 0:
  91                                 bot.connect()
  92
  93         #module functions
  94         def modlist(self): pass
  95         def hasmod(self, name): pass
  96         def loadmod(self, name): pass
  97         def unloadmod(self, name): pass
  98         def reloadmod(self, name): pass
  99
 100         #bind functions
 101         def bind(self, word, handler): pass
 102         def addbind(self, word, handler): pass
 103         def rmbind(self, word, handler): pass
 104         def getbind(self, word, handler): pass
 105
 106
 107 main = Erebus()
 108
 109 def setup():
 110         main.newbot('Erebus', 'erebus', None, 'irc.quakenet.org', 6667, 'Erebus', ['#dimetest'])
 111         main.bot('erebus').connect()
 112
 113 def loop():
 114         poready = main.poll()
 115         for (fileno,mask) in poready:
 116                 main.fd(fileno).getdata()
 117
 118 if __name__ == '__main__':
 119         setup()
 120         while True: loop()