From 14a9e4a6a4af8ad99081385c2ba97a908a4cba19 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Sun, 2 Sep 2018 11:50:32 +0100
Subject: Also optionally (attempt to) kick users that send malicious urls in
 check_urls.py

---
 modules/check_urls.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'modules')

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!" %
-- 
cgit v1.3.1-10-gc9f91