mirror of
https://github.com/microsoft/vscode.git
synced 2026-02-21 02:11:11 +00:00
0ac8174fbc
<!-- Thank you for submitting a Pull Request. Please: * Read our Pull Request guidelines: https://github.com/microsoft/vscode/wiki/How-to-Contribute#pull-requests * Associate an issue with the Pull Request. * Ensure that the code is up-to-date with the `main` branch. * Include a description of the proposed changes and how to test them. -->
67 lines
2.8 KiB
TypeScript
67 lines
2.8 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import { Emitter } from '../../../base/common/event.js';
|
|
import { IExtensionDescription } from '../../../platform/extensions/common/extensions.js';
|
|
import { ExtHostNotebookRenderersShape, IMainContext, MainContext, MainThreadNotebookRenderersShape } from './extHost.protocol.js';
|
|
import { ExtHostNotebookController } from './extHostNotebook.js';
|
|
import { ExtHostNotebookEditor } from './extHostNotebookEditor.js';
|
|
import * as vscode from 'vscode';
|
|
|
|
|
|
export class ExtHostNotebookRenderers implements ExtHostNotebookRenderersShape {
|
|
private readonly _rendererMessageEmitters = new Map<string /* rendererId */, Emitter<{ editor: vscode.NotebookEditor; message: unknown }>>();
|
|
private readonly proxy: MainThreadNotebookRenderersShape;
|
|
|
|
constructor(mainContext: IMainContext, private readonly _extHostNotebook: ExtHostNotebookController) {
|
|
this.proxy = mainContext.getProxy(MainContext.MainThreadNotebookRenderers);
|
|
}
|
|
|
|
public $postRendererMessage(editorId: string, rendererId: string, message: unknown): void {
|
|
const editor = this._extHostNotebook.getEditorById(editorId);
|
|
this._rendererMessageEmitters.get(rendererId)?.fire({ editor: editor.apiEditor, message });
|
|
}
|
|
|
|
public createRendererMessaging(manifest: IExtensionDescription, rendererId: string): vscode.NotebookRendererMessaging {
|
|
if (!manifest.contributes?.notebookRenderer?.some(r => r.id === rendererId)) {
|
|
throw new Error(`Extensions may only call createRendererMessaging() for renderers they contribute (got ${rendererId})`);
|
|
}
|
|
|
|
const messaging: vscode.NotebookRendererMessaging = {
|
|
onDidReceiveMessage: (listener, thisArg, disposables) => {
|
|
return this.getOrCreateEmitterFor(rendererId).event(listener, thisArg, disposables);
|
|
},
|
|
postMessage: (message, editorOrAlias) => {
|
|
if (ExtHostNotebookEditor.apiEditorsToExtHost.has(message)) { // back compat for swapped args
|
|
[message, editorOrAlias] = [editorOrAlias, message];
|
|
}
|
|
|
|
const extHostEditor = editorOrAlias && ExtHostNotebookEditor.apiEditorsToExtHost.get(editorOrAlias);
|
|
return this.proxy.$postMessage(extHostEditor?.id, rendererId, message);
|
|
},
|
|
};
|
|
|
|
return messaging;
|
|
}
|
|
|
|
private getOrCreateEmitterFor(rendererId: string) {
|
|
let emitter = this._rendererMessageEmitters.get(rendererId);
|
|
if (emitter) {
|
|
return emitter;
|
|
}
|
|
|
|
emitter = new Emitter({
|
|
onDidRemoveLastListener: () => {
|
|
emitter?.dispose();
|
|
this._rendererMessageEmitters.delete(rendererId);
|
|
}
|
|
});
|
|
|
|
this._rendererMessageEmitters.set(rendererId, emitter);
|
|
|
|
return emitter;
|
|
}
|
|
}
|