139 lines
5.8 KiB
Python
139 lines
5.8 KiB
Python
|
from irclib import irclib
|
||
|
import re
|
||
|
|
||
|
class Module(object):
|
||
|
# Base class used for creating classes that have real functionality.
|
||
|
|
||
|
def __init__(self, config, server, modlist, rehash):
|
||
|
# Constructor for a feature module. Inheritors should not do anything special
|
||
|
# here, instead they should implement register_handlers and do, or else this will
|
||
|
# be a very uneventful affair.
|
||
|
#
|
||
|
# Classes that are interested in allowing an indirect call to their do routine
|
||
|
# should add themselves to modlist inside their __init__. This will allow other
|
||
|
# modules to call do and see if anything can handle text they may have seen (such
|
||
|
# as in recursive commands).
|
||
|
|
||
|
self.config = config
|
||
|
self.modlist = modlist
|
||
|
self.register_handlers(server)
|
||
|
self.botserver = config.get('IRC', 'server')
|
||
|
self.botport = config.getint('IRC', 'port')
|
||
|
self.botnick = config.get('IRC', 'nick')
|
||
|
self.botircname = config.get('IRC', 'name')
|
||
|
self.rehash = rehash
|
||
|
|
||
|
def register_handlers(self, server):
|
||
|
# This is called by __init__ and sets up server.add_global_handlers. Classes
|
||
|
# inheriting from Module should implement this and set up the appropriate handlers,
|
||
|
# e.g.:
|
||
|
#
|
||
|
# server.add_global_handler('privmsg', self.on_privmsg)
|
||
|
#
|
||
|
# Module.on_pubmsg and Module.on_privmsg are defined so far, the rest, you're on your
|
||
|
# own.
|
||
|
|
||
|
print "looks like someone forgot to implement register_handlers!"
|
||
|
|
||
|
def on_pubmsg(self, connection, event):
|
||
|
# Does some variable setup and initial sanity checking before calling Module.do,
|
||
|
# which should be implemented by subclasses and what can be ultimately responsible
|
||
|
# for the work. Of course, you are free to reimplement on_pubmsg on your own too.
|
||
|
|
||
|
nick = irclib.nm_to_n(event.source())
|
||
|
userhost = irclib.nm_to_uh(event.source())
|
||
|
replypath = event.target()
|
||
|
what = event.arguments()[0]
|
||
|
|
||
|
admin_unlocked = False
|
||
|
|
||
|
try:
|
||
|
if userhost == self.config.get('admin', 'userhost'):
|
||
|
admin_unlocked = True
|
||
|
except NoOptionError: pass
|
||
|
|
||
|
# only do commands if the bot has been addressed directly
|
||
|
addressed_pattern = '^' + connection.get_nickname() + '[:,]?\s+'
|
||
|
addressed_re = re.compile(addressed_pattern)
|
||
|
|
||
|
if not addressed_re.match(what):
|
||
|
return
|
||
|
else:
|
||
|
what = addressed_re.sub('', what)
|
||
|
|
||
|
if replypath is not None:
|
||
|
what = self.try_recursion(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
||
|
|
||
|
self.do(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
||
|
|
||
|
def on_privmsg(self, connection, event):
|
||
|
# Does some variable setup and initial sanity checking before calling Module.do,
|
||
|
# which should be implemented by subclasses and what can be ultimately responsible
|
||
|
# for the work. Of course, you are free to reimplement on_privmsg on your own too.
|
||
|
|
||
|
nick = irclib.nm_to_n(event.source())
|
||
|
userhost = irclib.nm_to_uh(event.source())
|
||
|
replypath = nick
|
||
|
what = event.arguments()[0]
|
||
|
|
||
|
admin_unlocked = False
|
||
|
|
||
|
try:
|
||
|
if userhost == self.config.get('admin', 'userhost'):
|
||
|
admin_unlocked = True
|
||
|
except NoOptionError: pass
|
||
|
|
||
|
if replypath is not None:
|
||
|
what = self.try_recursion(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
||
|
|
||
|
self.do(connection, event, nick, userhost, replypath, what, admin_unlocked)
|
||
|
|
||
|
def try_recursion(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
|
||
|
# Upon seeing a line intended for this module, see if there are subcommands
|
||
|
# that we should do what is basically a text replacement on. The intent is to
|
||
|
# allow things like the following:
|
||
|
#
|
||
|
# command arg1 [anothercommand arg1 arg2]
|
||
|
#
|
||
|
# where the output of anothercommand is command's arg2..n. It's mostly for
|
||
|
# amusement purposes, but maybe there are legitimate uses. This is intended to
|
||
|
# be attempted after you've determined the line should be handled by your module.
|
||
|
|
||
|
start_idx = what.find('[')
|
||
|
subcmd = what[start_idx+1:]
|
||
|
end_idx = subcmd.rfind(']')
|
||
|
subcmd = subcmd[:end_idx]
|
||
|
|
||
|
attempt = what
|
||
|
|
||
|
if start_idx == -1 or end_idx == -1:
|
||
|
# no nested commands at all if replypath is a real value, so don't do a damn thing
|
||
|
if replypath is not None:
|
||
|
return attempt
|
||
|
# no more replacements found, see if what we had is workable
|
||
|
else:
|
||
|
for module in self.modlist:
|
||
|
ret = module.do(connection, event, nick, userhost, None, attempt, admin_unlocked)
|
||
|
if ret is not None:
|
||
|
return ret
|
||
|
|
||
|
# if we got here, it's not workable. just return what we got
|
||
|
return attempt
|
||
|
else:
|
||
|
# we have a subcmd, see if there's another one nested
|
||
|
ret = self.try_recursion(connection, event, nick, userhost, None, subcmd, admin_unlocked)
|
||
|
if ret is not None:
|
||
|
blarg = attempt.replace('['+subcmd+']', ret)
|
||
|
if replypath is not None:
|
||
|
return blarg
|
||
|
else:
|
||
|
return self.try_recursion(connection, event, nick, userhost, None, blarg, admin_unlocked)
|
||
|
else:
|
||
|
return attempt
|
||
|
|
||
|
def do(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
|
||
|
# Implement this method in your subclass to have a fairly-automatic hook into
|
||
|
# IRC functionality. This is called by the default on_pubmsg and on_privmsg
|
||
|
|
||
|
print "looks like someone forgot to implement do!"
|