@sourcegraph/extensions-client-common/lib/messaging.js

import { fromEvent, Subscription } from 'rxjs';
import { filter, map } from 'rxjs/operators';
import { createMessageConnection } from 'sourcegraph/module/jsonrpc2/connection';
import { AbstractMessageReader, AbstractMessageWriter, } from 'sourcegraph/module/jsonrpc2/transport';
class SubscribableMessageReader extends AbstractMessageReader {
    constructor(subscribable) {
        super();
        this.pending = [];
        this.callback = null;
        this.subscription = new Subscription();
        this.subscription.add(subscribable.subscribe(message => {
            try {
                if (this.callback) {
                    this.callback(message);
                }
                else {
                    this.pending.push(message);
                }
            }
            catch (err) {
                this.fireError(err);
            }
        }));
    }
    listen(callback) {
        if (this.callback) {
            throw new Error('callback is already set');
        }
        this.callback = callback;
        while (this.pending.length !== 0) {
            callback(this.pending.pop());
        }
    }
    unsubscribe() {
        super.unsubscribe();
        this.subscription.unsubscribe();
    }
}
class CallbackMessageWriter extends AbstractMessageWriter {
    constructor(callback) {
        super();
        this.callback = callback;
    }
    write(message) {
        this.callback(message);
    }
    unsubscribe() {
        super.unsubscribe();
    }
}
/**
 * Connects the Sourcegraph extension registry page to a client (such as a browser extension) or vice versa.
 */
function connectAs(source) {
    const messageConnection = createMessageConnection({
        reader: new SubscribableMessageReader(fromEvent(window, 'message').pipe(
        // Filter to relevant messages, ignoring our own
        filter(m => m.data && m.data.source && m.data.source !== source && m.data.message), map(m => m.data.message))),
        writer: new CallbackMessageWriter(message => {
            window.postMessage({ source, message }, '*');
        }),
    });
    messageConnection.listen();
    return new Promise(resolve => {
        messageConnection.onNotification('Ping', () => {
            messageConnection.sendNotification('Pong');
            resolve(messageConnection);
        });
        messageConnection.onNotification('Pong', () => {
            resolve(messageConnection);
        });
        messageConnection.sendNotification('Ping');
    });
}
/**
 * Connects the client (such as a browser extension) to a Sourcegraph extension registry page.
 */
export function connectAsClient() {
    return connectAs('Client').then(connection => ({
        onEditSetting: callback => {
            connection.onRequest('EditSetting', callback);
        },
        onGetSettings: callback => {
            connection.onRequest('GetSettings', callback);
        },
        sendSettings: settings => {
            connection.sendNotification('Settings', settings);
        },
        rawConnection: connection,
    }));
}
/**
 * Connects the Sourcegraph extension registry page to a client (such as a browser extension).
 */
export function connectAsPage() {
    return connectAs('Page').then(connection => ({
        onSettings: callback => {
            connection.onNotification('Settings', callback);
        },
        editSetting: callback => connection.sendRequest('EditSetting', callback),
        getSettings: () => connection.sendRequest('GetSettings'),
        rawConnection: connection,
    }));
}
//# sourceMappingURL=messaging.js.map