fix(pb-mixin): fix bug in wait method introduced by 189d5a1

eeditiones · May 19, 2024 · afb665d · afb665d
1 parent b16730a
commit afb665d
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/src/pb-mixin.js b/src/pb-mixin.js
@@ -213,11 +213,7 @@ export const pbMixin = (superclass) => class PbMixin extends superclass {
      * @param {Function} callback function to be called when all components are ready
      */
     wait(callback) {
-        if (!this.waitFor) {
-            callback();
-            return;
-        }
-        window.addEventListener('DOMContentLoaded', () => {
+        const _checkAndWait = () => {
             const targetNodes = Array.from(document.querySelectorAll(this.waitFor));
             const targets = targetNodes.filter(target => this.emitsOnSameChannel(target));
             const targetCount = targets.length;
@@ -247,7 +243,19 @@ export const pbMixin = (superclass) => class PbMixin extends superclass {
                     }
                 });
             });
-        });
+        }
+
+        if (!this.waitFor) {
+            callback();
+            return;
+        }
+        if (document.readyState === 'loading') {
+            document.addEventListener('DOMContentLoaded', () => {
+                _checkAndWait();
+            });
+        } else {
+            _checkAndWait();
+        }
     }
 
     /**