Function initializeWidgetComms

src/test/datascience/widgets/commUtils.ts:12–37

(disposables: IDisposable[])

import colors from 'colors';

export function initializeWidgetComms(disposables: IDisposable[]): Utils {
    const messageChannel = notebooks.createRendererMessaging('deepnote-ipywidget-renderer');
    if (!messageChannel) {
        throw new Error('No Widget renderer comms channel');
    }
    const deferred = createDeferred<NotebookEditor>();
    // eslint-disable-next-line @typescript-eslint/no-use-before-define
    const utils = new Utils(messageChannel, deferred.promise);
    disposables.push(utils);
    logger.ci(`Adding comm message handler`);
    const disposable = messageChannel.onDidReceiveMessage(async ({ editor, message }) => {
        if (message && message.command === 'log') {
            const messageToLog = message.category === 'error' ? colors.red(message.message) : message.message;
            const category = message.category ? ` (${message.category})` : '';
            logger.info(`${colors.yellow('Widget renderer')}${category}: ${messageToLog}`);
        }
        if (message && message.command === 'INIT') {
            deferred.resolve(editor);
            // Redirect all of console.log, console.warn & console.error messages from
            // renderer to the extension host.
            messageChannel.postMessage({ command: 'hijackLogging' }, editor).then(noop, noop);
        }
    });
    disposables.push(disposable);
    return utils;
}

export class Utils {
    private readonly disposables: IDisposable[] = [];