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

inject: rewrite evalOnNewDoc to return remove func

Browse Source
This commit is contained in:
CanadaHonk 2023-04-05 15:35:41 +01:00
parent 235301f566
commit 04b31831fd
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/launcher/inject.js
View File

@ -98,13 +98,25 @@ export default async (CDP, proc, injectionType = 'browser', { dataPath, browserN
return reply.result?.value ?? reply;
};
const evalOnNewDocument = async source => {
const { identifier } = await CDP.sendMessage('Page.addScriptToEvaluateOnNewDocument', {
source
}, sessionId);
return async () => {
await CDP.sendMessage('Page.removeScriptToEvaluateOnNewDocument', {
identifier
}, sessionId);
};
};
const [ ipcMessageCallback, injectIPC, IPC ] = await IPCApi({
browserName,
browserInfo,
browserType
}, {
evalInWindow,
evalOnNewDocument: source => CDP.sendMessage('Page.addScriptToEvaluateOnNewDocument', { source }, sessionId)
evalOnNewDocument
});
onWindowMessage = ipcMessageCallback;