Update Storage.cs

Author: IntoTheDev

Runtime/Storage.cs

@@ -1,6 +1,9 @@
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Linq;
 using UnityEngine;
+#if UNITY_EDITOR && ODIN_INSPECTOR
+using Sirenix.Utilities.Editor;
+#endif
 
 namespace ToolBox.Loader
 {
@@ -12,7 +15,7 @@ public static class Storage
 		private static void Setup()
 		{
 #if UNITY_EDITOR
-			var assets = Resources.FindObjectsOfTypeAll<ScriptableObject>();
+			var assets = GetAssets();
 #else
 			var assets = Resources.LoadAll<ScriptableObject>("");
 #endif
@@ -25,7 +28,7 @@ public static T Get<T>() where T : ScriptableObject, ILoadable
 		{
 #if UNITY_EDITOR
 			if (!Application.isPlaying)
-				_assets = Resources.FindObjectsOfTypeAll<ScriptableObject>();
+				_assets = GetAssets();
 #endif
 
 			for (int i = 0; i < _assets.Length; i++)
@@ -39,11 +42,22 @@ public static IEnumerable<T> GetAll<T>() where T : ScriptableObject, ILoadable
 		{
 #if UNITY_EDITOR
 			if (!Application.isPlaying)
-				_assets = Resources.FindObjectsOfTypeAll<ScriptableObject>();
+				_assets = GetAssets();
 #endif
 
 			return _assets.Where(a => a is T).Cast<T>();
 		}
+
+		private static ScriptableObject[] GetAssets()
+		{
+#if ODIN_INSPECTOR
+			var assets = AssetUtilities.GetAllAssetsOfType<ScriptableObject>().ToArray();
+#else
+			var assets = Resources.FindObjectsOfTypeAll<ScriptableObject>();
+#endif
+
+			return assets;
+		}
 	}
 
 	public interface ILoadable { }