erebus.py

   1 #!/usr/bin/python
   2
   3 #TODO: tons
   4
   5 import os, 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                 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                 print "newfd(Erebus(), %r, %r)" % (obj, fileno)
  77                 self.fds[fileno] = obj
  78                 if self.potype == "poll":
  79                         self.po.register(fileno, select.POLLIN)
  80                 elif self.potype == "select":
  81                         self.fdlist.append(fileno)
  82
  83         def bot(self, name): #get Bot() by name (nick)
  84                 return self.bots[name.lower()]
  85         def fd(self, fileno): #get Bot() by fd/fileno
  86                 return self.fds[fileno]
  87
  88         def user(self, nick): #TODO #get User() by nick
  89                 return self.User(nick.lower())
  90         def channel(self, name): #TODO #get Channel() by name
  91                 return self.Channel(name.lower())
  92
  93         def poll(self):
  94                 if self.potype == "poll":
  95                         return [fd for (fd, ev) in self.po.poll()]
  96                 elif self.potype == "select":
  97                         return select.select(self.fdlist, [], [])[0]
  98
  99         def connectall(self):
 100                 for bot in self.bots.itervalues():
 101                         if bot.conn.state == 0:
 102                                 bot.connect()
 103
 104         #module functions
 105         def modlist(self): pass
 106         def hasmod(self, name): pass
 107         def loadmod(self, name): pass
 108         def unloadmod(self, name): pass
 109         def reloadmod(self, name): pass
 110
 111         #bind functions
 112         def bind(self, word, handler): pass
 113         def addbind(self, word, handler): pass
 114         def rmbind(self, word, handler): pass
 115         def getbind(self, word, handler): pass
 116
 117
 118 main = Erebus()
 119
 120 def setup():
 121         main.newbot('Erebus', 'erebus', None, 'irc.quakenet.org', 6667, 'Erebus', ['#dimetest'])
 122         main.bot('erebus').connect()
 123
 124 def loop():
 125         poready = main.poll()
 126         for fileno in poready:
 127                 main.fd(fileno).getdata()
 128
 129 if __name__ == '__main__':
 130         setup()
 131         while True: loop()