Additional utility class responsible for caching scene-related data

Author: arimger

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

@@ -0,0 +1,9 @@
+namespace Toolbox.Serialization
+{
+    internal class SceneData
+    {
+        public string SceneName { get; set; }
+        public string ScenePath { get; set; }
+        public int BuildIndex { get; set; }
+    }
+}

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

@@ -0,0 +1,71 @@
+using System.Collections.Generic;
+#if UNITY_EDITOR
+using UnityEditor;
+#endif
+using UnityEngine;
+
+namespace Toolbox.Serialization
+{
+    internal static class SceneSerializationUtility
+    {
+#if UNITY_EDITOR
+        private readonly static Dictionary<SceneAsset, SceneData> cachedScenes = new Dictionary<SceneAsset, SceneData>();
+        private static bool isInitialized;
+
+
+        [InitializeOnLoadMethod]
+        private static void Initialize()
+        {
+            if (isInitialized)
+            {
+                return;
+            }
+
+            UnityEngine.Debug.Log("INIT");
+            UpdateAllIndexes();
+
+            EditorBuildSettings.sceneListChanged -= UpdateAllIndexes;
+            EditorBuildSettings.sceneListChanged += UpdateAllIndexes;
+            isInitialized = true;
+        }
+
+        private static void UpdateAllIndexes()
+        {
+            cachedScenes.Clear();
+            UnityEngine.Debug.Log("UPDATE INDEXES");
+            var buildIndex = -1;
+            foreach (var scene in EditorBuildSettings.scenes)
+            {
+                if (!scene.enabled)
+                {
+                    continue;
+                }
+
+                buildIndex++;
+                var sceneAsset = AssetDatabase.LoadAssetAtPath<SceneAsset>(scene.path);
+                if (sceneAsset != null)
+                {
+                    cachedScenes.Add(sceneAsset, new SceneData()
+                    {
+                        BuildIndex = buildIndex,
+                        SceneName = sceneAsset.name,
+                        ScenePath = scene.path
+                    });
+                }
+            }
+        }
+
+
+        public static bool TryGetSceneData(SceneAsset sceneAsset, out SceneData data)
+        {
+            if (!sceneAsset || !cachedScenes.TryGetValue(sceneAsset, out data))
+            {
+                data = null;
+                return false;
+            }
+
+            return true;
+        }
+#endif
+    }
+}

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

@@ -1,5 +1,5 @@
 using System;
-using System.Collections.Generic;
+using Toolbox.Serialization;
 #if UNITY_EDITOR
 using UnityEditor;
 #endif
@@ -13,8 +13,6 @@ namespace UnityEngine
     public class SerializedScene : ISerializationCallbackReceiver
     {
 #if UNITY_EDITOR
-        private readonly static Dictionary<SceneAsset, int> cachedBuildIndexes = new Dictionary<SceneAsset, int>();
-
         [SerializeField]
         private SceneAsset sceneReference;
 #endif
@@ -26,87 +24,40 @@ public class SerializedScene : ISerializationCallbackReceiver
         private int buildIndex;
 
 
-        void ISerializationCallbackReceiver.OnAfterDeserialize()
-        { }
-
         void ISerializationCallbackReceiver.OnBeforeSerialize()
         {
-#if UNITY_EDITOR
-            TryGetBuildIndex(sceneReference, out buildIndex);
-            TryGetSceneName(sceneReference, out sceneName);
-            TryGetScenePath(sceneReference, out scenePath);
-#endif
-        }
-
-        //TODO:
-        // 1 - move it to other class
-        // 2 - try cache indexes in runtime but only if needed
-        // 3 - update indexes before build
-#if UNITY_EDITOR
-        [InitializeOnLoadMethod, RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.BeforeSceneLoad)]
-        private static void Initialize()
-        {
-            UpdateAllIndexes();
-
-            EditorBuildSettings.sceneListChanged -= UpdateAllIndexes;
-            EditorBuildSettings.sceneListChanged += UpdateAllIndexes;
+            UnityEngine.Debug.Log("BEFORE");
+            UpdateProperties();
         }
 
-        private static void UpdateAllIndexes()
+        void ISerializationCallbackReceiver.OnAfterDeserialize()
         {
-            cachedBuildIndexes.Clear();
-
-            var buildIndex = -1;
-            foreach (var scene in EditorBuildSettings.scenes)
-            {
-                if (scene.enabled)
-                {
-                    buildIndex++;
-                    var sceneAsset = AssetDatabase.LoadAssetAtPath<SceneAsset>(scene.path);
-                    if (sceneAsset != null)
-                    {
-                        cachedBuildIndexes.Add(sceneAsset, buildIndex);
-                    }
-                }
-            }
+            UnityEngine.Debug.Log("AFTER");
+            UpdateProperties();
         }
 
-        private static bool TryGetBuildIndex(SceneAsset sceneAsset, out int index)
-        {
-            if (sceneAsset == null || !cachedBuildIndexes.TryGetValue(sceneAsset, out index))
-            {
-                index = -1;
-                return false;
-            }
 
-            return true;
-        }
-
-        private static bool TryGetSceneName(SceneAsset sceneAsset, out string name)
+        private void UpdateProperties()
         {
-            if (sceneAsset == null)
+#if UNITY_EDITOR
+            if (SceneSerializationUtility.TryGetSceneData(sceneReference, out var sceneData))
             {
-                name = null;
-                return false;
+                UnityEngine.Debug.Log("UPDATE");
+                SceneName = sceneData.SceneName;
+                ScenePath = sceneData.ScenePath;
+                BuildIndex = sceneData.BuildIndex;
             }
-
-            name = sceneAsset.name;
-            return true;
-        }
-
-        public static bool TryGetScenePath(SceneAsset sceneAsset, out string path)
-        {
-            if (sceneAsset == null)
+            else
             {
-                path = null;
-                return false;
+                UnityEngine.Debug.Log("RESET");
+                SceneName = string.Empty;
+                ScenePath = string.Empty;
+                BuildIndex = -1;
             }
-
-            path = AssetDatabase.GetAssetPath(sceneAsset);
-            return true;
+#endif
         }
 
-
+#if UNITY_EDITOR
         public SceneAsset SceneReference
         {
             get => sceneReference;