add kernel service and add extension host logic for it

src/vs/workbench/api/browser/mainThreadNotebookKernels.ts

@@ -0,0 +1,129 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Emitter, Event } from 'vs/base/common/event';
+import { IDisposable } from 'vs/base/common/lifecycle';
+import { URI } from 'vs/base/common/uri';
+import { ICommandService } from 'vs/platform/commands/common/commands';
+import { extHostNamedCustomer } from 'vs/workbench/api/common/extHostCustomers';
+import { ICell } from 'vs/workbench/contrib/notebook/common/notebookCommon';
+import { INotebookKernel2, INotebookKernel2ChangeEvent, INotebookKernelService } from 'vs/workbench/contrib/notebook/common/notebookKernelService';
+import { NotebookSelector } from 'vs/workbench/contrib/notebook/common/notebookSelector';
+import { ExtHostContext, ExtHostNotebookKernelsShape, IExtHostContext, INotebookKernelDto2, MainContext, MainThreadNotebookKernelsShape } from '../common/extHost.protocol';
+
+abstract class MainThreadKernel implements INotebookKernel2 {
+
+	private readonly _onDidChange = new Emitter<INotebookKernel2ChangeEvent>();
+	readonly onDidChange: Event<INotebookKernel2ChangeEvent> = this._onDidChange.event;
+
+	readonly id: string;
+	readonly selector: NotebookSelector;
+
+	label: string;
+	detail?: string;
+	description?: string;
+	isPreferred?: boolean;
+	supportedLanguages: string[];
+	hasExecutionOrder: boolean;
+	preloads?: URI[];
+
+	constructor(data: INotebookKernelDto2) {
+		this.id = data.id;
+		this.selector = data.selector;
+		this.label = data.label;
+		this.detail = data.detail;
+		this.description = data.description;
+		this.isPreferred = data.isPreferred;
+		this.supportedLanguages = data.supportedLanguages;
+		this.hasExecutionOrder = data.hasExecutionOrder ?? false;
+		this.preloads = data.preloads;
+	}
+
+
+	update(data: Partial<INotebookKernelDto2>) {
+		const event: INotebookKernel2ChangeEvent = Object.create(null);
+		if (data.label !== undefined) {
+			this.label = data.label;
+			event.label = true;
+		}
+		if (data.detail !== undefined) {
+			this.detail = data.detail;
+			event.detail = true;
+		}
+		if (data.description !== undefined) {
+			this.description = data.description;
+			event.description = true;
+		}
+		if (data.isPreferred !== undefined) {
+			this.isPreferred = data.isPreferred;
+			event.isPreferred = true;
+		}
+		if (data.supportedLanguages !== undefined) {
+			this.supportedLanguages = data.supportedLanguages;
+			event.supportedLanguages = true;
+		}
+		if (data.hasExecutionOrder !== undefined) {
+			this.hasExecutionOrder = data.hasExecutionOrder;
+			event.hasExecutionOrder = true;
+		}
+		this._onDidChange.fire(event);
+	}
+
+	abstract setSelected(value: boolean): void;
+	abstract executeCells(cells: ICell[]): void;
+}
+
+@extHostNamedCustomer(MainContext.MainThreadNotebookKernels)
+export class MainThreadNotebookKernels implements MainThreadNotebookKernelsShape {
+
+	private readonly _kernels = new Map<number, [kernel: MainThreadKernel, registraion: IDisposable]>();
+	private readonly _proxy: ExtHostNotebookKernelsShape;
+
+	constructor(
+		extHostContext: IExtHostContext,
+		@INotebookKernelService private readonly _notebookKernelService: INotebookKernelService,
+		@ICommandService private readonly _commandService: ICommandService,
+	) {
+		this._proxy = extHostContext.getProxy(ExtHostContext.ExtHostNotebookKernels);
+	}
+
+	dispose(): void {
+		for (let [, registration] of this._kernels.values()) {
+			registration.dispose();
+		}
+	}
+
+	$addKernel(handle: number, data: INotebookKernelDto2): void {
+		const that = this;
+		const kernel = new class extends MainThreadKernel {
+			setSelected(value: boolean): void {
+				that._proxy.$acceptSelection(handle, value);
+			}
+			executeCells(cells: ICell[]): void {
+				// todo@jrieken push down to INotebookKernel2?
+				if (data.executeCommand) {
+					that._commandService.executeCommand(data.executeCommand.id, cells);
+				}
+			}
+		}(data);
+		const disposable = this._notebookKernelService.addKernel(kernel);
+		this._kernels.set(handle, [kernel, disposable]);
+	}
+
+	$updateKernel(handle: number, data: Partial<INotebookKernelDto2>): void {
+		const tuple = this._kernels.get(handle);
+		if (tuple) {
+			tuple[0].update(data);
+		}
+	}
+
+	$removeKernel(handle: number): void {
+		const tuple = this._kernels.get(handle);
+		if (tuple) {
+			tuple[1].dispose();
+			this._kernels.delete(handle);
+		}
+	}
+}