148 lines
4.0 KiB
Python
Executable File
148 lines
4.0 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
network = "irc.synirc.net"
|
|
nick = "skybot"
|
|
channel = "#cobol"
|
|
|
|
import sys
|
|
import os
|
|
import glob
|
|
import imp
|
|
import re
|
|
import thread
|
|
import Queue
|
|
import collections
|
|
|
|
sys.path += ['plugins'] # so 'import hook' works without duplication
|
|
|
|
import irc
|
|
import yaml
|
|
|
|
os.chdir(sys.path[0]) # do stuff relative to the installation directory
|
|
|
|
class Bot(object):
|
|
def __init__(self, nick, channel, network):
|
|
self.nick = nick
|
|
self.channel = channel
|
|
self.network = network
|
|
|
|
bot = Bot(nick, channel, network)
|
|
|
|
print 'Loading plugins'
|
|
typs = '|'.join('command filter event'.split())
|
|
magic_re = re.compile(r'^\s*#!(%s)(?:: +(\S+) *(\S.*)?)?\s*$' % typs)
|
|
|
|
def reload_plugins(mtime=[0]):
|
|
new_mtime = os.stat('plugins')
|
|
if new_mtime == mtime[0]:
|
|
return
|
|
|
|
bot.plugs = collections.defaultdict(lambda: [])
|
|
|
|
for filename in glob.glob("plugins/*.py"):
|
|
shortname = os.path.splitext(os.path.basename(filename))[0]
|
|
try:
|
|
plugin = imp.load_source(shortname, filename)
|
|
for obj in vars(plugin).itervalues():
|
|
if hasattr(obj, '_skybot_hook'): #check for magic
|
|
for type, data in obj._skybot_hook:
|
|
bot.plugs[type] += [data]
|
|
except Exception, e:
|
|
print ' error:', e
|
|
|
|
mtime[0] = new_mtime
|
|
|
|
reload_plugins()
|
|
|
|
print ' plugin listing:'
|
|
for type, plugs in sorted(bot.plugs.iteritems()):
|
|
print ' %s:' % type
|
|
for plug in plugs:
|
|
out = ' %s:%s:%s' % (plug[0])
|
|
print out,
|
|
if len(plug) == 3 and 'hook' in plug[2]:
|
|
print '%s%s' % (' ' * (35 - len(out)), plug[2]['hook'])
|
|
else:
|
|
print
|
|
print
|
|
|
|
print 'Connecting to IRC'
|
|
bot.irc = irc.irc(network, nick)
|
|
bot.irc.join(channel)
|
|
bot.commandprefix = r'^(?:[.!]|'+nick+r'[:,]*\s*)'
|
|
bot.persist_dir = os.path.abspath('persist')
|
|
|
|
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
|
|
if command == "PRIVMSG":
|
|
self.chan = paraml[0]
|
|
else:
|
|
self.chan = ""
|
|
|
|
class FakeBot(object):
|
|
def __init__(self, bot, input, func):
|
|
self.bot = bot
|
|
self.persist_dir = bot.persist_dir
|
|
self.network = bot.network
|
|
self.input = input
|
|
self.msg = bot.irc.msg
|
|
self.cmd = bot.irc.cmd
|
|
self.join = bot.irc.join
|
|
self.func = func
|
|
self.doreply = True
|
|
self.chan = input.chan
|
|
|
|
def say(self, msg):
|
|
self.bot.irc.msg(self.input.paraml[0], msg)
|
|
|
|
def reply(self, msg):
|
|
self.say(self.input.nick + ': ' + msg)
|
|
|
|
def run(self):
|
|
ac = self.func.func_code.co_argcount
|
|
if ac == 2:
|
|
out = self.func(self, self.input)
|
|
elif ac == 1:
|
|
out = self.func(self.input.inp)
|
|
if out is not None:
|
|
if self.doreply:
|
|
self.reply(unicode(out))
|
|
else:
|
|
self.say(unicode(out))
|
|
|
|
while True:
|
|
try:
|
|
out = bot.irc.out.get(timeout=1)
|
|
reload_plugins()
|
|
printed = False
|
|
for csig, func, args in (bot.plugs['command'] + bot.plugs['event']):
|
|
input = Input(*out)
|
|
for fsig, sieve in bot.plugs['sieve']:
|
|
try:
|
|
input = sieve(bot, input, func, args)
|
|
except Exception, e:
|
|
print 'filter error:', e
|
|
input = None
|
|
if input == None:
|
|
break
|
|
if input == None:
|
|
continue
|
|
if not printed:
|
|
print '<<<', input.raw
|
|
printed = True
|
|
thread.start_new_thread(FakeBot(bot, input, func).run, ())
|
|
except Queue.Empty:
|
|
pass
|