fix: component icons will no longer be displayed in the scene view (Unity 2021.2 or later)

Author: mob-sakai

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

@@ -3,11 +3,16 @@
 using UnityEditor;
 using UnityEngine;
 using Object = UnityEngine.Object;
-#if UNITY_EDITOR && UNITY_2021_2_OR_NEWER
+#if UNITY_EDITOR
+using System.IO;
+using System.Linq;
+using System.Reflection;
+#if UNITY_2021_2_OR_NEWER
 using UnityEditor.SceneManagement;
-#elif UNITY_EDITOR
+#else
 using UnityEditor.Experimental.SceneManagement;
 #endif
+#endif
 
 namespace Coffee.UIEffectInternal
 {
@@ -72,5 +77,56 @@ public static T[] GetAllComponentsInPrefabStage<T>() where T : Component
 
         public static bool isBatchOrBuilding => Application.isBatchMode || BuildPipeline.isBuildingPlayer;
 #endif
+
+        [Conditional("UNITY_EDITOR")]
+        public static void QueuePlayerLoopUpdate()
+        {
+#if UNITY_EDITOR
+            if (!EditorApplication.isPlaying)
+            {
+                EditorApplication.QueuePlayerLoopUpdate();
+            }
+#endif
+        }
     }
+
+#if !UNITY_2021_2_OR_NEWER
+    [AttributeUsage(AttributeTargets.Class)]
+    [Conditional("UNITY_EDITOR")]
+    internal class IconAttribute : Attribute
+    {
+        private readonly string _path;
+
+        public IconAttribute(string path)
+        {
+            _path = path;
+        }
+
+#if UNITY_EDITOR
+        private static Action<Object, Texture2D> s_SetIconForObject = typeof(EditorGUIUtility)
+            .GetMethod("SetIconForObject", BindingFlags.Static | BindingFlags.NonPublic)
+            .CreateDelegate(typeof(Action<Object, Texture2D>), null) as Action<Object, Texture2D>;
+
+        [InitializeOnLoadMethod]
+        private static void InitializeOnLoadMethod()
+        {
+            if (Misc.isBatchOrBuilding) return;
+
+            var types = TypeCache.GetTypesWithAttribute<IconAttribute>();
+            var scripts = MonoImporter.GetAllRuntimeMonoScripts();
+            foreach (var type in types)
+            {
+                var script = scripts.FirstOrDefault(x => x.GetClass() == type);
+                if (!script) continue;
+
+                var path = type.GetCustomAttribute<IconAttribute>()?._path;
+                var icon = AssetDatabase.LoadAssetAtPath<Texture2D>(path);
+                if (!icon) continue;
+
+                s_SetIconForObject(script, icon);
+            }
+        }
+#endif
+    }
+#endif
 }

Packages/src/Runtime/UIEffect.cs

@@ -5,6 +5,7 @@
 
 namespace Coffee.UIEffects
 {
+    [Icon("Packages/com.coffee.ui-effect/Editor/UIEffectIconIcon.png")]
     [ExecuteAlways]
     [DisallowMultipleComponent]
     public class UIEffect : UIEffectBase

Packages/src/Runtime/UIEffect.cs.meta

@@ -8,6 +8,7 @@
 
 namespace Coffee.UIEffects
 {
+    [Icon("Packages/com.coffee.ui-effect/Editor/UIEffectIconIcon.png")]
     public class UIEffectProjectSettings : PreloadedProjectSettings<UIEffectProjectSettings>
     {
         [Tooltip(

Packages/src/Runtime/UIEffectProjectSettings.cs

@@ -3,6 +3,7 @@
 
 namespace Coffee.UIEffects
 {
+    [Icon("Packages/com.coffee.ui-effect/Editor/UIEffectIconIcon.png")]
     public class UIEffectReplica : UIEffectBase
     {
         [SerializeField] private UIEffect m_Target;

Packages/src/Runtime/UIEffectProjectSettings.cs.meta

@@ -1,10 +1,12 @@
 using System;
+using Coffee.UIEffectInternal;
 using UnityEngine;
 using UnityEngine.Events;
 using UnityEngine.Serialization;
 
 namespace Coffee.UIEffects
 {
+    [Icon("Packages/com.coffee.ui-effect/Editor/UIEffectIconIcon.png")]
     [ExecuteAlways]
     [RequireComponent(typeof(UIEffectBase))]
     public class UIEffectTweener : MonoBehaviour