mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-11-19 17:09:27 +01:00
794 lines
30 KiB
Python
794 lines
30 KiB
Python
#!/usr/bin/env python
|
|
|
|
###
|
|
# Copyright (c) 2002, Jeremiah Fincher
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions, and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above copyright notice,
|
|
# this list of conditions, and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# * Neither the name of the author of this software nor the name of
|
|
# contributors to this software may be used to endorse or promote products
|
|
# derived from this software without specific prior written consent.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
###
|
|
|
|
"""
|
|
This module contains the basic callbacks for handling PRIVMSGs. Both Privmsg
|
|
and PrivmsgRegexp classes are provided; for offering callbacks based on
|
|
commands and their arguments (much like *nix command line programs) use the
|
|
Privmsg class; for offering callbacks based on regular expressions, use the
|
|
PrivmsgRegexp class. Read their respective docstrings for more information on
|
|
how to use them.
|
|
"""
|
|
|
|
import fix
|
|
|
|
import re
|
|
import copy
|
|
import sets
|
|
import time
|
|
import shlex
|
|
import getopt
|
|
import string
|
|
import inspect
|
|
import textwrap
|
|
import threading
|
|
from cStringIO import StringIO
|
|
|
|
import conf
|
|
import debug
|
|
import utils
|
|
import world
|
|
import ircdb
|
|
import irclib
|
|
import ircmsgs
|
|
import ircutils
|
|
|
|
###
|
|
# Privmsg: handles privmsg commands in a standard fashion.
|
|
###
|
|
def addressed(nick, msg):
|
|
"""If msg is addressed to 'name', returns the portion after the address.
|
|
Otherwise returns the empty string.
|
|
"""
|
|
nick = ircutils.toLower(nick)
|
|
if ircutils.nickEqual(msg.args[0], nick):
|
|
if msg.args[1][0] in conf.prefixChars:
|
|
return msg.args[1][1:].strip()
|
|
else:
|
|
return msg.args[1].strip()
|
|
elif ircutils.toLower(msg.args[1]).startswith(nick):
|
|
try:
|
|
(maybeNick, rest) = msg.args[1].split(None, 1)
|
|
while not ircutils.isNick(maybeNick):
|
|
maybeNick = maybeNick[:-1]
|
|
if ircutils.nickEqual(maybeNick, nick):
|
|
return rest
|
|
else:
|
|
return ''
|
|
except ValueError: # split didn't work.
|
|
return ''
|
|
elif msg.args[1] and msg.args[1][0] in conf.prefixChars:
|
|
return msg.args[1][1:].strip()
|
|
else:
|
|
return ''
|
|
|
|
def canonicalName(command):
|
|
"""Turn a command into its canonical form.
|
|
|
|
Currently, this makes everything lowercase and removes all dashes and
|
|
underscores.
|
|
"""
|
|
return command.translate(string.ascii, '\t -_').lower()
|
|
|
|
def reply(msg, s, prefixName=True, private=False):
|
|
"""Makes a reply to msg with the payload s"""
|
|
s = ircutils.safeArgument(s)
|
|
if ircutils.isChannel(msg.args[0]) and not private:
|
|
if prefixName:
|
|
m = ircmsgs.privmsg(msg.args[0], '%s: %s' % (msg.nick, s))
|
|
else:
|
|
m = ircmsgs.privmsg(msg.args[0], s)
|
|
else:
|
|
m = ircmsgs.privmsg(msg.nick, s)
|
|
return m
|
|
|
|
def error(msg, s):
|
|
"""Makes an error reply to msg with the appropriate error payload."""
|
|
return reply(msg, 'Error: ' + s)
|
|
|
|
class RateLimiter:
|
|
"""This class is used to rate limit replies to certain people, in order to
|
|
prevent abuse of the bot. Basically, you put messages in with the .put
|
|
method, and then take a message out with the .get method. .get may return
|
|
None if there is no message waiting that isn't being rate limited.
|
|
"""
|
|
# lastRequest must be class-global, so each instance of it uses the same
|
|
# information. Otherwise, if it was an instance variable, then rate
|
|
# limiting would only work within a single plugin.
|
|
lastRequest = {}
|
|
def __init__(self):
|
|
self.limited = []
|
|
self.unlimited = []
|
|
|
|
def get(self):
|
|
"""Returns the next un-ratelimited message, or the next rate-limited
|
|
message whose time has come up."""
|
|
if self.unlimited:
|
|
return self.unlimited.pop(0)
|
|
elif self.limited:
|
|
for i in range(len(self.limited)):
|
|
msg = self.limited[i]
|
|
if not self._limit(msg, penalize=False):
|
|
return self.limited.pop(i)
|
|
return None
|
|
else:
|
|
return None
|
|
|
|
def put(self, msg):
|
|
"""Puts a message in for possible ratelimiting."""
|
|
t = self._limit(msg)
|
|
if t and not world.testing:
|
|
s = 'Limiting message from %s for %s seconds' % (msg.prefix, t)
|
|
debug.msg(s, 'normal')
|
|
self.limited.append(msg)
|
|
else:
|
|
self.unlimited.append(msg)
|
|
|
|
def _limit(self, msg, penalize=True):
|
|
if msg.prefix and ircutils.isUserHostmask(msg.prefix):
|
|
(nick, user, host) = ircutils.splitHostmask(msg.prefix)
|
|
key = '@'.join((user, host))
|
|
now = time.time()
|
|
if ircdb.checkCapabilities(msg.prefix, ('owner', 'admin')):
|
|
return 0
|
|
if key in self.lastRequest:
|
|
# Here's how we throttle requests. We keep a dictionary of
|
|
# (lastRequest, wait period) tuples. When a request arrives,
|
|
# we check to see if we have a lastRequest tuple, and if so,
|
|
# we check to make sure that lastRequest was more than wait
|
|
# seconds ago. If not, we're getting flooded, and we set
|
|
# the lastRequest time to the current time and increment wait,
|
|
# thus making it even harder for the flooder to get us to
|
|
# send them messages.
|
|
(t, wait) = self.lastRequest[key]
|
|
if now - t <= wait:
|
|
if penalize:
|
|
newWait = wait + conf.throttleTime
|
|
else:
|
|
newWait = wait - (now - t)
|
|
self.lastRequest[key] = (now, newWait)
|
|
return newWait
|
|
else:
|
|
self.lastRequest[key] = (now, conf.throttleTime)
|
|
return 0
|
|
else:
|
|
self.lastRequest[key] = (now, conf.throttleTime)
|
|
return 0
|
|
else:
|
|
return 0
|
|
|
|
|
|
class Error(Exception):
|
|
"""Generic class for errors in Privmsg callbacks."""
|
|
pass
|
|
|
|
class ArgumentError(Error):
|
|
"""The bot replies with a help message when this is raised."""
|
|
pass
|
|
|
|
class CannotNest(Error):
|
|
"""Exception to be raised by commands that cannot be nested."""
|
|
pass
|
|
|
|
class Tokenizer:
|
|
# This will be used as a global environment to evaluate strings in.
|
|
# Evaluation is, of course, necessary in order to allowed escaped
|
|
# characters to be properly handled.
|
|
#
|
|
# These are the characters valid in a token. Everything printable except
|
|
# double-quote, left-bracket, and right-bracket.
|
|
validChars = string.ascii[33:].translate(string.ascii, '"[]')
|
|
def __init__(self, tokens=''):
|
|
# Add a '|' to tokens to have the pipe syntax.
|
|
self.validChars = self.validChars.translate(string.ascii, tokens)
|
|
|
|
def _handleToken(self, token):
|
|
while token and token[0] == '"' and token[-1] == token[0]:
|
|
if len(token) > 1:
|
|
token = token[1:-1].decode('string_escape') # 2.3+
|
|
else:
|
|
break
|
|
return token
|
|
|
|
def _insideBrackets(self, lexer):
|
|
ret = []
|
|
while True:
|
|
token = lexer.get_token()
|
|
if not token:
|
|
raise SyntaxError, 'Missing "]"'
|
|
elif token == ']':
|
|
return ret
|
|
elif token == '[':
|
|
ret.append(self._insideBrackets(lexer))
|
|
else:
|
|
ret.append(self._handleToken(token))
|
|
return ret
|
|
|
|
def tokenize(self, s):
|
|
"""Tokenizes a string according to supybot's nested argument format."""
|
|
lexer = shlex.shlex(StringIO(s))
|
|
lexer.commenters = ''
|
|
lexer.quotes = '"'
|
|
lexer.wordchars = self.validChars
|
|
args = []
|
|
ends = []
|
|
while True:
|
|
token = lexer.get_token()
|
|
#debug.printf(repr(token))
|
|
if not token:
|
|
break
|
|
elif token == '|':
|
|
if not args:
|
|
raise SyntaxError, '"|" with nothing preceding'
|
|
ends.append(args)
|
|
args = []
|
|
elif token == '[':
|
|
args.append(self._insideBrackets(lexer))
|
|
elif token == ']':
|
|
raise SyntaxError, 'Spurious "["'
|
|
else:
|
|
args.append(self._handleToken(token))
|
|
if ends:
|
|
if not args:
|
|
raise SyntaxError, '"|" with nothing following'
|
|
args.append(ends.pop())
|
|
while ends:
|
|
args[-1].append(ends.pop())
|
|
return args
|
|
|
|
_lastTokenized = None
|
|
_lastTokenizeResult = None
|
|
def tokenize(s):
|
|
"""A utility function to create a Tokenizer and tokenize a string."""
|
|
global _lastTokenized, _lastTokenizeResult
|
|
start = time.time()
|
|
try:
|
|
if s != _lastTokenized:
|
|
_lastTokenized = s
|
|
if conf.enablePipeSyntax:
|
|
tokens = '|'
|
|
else:
|
|
tokens = ''
|
|
_lastTokenizeResult = Tokenizer(tokens).tokenize(s)
|
|
except ValueError, e:
|
|
_lastTokenized = None
|
|
_lastTokenizedResult = None
|
|
raise SyntaxError, str(e)
|
|
#debug.msg('tokenize took %s seconds.' % (time.time() - start), 'verbose')
|
|
return copy.deepcopy(_lastTokenizeResult)
|
|
|
|
def getCommands(tokens):
|
|
"""Given tokens as output by tokenize, returns the command names."""
|
|
L = []
|
|
if tokens and isinstance(tokens, list):
|
|
L.append(tokens[0])
|
|
for elt in tokens:
|
|
L.extend(getCommands(elt))
|
|
return L
|
|
|
|
def findCallbackForCommand(irc, commandName):
|
|
"""Given a command name and an Irc object, returns the callback that
|
|
command is in. Returns None if there is no callback with that command."""
|
|
for callback in irc.callbacks:
|
|
if not isinstance(callback, PrivmsgRegexp):
|
|
if hasattr(callback, 'isCommand'):
|
|
if callback.isCommand(commandName):
|
|
return callback
|
|
return None
|
|
|
|
class IrcObjectProxy:
|
|
"A proxy object to allow proper nested of commands (even threaded ones)."
|
|
def __init__(self, irc, msg, args):
|
|
#debug.printf('__init__: %s' % args)
|
|
if not args:
|
|
irc.reply(msg, '[]')
|
|
else:
|
|
self.irc = irc
|
|
self.msg = msg
|
|
self.args = args
|
|
self.counter = 0
|
|
self.finalEvaled = False
|
|
self.action = False
|
|
self.private = False
|
|
self.prefixName = True
|
|
self.noLengthCheck = False
|
|
world.commandsProcessed += 1
|
|
self.evalArgs()
|
|
|
|
def evalArgs(self):
|
|
while self.counter < len(self.args):
|
|
if type(self.args[self.counter]) == str:
|
|
self.counter += 1
|
|
else:
|
|
IrcObjectProxy(self, self.msg, self.args[self.counter])
|
|
return
|
|
self.finalEval()
|
|
|
|
def finalEval(self):
|
|
if self.finalEvaled:
|
|
raise ValueError, 'finalEval called twice. Odd.'
|
|
self.finalEvaled = True
|
|
originalName = self.args.pop(0)
|
|
name = canonicalName(originalName)
|
|
cb = findCallbackForCommand(self, name)
|
|
try:
|
|
if cb is not None:
|
|
anticap = ircdb.makeAntiCapability(name)
|
|
#debug.printf('Checking for %s' % anticap)
|
|
if ircdb.checkCapability(self.msg.prefix, anticap):
|
|
#debug.printf('Being prevented with anticap')
|
|
debug.msg('Preventing %s from calling %s' % \
|
|
(self.msg.nick, name), 'normal')
|
|
s = conf.replyNoCapability % name
|
|
self.error(self.msg, s, private=True)
|
|
return
|
|
recipient = self.msg.args[0]
|
|
if ircutils.isChannel(recipient):
|
|
chancap = ircdb.makeChannelCapability(recipient, anticap)
|
|
#debug.printf('Checking for %s' % chancap)
|
|
if ircdb.checkCapability(self.msg.prefix, chancap):
|
|
#debug.printf('Being prevented with chancap')
|
|
debug.msg('Preventing %s from calling %s' % \
|
|
(self.msg.nick, name), 'normal')
|
|
s = conf.replyNoCapability % name
|
|
self.error(self.msg, s, private=True)
|
|
return
|
|
command = getattr(cb, name)
|
|
if cb.threaded:
|
|
t = CommandThread(cb.callCommand, command,
|
|
self, self.msg, self.args)
|
|
t.start()
|
|
else:
|
|
cb.callCommand(command, self, self.msg, self.args)
|
|
else:
|
|
self.args.insert(0, originalName)
|
|
if not isinstance(self.irc, irclib.Irc):
|
|
# If self.irc is an actual irclib.Irc, then this is the
|
|
# first command given, and should be ignored as usual.
|
|
self.reply(self.msg, '[%s]' % ' '.join(self.args))
|
|
except (getopt.GetoptError, ArgumentError):
|
|
if hasattr(command, '__doc__'):
|
|
s = '%s %s' % (name, command.__doc__.splitlines()[0])
|
|
else:
|
|
s = 'Invalid arguments for %s.' % name
|
|
self.reply(self.msg, s)
|
|
except CannotNest, e:
|
|
if not isinstance(self.irc, irclib.Irc):
|
|
self.error(self.msg, 'Command %r cannot be nested.' % name)
|
|
except (SyntaxError, Error), e:
|
|
self.reply(self.msg, debug.exnToString(e))
|
|
except Exception, e:
|
|
debug.recoverableException()
|
|
self.error(self.msg, debug.exnToString(e))
|
|
|
|
def reply(self, msg, s, noLengthCheck=False, prefixName=True,
|
|
action=False, private=False):
|
|
"""reply(msg, text) -> replies to msg with text
|
|
|
|
Keyword arguments:
|
|
noLengthCheck=False: True if the length shouldn't be checked
|
|
(used for 'more' handling)
|
|
prefixName=True: False if the nick shouldn't be prefixed to the
|
|
reply.
|
|
action=False: True if the reply should be an action.
|
|
private=False: True if the reply should be in private.
|
|
"""
|
|
# These use |= or &= based on whether or not they default to True or
|
|
# False. Those that default to True use &=; those that default to
|
|
# False use |=.
|
|
self.action |= action
|
|
self.private |= private
|
|
self.prefixName &= prefixName
|
|
self.noLengthCheck |= noLengthCheck
|
|
if self.finalEvaled:
|
|
if isinstance(self.irc, self.__class__):
|
|
self.irc.reply(msg, s, self.noLengthCheck, self.prefixName,
|
|
self.action, self.private)
|
|
elif self.noLengthCheck:
|
|
self.irc.queueMsg(reply(msg, s, self.prefixName))
|
|
elif self.action:
|
|
self.irc.queueMsg(ircmsgs.action(msg.args[0], s))
|
|
else:
|
|
# The size of a PRIVMSG is:
|
|
# 1 for the colon
|
|
# len(prefix)
|
|
# 1 for the space
|
|
# 7 for the PRIVMSG
|
|
# 1 for the space
|
|
# len(target)
|
|
# 1 for the space
|
|
# 1 for the colon
|
|
# len(payload)
|
|
# 2 for the \r\n
|
|
# So non-variable stuff it's 1+1+7+1+1+1+2, or 14
|
|
# We'll estimate the channel length at 30, and we'll know the
|
|
# prefix length exactly. We also might append the string
|
|
# " (more)" to the end, so that's 7 more characters.
|
|
# 512 - 51 == 461.
|
|
s = ircutils.safeArgument(s)
|
|
allowedLength = 450 - len(self.irc.prefix)
|
|
msgs = textwrap.wrap(s, allowedLength-30) # -30 is for "nick:"
|
|
msgs.reverse()
|
|
response = msgs.pop()
|
|
if msgs:
|
|
response += ' \x02(%s)\x0F' % \
|
|
utils.nItems(len(msgs), 'message', 'more')
|
|
mask = msg.prefix.split('!', 1)[1]
|
|
Privmsg._mores[mask] = msgs
|
|
private = self.private or not ircutils.isChannel(msg.args[0])
|
|
Privmsg._mores[msg.nick] = (private, msgs)
|
|
if self.private:
|
|
self.irc.queueMsg(ircmsgs.privmsg(msg.nick, response))
|
|
else:
|
|
self.irc.queueMsg(reply(msg, response, self.prefixName))
|
|
else:
|
|
self.args[self.counter] = s
|
|
self.evalArgs()
|
|
|
|
def error(self, msg, s, private=False):
|
|
"""error(msg, text) -> replies to msg with an error message of text.
|
|
|
|
Keyword arguments:
|
|
private=False: True if the error should be given in private.
|
|
"""
|
|
if isinstance(self.irc, self.__class__):
|
|
self.irc.error(msg, s, private)
|
|
else:
|
|
s = 'Error: ' + s
|
|
if private or conf.errorReplyPrivate:
|
|
self.irc.queueMsg(ircmsgs.privmsg(msg.nick, s))
|
|
else:
|
|
self.irc.queueMsg(reply(msg, s))
|
|
|
|
def killProxy(self):
|
|
"""Kills this proxy object and all its parents."""
|
|
if not isinstance(self.irc, irclib.Irc):
|
|
self.irc.killProxy()
|
|
self.__dict__ = {}
|
|
|
|
def getRealIrc(self):
|
|
"""Returns the real irclib.Irc object underlying this proxy chain."""
|
|
if isinstance(self.irc, irclib.Irc):
|
|
return self.irc
|
|
else:
|
|
return self.irc.getRealIrc()
|
|
|
|
def __getattr__(self, attr):
|
|
return getattr(self.irc, attr)
|
|
|
|
|
|
class CommandThread(threading.Thread):
|
|
"""Just does some extra logging and error-recovery for commands that need
|
|
to run in threads.
|
|
"""
|
|
def __init__(self, callCommand, command, irc, msg, args, *L):
|
|
self.command = command
|
|
world.threadsSpawned += 1
|
|
try:
|
|
self.commandName = command.im_func.func_name
|
|
except AttributeError:
|
|
self.commandName = command.__name__
|
|
try:
|
|
self.className = command.im_class.__name__
|
|
except AttributeError:
|
|
self.className = '<unknown>'
|
|
name = '%s.%s with args %r' % (self.className, self.commandName, args)
|
|
threading.Thread.__init__(self, target=callCommand, name=name,
|
|
args=(command, irc, msg, args)+L)
|
|
debug.msg('Spawning thread %s' % name, 'verbose')
|
|
self.irc = irc
|
|
self.msg = msg
|
|
self.setDaemon(True)
|
|
|
|
def run(self):
|
|
try:
|
|
threading.Thread.run(self)
|
|
except (getopt.GetoptError, ArgumentError):
|
|
if hasattr(self.command, '__doc__'):
|
|
help = self.command.__doc__.splitlines()[0]
|
|
s = '%s %s' % (self.commandName, help)
|
|
else:
|
|
s = 'Invalid arguments for %s.' % self.commandName
|
|
self.irc.reply(self.msg, s)
|
|
except CannotNest:
|
|
if not isinstance(self.irc.irc, irclib.Irc):
|
|
s = 'Command %r cannot be nested.' % self.commandName
|
|
self.irc.error(self.msg, s)
|
|
except (SyntaxError, Error), e:
|
|
self.irc.reply(self.msg, debug.exnToString(e))
|
|
except Exception, e:
|
|
debug.recoverableException()
|
|
self.irc.error(self.msg, debug.exnToString(e))
|
|
|
|
|
|
class ConfigIrcProxy(object):
|
|
"""Used as a proxy Irc object during configuration. """
|
|
def __init__(self, irc):
|
|
self.__dict__['irc'] = irc
|
|
|
|
def reply(self, msg, s, *args):
|
|
return None
|
|
|
|
def error(self, msg, s, *args):
|
|
debug.msg('ConfigIrcProxy saw an error: %s' % s, 'normal')
|
|
|
|
def getRealIrc(self):
|
|
irc = self.__dict__['irc']
|
|
while(hasattr(irc, 'getRealIrc')):
|
|
irc = irc.getRealIrc()
|
|
return irc
|
|
|
|
def __getattr__(self, attr):
|
|
return getattr(self.getRealIrc(), attr)
|
|
|
|
def __setattr__(self, attr, value):
|
|
setattr(self.getRealIrc(), attr, value)
|
|
|
|
|
|
class Privmsg(irclib.IrcCallback):
|
|
"""Base class for all Privmsg handlers."""
|
|
threaded = False
|
|
public = True
|
|
commandArgs = ['self', 'irc', 'msg', 'args']
|
|
_mores = {} # This must be class-scope, so all subclasses use the same one.
|
|
def __init__(self):
|
|
self.rateLimiter = RateLimiter()
|
|
self.Proxy = IrcObjectProxy
|
|
|
|
def configure(self, irc):
|
|
fakeIrc = ConfigIrcProxy(irc)
|
|
for args in conf.commandsOnStart:
|
|
args = args[:]
|
|
command = args.pop(0)
|
|
if self.isCommand(command):
|
|
#debug.printf('%s: %r' % (command, args))
|
|
method = getattr(self, command)
|
|
line = '%s %s' % (command, ' '.join(map(utils.dqrepr, args)))
|
|
msg = ircmsgs.privmsg(fakeIrc.nick, line, fakeIrc.prefix)
|
|
try:
|
|
world.startup = True
|
|
method(fakeIrc, msg, args)
|
|
finally:
|
|
world.startup = False
|
|
|
|
def __call__(self, irc, msg):
|
|
irclib.IrcCallback.__call__(self, irc, msg)
|
|
# Now, if there's anything in the rateLimiter...
|
|
msg = self.rateLimiter.get()
|
|
while msg:
|
|
s = addressed(irc.nick, msg)
|
|
try:
|
|
args = tokenize(s)
|
|
self.Proxy(irc, msg, args)
|
|
except SyntaxError, e:
|
|
irc.queueMsg(reply(msg, str(e)))
|
|
msg = self.rateLimiter.get()
|
|
|
|
def isCommand(self, methodName):
|
|
# This function is ugly, but I don't want users to call methods like
|
|
# doPrivmsg or __init__ or whatever, and this is good to stop them.
|
|
if hasattr(self, methodName):
|
|
method = getattr(self, methodName)
|
|
if inspect.ismethod(method):
|
|
code = method.im_func.func_code
|
|
return inspect.getargs(code)[0] == self.commandArgs
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
def callCommand(self, f, irc, msg, *L):
|
|
# Exceptions aren't caught here because IrcObjectProxy.finalEval
|
|
# catches them and does The Right Thing.
|
|
start = time.time()
|
|
f(irc, msg, *L)
|
|
elapsed = time.time() - start
|
|
funcname = '%s.%s' % (f.im_class.__name__, f.im_func.func_name)
|
|
debug.msg('%s took %s seconds' % (funcname, elapsed), 'verbose')
|
|
|
|
def doPrivmsg(self, irc, msg, rateLimit=True):
|
|
s = addressed(irc.nick, msg)
|
|
#debug.printf('Privmsg.doPrivmsg: s == %r' % s)
|
|
if s:
|
|
recipient = msg.args[0]
|
|
if ircdb.checkIgnored(msg.prefix, recipient):
|
|
debug.msg('Privmsg.doPrivmsg: ignoring %s.' % msg.prefix)
|
|
return
|
|
try:
|
|
args = tokenize(s)
|
|
except SyntaxError, e:
|
|
irc.queueMsg(reply(msg, debug.exnToString(e)))
|
|
return
|
|
if args and isinstance(args[0], str):
|
|
args[0] = canonicalName(args[0])
|
|
if self.isCommand(args[0]):
|
|
if rateLimit:
|
|
self.rateLimiter.put(msg)
|
|
msg = self.rateLimiter.get()
|
|
if msg:
|
|
if conf.replyWhenNotCommand:
|
|
for command in getCommands(args):
|
|
command = canonicalName(command)
|
|
if not findCallbackForCommand(irc, command):
|
|
return
|
|
self.Proxy(irc, msg, args)
|
|
|
|
|
|
class IrcObjectProxyRegexp:
|
|
def __init__(self, irc, *args):
|
|
self.irc = irc
|
|
|
|
def error(self, msg, s, private=False):
|
|
private = private or conf.errorReplyPrivate
|
|
self.reply(msg, 'Error: ' + s, private=private)
|
|
|
|
def reply(self, msg, s, prefixName=True, action=False, private=False,):
|
|
if action:
|
|
self.irc.queueMsg(ircmsgs.action(ircutils.replyTo(msg), s))
|
|
else:
|
|
self.irc.queueMsg(reply(msg, s, private=private,
|
|
prefixName=prefixName))
|
|
|
|
def __getattr__(self, attr):
|
|
return getattr(self.irc, attr)
|
|
|
|
|
|
class PrivmsgRegexp(Privmsg):
|
|
"""A class to allow a person to create regular expression callbacks.
|
|
|
|
Much more primitive, but more flexible than the 'normal' method of using
|
|
the Privmsg class and its lexer, PrivmsgRegexp allows you to write
|
|
callbacks that aren't addressed to the bot, for instance. There are, of
|
|
course, several other possibilities. Callbacks are registered with a
|
|
string (the regular expression) and a function to be called (with the Irc
|
|
object, the IrcMsg object, and the match object) when the regular
|
|
expression matches. Callbacks must have the signature (self, irc, msg,
|
|
match) to be counted as such.
|
|
|
|
A class-level flags attribute is used to determine what regexp flags to
|
|
compile the regular expressions with. By default, it's re.I, which means
|
|
regular expressions are by default case-insensitive.
|
|
|
|
If you have a standard command-type callback, though, Privmsg is a much
|
|
better class to use, at the very least for consistency's sake, but also
|
|
because it's much more easily coded and maintained.
|
|
"""
|
|
threaded = False # Again, like Privmsg...
|
|
flags = re.I
|
|
onlyFirstMatch = False
|
|
commandArgs = ['self', 'irc', 'msg', 'match']
|
|
def __init__(self):
|
|
Privmsg.__init__(self)
|
|
self.Proxy = IrcObjectProxyRegexp
|
|
self.res = []
|
|
#for name, value in self.__class__.__dict__.iteritems():
|
|
for name, value in self.__class__.__dict__.items():
|
|
value = getattr(self, name)
|
|
if self.isCommand(name):
|
|
try:
|
|
r = re.compile(value.__doc__, self.flags)
|
|
self.res.append((r, value))
|
|
except re.error, e:
|
|
s = '%s.%s has an invalid regexp %s: %s' % \
|
|
(self.__class__.__name__, name,
|
|
value.__doc__, debug.exnToString(e))
|
|
debug.msg(s)
|
|
self.res.sort(lambda (r1, m1), (r2, m2): cmp(m1.__name__, m2.__name__))
|
|
|
|
def callCommand(self, irc, msg, *L):
|
|
try:
|
|
Privmsg.callCommand(self, irc, msg, *L)
|
|
except Exception, e:
|
|
debug.recoverableException()
|
|
irc.error(msg, debug.exnToString(e))
|
|
|
|
def doPrivmsg(self, irc, msg):
|
|
if ircdb.checkIgnored(msg.prefix, msg.args[0]):
|
|
debug.msg('PrivmsgRegexp.doPrivmsg: ignoring %s' % msg.prefix)
|
|
return
|
|
fed = False
|
|
for (r, method) in self.res:
|
|
spans = sets.Set()
|
|
for m in r.finditer(msg.args[1]):
|
|
# There's a bug in finditer: http://www.python.org/sf/817234
|
|
if m.span() in spans:
|
|
break
|
|
else:
|
|
spans.add(m.span())
|
|
if not fed:
|
|
fed = True
|
|
self.rateLimiter.put(msg)
|
|
msg = self.rateLimiter.get()
|
|
if msg:
|
|
self.callCommand(method, irc, msg, m)
|
|
if self.onlyFirstMatch:
|
|
return
|
|
|
|
|
|
class PrivmsgCommandAndRegexp(Privmsg):
|
|
"""Same as Privmsg, except allows the user to also include regexp-based
|
|
callbacks. All regexp-based callbacks must be specified in a sets.Set
|
|
(or list) attribute "regexps".
|
|
"""
|
|
flags = re.I
|
|
regexps = () # Use sets.Set() in your own callbacks.
|
|
addressedRegexps = () # Ditto on the sets.Sets() idea.
|
|
def __init__(self):
|
|
Privmsg.__init__(self)
|
|
self.res = []
|
|
self.addressedRes = []
|
|
for name in self.regexps:
|
|
method = getattr(self, name)
|
|
r = re.compile(method.__doc__, self.flags)
|
|
self.res.append((r, method))
|
|
for name in self.addressedRegexps:
|
|
method = getattr(self, name)
|
|
r = re.compile(method.__doc__, self.flags)
|
|
self.addressedRes.append((r, method))
|
|
self.res.sort(lambda (r1, m1), (r2, m2): cmp(m1.__name__, m2.__name__))
|
|
self.addressedRes.sort(lambda (r1, m1), (r2, m2): cmp(m1.__name__,
|
|
m2.__name__))
|
|
def callCommand(self, f, irc, msg, *L):
|
|
try:
|
|
Privmsg.callCommand(self, f, irc, msg, *L)
|
|
except Exception, e:
|
|
irc.error(msg, debug.exnToString(e))
|
|
debug.recoverableException()
|
|
|
|
def doPrivmsg(self, irc, msg):
|
|
if ircdb.checkIgnored(msg.prefix, msg.args[0]):
|
|
return
|
|
fed = False
|
|
for (r, method) in self.res:
|
|
for m in r.finditer(msg.args[1]):
|
|
if not fed:
|
|
fed = True
|
|
self.rateLimiter.put(msg)
|
|
msg = self.rateLimiter.get()
|
|
if msg:
|
|
proxy = IrcObjectProxyRegexp(irc)
|
|
self.callCommand(method, proxy, msg, m)
|
|
s = addressed(irc.nick, msg)
|
|
if s:
|
|
for (r, method) in self.addressedRes:
|
|
for m in r.finditer(s):
|
|
if not fed:
|
|
fed = True
|
|
self.rateLimiter.put(msg)
|
|
msg = self.rateLimiter.get()
|
|
if msg:
|
|
proxy = IrcObjectProxyRegexp(irc)
|
|
self.callCommand(method, proxy, msg, m)
|
|
Privmsg.doPrivmsg(self, irc, msg, rateLimit=(not fed))
|
|
|
|
|
|
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78:
|