diff options
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/check_urls.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/modules/check_urls.py b/modules/check_urls.py index d9fe05bc..7d39294e 100644 --- a/modules/check_urls.py +++ b/modules/check_urls.py @@ -13,8 +13,12 @@ class Module(object): events.on("received.message.channel").hook(self.message) events.on("postboot").on("configure").on( "channelset").assure_call(setting="check-urls", - help="Enable/Disable automatically checking for malicious urls", + help="Enable/Disable automatically checking for malicious URLs", validate=Utils.bool_or_none) + events.on("postboot").on("configure").on( + "channelset").assure_call(setting="check-urls-kick", + help="Enable/Disable automatically kicking users that send " + "malicious URLs", validate=Utils.bool_or_none) def message(self, event): match = RE_URL.search(event["message"]) @@ -25,8 +29,11 @@ class Module(object): "apikey": self.bot.config["virustotal-api-key"], "resource": url}, json=True) - if page: - if page.get("positives", 0) > 1: + if page and page.get("positives", 0) > 1: + if event["channel"].get_setting("check-urls-kick", False): + event["channel"].send_kick(event["user"].nickname, + "Don't send malicious URLs!") + else: self.events.on("send.stdout").call( module_name="CheckURL", target=event["channel"], message="%s just send a malicous URL!" % |