Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Added highlight spam detection/prevention logic to channel_op

Browse source
This commit is contained in:
jesopo 2017-12-26 11:50:18 +00:00
parent 0cc72d5d66
commit 14ffa6716d
2 changed files with 25 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
Utils.py
View file

@ -183,6 +183,10 @@ def bool_or_none(s):
return True return True
elif s in IS_FALSE: elif s in IS_FALSE:
return False return False
def int_or_none(s):
stripped_s = s.lstrip("0")
if stripped_s.isdigit():
return int(stripped_s)
def get_closest_setting(event, setting, default=None): def get_closest_setting(event, setting, default=None):
server = event["server"] server = event["server"]

22
modules/channel_op.py
View file

@ -1,4 +1,4 @@
import Utils
class Module(object): class Module(object):
_name = "Channel Op" _name = "Channel Op"
@ -21,6 +21,16 @@ class Module(object):
).hook(self.voice, channel_only=True, require_mode="o") ).hook(self.voice, channel_only=True, require_mode="o")
bot.events.on("received").on("command").on("devoice" bot.events.on("received").on("command").on("devoice"
).hook(self.devoice, channel_only=True, require_mode="o") ).hook(self.devoice, channel_only=True, require_mode="o")
bot.events.on("received").on("message").on("channel").hook(self.highlight_spam)
bot.events.on("postboot").on("configure").on(
"channelset").call(setting="highlight-spam-threshold",
help="Set the number of nicknames in a message that qualifies as spam",
validate=Utils.int_or_none)
bot.events.on("postboot").on("configure").on(
"channelset").call(setting="highlight-spam-protection",
help="Enable/Disable highligh spam protection",
validate=Utils.bool_or_none)
def kick(self, event): def kick(self, event):
target = event["args_split"][0] target = event["args_split"][0]
@ -69,3 +79,13 @@ class Module(object):
target = event["user"].nickname if not event["args_split"] else event[ target = event["user"].nickname if not event["args_split"] else event[
"args_split"][0] "args_split"][0]
event["target"].send_mode("-v", target) event["target"].send_mode("-v", target)
def highlight_spam(self, event):
nicknames = list(map(lambda user: user.nickname, event["channel"].users)
) + [event["server"].nickname]
if len(set(nicknames) & set(event["message_split"])) >= event["channel"].get_setting(
"highlight-spam-threshold", 10):
if event["channel"].get_setting("highlight-spam-protection", False):
if not event["channel"].mode_or_above(event["user"].nickname, "v"):
event["channel"].send_kick(event["user"].nickname,
"highlight spam detected")