Limnoria/plugins/Ctcp.py

179 lines
7.5 KiB
Python
Raw Normal View History

2003-03-12 07:26:59 +01:00
###
# Copyright (c) 2002-2004, Jeremiah Fincher
2003-03-12 07:26:59 +01:00
# 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.
###
"""
2003-08-20 18:26:23 +02:00
Handles standard CTCP responses to PING, TIME, SOURCE, VERSION, USERINFO,
and FINGER.
"""
2003-11-25 09:23:47 +01:00
__revision__ = "$Id$"
2004-07-24 07:18:26 +02:00
import supybot.plugins as plugins
2003-03-12 07:26:59 +01:00
import os
import sys
import time
2004-07-24 07:18:26 +02:00
import supybot.conf as conf
2004-09-17 06:56:38 +02:00
import supybot.utils as utils
from supybot.commands import *
2004-07-24 07:18:26 +02:00
import supybot.ircmsgs as ircmsgs
import supybot.ircutils as ircutils
2004-09-17 06:56:38 +02:00
import supybot.privmsgs as privmsgs
import supybot.registry as registry
2004-09-17 06:56:38 +02:00
import supybot.schedule as schedule
2004-07-24 07:18:26 +02:00
import supybot.callbacks as callbacks
2003-03-12 07:26:59 +01:00
conf.registerPlugin('Ctcp')
conf.registerGlobalValue(conf.supybot.abuse.flood, 'ctcp',
registry.Boolean(True, """Determines whether the bot will defend itself
against CTCP flooding."""))
conf.registerGlobalValue(conf.supybot.abuse.flood.ctcp, 'maximum',
registry.PositiveInteger(5, """Determines how many CTCP messages (not
including actions) the bot will reply to from a given user in a minute.
If a user sends more than this many CTCP messages in a 60 second period,
the bot will ignore CTCP messages from this user for
supybot.abuse.flood.ctcp.punishment seconds."""))
conf.registerGlobalValue(conf.supybot.abuse.flood.ctcp, 'punishment',
registry.PositiveInteger(300, """Determines how many seconds the bot will
ignore CTCP messages from users who flood it with CTCP messages."""))
2004-09-17 06:56:38 +02:00
conf.registerGlobalValue(conf.supybot.plugins.Ctcp, 'versionWait',
registry.PositiveInteger(10, """Determines how many seconds the bot will
wait after getting a version command (not a CTCP VERSION, but an actual
call of the command in this plugin named "version") before replying with
the results it has collected."""))
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
class Ctcp(callbacks.PrivmsgCommandAndRegexp):
2003-03-12 07:26:59 +01:00
public = False
def __init__(self):
self.__parent = super(Ctcp, self)
self.__parent.__init__()
self.ignores = ircutils.IrcDict()
self.floods = ircutils.FloodQueue(60)
def callCommand(self, name, irc, msg, *L, **kwargs):
if conf.supybot.abuse.flood.ctcp():
now = time.time()
for (ignore, expiration) in self.ignores.items():
if expiration < now:
del self.ignores[ignore]
elif ircutils.hostmaskPatternEqual(ignore, msg.prefix):
return
self.floods.enqueue(msg)
max = conf.supybot.abuse.flood.ctcp.maximum()
if self.floods.len(msg) > max:
expires = conf.supybot.abuse.flood.ctcp.punishment()
self.log.warning('Apparent CTCP flood from %s, '
'ignoring CTCP messages for %s seconds.',
msg.prefix, expires)
ignoreMask = '*!%s@%s' % (msg.user, msg.host)
self.ignores[ignoreMask] = now + expires
return
self.__parent.callCommand(name, irc, msg, *L, **kwargs)
def _reply(self, irc, msg, s):
s = '\x01%s\x01' % s
irc.reply(s, notice=True, private=True, to=msg.nick)
2004-09-17 06:56:38 +02:00
regexps = ('ctcpPing', 'ctcpVersion', 'ctcpUserinfo',
'ctcpTime', 'ctcpFinger', 'ctcpSource')
def ctcpPing(self, irc, msg, match):
2003-03-12 07:26:59 +01:00
"\x01PING (.*)\x01"
self.log.info('Received CTCP PING from %s', msg.prefix)
self._reply(irc, msg, 'PING %s' % match.group(1))
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
def ctcpVersion(self, irc, msg, match):
2003-03-12 07:26:59 +01:00
"\x01VERSION\x01"
self.log.info('Received CTCP VERSION from %s', msg.prefix)
self._reply(irc, msg, 'VERSION Supybot %s' % conf.version)
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
def ctcpUserinfo(self, irc, msg, match):
2003-03-12 07:26:59 +01:00
"\x01USERINFO\x01"
self.log.info('Received CTCP USERINFO from %s', msg.prefix)
self._reply(irc, msg, 'USERINFO')
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
def ctcpTime(self, irc, msg, match):
2003-03-12 07:26:59 +01:00
"\x01TIME\x01"
self.log.info('Received CTCP TIME from %s' % msg.prefix)
self._reply(irc, msg, time.ctime())
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
def ctcpFinger(self, irc, msg, match):
2003-03-12 07:26:59 +01:00
"\x01FINGER\x01"
self.log.info('Received CTCP FINGER from %s' % msg.prefix)
2004-09-13 05:18:35 +02:00
self._reply(irc, msg, 'Supybot, the best Python IRC bot in existence!')
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
def ctcpSource(self, irc, msg, match):
2003-03-12 07:26:59 +01:00
"\x01SOURCE\x01"
self.log.info('Received CTCP SOURCE from %s' % msg.prefix)
self._reply(irc, msg, 'http://www.sourceforge.net/projects/supybot/')
2003-03-12 07:26:59 +01:00
2004-09-17 06:56:38 +02:00
def doNotice(self, irc, msg):
if ircmsgs.isCtcp(msg):
try:
(version, payload) = msg.args[1][1:-1].split(None, 1)
except ValueError:
return
if version == 'VERSION':
self.versions.setdefault(payload, []).append(msg.nick)
def version(self, irc, msg, args, channel, optlist):
2004-09-17 06:56:38 +02:00
"""[<channel>] [--nicks]
Sends a CTCP VERSION to <channel>, returning the various
version strings returned. It waits for 10 seconds before returning
the versions received at that point. If --nicks is given, nicks are
associated with the version strings; otherwise, only the version
strings are given.
"""
self.versions = ircutils.IrcDict()
nicks = False
for (option, arg) in optlist:
if option == 'nicks':
nicks = True
2004-09-17 06:56:38 +02:00
irc.queueMsg(ircmsgs.privmsg(channel, '\x01VERSION\x01'))
def doReply():
if self.versions:
L = []
for (reply, nicks) in self.versions.iteritems():
if nicks:
L.append('%s responded with %s' %
(utils.commaAndify(nicks),
utils.quoted(reply)))
2004-09-17 06:56:38 +02:00
else:
L.append(reply)
irc.reply(utils.commaAndify(L))
else:
irc.reply('I received no version responses.')
wait = self.registryValue('versionWait')
schedule.addEvent(doReply, time.time()+wait)
version = wrap(version, ['channel', getopts({'nicks':''})])
2004-09-17 06:56:38 +02:00
2003-03-12 07:26:59 +01:00
Class = Ctcp
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: