From 5f047a45078b3c18ff925a4d521a20b9085de7c7 Mon Sep 17 00:00:00 2001
From: Mikhail <99481254+DCFApixels@users.noreply.github.com>
Date: Wed, 31 May 2023 04:09:55 +0800
Subject: [PATCH] refactoring/update debug utility

---
 src/Builtin/InjectSystem.cs  | 1 +
 src/Debug/EcsDebugUtility.cs | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/src/Builtin/InjectSystem.cs b/src/Builtin/InjectSystem.cs
index 88654f2..c7fdf6a 100644
--- a/src/Builtin/InjectSystem.cs
+++ b/src/Builtin/InjectSystem.cs
@@ -129,6 +129,7 @@ namespace DCFApixels.DragonECS
                     injectCallbacksRunner.OnPreInitInjectionAfter();
                     EcsRunner.Destroy(injectCallbacksRunner);
                 }
+                _injectedData = default;
             }
             public void OnPreInitInjectionBefore() { }
             public void OnPreInitInjectionAfter() => _injectController = null;
diff --git a/src/Debug/EcsDebugUtility.cs b/src/Debug/EcsDebugUtility.cs
index 341d969..e6f6bf3 100644
--- a/src/Debug/EcsDebugUtility.cs
+++ b/src/Debug/EcsDebugUtility.cs
@@ -54,5 +54,11 @@ namespace DCFApixels.DragonECS
             var atr = type.GetCustomAttribute<DebugColorAttribute>();
             return atr != null ? (atr.r, atr.g, atr.b) : ((byte)255, (byte)255, (byte)255);
         }
+
+        public static bool IsHidden<T>() => IsHidden(typeof(T));
+        public static bool IsHidden(Type type)
+        {
+            return type.GetCustomAttribute<DebugHideAttribute>() != null;
+        }
     }
 }