bitbot-3.11-fork/modules/ircv3_multiline.py

44 lines
1.7 KiB
Python
Raw Normal View History

2019-09-19 15:24:53 +00:00
from src import IRCLine, ModuleManager, utils
2019-06-16 08:53:12 +00:00
CAP = utils.irc.Capability(None, "draft/multiline", alias="multiline")
BATCH = utils.irc.BatchType(None, "draft/multiline")
TAG = utils.irc.MessageTag(None, "draft/multiline-concat")
2019-06-16 08:53:12 +00:00
2019-06-20 15:52:23 +00:00
@utils.export("cap", CAP)
2019-06-16 08:53:12 +00:00
class Module(ModuleManager.BaseModule):
@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 = IRCLine.IRCSendBatch("draft/multiline",
2019-06-16 09:03:28 +00:00
[target])
2019-06-16 08:53:12 +00:00
for line in lines:
2019-10-28 10:56:33 +00:00
line = IRCLine.ParsedLine("PRIVMSG", [target, line])
batch.add_line(line)
2019-06-16 08:53:12 +00:00
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:
message = line.args[1]
if TAG.present(line.tags):
last_message = ""
if messages:
last_message = messages.pop(-1)
message = last_message+message
messages.append(message)
target = event["batch"].args[0]
message = "\n".join(messages)
return [IRCLine.ParsedLine("PRIVMSG", [target, message],
source=event["batch"].source)]