From 92a03ecdb962b0b210c9df8d7dc38ab9d1ed9233 Mon Sep 17 00:00:00 2001
From: lnd3 <linus.cumselius@gmail.com>
Date: Tue, 29 Oct 2024 17:25:33 +0100
Subject: [PATCH] Add listener structure directly to component caches.

---
 packages/ecs/include/ecs/entityecs/ECSExt.h | 49 ++++++++++++++++++++-
 1 file changed, 47 insertions(+), 2 deletions(-)

diff --git a/packages/ecs/include/ecs/entityecs/ECSExt.h b/packages/ecs/include/ecs/entityecs/ECSExt.h
index 0fe68f7..601811f 100644
--- a/packages/ecs/include/ecs/entityecs/ECSExt.h
+++ b/packages/ecs/include/ecs/entityecs/ECSExt.h
@@ -182,17 +182,62 @@ namespace l::ecs {
 		}
 
 		virtual void receive(World*, const Events::OnComponentAssigned2& event) {
-			tryAdd(event.entity);
+			if (tryAdd(event.entity)) {
+				for (auto& listener : mEventListenersAssign) {
+					listener.second(event);
+				}
+			}
 		}
 
 		virtual void receive(World*, const Events::OnComponentRemoved2& event) {
-			tryRemove(event.entity);
+			if (tryRemove(event.entity)) {
+				for (auto& listener : mEventListenersRemove) {
+					listener.second(event);
+				}
+			}
+		}
+
+		template<class T>
+		int32_t registerListener(std::function<void(const T&)> listener) {
+			if constexpr (std::is_same<T, Events::OnComponentAssigned2>){
+				mEventListenersAssign.push_back({ mIdEnumerator, listener });
+			}
+			else if constexpr (std::is_same<T, Events::OnComponentRemoved2>){
+				mEventListenersRemove.push_back({ mIdEnumerator, listener });
+			}
+			else {
+				return -1;
+			}
+			return mIdEnumerator++;
 		}
 
+		template<class T>
+		void unregisterListener(int32_t id) {
+			if constexpr (std::is_same<T, Events::OnComponentAssigned2>){
+				std::erase_if(mEventListenersAssign, [cid = id](auto& listener) {
+					if (listener.first == cid) {
+						return true;
+					}
+					return false;
+					});
+			}
+			else if constexpr (std::is_same<T, Events::OnComponentRemoved2>){
+				std::erase_if(mEventListenersRemove, [cid = id](auto& listener) {
+					if (listener.first == cid) {
+						return true;
+					}
+					return false;
+					});
+			}
+		}
 	protected:
 
 		std::vector<std::tuple<Entity*, ComponentHandle<Types>...>> mComponents;
 		std::map<Entity*, std::tuple<ComponentHandle<Types>...>> mComponentMap;
+
+		int32_t mIdEnumerator = 0;
+		std::vector<std::pair<int32_t, std::function<void(const Events::OnComponentAssigned2&)>>> mEventListenersAssign;
+		std::vector<std::pair<int32_t, std::function<void(const Events::OnComponentRemoved2&)>>> mEventListenersRemove;
 	};
 
 	class EntitySystem2 : public l::ecs::EntitySystem