From 62702ec48158ce45aa0611a61f560ea5e32141fb Mon Sep 17 00:00:00 2001
From: SirLynix <lynix680@gmail.com>
Date: Thu, 22 Feb 2024 19:38:25 +0100
Subject: [PATCH] Network/ENetPacket: Add OnAcknowledged signal

---
 include/Nazara/Network/ENetPacket.hpp | 3 +++
 src/Nazara/Network/ENetPeer.cpp       | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/include/Nazara/Network/ENetPacket.hpp b/include/Nazara/Network/ENetPacket.hpp
index 6353309f3..4fdcaf625 100644
--- a/include/Nazara/Network/ENetPacket.hpp
+++ b/include/Nazara/Network/ENetPacket.hpp
@@ -11,6 +11,7 @@
 #include <Nazara/Network/NetPacket.hpp>
 #include <NazaraUtils/MemoryPool.hpp>
 #include <NazaraUtils/MovablePtr.hpp>
+#include <NazaraUtils/Signal.hpp>
 
 namespace Nz
 {
@@ -37,6 +38,8 @@ namespace Nz
 		NetPacket data;
 		std::size_t poolIndex;
 		std::size_t referenceCount = 0;
+
+		NazaraSignal(OnAcknowledged);
 	};
 
 	struct NAZARA_NETWORK_API ENetPacketRef
diff --git a/src/Nazara/Network/ENetPeer.cpp b/src/Nazara/Network/ENetPeer.cpp
index 7789c1a6d..ac13a5f5d 100644
--- a/src/Nazara/Network/ENetPeer.cpp
+++ b/src/Nazara/Network/ENetPeer.cpp
@@ -1029,7 +1029,10 @@ namespace Nz
 		ENetProtocolCommand commandNumber = static_cast<ENetProtocolCommand>(currentCommand->command.header.command & ENetProtocolCommand_Mask);
 
 		if (currentCommand->packet && wasSent)
+		{
 			m_reliableDataInTransit -= currentCommand->fragmentLength;
+			currentCommand->packet->OnAcknowledged();
+		}
 
 		commandList->erase(currentCommand);