h/bot.py

#!/usr/bin/python

network = "localhost"
nickname = "skybot"
channel = "#skybot"

import sys
import os
import glob
import imp
import re
import thread
import Queue
import copy

import irc
import yaml

os.chdir(sys.path[0]) #do stuff relative to the installation directory

class Bot(object):
    def __init__(self):
        self.plugins = {}
        self.commands = [] # fn, name, func, args
        self.listens = {} 
        self.filters = [] #fn, name, func
        self.daemons = {}

bot = Bot()
bot.nickname = nickname
bot.channel = channel
bot.network = network

print 'Loading plugins'
magic_re = re.compile(r'^\s*#(command|filter)(?:: +(\S+) *(\S.*)?)?\s*$')
for filename in glob.glob("plugins/*.py"):
    shortname = os.path.splitext(os.path.basename(filename))[0]
    try:
        bot.plugins[shortname] = imp.load_source(shortname, filename)
        plugin = bot.plugins[shortname]
        source = open(filename).read().split('\n')
        #this is a nasty hack, but it simplifies registration
        funcs = [x for x in dir(plugin) 
                 if str(type(getattr(plugin,x))) == "<type 'function'>"]
        for func in funcs:
            #read the line before the function definition, looking for a
            # comment that tells the bot about what it does
            func = getattr(plugin, func)
            lineno = func.func_code.co_firstlineno
            if lineno == 1:
                continue #can't have a line before the first line...
            m = magic_re.match(source[lineno-2])
            if m:
                typ, nam, rest = m.groups()
                if nam is None:
                    nam = func.__name__
                if rest is None:
                    rest = '\s*(.*)'
                if typ == 'command':
                    args = {'name': nam, 'hook': nam + rest}
                    bot.commands.append((filename, nam, func, args))
                if typ == 'filter':
                    bot.filters.append((filename, nam, func))
    except Exception, e:
        print e

if bot.daemons:
    print 'Running daemons'
    for daemon in bot.daemons.itervalues():
        thread.start_new_thread(daemon, ())

print 'Connecting to IRC'
bot.irc = irc.irc(network, nickname)
bot.irc.join(channel)
bot.commandprefix = '^(?:\.|'+nickname+'[:,]*\s*)'

print 'Running main loop'

class Input(object):
    def __init__(self, raw, prefix, command, 
            params, nick, user, host, paraml, msg):
        self.raw = raw
        self.prefix = prefix
        self.command = command
        self.params = params
        self.nick = nick
        self.user = user
        self.host = host
        self.paraml = paraml
        self.msg = msg

class FakeBot(object):
    def __init__(self, bot, input, func):
        self.bot = bot
        self.input = input
        self.msg = bot.irc.msg
        self.cmd = bot.irc.cmd
        self.func = func
        if input.command == "PRIVMSG":
            self.chan = input.paraml[0]
    
    def say(self, msg):
        self.bot.irc.msg(input.paraml[0], msg)

    def reply(self, msg):
        self.say(input.nick + ': ' + msg)

    def run(self):
        out = self.func(self, self.input)
        if out is not None:
            self.reply(unicode(out))

print bot.commands
print bot.filters

while True:
    try: 
        out = bot.irc.out.get(timeout=1)
        for fn, name, func, args in bot.commands:
            input = Input(*out)
            for fn, nam, filt in bot.filters:
                input = filt(bot, func, args, input)
                if input == None:
                    break
            if input == None:
                break
            thread.start_new_thread(FakeBot(bot, input, func).run, ())
    except Queue.Empty:
        pass