mirror of https://github.com/Mikaela/Limnoria.git
280 lines
9.9 KiB
Python
280 lines
9.9 KiB
Python
###
|
|
# Copyright (c) 2002-2004, 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.
|
|
###
|
|
|
|
"""
|
|
Handles configuration of the bot while it's running.
|
|
"""
|
|
|
|
import supybot
|
|
|
|
__revision__ = "$Id$"
|
|
__author__ = supybot.authors.jemfinch
|
|
|
|
import os
|
|
import getopt
|
|
import signal
|
|
|
|
import supybot.log as log
|
|
import supybot.conf as conf
|
|
import supybot.utils as utils
|
|
import supybot.world as world
|
|
import supybot.ircdb as ircdb
|
|
import supybot.ircutils as ircutils
|
|
import supybot.privmsgs as privmsgs
|
|
import supybot.registry as registry
|
|
import supybot.callbacks as callbacks
|
|
|
|
###
|
|
# Now, to setup the registry.
|
|
###
|
|
|
|
#class InvalidRegistryName(callbacks.Error):
|
|
# pass
|
|
|
|
def getWrapper(name):
|
|
parts = registry.split(name)
|
|
if not parts or parts[0] not in ('supybot', 'users'):
|
|
raise InvalidRegistryName, name
|
|
group = getattr(conf, parts.pop(0))
|
|
while parts:
|
|
try:
|
|
group = group.get(parts.pop(0))
|
|
# We'll catch registry.InvalidRegistryName and re-raise it here so
|
|
# that we have a useful error message for the user.
|
|
except (registry.NonExistentRegistryEntry,
|
|
registry.InvalidRegistryName):
|
|
raise registry.InvalidRegistryName, name
|
|
return group
|
|
|
|
def getCapability(name):
|
|
capability = 'owner' # Default to requiring the owner capability.
|
|
parts = registry.split(name)
|
|
while parts:
|
|
part = parts.pop()
|
|
if ircutils.isChannel(part):
|
|
# If a registry value has a channel in it, it requires a channel.op
|
|
# capability, or so we assume. We'll see if we're proven wrong.
|
|
capability = ircdb.makeChannelCapability(part, 'op')
|
|
### Do more later, for specific capabilities/sections.
|
|
return capability
|
|
|
|
def _reload():
|
|
ircdb.users.reload()
|
|
ircdb.channels.reload()
|
|
registry.open(world.registryFilename)
|
|
|
|
def _hupHandler(sig, frame):
|
|
log.info('Received SIGHUP, reloading configuration.')
|
|
_reload()
|
|
|
|
if os.name == 'posix':
|
|
signal.signal(signal.SIGHUP, _hupHandler)
|
|
|
|
|
|
class Config(callbacks.Privmsg):
|
|
def callCommand(self, name, irc, msg, *L, **kwargs):
|
|
#XXX For some reason, that confuses jamessan, InvalidRegistryName
|
|
# is not really being caught here, but registry.InvalidRegistryName
|
|
# is. So, we'll just re-raise registry.InvalidRegsitryName instead
|
|
# of using Config.InvalidRegistryName
|
|
try:
|
|
super(Config, self).callCommand(name, irc, msg, *L, **kwargs)
|
|
except registry.InvalidRegistryName, e:
|
|
irc.errorInvalid('configuration variable', str(e))
|
|
except registry.InvalidRegistryValue, e:
|
|
irc.error(str(e))
|
|
|
|
def _canonicalizeName(self, name):
|
|
if name.startswith('conf.'):
|
|
name = name[5:]
|
|
if not name.startswith('supybot') and not name.startswith('users'):
|
|
name = 'supybot.' + name
|
|
return name
|
|
|
|
def _list(self, name):
|
|
name = self._canonicalizeName(name)
|
|
group = getWrapper(name)
|
|
L = []
|
|
for (vname, v) in group._children.iteritems():
|
|
if hasattr(v, 'channelValue') and v.channelValue:
|
|
vname = '#' + vname
|
|
if v._added and not all(ircutils.isChannel, v._added):
|
|
vname = '@' + vname
|
|
L.append(vname)
|
|
utils.sortBy(str.lower, L)
|
|
return L
|
|
|
|
def list(self, irc, msg, args):
|
|
"""<group>
|
|
|
|
Returns the configuration variables available under the given
|
|
configuration <group>. Subgroups are indicated by a preceding @.
|
|
"""
|
|
name = privmsgs.getArgs(args)
|
|
L = self._list(name)
|
|
if L:
|
|
irc.reply(utils.commaAndify(L))
|
|
else:
|
|
irc.error('There don\'t seem to be any values in %s.' % name)
|
|
|
|
def search(self, irc, msg, args):
|
|
"""<word>
|
|
|
|
Searches for <word> in the current configuration variables.
|
|
"""
|
|
word = privmsgs.getArgs(args)
|
|
word = word.lower()
|
|
L = []
|
|
for (name, _) in conf.supybot.getValues(getChildren=True):
|
|
if word in name.lower():
|
|
possibleChannel = registry.split(name)[-1]
|
|
if not ircutils.isChannel(possibleChannel):
|
|
L.append(name)
|
|
if L:
|
|
irc.reply(utils.commaAndify(L))
|
|
else:
|
|
irc.reply('There were no matching configuration variables.')
|
|
|
|
def config(self, irc, msg, args):
|
|
"""<name> [<value>]
|
|
|
|
If <value> is given, sets the value of <name> to <value>. Otherwise,
|
|
returns the current value of <name>. You may omit the leading
|
|
"supybot." in the name if you so choose.
|
|
"""
|
|
if len(args) >= 2:
|
|
self._set(irc, msg, args)
|
|
else:
|
|
self._get(irc, msg, args)
|
|
|
|
def channel(self, irc, msg, args):
|
|
"""[<channel>] <name> [<value>]
|
|
|
|
If <value> is given, sets the channel configuration variable for <name>
|
|
to <value> for <channel>. Otherwise, returns the current channel
|
|
configuration value of <name>. <channel> is only necessary if the
|
|
message isn't sent in the channel itself."""
|
|
channel = privmsgs.getChannel(msg, args)
|
|
if not args:
|
|
raise callbacks.ArgumentError
|
|
args[0] = self._canonicalizeName(args[0])
|
|
wrapper = getWrapper(args[0])
|
|
if not wrapper.channelValue:
|
|
irc.error('That configuration variable is not a channel-specific '
|
|
'configuration variable.')
|
|
return
|
|
components = registry.split(args[0])
|
|
components.append(channel)
|
|
args[0] = registry.join(components)
|
|
self.config(irc, msg, args)
|
|
|
|
def _get(self, irc, msg, args):
|
|
"""<name>
|
|
|
|
Shows the current value of the configuration variable <name>.
|
|
"""
|
|
name = privmsgs.getArgs(args)
|
|
name = self._canonicalizeName(name)
|
|
wrapper = getWrapper(name)
|
|
if hasattr(wrapper, 'value'):
|
|
if not wrapper._private:
|
|
irc.reply(str(wrapper) or ' ')
|
|
else:
|
|
capability = getCapability(name)
|
|
if ircdb.checkCapability(msg.prefix, capability):
|
|
irc.reply(str(wrapper), private=True)
|
|
else:
|
|
irc.errorNoCapability(capability)
|
|
else:
|
|
irc.error('That registry variable has no value. Use the list '
|
|
'command in this plugin to see what values are '
|
|
'available in this group.')
|
|
|
|
def _set(self, irc, msg, args):
|
|
"""<name> <value>
|
|
|
|
Sets the current value of the configuration variable <name> to <value>.
|
|
"""
|
|
(name, value) = privmsgs.getArgs(args, required=2)
|
|
name = self._canonicalizeName(name)
|
|
capability = getCapability(name)
|
|
if ircdb.checkCapability(msg.prefix, capability):
|
|
wrapper = getWrapper(name)
|
|
wrapper.set(value)
|
|
irc.replySuccess()
|
|
else:
|
|
irc.errorNoCapability(capability)
|
|
|
|
def help(self, irc, msg, args):
|
|
"""<name>
|
|
|
|
Returns the description of the configuration variable <name>.
|
|
"""
|
|
name = privmsgs.getArgs(args)
|
|
name = self._canonicalizeName(name)
|
|
wrapper = getWrapper(name)
|
|
if hasattr(wrapper, 'help'):
|
|
s = wrapper.help()
|
|
if s:
|
|
if hasattr(wrapper, 'value') and not wrapper._private:
|
|
s += ' (Current value: %s)' % wrapper
|
|
irc.reply(s)
|
|
else:
|
|
irc.reply('That configuration group exists, but seems to have '
|
|
'no help. Try "config list %s" to see if it has '
|
|
'any children values.')
|
|
else:
|
|
irc.error('%s has no help.' % name)
|
|
|
|
def default(self, irc, msg, args):
|
|
"""<name>
|
|
|
|
Returns the default value of the configuration variable <name>.
|
|
"""
|
|
name = privmsgs.getArgs(args)
|
|
name = self._canonicalizeName(name)
|
|
wrapper = getWrapper(name)
|
|
v = wrapper.__class__(wrapper._default, '')
|
|
irc.reply(str(v))
|
|
|
|
def reload(self, irc, msg, args):
|
|
"""takes no arguments
|
|
|
|
Reloads the various configuration files (user database, channel
|
|
database, registry, etc.).
|
|
"""
|
|
_reload() # This was factored out for SIGHUP handling.
|
|
irc.replySuccess()
|
|
reload = privmsgs.checkCapability(reload, 'owner')
|
|
|
|
|
|
Class = Config
|
|
|
|
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78:
|