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

cdp: add internal close api

Browse Source
This commit is contained in:
CanadaHonk 2022-12-30 22:12:35 +00:00
parent b23351ae4a
commit 136048ebf9
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/lib/cdp.js
View File

@ -4,6 +4,8 @@ import { get } from 'http';
export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => { export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => {
let messageCallbacks = [], onReply = {}; let messageCallbacks = [], onReply = {};
const onMessage = msg => { const onMessage = msg => {
if (closed) return; // closed, ignore
msg = JSON.parse(msg); msg = JSON.parse(msg);
// log('received', msg); // log('received', msg);
@ -17,10 +19,13 @@ export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => {
for (const callback of messageCallbacks) callback(msg); for (const callback of messageCallbacks) callback(msg);
}; };
let _send; let closed = false;
let _send, _close;
let msgId = 0; let msgId = 0;
const sendMessage = async (method, params = {}, sessionId = undefined) => { const sendMessage = async (method, params = {}, sessionId = undefined) => {
if (closed) throw new Error('CDP connection closed');
const id = msgId++; const id = msgId++;
const msg = { const msg = {
@ -78,12 +83,16 @@ export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => {
const ws = new WebSocket(target.webSocketDebuggerUrl); const ws = new WebSocket(target.webSocketDebuggerUrl);
await new Promise(resolve => ws.on('open', resolve)); await new Promise(resolve => ws.on('open', resolve));
ws.on('message', data => onMessage(data));
_send = data => ws.send(data); _send = data => ws.send(data);
ws.on('message', data => onMessage(data)); _close = () => ws.close();
} else { } else {
let pending = ''; let pending = '';
pipeRead.on('data', buf => { pipeRead.on('data', buf => {
if (closed) return; // closed, ignore
let end = buf.indexOf('\0'); // messages are null separated let end = buf.indexOf('\0'); // messages are null separated
if (end === -1) { // no complete message yet if (end === -1) { // no complete message yet
@ -110,6 +119,8 @@ export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => {
pipeWrite.write(data); pipeWrite.write(data);
pipeWrite.write('\0'); pipeWrite.write('\0');
}; };
_close = () => {};
} }
return { return {
@ -121,6 +132,12 @@ export default async ({ pipe: { pipeWrite, pipeRead } = {}, port }) => {
messageCallbacks.push(callback); messageCallbacks.push(callback);
}, },
sendMessage
sendMessage,
close: () => {
closed = true;
_close();
}
}; };
}; };