Support kernel source actions update.

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

@@ -115,6 +115,8 @@ export class MainThreadNotebookKernels implements MainThreadNotebookKernelsShape
 	private readonly _kernels = new Map<number, [kernel: MainThreadKernel, registraion: IDisposable]>();
 	private readonly _kernelDetectionTasks = new Map<number, [task: MainThreadKernelDetectionTask, registraion: IDisposable]>();
 	private readonly _kernelSourceActionProviders = new Map<number, [provider: IKernelSourceActionProvider, registraion: IDisposable]>();
+	private readonly _kernelSourceActionProvidersEventRegistrations = new Map<number, IDisposable>();
+
 	private readonly _proxy: ExtHostNotebookKernelsShape;
 
 	private readonly _executions = new Map<number, INotebookCellExecution>();
@@ -314,7 +316,7 @@ export class MainThreadNotebookKernels implements MainThreadNotebookKernelsShape
 
 	// --- notebook kernel source action provider
 
-	async $addKernelSourceActionProvider(handle: number, notebookType: string): Promise<void> {
+	async $addKernelSourceActionProvider(handle: number, eventHandle: number, notebookType: string): Promise<void> {
 		const kernelSourceActionProvider: IKernelSourceActionProvider = {
 			viewType: notebookType,
 			provideKernelSourceActions: async () => {
@@ -336,15 +338,32 @@ export class MainThreadNotebookKernels implements MainThreadNotebookKernelsShape
 				});
 			}
 		};
+
+		if (typeof eventHandle === 'number') {
+			const emitter = new Emitter<void>();
+			this._kernelSourceActionProvidersEventRegistrations.set(eventHandle, emitter);
+			kernelSourceActionProvider.onDidChangeSourceActions = emitter.event;
+		}
+
 		const registration = this._notebookKernelService.registerKernelSourceActionProvider(notebookType, kernelSourceActionProvider);
 		this._kernelSourceActionProviders.set(handle, [kernelSourceActionProvider, registration]);
 	}
 
-	$removeKernelSourceActionProvider(handle: number): void {
+	$removeKernelSourceActionProvider(handle: number, eventHandle: number): void {
 		const tuple = this._kernelSourceActionProviders.get(handle);
 		if (tuple) {
 			tuple[1].dispose();
 			this._kernelSourceActionProviders.delete(handle);
 		}
+		if (typeof eventHandle === 'number') {
+			this._kernelSourceActionProvidersEventRegistrations.delete(eventHandle);
+		}
+	}
+
+	$emitNotebookKernelSourceActionsChangeEvent(eventHandle: number): void {
+		const emitter = this._kernelSourceActionProvidersEventRegistrations.get(eventHandle);
+		if (emitter instanceof Emitter) {
+			emitter.fire(undefined);
+		}
 	}
 }

src/vs/workbench/api/common/extHost.protocol.ts

@@ -1055,8 +1055,9 @@ export interface MainThreadNotebookKernelsShape extends IDisposable {
 	$addKernelDetectionTask(handle: number, notebookType: string): Promise<void>;
 	$removeKernelDetectionTask(handle: number): void;
 
-	$addKernelSourceActionProvider(handle: number, notebookType: string): Promise<void>;
-	$removeKernelSourceActionProvider(handle: number): void;
+	$addKernelSourceActionProvider(handle: number, eventHandle: number, notebookType: string): Promise<void>;
+	$removeKernelSourceActionProvider(handle: number, eventHandle: number): void;
+	$emitNotebookKernelSourceActionsChangeEvent(eventHandle: number): void;
 }
 
 export interface MainThreadNotebookRenderersShape extends IDisposable {

src/vs/workbench/api/common/extHostNotebookKernels.ts

@@ -299,16 +299,23 @@ export class ExtHostNotebookKernels implements ExtHostNotebookKernelsShape {
 
 	registerKernelSourceActionProvider(extension: IExtensionDescription, viewType: string, provider: vscode.NotebookKernelSourceActionProvider) {
 		const handle = this._kernelSourceActionProviderHandlePool++;
+		const eventHandle = typeof provider.onDidChangeNotebookKernelSourceActions === 'function' ? handle : undefined;
 		const that = this;
 
 		this._kernelSourceActionProviders.set(handle, provider);
 		this._logService.trace(`NotebookKernelSourceActionProvider[${handle}], CREATED by ${extension.identifier.value}`);
-		this._proxy.$addKernelSourceActionProvider(handle, viewType);
+		this._proxy.$addKernelSourceActionProvider(handle, handle, viewType);
+
+		let subscription: vscode.Disposable | undefined;
+		if (eventHandle !== undefined) {
+			subscription = provider.onDidChangeNotebookKernelSourceActions!(_ => this._proxy.$emitNotebookKernelSourceActionsChangeEvent(eventHandle));
+		}
 
 		return {
 			dispose: () => {
 				this._kernelSourceActionProviders.delete(handle);
-				that._proxy.$removeKernelSourceActionProvider(handle);
+				that._proxy.$removeKernelSourceActionProvider(handle, handle);
+				subscription?.dispose();
 			}
 		};
 	}

src/vs/workbench/contrib/notebook/browser/services/notebookKernelServiceImpl.ts

@@ -384,13 +384,19 @@ export class NotebookKernelService extends Disposable implements INotebookKernel
 		this._kernelSourceActionProviders.set(viewType, providers);
 		this._onDidChangeSourceActions.fire({ viewType: viewType });
 
+		const eventEmitterDisposable = provider.onDidChangeSourceActions?.(() => {
+			this._onDidChangeSourceActions.fire({ viewType: viewType });
+		});
+
 		return toDisposable(() => {
 			const providers = this._kernelSourceActionProviders.get(viewType) ?? [];
 			const idx = providers.indexOf(provider);
 			if (idx >= 0) {
 				providers.splice(idx, 1);
 				this._kernelSourceActionProviders.set(viewType, providers);
 			}
+
+			eventEmitterDisposable?.dispose();
 		});
 	}
 

src/vs/workbench/contrib/notebook/common/notebookKernelService.ts

@@ -87,6 +87,7 @@ export interface INotebookSourceActionChangeEvent {
 
 export interface IKernelSourceActionProvider {
 	readonly viewType: string;
+	onDidChangeSourceActions?: Event<void>;
 	provideKernelSourceActions(): Promise<INotebookKernelSourceAction[]>;
 }
 

src/vscode-dts/vscode.proposed.notebookKernelSource.d.ts

@@ -22,6 +22,10 @@ declare module 'vscode' {
 	}
 
 	export interface NotebookKernelSourceActionProvider {
+		/**
+		 * An optional event to signal that the kernel source actions have changed.
+		 */
+		onDidChangeNotebookKernelSourceActions?: Event<void>;
 		/**
 		 * Provide kernel source actions
 		 */