bitbot-3.11-fork/modules/ctcp.py

import datetime
from src import ModuleManager, Utils

@Utils.export("serverset", {"setting": "ctcp-responses",
    "help": "Set whether I respond to CTCPs on this server",
    "validate": Utils.bool_or_none})
class Module(ModuleManager.BaseModule):
    @Utils.hook("received.message.private")
    def private_message(self, event):
        if event["message"][0] == "\x01" and event["message"][-1] == "\x01":
            if event["server"].get_setting("ctcp-responses", True):
                ctcp_command = event["message_split"][0][1:].upper()
                if ctcp_command.endswith("\x01"):
                    ctcp_command = ctcp_command[:-1]
                ctcp_args = " ".join(event["message_split"][1:])[:-1]
                ctcp_args_split = ctcp_args.split(" ")

                ctcp_response = None
                if ctcp_command == "VERSION":
                    ctcp_response = self.bot.config.get("ctcp-version",
                        "BitBot (https://github.com/jesopo/bitbot)")
                elif ctcp_command == "SOURCE":
                    ctcp_response = self.bot.config.get("ctcp-source",
                        "https://github.com/jesopo/bitbot")
                elif ctcp_command == "PING":
                    ctcp_response = " ".join(ctcp_args_split)
                elif ctcp_command == "TIME":
                    ctcp_response = datetime.datetime.now().strftime("%c")

                if ctcp_response:
                    event["user"].send_ctcp_response(ctcp_command,
                        ctcp_response)