93 lines
4.1 KiB
Python
93 lines
4.1 KiB
Python
###
|
|
# Copyright (c) 2010, Valentin Lorentz
|
|
# 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.
|
|
|
|
###
|
|
|
|
import re
|
|
import supybot.conf as conf
|
|
import supybot.registry as registry
|
|
|
|
try:
|
|
from supybot.i18n import PluginInternationalization
|
|
from supybot.i18n import internationalizeDocstring
|
|
_ = PluginInternationalization('AttackProtector')
|
|
except:
|
|
# This are useless functions that's allow to run the plugin on a bot
|
|
# without the i18n plugin
|
|
_ = lambda x:x
|
|
internationalizeDocstring = lambda x:x
|
|
|
|
def configure(advanced):
|
|
# This will be called by supybot to configure this module. advanced is
|
|
# a bool that specifies whether the user identified himself as an advanced
|
|
# user or not. You should effect your configuration by manipulating the
|
|
# registry as appropriate.
|
|
from supybot.questions import expect, anything, something, yn
|
|
conf.registerPlugin('AttackProtector', True)
|
|
|
|
class XpY(registry.String):
|
|
"""Value must be in the format <number>p<seconds>."""
|
|
_re = re.compile('(?P<number>[0-9]+)p(?P<seconds>[0-9]+)')
|
|
def setValue(self, v):
|
|
if self._re.match(v):
|
|
registry.String.setValue(self, v)
|
|
else:
|
|
self.error()
|
|
XpY = internationalizeDocstring(XpY)
|
|
|
|
class Punishment(registry.OnlySomeStrings):
|
|
validStrings = ('ban', 'kick', 'kban', 'mode+i')
|
|
Punishment = internationalizeDocstring(Punishment)
|
|
|
|
AttackProtector = conf.registerPlugin('AttackProtector')
|
|
# This is where your configuration variables (if any) should go. For example:
|
|
# conf.registerGlobalValue(AttackProtector, 'someConfigVariableName',
|
|
# registry.Boolean(False, """Help for someConfigVariableName."""))
|
|
|
|
conf.registerGlobalValue(AttackProtector, 'delay',
|
|
registry.Integer(10, _("""Determines how long (in seconds) the plugin will
|
|
wait before being enabled. A too low value makes the bot believe that
|
|
its incoming messages 'flood' on connection is an attack.""")))
|
|
|
|
kinds = {'join': ['5p10', 'ban'],
|
|
'part': ['4p5', 'ban'],
|
|
'nick': ['7p300', 'ban'],
|
|
'message': ['10p20', 'kick'],
|
|
'groupjoin': ['20p10', 'mode+i']}
|
|
for kind in kinds:
|
|
data = kinds[kind]
|
|
conf.registerGroup(AttackProtector, kind)
|
|
conf.registerChannelValue(getattr(AttackProtector, kind), 'detection',
|
|
XpY(data[0], _("""In the format XpY, where X is the number of %s per
|
|
Y seconds that triggers the punishment.""") % kind))
|
|
conf.registerChannelValue(getattr(AttackProtector, kind), 'punishment',
|
|
Punishment(data[1], _("""Determines the pushiment applyed when a
|
|
%s flood is detected.""") % kind))
|
|
|
|
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:
|