added fail guard

ErnSur · Jan 5, 2024 · 819364f · 819364f
1 parent 962d2a3
commit 819364f
Showing 1 changed file with 21 additions and 8 deletions.
diff --git a/Packages/com.quickeye.utility/Editor/AssemblyReloading/StatusBarExtender.cs b/Packages/com.quickeye.utility/Editor/AssemblyReloading/StatusBarExtender.cs
@@ -9,8 +9,10 @@ namespace QuickEye.Utility.Editor.AssemblyReloading
 {
     internal static class StatusBarExtender
     {
+        private const int MaxRetries = 5;
         public static event Action<VisualElement> StatusBarCreated; 
         private static GUIView _statusBarInstance;
+        private static int _retries;
 
         [InitializeOnLoadMethod]
         private static void Initialize()
@@ -20,18 +22,29 @@ private static void Initialize()
 
         private static void OnUpdate()
         {
-            if (_statusBarInstance != null)
+            if (_statusBarInstance != null || _retries >= MaxRetries)
                 return;
             _statusBarInstance = Resources.FindObjectsOfTypeAll<AppStatusBar>().FirstOrDefault();
             if(_statusBarInstance == null)
-                return; // todo: prevent FindObjectsOfTypeAll on every tick
-
-            var visualTree = _statusBarInstance.windowBackend.visualTree as VisualElement;
-            visualTree.styleSheets.Add(UIElementsEditorUtility.GetCommonDarkStyleSheet());
+            {
+                _retries++;
+                return;
+            }
+
+            try
+            {
+                var visualTree = _statusBarInstance.windowBackend.visualTree as VisualElement;
+                visualTree.styleSheets.Add(UIElementsEditorUtility.GetCommonDarkStyleSheet());
 
-            var originalBar = visualTree.Q<IMGUIContainer>();
-            originalBar.AddToClassList("status-bar");
-            StatusBarCreated?.Invoke(visualTree);
+                var originalBar = visualTree.Q<IMGUIContainer>();
+                originalBar.AddToClassList("status-bar");
+                StatusBarCreated?.Invoke(visualTree);
+            }
+            catch (Exception e)
+            {
+                Debug.LogError($"Failed to extend status bar: {e}");
+                _retries = MaxRetries;
+            }
         }
     }
 }