From c212d70b68dff3d3f5ce68254dd6d8db7bb81e25 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Sun, 2 Jun 2019 15:18:09 +0100
Subject: Hold on to BATCH args, allow event hooks to manipulate batches

---
 modules/line_handler/__init__.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

(limited to 'modules')

diff --git a/modules/line_handler/__init__.py b/modules/line_handler/__init__.py
index 70e53d43..73057b55 100644
--- a/modules/line_handler/__init__.py
+++ b/modules/line_handler/__init__.py
@@ -182,7 +182,10 @@ class Module(ModuleManager.BaseModule):
 
         if modifier == "+":
             batch_type = event["args"][1]
-            batch = utils.irc.IRCBatch(identifier, batch_type, event["tags"])
+            args = event["args"][2:]
+
+            batch = utils.irc.IRCBatch(identifier, batch_type, args,
+                event["tags"])
             event["server"].batches[identifier] = batch
 
             self.events.on("received.batch.start").call(batch=batch,
@@ -191,10 +194,17 @@ class Module(ModuleManager.BaseModule):
             batch = event["server"].batches[identifier]
             del event["server"].batches[identifier]
 
-            self.events.on("received.batch.end").call(batch=batch,
+            lines = batch.get_lines()
+
+            results = self.events.on("received.batch.end").call(batch=batch,
                 server=event["server"])
 
-            for line in batch.lines:
+            for result in results:
+                if not result == None:
+                    lines = result
+                    break
+
+            for line in lines:
                 self._handle(event["server"], line)
 
     # IRCv3 CHGHOST, a user's username and/or hostname has changed
-- 
cgit v1.3.1-10-gc9f91