Bugfix: Endless exception loop when exiting prefab mode annulusgames#223

Author: duynt-hypersol

src/LitMotion/Assets/LitMotion.Animation/Editor/LitMotionAnimationEditor.cs

@@ -3,6 +3,7 @@
 using UnityEditor.UIElements;
 using System.Collections.Generic;
 using UnityEngine;
+using UnityEditor.SceneManagement;
 
 namespace LitMotion.Animation.Editor
 {
@@ -192,7 +193,14 @@ VisualElement CreateDebugPanel()
                     flexGrow = 1f,
                 }
             };
-            var playButton = new Button(() => ((LitMotionAnimation)target).Play())
+            var playButton = new Button(() => {
+                ((LitMotionAnimation)target).Play();
+                if (PrefabStageUtility.GetCurrentPrefabStage() != null)
+                {
+                    PrefabStage.prefabStageClosing -= OnPrefabStageClosing;
+                    PrefabStage.prefabStageClosing += OnPrefabStageClosing;
+                }
+            })
             {
                 text = "Play",
                 style = {
@@ -345,5 +353,14 @@ bool IsActive()
         {
             return !((LitMotionAnimation)target).IsActive;
         }
+
+        void OnPrefabStageClosing(PrefabStage stage)
+        {
+            PrefabStage.prefabStageClosing -= OnPrefabStageClosing;
+            foreach (var i in stage.prefabContentsRoot.GetComponentsInChildren<LitMotionAnimation>(true))
+            {
+                i.Stop();
+            }
+        }
     }
 }