@fanoutio/grip/build/data/websocket/WebSocketMessageFormat.js

GRIP Interface Library

import { encodeBytesToBase64String } from '../../utilities/index.js';
// The WebSocketMessageFormat class is the format used to publish data to
// WebSocket clients connected to GRIP proxies.
export class WebSocketMessageFormat {
    content;
    close;
    code;
    constructor(content = null, close = false, code) {
        // Initialize with either the message content or a boolean indicating that
        // the streaming connection should be closed. If neither the content nor
        // the boolean flag is set then an error will be thrown.
        if (content == null && !close) {
            throw new Error('WebSocketMessageFormat requires content.');
        }
        this.content = content;
        this.close = close;
        this.code = code;
    }
    // The name used when publishing this format.
    name() {
        return 'ws-message';
    }
    // Exports the message in the required format depending on whether the
    // message content is a buffer or not, or whether the connection should
    // be closed.
    export() {
        const obj = {};
        if (this.close) {
            obj['action'] = 'close';
            if (this.code != null) {
                obj['code'] = this.code;
            }
        }
        else {
            if (this.content instanceof Uint8Array) {
                obj['content-bin'] = encodeBytesToBase64String(this.content);
            }
            else if (this.content != null) {
                obj['content'] = this.content.toString();
            }
        }
        return obj;
    }
}