From c3bcabbe3c54b1cce68464fd4b022f58346d53b7 Mon Sep 17 00:00:00 2001
From: JakobEdurino <139774524+JakobEdurino@users.noreply.github.com>
Date: Mon, 17 Jun 2024 16:08:12 +0200
Subject: [PATCH] Add Undo Functionality

---
 Editor/HierarchyDrawer.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Editor/HierarchyDrawer.cs b/Editor/HierarchyDrawer.cs
index efcfbb0..32bb4a2 100644
--- a/Editor/HierarchyDrawer.cs
+++ b/Editor/HierarchyDrawer.cs
@@ -521,12 +521,12 @@ static void DrawCore(int instanceID, Rect selectionRect)
                 var isActive = GUI.Toggle(r, wasActive, "");
                 if (wasActive != isActive)
                 {
-                    go.SetActive(isActive);
                     if (EditorApplication.isPlaying == false)
                     {
                         UnityEditor.SceneManagement.EditorSceneManager.MarkSceneDirty(go.scene);
-                        EditorUtility.SetDirty(go);
+                        Undo.RecordObject(go, isActive ? "Enabled GameObject" : "Disabled GameObject" );
                     }
+                    go.SetActive(isActive);
                 }
             }
             #endregion