X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;ds=inline;f=include%2Fmodules%2Fircv3_batch.h;h=a5319dd2dcabf6b115323f2468be22bd61480c16;hb=c05f81cac83e80c7727594e3929e0709eccca689;hp=841554bdbc5db2f4c00d26ed2bdbc4f87056db42;hpb=87b1461e2a4710a38b32186c2582da9fe9bb3804;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/include/modules/ircv3_batch.h b/include/modules/ircv3_batch.h
index 841554bdb..a5319dd2d 100644
--- a/include/modules/ircv3_batch.h
+++ b/include/modules/ircv3_batch.h
@@ -1,7 +1,8 @@
 /*
  * InspIRCd -- Internet Relay Chat Daemon
  *
- *   Copyright (C) 2016 Attila Molnar <attilamolnar@hush.com>
+ *   Copyright (C) 2018-2020 Sadie Powell <sadie@witchery.services>
+ *   Copyright (C) 2018 Attila Molnar <attilamolnar@hush.com>
  *
  * This file is part of InspIRCd.  InspIRCd is free software: you can
  * redistribute it and/or modify it under the terms of the GNU General Public
@@ -89,6 +90,7 @@ class IRCv3::Batch::Batch
 	unsigned int bit;
 	BatchInfo* batchinfo;
 	ClientProtocol::Message* batchstartmsg;
+	ClientProtocol::Message* batchendmsg;
 
 	void Setup(unsigned int b)
 	{
@@ -110,6 +112,7 @@ class IRCv3::Batch::Batch
 		, type(Type)
 		, batchinfo(NULL)
 		, batchstartmsg(NULL)
+		, batchendmsg(NULL)
 	{
 	}
 
@@ -158,6 +161,15 @@ class IRCv3::Batch::Batch
 	 */
 	ClientProtocol::Message& GetBatchStartMessage() { return *batchstartmsg; }
 
+	/** Get the batch end client protocol message.
+	 * The returned message object can be manipulated to add extra parameters or labels to the message. The first
+	 * parameter of the message is the batch reference tag generated by the module providing batch support.
+	 * If the batch type string was specified, it will be the second parameter of the message.
+	 * May only be called if IsRunning() == true.
+	 * @return Mutable batch end client protocol message.
+	 */
+	ClientProtocol::Message& GetBatchEndMessage() { return *batchendmsg; }
+
 	friend class ManagerImpl;
 };