bitbot-3.11-fork/modules/ctcp.py

import datetime

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

    def private_message(self, event):
        if event["message"][0] == "\x01" and event["message"][-1] == "\x01":
            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)