bitbot-3.11-fork/modules/to.py

from src import EventManager, ModuleManager, utils

class Module(ModuleManager.BaseModule):
    @utils.hook("received.message.channel", priority=EventManager.PRIORITY_HIGH)
    def channel_message(self, event):
        messages = event["channel"].get_user_setting(event["user"].get_id(),
            "to", [])
        for nickname, message, timestamp in messages:
            timestamp_parsed = utils.iso8601_parse(timestamp)
            timestamp_human = utils.datetime_human(timestamp_parsed)
            event["channel"].send_message("%s: <%s> %s (at %s UTC)" % (
                event["user"].nickname, nickname, message, timestamp_human))
        if messages:
            event["channel"].del_user_setting(event["user"].get_id(), "to")

    @utils.hook("received.command.to", min_args=2, channel_only=True)
    def to(self, event):
        """
        :help: Relay a message to a user the next time they talk in this
            channel
        :usage: <nickname> <message>
        """
        target_user = event["server"].get_user(event["args_split"][0])
        messages = event["target"].get_user_setting(target_user.get_id(),
            "to", [])

        if len(messages) == 5:
            raise utils.EventError("Users can only have 5 messages stored")

        messages.append([event["user"].nickname,
            " ".join(event["args_split"][1:]),
            utils.iso8601_format_now()])
        event["target"].set_user_setting(target_user.get_id(),
            "to", messages)
        event["stdout"].write("Message saved")