130 lines
3.7 KiB
Python
Executable File
130 lines
3.7 KiB
Python
Executable File
#!/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
|