Handle removing and reimporting Scene assets

Author: arimger

Assets/Editor Toolbox/Runtime/Serialization/SceneSerializationProcessor.cs

@@ -0,0 +1,42 @@
+using System.IO;
+#if UNITY_EDITOR
+using UnityEditor;
+#endif
+
+namespace Toolbox.Serialization
+{
+#if UNITY_EDITOR
+    public class SceneSerializationProcessor : AssetPostprocessor
+    {
+#if UNITY_2021_3_OR_NEWER
+        internal static void OnPostprocessAllAssets(string[] importedAssets, string[] deletedAssets, string[] movedAssets, string[] movedFromAssetPaths, bool didDomainReload)
+#else
+        internal static void OnPostprocessAllAssets(string[] importedAssets, string[] deletedAssets, string[] movedAssets, string[] movedFromAssetPaths)
+#endif
+        {
+            foreach (string path in importedAssets)
+            {
+                if (IsSceneAsset(path))
+                {
+                    SceneSerializationUtility.RefreshCache();
+                    return;
+                }
+            }
+
+            foreach (string path in deletedAssets)
+            {
+                if (IsSceneAsset(path))
+                {
+                    SceneSerializationUtility.RefreshCache();
+                    return;
+                }
+            }
+        }
+
+        private static bool IsSceneAsset(string path)
+        {
+            return Path.GetExtension(path) == ".unity";
+        }
+    }
+#endif
+}

Assets/Editor Toolbox/Runtime/Serialization/SceneSerializationProcessor.cs.meta

@@ -27,7 +27,7 @@ private static void Initialize()
             isInitialized = true;
         }
 
-        private static void ConfirmCache()
+        internal static void ConfirmCache()
         {
             //NOTE: refresh data only if the cache is empty,
             //it probably means that it's our first time when we are updating it
@@ -37,7 +37,7 @@ private static void ConfirmCache()
             }
         }
 
-        private static void RefreshCache()
+        internal static void RefreshCache()
         {
             cachedScenes.Clear();
             var buildIndex = -1;
@@ -48,23 +48,30 @@ private static void RefreshCache()
                     continue;
                 }
 
-                buildIndex++;
-                var sceneIndex = scene.enabled ? buildIndex : InvalidSceneIndex;
                 var sceneAsset = EditorGUIUtility.Load(scene.path) as SceneAsset;
-                if (sceneAsset != null)
+                if (sceneAsset == null)
                 {
-                    if (cachedScenes.ContainsKey(sceneAsset))
-                    {
-                        continue;
-                    }
-
-                    cachedScenes.Add(sceneAsset, new SceneData()
-                    {
-                        BuildIndex = buildIndex,
-                        SceneName = sceneAsset.name,
-                        ScenePath = scene.path
-                    });
+                    continue;
+                }
+
+                var sceneIndex = InvalidSceneIndex;
+                if (scene.enabled)
+                {
+                    buildIndex++;
+                    sceneIndex = buildIndex;
+                }
+
+                if (cachedScenes.ContainsKey(sceneAsset))
+                {
+                    continue;
                 }
+
+                cachedScenes.Add(sceneAsset, new SceneData()
+                {
+                    BuildIndex = sceneIndex,
+                    SceneName = sceneAsset.name,
+                    ScenePath = scene.path
+                });
             }
         }