bitbot-3.11-fork/modules/ircv3_multiline.py
2019-06-16 17:20:31 +01:00

38 lines
1.4 KiB
Python

from src import ModuleManager, utils
CAP = utils.irc.Capability(None, "bitbot.dev/multiline")
BATCH = utils.irc.BatchType(None, "bitbot.dev/multiline")
class Module(ModuleManager.BaseModule):
@utils.hook("received.cap.ls")
@utils.hook("received.cap.new")
def on_cap(self, event):
return CAP.copy()
@utils.hook("preprocess.send.privmsg")
def preprocess_send_privmsg(self, event):
if len(event["line"].args) > 1:
if ("\n" in event["line"].args[1] and
event["server"].has_capability(CAP)):
event["line"].invalidate()
target = event["line"].args[0]
lines = event["line"].args[1].split("\n")
batch = utils.irc.IRCSendBatch("bitbot.dev/multiline",
[target])
for line in lines:
batch.add_line(utils.irc.protocol.privmsg(target, line))
for line in batch.get_lines():
event["server"].send(line)
@utils.hook("received.batch.end")
def batch_end(self, event):
if BATCH.match(event["batch"].type):
messages = []
lines = event["batch"].get_lines()
for line in lines:
messages.append(line.args[1])
target = event["batch"].args[0]
message = "\n".join(messages)
return [IRCLine.ParsedLine("PRIVMSG", [target, message])]