Add WaitFrameBeforeRebuild property

Author: ivan-trackman

Editor/Inspector/FigmaInspector.cs

@@ -33,6 +33,7 @@ public class FigmaInspector : Editor
         SerializedProperty title;
         SerializedProperty filter;
         SerializedProperty reorder;
+        SerializedProperty waitFrameBeforeRebuild;
         SerializedProperty fontsDirs;
         UIDocument document;
         #endregion
@@ -51,6 +52,7 @@ void OnEnable()
             title = serializedObject.FindProperty(nameof(title));
             filter = serializedObject.FindProperty(nameof(filter));
             reorder = serializedObject.FindProperty(nameof(reorder));
+            waitFrameBeforeRebuild = serializedObject.FindProperty(nameof(waitFrameBeforeRebuild));
             fontsDirs = serializedObject.FindProperty(nameof(fontsDirs));
             document = ((MonoBehaviour)target).GetComponent<UIDocument>();
         }
@@ -139,6 +141,7 @@ void OnFigmaGUI()
             EditorGUILayout.PropertyField(reorder, new GUIContent("De-root and Re-order Hierarchy"));
             EditorGUILayout.PropertyField(filter, new GUIContent("Filter by Path"));
             EditorGUILayout.PropertyField(fontsDirs, new GUIContent("Additional Fonts Directories"));
+            EditorGUILayout.PropertyField(waitFrameBeforeRebuild, new GUIContent("Wait Frame Before Rebuild"));
             EditorGUI.BeginDisabledGroup(true);
 
             if (document && document.visualTreeAsset)

Runtime/Figma.cs

@@ -15,6 +15,7 @@ public class Figma : MonoBehaviour
         [SerializeField] string title;
         [SerializeField] bool filter;
         [SerializeField] bool reorder;
+        [SerializeField] bool waitFrameBeforeRebuild = true;
         [SerializeField] string[] fontsDirs;
         #endregion
 
@@ -36,7 +37,7 @@ async void OnEnable()
 
             if (!Application.isPlaying) return;
 
-            await new WaitForEndOfFrame();
+            if (waitFrameBeforeRebuild) await new WaitForEndOfFrame();
             VisualElementMetadata.Rebuild(elements);
 
             VisualElement root = document.rootVisualElement.Q(UxmlAttribute.prefix);