bitbot-3.11-fork/modules/commands.py

import re
import Utils

STR_MORE = "%s (more...)" % Utils.FONT_RESET
STR_CONTINUED = "(...continued) "

OUT_CUTOFF = 400

REGEX_CUTOFF = re.compile("^.{1,%d}(?:\s|$)" % OUT_CUTOFF)

class Out(object):
    def __init__(self, module_name, target):
        self.module_name = module_name
        self.target = target
        self._text = ""
        self.written = False
    def write(self, text):
        self._text += text
        self.written = True
        return self
    def send(self):
        if self.has_text():
            text = self._text
            text_encoded = text.encode("utf8")
            if len(text_encoded) > OUT_CUTOFF:
                text = "%s%s" % (text_encoded[:OUT_CUTOFF].decode("utf8"
                    ).rstrip(), STR_MORE)
                self._text = "%s%s" % (STR_CONTINUED, text_encoded[OUT_CUTOFF:
                    ].decode("utf8").lstrip())
            self.target.send_message(text, prefix="[%s] " % self.prefix())
    def set_prefix(self, prefix):
        self.module_name = prefix
    def has_text(self):
        return bool(self._text)

class StdOut(Out):
    def prefix(self):
        return "%s%s%s" % (Utils.color(Utils.COLOR_GREEN),
            self.module_name, Utils.FONT_RESET)
class StdErr(Out):
    def prefix(self):
        return "%s!%s%s" % (Utils.color(Utils.COLOR_RED),
            self.module_name, Utils.FONT_RESET)

class Module(object):
    def __init__(self, bot):
        self.bot = bot
        bot.events.on("received").on("message").on("channel").hook(
            self.channel_message)
        bot.events.on("received").on("message").on("private").hook(
            self.private_message)
        bot.events.on("received").on("command").on("help").hook(self.help,
            help="Show help for commands", usage="<command>")
        bot.events.on("received").on("command").on("usage").hook(self.usage,
            help="Show usage help for commands", min_args=1,
            usage="<command>")
        bot.events.on("received").on("command").on("more").hook(self.more,
            help="Get more output from the last command", skip_out=True)

        bot.events.on("postboot").on("configure").on(
            "channelset").call(setting="command-prefix",
            help="Set the command prefix used in this channel")

        bot.events.on("new").on("user", "channel").hook(self.new)
        bot.events.on("send").on("stdout").hook(self.send_stdout)
        bot.events.on("send").on("stderr").hook(self.send_stderr)

    def new(self, event):
        if "user" in event:
            target = event["user"]
        else:
            target = event["channel"]
        target.last_stdout = None
        target.last_stderr = None

    def has_command(self, command):
        return command.lower() in self.bot.events.on("received").on(
            "command").get_children()
    def get_hook(self, command):
        return self.bot.events.on("received").on("command").on(command
            ).get_hooks()[0]

    def is_highlight(self, server, s):
        return s.lower() == server.nickname_lower or (s.lower().startswith(
            server.nickname_lower) and len(s) == len(server.nickname_lower
            )+1 and s[-1] in [":", ","])

    def message(self, event, command, args_index=1):
        if self.has_command(command):
            hook = self.get_hook(command)
            is_channel = False

            if "channel" in event:
                target = event["channel"]
                is_channel = True
            else:
                target = event["user"]
            if not is_channel and hook.kwargs.get("channel_only"):
                return
            if is_channel and hook.kwargs.get("private_only"):
                return

            log = target.log

            module_name = hook.function.__self__._name
            stdout, stderr = StdOut(module_name, target), StdErr(module_name,
                target)

            returns = self.bot.events.on("preprocess").on("command"
                ).call(hook=hook, user=event["user"], server=event["server"],
                target=target, is_channel=is_channel)
            for returned in returns:
                if returned:
                    stderr.write(returned).send()
                    log.skip_next()
                    return
            args_split = list(filter(None, event["message_split"][args_index:]))
            min_args = hook.kwargs.get("min_args")
            if min_args and len(args_split) < min_args:
                if "usage" in hook.kwargs:
                    stderr.write("Not enough arguments, usage: %s %s" % (
                        command, hook.kwargs["usage"])).send()
                else:
                    stderr.write("Not enough arguments (minimum: %d)" % min_args
                       ).send()
            else:
                args = " ".join(args_split)
                server = event["server"]
                user = event["user"]
                self.bot.events.on("received").on("command").on(command).call(
                    1, user=user, server=server, target=target, log=log,
                    args=args, args_split=args_split, stdout=stdout, stderr=stderr,
                    command=command.lower(), is_channel=is_channel)
                if not hook.kwargs.get("skip_out", False):
                    stdout.send()
                    stderr.send()
                    target.last_stdout = stdout
                    target.last_stderr = stderr
            log.skip_next()


    def channel_message(self, event):
        command_prefix = event["channel"].get_setting("command-prefix",
            event["server"].get_setting("command-prefix", "!"))
        if event["message_split"][0].startswith(command_prefix):
            command = event["message_split"][0].replace(
                command_prefix, "", 1).lower()
            self.message(event, command)
        elif len(event["message_split"]) > 1 and self.is_highlight(
                event["server"], event["message_split"][0]):
            command = event["message_split"][1].lower()
            self.message(event, command, 2)

    def private_message(self, event):
        if event["message_split"]:
            command = event["message_split"][0].lower()
            self.message(event, command)

    def help(self, event):
        if event["args"]:
            command = event["args_split"][0].lower()
            if command in self.bot.events.on("received").on(
                    "command").get_children():
                hooks = self.bot.events.on("received").on("command").on(command).get_hooks()
                if hooks and "help" in hooks[0].kwargs:
                    event["stdout"].write("%s: %s" % (command, hooks[0].kwargs["help"]))
                else:
                    event["stderr"].write("No help available for %s" % command)
            else:
                event["stderr"].write("Unknown command '%s'" % command)
        else:
            help_available = []
            for child in self.bot.events.on("received").on("command").get_children():
                hooks = self.bot.events.on("received").on("command").on(child).get_hooks()
                if hooks and "help" in hooks[0].kwargs:
                    help_available.append(child)
            help_available = sorted(help_available)
            event["stdout"].write("Commands: %s" % ", ".join(help_available))

    def usage(self, event):
        command = event["args_split"][0].lower()
        if command in self.bot.events.on("received").on(
                "command").get_children():
            hooks = self.bot.events.on("received").on("command").on(command).get_hooks()
            if hooks and "usage" in hooks[0].kwargs:
                event["stdout"].write("Usage: %s %s" % (command, hooks[0].kwargs["usage"]))
            else:
                event["stderr"].write("No usage help available for %s" % command)
        else:
            event["stderr"].write("Unknown command '%s'" % command)

    def more(self, event):
        if event["target"].last_stdout and event["target"].last_stdout.has_text():
            event["target"].last_stdout.send()

    def send_stdout(self, event):
        stdout = StdOut(event["module_name"], event["target"])
        stdout.write(event["message"]).send()
        if stdout.has_text():
            event["target"].last_stdout = stdout
    def send_stderr(self, event):
        stderr = StdErr(event["module_name"], event["target"])
        stderr.write(event["message"]).send()
        if stderr.has_text():
            event["target"].last_stderr = stderr