Delay frame w unitask when regenerating children, closes #3

Editor/NeilSarkar.VioletUI.Editor.asmdef

@@ -1,7 +1,8 @@
 {
     "name": "NeilSarkar.VioletUI.Editor.asmdef",
     "references": [
-        "GUID:6f301220edcc84749a65b36e4025ea3a"
+        "GUID:6f301220edcc84749a65b36e4025ea3a",
+        "GUID:f51ebe6a0ceec4240a699833d6309b23"
     ],
     "includePlatforms": [
         "Editor"

Editor/PrefabListener.cs

@@ -1,3 +1,4 @@
+using System.Collections;
 using System.Collections.Generic;
 using UnityEditor;
 using UnityEngine;
@@ -36,11 +37,12 @@ static void UpdateRepeatViews(string prefabPath) {
 				if (processed.Contains(go.gameObject)) { continue; }
 
 				if (PrefabUtility.GetPrefabAssetPathOfNearestInstanceRoot(repeatView.ViewPrefab) == prefabPath) {
-					repeatView.RegenerateChildren();
+					_ = repeatView.RegenerateChildren();
 					processed.Add(go.gameObject);
 					Violet.Log($"Updated RepeatView on {go.gameObject.name}");
 				}
 			}
 		}
+
 	}
 }

Runtime/IRepeatView.cs

@@ -1,6 +1,7 @@
 using UnityEngine;
+using Cysharp.Threading.Tasks;
 
 public interface IRepeatView {
-	void RegenerateChildren();
+	UniTask RegenerateChildren();
 	GameObject ViewPrefab { get; }
 }

Runtime/Views/RepeatView.cs

@@ -3,6 +3,7 @@
 using Dispatch;
 using UnityEngine;
 using System.Collections;
+using Cysharp.Threading.Tasks;
 #if UNITY_EDITOR
 using UnityEditor;
 #endif
@@ -67,15 +68,11 @@ GameObject CreateChild(int index) {
 		/// Call RegenerateChildren in editor only to regenerate the children
 		/// when the child prefab is updated
 		/// </summary>
-		public void RegenerateChildren() {
-			StartCoroutine(RegenerateChildrenNextFrame());
-		}
-
-		IEnumerator RegenerateChildrenNextFrame() {
+		public async UniTask RegenerateChildren() {
+#if UNITY_EDITOR
 			// we have to wait a frame because PrefabListener is run as the asset is saving
 			// and the menu prefab won't be updated until the next frame
-#if UNITY_EDITOR
-			yield return null;
+			await UniTask.DelayFrame(1);
 			Transform t = transform;
 			VioletScreen screen = default;
 			while (t != null) {
@@ -86,7 +83,7 @@ IEnumerator RegenerateChildrenNextFrame() {
 
 			if (screen == null) {
 				Violet.LogError($"Can't regenerate children bc screen is null. name={gameObject.name} parent={transform.parent}");
-				yield break;
+				return;
 			}
 
 			var wasEditing = screen.isEditing;