103 lines
4.1 KiB
TypeScript
103 lines
4.1 KiB
TypeScript
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
import { ILogger, LogLevel } from "./ILogger";
|
|
import { ITransport, TransferFormat } from "./ITransport";
|
|
import { WebSocketConstructor } from "./Polyfills";
|
|
import { Arg, getDataDetail } from "./Utils";
|
|
|
|
/** @private */
|
|
export class WebSocketTransport implements ITransport {
|
|
private readonly logger: ILogger;
|
|
private readonly accessTokenFactory: (() => string | Promise<string>) | undefined;
|
|
private readonly logMessageContent: boolean;
|
|
private readonly webSocketConstructor: WebSocketConstructor;
|
|
private webSocket?: WebSocket;
|
|
|
|
public onreceive: ((data: string | ArrayBuffer) => void) | null;
|
|
public onclose: ((error?: Error) => void) | null;
|
|
|
|
constructor(accessTokenFactory: (() => string | Promise<string>) | undefined, logger: ILogger,
|
|
logMessageContent: boolean, webSocketConstructor: WebSocketConstructor) {
|
|
this.logger = logger;
|
|
this.accessTokenFactory = accessTokenFactory;
|
|
this.logMessageContent = logMessageContent;
|
|
this.webSocketConstructor = webSocketConstructor;
|
|
|
|
this.onreceive = null;
|
|
this.onclose = null;
|
|
}
|
|
|
|
public async connect(url: string, transferFormat: TransferFormat): Promise<void> {
|
|
Arg.isRequired(url, "url");
|
|
Arg.isRequired(transferFormat, "transferFormat");
|
|
Arg.isIn(transferFormat, TransferFormat, "transferFormat");
|
|
|
|
this.logger.log(LogLevel.Trace, "(WebSockets transport) Connecting");
|
|
|
|
if (this.accessTokenFactory) {
|
|
const token = await this.accessTokenFactory();
|
|
if (token) {
|
|
url += (url.indexOf("?") < 0 ? "?" : "&") + `access_token=${encodeURIComponent(token)}`;
|
|
}
|
|
}
|
|
|
|
return new Promise<void>((resolve, reject) => {
|
|
url = url.replace(/^http/, "ws");
|
|
const webSocket = new this.webSocketConstructor(url);
|
|
if (transferFormat === TransferFormat.Binary) {
|
|
webSocket.binaryType = "arraybuffer";
|
|
}
|
|
|
|
// tslint:disable-next-line:variable-name
|
|
webSocket.onopen = (_event: Event) => {
|
|
this.logger.log(LogLevel.Information, `WebSocket connected to ${url}`);
|
|
this.webSocket = webSocket;
|
|
resolve();
|
|
};
|
|
|
|
webSocket.onerror = (event: Event) => {
|
|
const error = (event instanceof ErrorEvent) ? event.error : null;
|
|
reject(error);
|
|
};
|
|
|
|
webSocket.onmessage = (message: MessageEvent) => {
|
|
this.logger.log(LogLevel.Trace, `(WebSockets transport) data received. ${getDataDetail(message.data, this.logMessageContent)}.`);
|
|
if (this.onreceive) {
|
|
this.onreceive(message.data);
|
|
}
|
|
};
|
|
|
|
webSocket.onclose = (event: CloseEvent) => {
|
|
// webSocket will be null if the transport did not start successfully
|
|
this.logger.log(LogLevel.Trace, "(WebSockets transport) socket closed.");
|
|
if (this.onclose) {
|
|
if (event.wasClean === false || event.code !== 1000) {
|
|
this.onclose(new Error(`Websocket closed with status code: ${event.code} (${event.reason})`));
|
|
} else {
|
|
this.onclose();
|
|
}
|
|
}
|
|
};
|
|
});
|
|
}
|
|
|
|
public send(data: any): Promise<void> {
|
|
if (this.webSocket && this.webSocket.readyState === this.webSocketConstructor.OPEN) {
|
|
this.logger.log(LogLevel.Trace, `(WebSockets transport) sending data. ${getDataDetail(data, this.logMessageContent)}.`);
|
|
this.webSocket.send(data);
|
|
return Promise.resolve();
|
|
}
|
|
|
|
return Promise.reject("WebSocket is not in the OPEN state");
|
|
}
|
|
|
|
public stop(): Promise<void> {
|
|
if (this.webSocket) {
|
|
this.webSocket.close();
|
|
this.webSocket = undefined;
|
|
}
|
|
return Promise.resolve();
|
|
}
|
|
}
|