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.
61 lines
1.8 KiB
TypeScript
61 lines
1.8 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 * as NodeWebSocket from 'ws';
|
|
import type {ConnectionTransport} from './ConnectionTransport';
|
|
|
|
export class WebSocketTransport implements ConnectionTransport {
|
|
static create(url: string): Promise<WebSocketTransport> {
|
|
return new Promise((resolve, reject) => {
|
|
const ws = new NodeWebSocket(url, [], {
|
|
perMessageDeflate: false,
|
|
maxPayload: 256 * 1024 * 1024, // 256Mb
|
|
});
|
|
|
|
ws.addEventListener('open', () => resolve(new WebSocketTransport(ws)));
|
|
ws.addEventListener('error', reject);
|
|
});
|
|
}
|
|
|
|
_ws: NodeWebSocket;
|
|
onmessage?: (message: string) => void;
|
|
onclose?: () => void;
|
|
|
|
constructor(ws: NodeWebSocket) {
|
|
this._ws = ws;
|
|
this._ws.addEventListener('message', event => {
|
|
if (this.onmessage)
|
|
this.onmessage.call(null, event.data);
|
|
});
|
|
this._ws.addEventListener('close', () => {
|
|
if (this.onclose)
|
|
this.onclose.call(null);
|
|
});
|
|
// Silently ignore all errors - we don't know what to do with them.
|
|
this._ws.addEventListener('error', () => {});
|
|
this.onmessage = null;
|
|
this.onclose = null;
|
|
}
|
|
|
|
send(message): void {
|
|
this._ws.send(message);
|
|
}
|
|
|
|
close(): void {
|
|
this._ws.close();
|
|
}
|
|
}
|
|
|