bitbot-3.11-fork/modules/check_urls.py

#--require-config virustotal-api-key

import re
from src import Utils

URL_VIRUSTOTAL = "https://www.virustotal.com/vtapi/v2/url/report"
RE_URL = re.compile(r"https?://\S+", re.I)

class Module(object):
    def __init__(self, bot, events, exports):
        self.bot = bot
        self.events = events
        events.on("received.message.channel").hook(self.message)

        exports.add("channelset", {"setting": "check-urls",
            "help": "Enable/Disable automatically checking for "
            "malicious URLs", "validate": Utils.bool_or_none})
        exports.add("serverset", {"setting": "check-urls",
            "help": "Enable/Disable automatically checking for "
            "malicious URLs", "validate": Utils.bool_or_none})
        exports.add("channelset", {"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"])
        if match and event["channel"].get_setting("check-urls",
                event["server"].get_setting("check-urls", False)):
            url = match.group(0)
            page = Utils.get_url(URL_VIRUSTOTAL, get_params={
                "apikey": self.bot.config["virustotal-api-key"],
                "resource": url}, json=True)

            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!" %
                        event["user"].nickname)