inject: rewrite evalOnNewDoc to return remove func
This commit is contained in:
parent
235301f566
commit
04b31831fd
@ -98,13 +98,25 @@ export default async (CDP, proc, injectionType = 'browser', { dataPath, browserN
|
|||||||
return reply.result?.value ?? reply;
|
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({
|
const [ ipcMessageCallback, injectIPC, IPC ] = await IPCApi({
|
||||||
browserName,
|
browserName,
|
||||||
browserInfo,
|
browserInfo,
|
||||||
browserType
|
browserType
|
||||||
}, {
|
}, {
|
||||||
evalInWindow,
|
evalInWindow,
|
||||||
evalOnNewDocument: source => CDP.sendMessage('Page.addScriptToEvaluateOnNewDocument', { source }, sessionId)
|
evalOnNewDocument
|
||||||
});
|
});
|
||||||
onWindowMessage = ipcMessageCallback;
|
onWindowMessage = ipcMessageCallback;
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user