Add Undo Functionality

JakobEdurino · web-flow · commit c3bcabbe3c54 · 2024-06-17T16:08:12.000+02:00
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -521,12 +521,12 @@ static void DrawCore(int instanceID, Rect selectionRect)`
`521`	`521`	`var isActive = GUI.Toggle(r, wasActive, "");`
`522`	`522`	`if (wasActive != isActive)`
`523`	`523`	`{`
`524`		`- go.SetActive(isActive);`
`525`	`524`	`if (EditorApplication.isPlaying == false)`
`526`	`525`	`{`
`527`	`526`	`UnityEditor.SceneManagement.EditorSceneManager.MarkSceneDirty(go.scene);`
`528`		`- EditorUtility.SetDirty(go);`
	`527`	`+ Undo.RecordObject(go, isActive ? "Enabled GameObject" : "Disabled GameObject" );`
`529`	`528`	`}`
	`529`	`+ go.SetActive(isActive);`
`530`	`530`	`}`
`531`	`531`	`}`
`532`	`532`	`#endregion`