from src import ModuleManager, utils

TAG = utils.irc.MessageTag("msgid", "draft/msgid")

class Module(ModuleManager.BaseModule):
    def _on_channel(self, channel, tags):
        msgid = TAG.get_value(tags)
        if not msgid == None:
            channel.set_setting("last-msgid", msgid)

    @utils.hook("received.message.channel")
    @utils.hook("send.message.channel")
    @utils.hook("received.notice.channel")
    @utils.hook("send.notice.channel")
    @utils.hook("received.tagmsg.channel")
    @utils.hook("send.tagmsg.channel")
    def on_channel(self, event):
        self._on_channel(event["channel"], event["tags"])

    @utils.hook("received.ctcp.request")
    @utils.hook("received.ctcp.response")
    def ctcp(self, event):
        if event["is_channel"]:
            self._on_channel(event["target"], event["tags"])

    @utils.hook("postprocess.command")
    def postprocess_command(self, event):
        msgid = TAG.get_value(event["line"].tags)
        if msgid:
            event["stdout"].tags["+draft/reply"] = msgid
            event["stderr"].tags["+draft/reply"] = msgid