Workaround handling for Unity being dumb

- Workaround handling for when Unity decides that a scriptable object is not the type it is. Which Unity claims to have fixed in 2019.4, but clearly hasn't and instead made it worse https://forum.unity.com/threads/assetdatabase-findassets-not-working.587062/
MerlinVR · Aug 6, 2021 · 7ddf44f · 7ddf44f
1 parent 47a8bda
commit 7ddf44f
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/Assets/UdonSharp/Editor/UdonSharpProgramAsset.cs b/Assets/UdonSharp/Editor/UdonSharpProgramAsset.cs
@@ -4,6 +4,7 @@
 using System.Collections.Generic;
 using System.Collections.Immutable;
 using System.IO;
+using System.Linq;
 using System.Reflection;
 using UdonSharp.Compiler;
 using UdonSharpEditor;
@@ -244,13 +245,39 @@ public static UdonSharpProgramAsset[] GetAllUdonSharpPrograms()
         {
             if (_programAssetCache == null)
             {
-                string[] udonSharpDataAssets = AssetDatabase.FindAssets($"t:{typeof(UdonSharpProgramAsset).Name}");
+                string[] udonSharpDataAssets = AssetDatabase.FindAssets($"t:{nameof(UdonSharpProgramAsset)}");
 
                 _programAssetCache = new UdonSharpProgramAsset[udonSharpDataAssets.Length];
 
                 for (int i = 0; i < _programAssetCache.Length; ++i)
-                {
                     _programAssetCache[i] = AssetDatabase.LoadAssetAtPath<UdonSharpProgramAsset>(AssetDatabase.GUIDToAssetPath(udonSharpDataAssets[i]));
+
+                bool neededFallback = false;
+                UdonSharpProgramAsset[] fallbackAssets = Resources.FindObjectsOfTypeAll<UdonSharpProgramAsset>();
+
+                foreach (UdonSharpProgramAsset fallbackAsset in fallbackAssets)
+                {
+                    if (!_programAssetCache.Contains(fallbackAsset))
+                    {
+                        Debug.LogWarning($"Repairing program asset {fallbackAsset} which Unity has broken");
+                        neededFallback = true;
+
+                        AssetDatabase.ImportAsset(AssetDatabase.GetAssetPath(fallbackAsset), ImportAssetOptions.ForceUpdate);
+                    }
+                }
+
+                if (neededFallback)
+                {
+                    udonSharpDataAssets = AssetDatabase.FindAssets($"t:{nameof(UdonSharpProgramAsset)}");
+
+                    _programAssetCache = new UdonSharpProgramAsset[udonSharpDataAssets.Length];
+
+                    for (int i = 0; i < _programAssetCache.Length; ++i)
+                    {
+                        _programAssetCache[i] =
+                            AssetDatabase.LoadAssetAtPath<UdonSharpProgramAsset>(
+                                AssetDatabase.GUIDToAssetPath(udonSharpDataAssets[i]));
+                    }
                 }
             }