mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-11-05 10:39:22 +01:00
111 lines
4.1 KiB
Python
111 lines
4.1 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.
|
|
###
|
|
|
|
from fix import *
|
|
|
|
import new
|
|
|
|
import conf
|
|
import debug
|
|
import world
|
|
import ircdb
|
|
import ircmsgs
|
|
import ircutils
|
|
import callbacks
|
|
|
|
def getChannel(msg, args):
|
|
"""Returns the channel the msg came over or the channel given in args.
|
|
|
|
If the channel was given in args, args is modified (the channel is
|
|
removed).
|
|
"""
|
|
if args and ircutils.isChannel(args[0]):
|
|
return args.pop(0)
|
|
elif ircutils.isChannel(msg.args[0]):
|
|
return msg.args[0]
|
|
else:
|
|
raise callbacks.Error, 'Command must be sent in a channel or ' \
|
|
'include a channel in its arguments.'
|
|
|
|
def getArgs(args, needed=1, optional=0):
|
|
"""Take the needed arguments from args.
|
|
|
|
Always returns a list of size needed + optional, filling it with however
|
|
many empty strings is necessary to fill the tuple to the right size.
|
|
|
|
If there aren't enough args even to satisfy needed, raise an error and
|
|
let the caller handle sending the help message.
|
|
"""
|
|
if len(args) < needed:
|
|
raise callbacks.ArgumentError
|
|
if len(args) < needed + optional:
|
|
ret = list(args) + ([''] * (needed + optional - len(args)))
|
|
elif len(args) >= needed + optional:
|
|
ret = list(args[:needed + optional - 1])
|
|
ret.append(' '.join(args[needed + optional - 1:]))
|
|
if len(ret) == 1:
|
|
return ret[0]
|
|
else:
|
|
return ret
|
|
|
|
def checkCapability(f, capability):
|
|
"""Makes sure a user has a certain capability before a command will run."""
|
|
def newf(self, irc, msg, args):
|
|
if ircdb.checkCapability(msg.prefix, capability):
|
|
f(self, irc, msg, args)
|
|
else:
|
|
irc.error(msg, conf.replyNoCapability % capability)
|
|
newf.__doc__ = f.__doc__
|
|
return newf
|
|
|
|
def thread(f):
|
|
"""Makes sure a command spawns a thread when called."""
|
|
def newf(self, irc, msg, args):
|
|
ff = new.instancemethod(f, self, self.__class__)
|
|
t = callbacks.CommandThread(self.callCommand, ff, irc, msg, args)
|
|
t.start()
|
|
newf.__doc__ = f.__doc__
|
|
return newf
|
|
|
|
class CapabilityCheckingPrivmsg(callbacks.Privmsg):
|
|
"""A small subclass of callbacks.Privmsg that checks self.capability
|
|
before allowing any command to be called.
|
|
"""
|
|
capability = '' # To satisfy PyChecker
|
|
def callCommand(self, f, irc, msg, args):
|
|
if ircdb.checkCapability(msg.prefix, self.capability):
|
|
callbacks.Privmsg.callCommand(self, f, irc, msg, args)
|
|
else:
|
|
irc.error(msg, conf.replyNoCapability % self.capability)
|
|
|
|
|
|
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78:
|