Limnoria/plugins/Topic.py

305 lines
12 KiB
Python
Raw Normal View History

2003-03-12 07:26:59 +01:00
#!/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.
###
"""
Provides commands for manipulating channel topics.
"""
2003-11-25 09:23:47 +01:00
__revision__ = "$Id$"
import plugins
2003-03-12 07:26:59 +01:00
import re
import random
2003-10-11 23:03:02 +02:00
import conf
2003-06-02 08:41:35 +02:00
import utils
2003-03-12 07:26:59 +01:00
import ircdb
import ircmsgs
2003-12-02 17:26:08 +01:00
import plugins
2003-03-12 07:26:59 +01:00
import privmsgs
2004-01-26 20:17:59 +01:00
import registry
2003-03-12 07:26:59 +01:00
import callbacks
2004-01-26 20:17:59 +01:00
conf.registerPlugin('Topic')
conf.registerChannelValue(conf.supybot.plugins.Topic, 'separator',
registry.StringSurroundedBySpaces(' || ', """Determines what separator is
used between individually added topics in the channel topic."""))
class Topic(callbacks.Privmsg):
2003-03-12 07:26:59 +01:00
topicFormatter = '%s (%s)'
topicUnformatter = re.compile('(.*) \((\S+)\)')
2003-12-02 17:26:08 +01:00
def _splitTopic(self, topic, channel):
2004-01-26 20:17:59 +01:00
separator = self.registryValue('separator', channel)
2003-12-02 17:26:08 +01:00
return filter(None, topic.split(separator))
def _joinTopic(self, topics, channel):
2004-01-26 20:17:59 +01:00
separator = self.registryValue('separator', channel)
2003-12-02 17:26:08 +01:00
return separator.join(topics)
def _unformatTopic(self, topic, channel):
m = self.topicUnformatter.match(topic)
if m:
return (m.group(1), m.group(2))
else:
return (topic, '')
2003-10-21 06:03:25 +02:00
def add(self, irc, msg, args, channel):
"""[<channel>] <topic>
Adds <topic> to the topics for <channel>. <channel> is only necessary
if the message isn't sent in the channel itself.
"""
2003-03-12 07:26:59 +01:00
topic = privmsgs.getArgs(args)
2004-01-26 20:17:59 +01:00
separator = self.registryValue('separator', channel)
2003-12-02 17:26:08 +01:00
if separator in topic:
s = 'You can\'t have %s in your topic' % separator
irc.error(s)
return
currentTopic = irc.state.getTopic(channel)
try:
name = ircdb.users.getUser(msg.prefix).name
except KeyError:
name = msg.nick
formattedTopic = self.topicFormatter % (topic, name)
if currentTopic:
2003-12-02 17:26:08 +01:00
newTopic = self._joinTopic([currentTopic, formattedTopic], channel)
2003-03-12 07:26:59 +01:00
else:
newTopic = formattedTopic
irc.queueMsg(ircmsgs.topic(channel, newTopic))
2003-10-21 06:03:25 +02:00
add = privmsgs.checkChannelCapability(add, 'topic')
2003-03-12 07:26:59 +01:00
2003-10-21 06:03:25 +02:00
def shuffle(self, irc, msg, args, channel):
"""[<channel>]
Shuffles the topics in <channel>. <channel> is only necessary if the
message isn't sent in the channel itself.
"""
newtopic = irc.state.getTopic(channel)
2003-12-02 17:26:08 +01:00
topics = self._splitTopic(irc.state.getTopic(channel), channel)
if len(topics) == 0 or len(topics) == 1:
irc.error('I can\'t shuffle 1 or fewer topics.')
return
elif len(topics) == 2:
topics.reverse()
2003-12-02 17:26:08 +01:00
newtopic = self._joinTopic(topics, channel)
else:
random.shuffle(topics)
2003-12-02 17:26:08 +01:00
newtopic = self._joinTopic(topics, channel)
while newtopic == irc.state.getTopic(channel):
random.shuffle(topics)
2003-12-02 17:26:08 +01:00
newtopic = self._joinTopic(topics, channel)
irc.queueMsg(ircmsgs.topic(channel, newtopic))
2003-10-21 06:03:25 +02:00
shuffle = privmsgs.checkChannelCapability(shuffle, 'topic')
2003-12-09 15:05:32 +01:00
def reorder(self, irc, msg, args, channel):
"""[<channel>] <number> [<number> ...]
Reorders the topics from <channel> in the order of the specified
<number> arguments. <number> is a one-based index into the topics.
<channel> is only necessary if the message isn't sent in the channel
itself.
"""
topics = self._splitTopic(irc.state.getTopic(channel), channel)
num = len(topics)
if num == 0 or num == 1:
irc.error('I cannot reorder 1 or fewer topics.')
return
if len(args) != num:
irc.error('All topic numbers must be specified.')
return
order = privmsgs.getArgs(args, required=num)
2003-12-09 15:05:32 +01:00
if topics:
for i,p in enumerate(order):
try:
p = int(p)
if p > 0:
order[i] = p - 1
elif p == 0:
irc.error('0 is not a valid topic number.')
2003-12-09 15:05:32 +01:00
return
else:
order[i] = num + p
except ValueError:
irc.error('The positions must be valid integers.')
2003-12-09 15:05:32 +01:00
return
2004-04-20 12:04:09 +02:00
if sorted(order) != range(num):
irc.error('Duplicate topic numbers cannot be specified.')
return
2003-12-09 15:05:32 +01:00
try:
newtopics = [topics[i] for i in order]
newtopic = self._joinTopic(newtopics, channel)
irc.queueMsg(ircmsgs.topic(channel, newtopic))
except IndexError:
irc.error('An invalid topic number was specified.')
2003-12-09 15:05:32 +01:00
else:
irc.error('There are no topics to reorder.')
2003-12-09 15:05:32 +01:00
reorder = privmsgs.checkChannelCapability(reorder, 'topic')
def list(self, irc, msg, args, channel):
"""[<channel>] <number>
Returns a list of the topics in <channel>, prefixed by their indexes.
Mostly useful for topic reordering. <channel> is only necessary if the
message isn't sent in the channel itself.
"""
topics = self._splitTopic(irc.state.getTopic(channel), channel)
L = []
for (i, t) in enumerate(topics):
(t, _) = self._unformatTopic(t, channel)
L.append('%s: %s' % (i+1, utils.ellipsisify(t, 30)))
s = utils.commaAndify(L)
irc.reply(s)
list = privmsgs.channel(list)
2003-10-21 06:03:25 +02:00
def get(self, irc, msg, args, channel):
"""[<channel>] <number>
Returns topic number <number> from <channel>. <number> is a one-based
index into the topics. <channel> is only necessary if the message
isn't sent in the channel itself.
"""
number = privmsgs.getArgs(args)
try:
number = int(number)
if number > 0:
number -= 1
elif number == 0:
irc.error('That\'s not a valid topic number.')
return
except ValueError:
irc.error('The argument must be a valid integer.')
return
2003-12-02 17:26:08 +01:00
topics = self._splitTopic(irc.state.getTopic(channel), channel)
if topics:
try:
irc.reply(self._unformatTopic(topics[number], channel)[0])
except IndexError:
irc.error('That\'s not a valid topic.')
else:
irc.error('There are no topics to get.')
2003-10-21 06:03:25 +02:00
get = privmsgs.channel(get)
2003-08-20 18:26:23 +02:00
2003-10-21 06:03:25 +02:00
def change(self, irc, msg, args, channel):
"""[<channel>] <number> <regexp>
Changes the topic number <number> on <channel> according to the regular
expression <regexp>. <number> is the one-based index into the topics;
<regexp> is a regular expression of the form
s/regexp/replacement/flags. <channel> is only necessary if the message
isn't sent in the channel itself.
"""
(number, regexp) = privmsgs.getArgs(args, required=2)
2003-03-28 18:30:10 +01:00
try:
number = int(number)
if number > 0:
number -= 1
elif number == 0:
irc.error('That\'s not a valid topic number.')
return
2003-03-28 18:30:10 +01:00
except ValueError:
irc.error('The <number> argument must be a number.')
2003-03-28 18:30:10 +01:00
return
try:
replacer = utils.perlReToReplacer(regexp)
except ValueError, e:
irc.error('The regexp wasn\'t valid: %s' % e.args[0])
2003-09-24 08:16:29 +02:00
return
except re.error, e:
irc.error(utils.exnToString(e))
2003-03-28 18:30:10 +01:00
return
2003-12-02 17:26:08 +01:00
topics = self._splitTopic(irc.state.getTopic(channel), channel)
if not topics:
irc.error('There are no topics to change.')
return
2003-03-28 18:30:10 +01:00
topic = topics.pop(number)
(topic, name) = self._unformatTopic(topic, channel)
2003-04-16 10:34:44 +02:00
try:
senderName = ircdb.users.getUser(msg.prefix).name
2003-04-16 10:34:44 +02:00
except KeyError:
senderName = msg.nick
# irc.errorNoUser()
# return
if name and name != senderName and \
2003-08-19 03:20:56 +02:00
not ircdb.checkCapabilities(msg.prefix, ('op', 'admin')):
irc.error('You can only modify your own topics.')
2003-03-28 18:30:10 +01:00
return
newTopic = self.topicFormatter % (replacer(topic), name)
if number < 0:
number = len(topics)+1+number
2003-03-28 18:30:10 +01:00
topics.insert(number, newTopic)
2003-12-02 17:26:08 +01:00
newTopic = self._joinTopic(topics, channel)
2003-03-28 18:30:10 +01:00
irc.queueMsg(ircmsgs.topic(channel, newTopic))
2003-10-21 06:03:25 +02:00
change = privmsgs.checkChannelCapability(change, 'topic')
2003-08-20 18:26:23 +02:00
2003-10-21 06:03:25 +02:00
def remove(self, irc, msg, args, channel):
"""[<channel>] <number>
Removes topic <number> from the topic for <channel> Topics are
numbered starting from 1; you can also use negative indexes to refer
to topics starting the from the end of the topic. <channel> is only
necessary if the message isn't sent in the channel itself.
"""
2003-03-12 07:26:59 +01:00
try:
number = int(privmsgs.getArgs(args))
if number > 0:
number -= 1
elif number == 0:
irc.error('That\'s not a valid topic number.')
return
2003-03-12 07:26:59 +01:00
except ValueError:
irc.error('The argument must be a number.')
2003-03-12 07:26:59 +01:00
return
2003-12-02 17:26:08 +01:00
topics = self._splitTopic(irc.state.getTopic(channel), channel)
try:
topic = topics.pop(number)
except IndexError:
irc.error('That\'s not a valid topic number.')
return
(topic, name) = self._unformatTopic(topic, channel)
try:
username = ircdb.users.getUser(msg.prefix).name
except KeyError:
username = msg.nick
if name and name != username and \
not ircdb.checkCapabilities(msg.prefix, ('op', 'admin')):
irc.error('You can only remove your own topics.')
return
2003-12-02 17:26:08 +01:00
newTopic = self._joinTopic(topics, channel)
irc.queueMsg(ircmsgs.topic(channel, newTopic))
2003-10-21 06:03:25 +02:00
remove = privmsgs.checkChannelCapability(remove, 'topic')
2003-03-12 07:26:59 +01:00
2003-10-19 17:48:44 +02:00
Class = Topic
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: