github/gluon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

inject: add window.loaded promise

Browse Source
This commit is contained in:
Oj 2023-01-05 18:41:45 +00:00
parent 785778d450
commit 44f6489a52
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/launcher/inject.js
View File

@ -1,10 +1,12 @@
import IPCApi from '../lib/ipc.js';
export default async (CDP, proc, injectionType = 'browser', { browserName } = { browserName: 'unknown' }) => {
let pageLoadCallback = () => {}, onWindowMessage = () => {};
let pageLoadCallback, pageLoadPromise = new Promise(res => pageLoadCallback = res);
let frameLoadCallback = () => {}, onWindowMessage = () => {};
CDP.onMessage(msg => {
if (msg.method === 'Runtime.bindingCalled' && msg.params.name === '_gluonSend') onWindowMessage(JSON.parse(msg.params.payload));
if (msg.method === 'Page.frameStoppedLoading') pageLoadCallback(msg.params);
if (msg.method === 'Page.frameStoppedLoading') frameLoadCallback(msg.params);
if (msg.method === 'Page.loadEventFired') pageLoadCallback();
if (msg.method === 'Runtime.executionContextCreated') injectIPC(); // ensure IPC injection again
});
@ -42,13 +44,17 @@ export default async (CDP, proc, injectionType = 'browser', { browserName } = {
}, {
evalInWindow,
evalOnNewDocument: source => CDP.sendMessage('Page.addScriptToEvaluateOnNewDocument', { source }, sessionId),
pageLoadPromise: new Promise(res => pageLoadCallback = res)
pageLoadPromise: new Promise(res => frameLoadCallback = res)
});
onWindowMessage = ipcMessageCallback;
log('finished setup');
evalInWindow('document.readyState').then(readyState => {
if (readyState === 'complete' || readyState === 'ready') pageLoadCallback();
});
const generateVersionInfo = (name, version) => ({
name,
version,
@ -58,6 +64,7 @@ export default async (CDP, proc, injectionType = 'browser', { browserName } = {
return {
window: {
eval: evalInWindow,
loaded: pageLoadPromise
},
ipc: IPC,