fix: ensure onStylesLoaded callback is fired consistently

measuredco · Apr 16, 2024 · ae0f903 · ae0f903
1 parent e50ef03
commit ae0f903
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/src/AutoFrameComponent.tsx b/src/AutoFrameComponent.tsx
@@ -202,16 +202,9 @@ const CopyHostStyles = ({
 
     const collectedStyles = collectStyles(parentDocument);
 
-    let mountedCounter = 0;
-
     Promise.all(
       collectedStyles.map(async (styleNode) => {
-        const mirror = await mirrorEl(styleNode, () => {
-          mountedCounter += 1;
-          if (mountedCounter === collectedStyles.length) {
-            onStylesLoaded();
-          }
-        });
+        const mirror = await mirrorEl(styleNode);
 
         if (!mirror) return;
 
@@ -237,6 +230,8 @@ const CopyHostStyles = ({
 
         hashes[elHash] = true;
       });
+
+      onStylesLoaded();
     });
 
     return () => {