github/puppeteer
0
0
Fork 0
mirror of https://github.com/puppeteer/puppeteer synced 2024-06-14 14:02:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify waitFor code (#92)

Browse Source
This commit is contained in:
Pavel Feldman 2017-07-18 12:16:19 -07:00 committed by Andrey Lushnikov
parent 5ca92ffb71
commit bef9982687
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
lib/FrameManager.js
View File

@ -178,25 +178,28 @@ class FrameManager extends EventEmitter {
* @return {!Promise<undefined>}
*/
async _waitForInFrame(selector, frame) {
let code = selector => {
let promise = new Promise(fulfill => {
function code(selector) {
if (document.querySelector(selector))
return Promise.resolve();
let callback;
const result = new Promise(fulfill => callback = fulfill);
const mo = new MutationObserver((mutations, observer) => {
if (document.querySelector(selector)) {
fulfill();
observer.disconnect();
callback();
return;
}
new MutationObserver((mutations, observer) => {
if (document.querySelector(selector)) {
observer.disconnect();
fulfill();
return;
}
}).observe(document.documentElement, {
childList: true,
subtree: true
});
});
return promise;
};
mo.observe(document.documentElement, {
childList: true,
subtree: true
});
return result;
}
let contextId = undefined;
if (!frame.isMainFrame()) {
contextId = this._frameIdToExecutionContextId.get(frame._id);