ae7483d5cf
* chore: remove src/externs.d.ts It defined global types that we don't want to use, and instead we move to using interfaces that we import and reference just like with any other interface. This means other than Protocol (which I think is fine to leave as is), there are no other magic global types and you have to import any types or interfaces that you want.
74 lines
2.3 KiB
TypeScript
74 lines
2.3 KiB
TypeScript
/**
|
|
* Copyright 2018 Google Inc. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
import {helper, debugError, PuppeteerEventListener} from './helper';
|
|
import type {ConnectionTransport} from './ConnectionTransport';
|
|
|
|
export class PipeTransport implements ConnectionTransport {
|
|
_pipeWrite: NodeJS.WritableStream;
|
|
_pendingMessage: string;
|
|
_eventListeners: PuppeteerEventListener[];
|
|
|
|
onclose?: () => void;
|
|
onmessage?: () => void;
|
|
|
|
constructor(pipeWrite: NodeJS.WritableStream, pipeRead: NodeJS.ReadableStream) {
|
|
this._pipeWrite = pipeWrite;
|
|
this._pendingMessage = '';
|
|
this._eventListeners = [
|
|
helper.addEventListener(pipeRead, 'data', buffer => this._dispatch(buffer)),
|
|
helper.addEventListener(pipeRead, 'close', () => {
|
|
if (this.onclose)
|
|
this.onclose.call(null);
|
|
}),
|
|
helper.addEventListener(pipeRead, 'error', debugError),
|
|
helper.addEventListener(pipeWrite, 'error', debugError),
|
|
];
|
|
this.onmessage = null;
|
|
this.onclose = null;
|
|
}
|
|
|
|
send(message: string): void {
|
|
this._pipeWrite.write(message);
|
|
this._pipeWrite.write('\0');
|
|
}
|
|
|
|
_dispatch(buffer: Buffer): void {
|
|
let end = buffer.indexOf('\0');
|
|
if (end === -1) {
|
|
this._pendingMessage += buffer.toString();
|
|
return;
|
|
}
|
|
const message = this._pendingMessage + buffer.toString(undefined, 0, end);
|
|
if (this.onmessage)
|
|
this.onmessage.call(null, message);
|
|
|
|
let start = end + 1;
|
|
end = buffer.indexOf('\0', start);
|
|
while (end !== -1) {
|
|
if (this.onmessage)
|
|
this.onmessage.call(null, buffer.toString(undefined, start, end));
|
|
start = end + 1;
|
|
end = buffer.indexOf('\0', start);
|
|
}
|
|
this._pendingMessage = buffer.toString(undefined, start);
|
|
}
|
|
|
|
close(): void {
|
|
this._pipeWrite = null;
|
|
helper.removeEventListeners(this._eventListeners);
|
|
}
|
|
}
|