fix: TextMeshPro objects appeared as black blocks when saving prefabs in prefab mode

close #285

Author: mob-sakai

Packages/src/Runtime/Internal/Utilities/Misc.cs

@@ -0,0 +1,76 @@
+using System;
+using System.Diagnostics;
+using UnityEditor;
+using UnityEngine;
+using Object = UnityEngine.Object;
+#if UNITY_EDITOR && UNITY_2021_2_OR_NEWER
+using UnityEditor.SceneManagement;
+#elif UNITY_EDITOR
+using UnityEditor.Experimental.SceneManagement;
+#endif
+
+namespace Coffee.UIEffectInternal
+{
+    internal static class Misc
+    {
+        public static T[] FindObjectsOfType<T>() where T : Object
+        {
+#if UNITY_2023_1_OR_NEWER
+            return Object.FindObjectsByType<T>(FindObjectsInactive.Include, FindObjectsSortMode.None);
+#else
+            return Object.FindObjectsOfType<T>();
+#endif
+        }
+
+        public static void Destroy(Object obj)
+        {
+            if (!obj) return;
+#if UNITY_EDITOR
+            if (!Application.isPlaying)
+            {
+                Object.DestroyImmediate(obj);
+            }
+            else
+#endif
+            {
+                Object.Destroy(obj);
+            }
+        }
+
+        public static void DestroyImmediate(Object obj)
+        {
+            if (!obj) return;
+#if UNITY_EDITOR
+            if (Application.isEditor)
+            {
+                Object.DestroyImmediate(obj);
+            }
+            else
+#endif
+            {
+                Object.Destroy(obj);
+            }
+        }
+
+        [Conditional("UNITY_EDITOR")]
+        public static void SetDirty(Object obj)
+        {
+#if UNITY_EDITOR
+            if (!obj) return;
+            EditorUtility.SetDirty(obj);
+#endif
+        }
+
+#if UNITY_EDITOR
+        public static T[] GetAllComponentsInPrefabStage<T>() where T : Component
+        {
+            if (!PrefabStageUtility.GetCurrentPrefabStage()) return Array.Empty<T>();
+
+            var prefabStage = PrefabStageUtility.GetCurrentPrefabStage();
+            if (!prefabStage) return Array.Empty<T>();
+
+            return prefabStage.prefabContentsRoot.GetComponentsInChildren<T>(true);
+        }
+#endif
+    }
+}

Packages/src/Runtime/Internal/Utilities/Misc.cs.meta

@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Linq;
 using System.Runtime.CompilerServices;
 using Coffee.UIEffectInternal;
 using UnityEngine;
@@ -245,11 +246,8 @@ private static void OnPostprocessAllAssets(string[] _, string[] __, string[] ___
                 if (Application.isBatchMode || BuildPipeline.isBuildingPlayer) return;
 
                 s_ShaderNameCache.Clear();
-#if UNITY_2021_3 || UNITY_2022_2_OR_NEWER
-                foreach (var effect in FindObjectsByType<UIEffectBase>(FindObjectsSortMode.None))
-#else
-                foreach (var effect in FindObjectsOfType<UIEffectBase>())
-#endif
+                foreach (var effect in Misc.FindObjectsOfType<UIEffectBase>()
+                             .Concat(Misc.GetAllComponentsInPrefabStage<UIEffectBase>()))
                 {
                     if (!effect.isActiveAndEnabled) continue;
                     if (!(effect.graphic is TextMeshProUGUI tmp) || !tmp.isActiveAndEnabled) continue;