From a3f1284a27b1a4d17b579d4ee3149995abd3a4d1 Mon Sep 17 00:00:00 2001
From: Mikhail <99481254+DCFApixels@users.noreply.github.com>
Date: Sat, 7 Sep 2024 21:20:29 +0800
Subject: [PATCH] add abstract class EcsModule

---
 src/EcsPipeline.cs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/EcsPipeline.cs b/src/EcsPipeline.cs
index e7ecf07..befaa9a 100644
--- a/src/EcsPipeline.cs
+++ b/src/EcsPipeline.cs
@@ -228,6 +228,15 @@ namespace DCFApixels.DragonECS
     {
         void Import(EcsPipeline.Builder b);
     }
+    public abstract class EcsModule : IEcsModule
+    {
+        public abstract void Import(EcsPipeline.Builder b);
+    }
+    public abstract class EcsModule<T> : IInjectionUnit
+    {
+        void IInjectionUnit.OnInitInjectionBranch(InjectionBranchIniter initer) { initer.AddNode<T>(); }
+        public EcsModule() { if (GetType() != typeof(T)) { Throw.UndefinedException(); } }
+    }
 
     #region Extensions
     public static partial class EcsPipelineExtensions