Fix support for IPyWidgets in Interactive Window

news/2 Fixes/4203.md

@@ -0,0 +1 @@
+Fix support for IPyWidgets in Interactive Window.

src/client/datascience/ipywidgets/commonMessageCoordinator.ts

@@ -62,10 +62,8 @@ export class CommonMessageCoordinator {
         this.jupyterOutput = this.serviceContainer.get<IOutputChannel>(IOutputChannel, JUPYTER_OUTPUT_CHANNEL);
     }
 
-    public static async create(identity: Uri, serviceContainer: IServiceContainer): Promise<CommonMessageCoordinator> {
-        const result = new CommonMessageCoordinator(identity, serviceContainer);
-        await result.initialize();
-        return result;
+    public static create(identity: Uri, serviceContainer: IServiceContainer): CommonMessageCoordinator {
+        return new CommonMessageCoordinator(identity, serviceContainer);
     }
 
     public dispose() {
@@ -94,6 +92,19 @@ export class CommonMessageCoordinator {
         this.getIPyWidgetScriptSource()?.onMessage(message, payload);
     }
 
+    public async initialize() {
+        const dispatcher = this.getIPyWidgetMessageDispatcher();
+        const promises = [];
+        if (dispatcher) {
+            promises.push(dispatcher.initialize());
+        }
+        const scriptSource = this.getIPyWidgetScriptSource();
+        if (scriptSource) {
+            promises.push(scriptSource.initialize());
+        }
+        return Promise.all(promises);
+    }
+
     private hash(s: string): string {
         return this.hashFn().update(s).digest('hex');
     }
@@ -164,6 +175,9 @@ export class CommonMessageCoordinator {
             this.ipyWidgetMessageDispatcher = this.serviceContainer
                 .get<IPyWidgetMessageDispatcherFactory>(IPyWidgetMessageDispatcherFactory)
                 .create(this.identity);
+            this.disposables.push(
+                this.ipyWidgetMessageDispatcher.postMessage(this.postEmitter.fire.bind(this.postEmitter))
+            );
         }
         return this.ipyWidgetMessageDispatcher;
     }
@@ -183,23 +197,11 @@ export class CommonMessageCoordinator {
                 this.serviceContainer.get<IPersistentStateFactory>(IPersistentStateFactory),
                 this.serviceContainer.get<IExtensionContext>(IExtensionContext)
             );
+            this.disposables.push(this.ipyWidgetScriptSource.postMessage(this.postEmitter.fire.bind(this.postEmitter)));
+            this.disposables.push(
+                this.ipyWidgetScriptSource.postInternalMessage(this.postEmitter.fire.bind(this.postEmitter))
+            );
         }
         return this.ipyWidgetScriptSource;
     }
-
-    private async initialize() {
-        const dispatcher = this.getIPyWidgetMessageDispatcher();
-        const promises = [];
-        if (dispatcher) {
-            this.disposables.push(dispatcher.postMessage(this.postEmitter.fire.bind(this.postEmitter)));
-            promises.push(dispatcher.initialize());
-        }
-        const scriptSource = this.getIPyWidgetScriptSource();
-        if (scriptSource) {
-            this.disposables.push(scriptSource.postMessage(this.postEmitter.fire.bind(this.postEmitter)));
-            this.disposables.push(scriptSource.postInternalMessage(this.postEmitter.fire.bind(this.postEmitter)));
-            promises.push(scriptSource.initialize());
-        }
-        return Promise.all(promises);
-    }
 }

src/client/datascience/ipywidgets/notebookIPyWidgetCoordinator.ts

@@ -38,7 +38,8 @@ export class NotebookIPyWidgetCoordinator implements INotebookKernelResolver {
         // entire VS code session, we have a map of notebook document to message coordinator
         let promise = this.messageCoordinators.get(document.uri.toString());
         if (!promise) {
-            promise = CommonMessageCoordinator.create(document.uri, this.serviceContainer);
+            const coordinator = CommonMessageCoordinator.create(document.uri, this.serviceContainer);
+            const promise = coordinator.initialize().then(() => coordinator);
             this.messageCoordinators.set(document.uri.toString(), promise);
         }
         return Cancellation.race(() => promise!.then(this.attachCoordinator.bind(this, document, webview)), token);

src/client/datascience/ipywidgets/webviewIPyWidgetCoordinator.ts

@@ -62,7 +62,7 @@ export class WebviewIPyWidgetCoordinator implements IInteractiveWindowListener {
         // There should be an instance of the WebviewMessageCoordinator per notebook webview or interactive window. Create
         // the message coordinator as soon as we're sure what notebook we're in.
         this.notebookIdentity = args.resource;
-        this.messageCoordinator = await CommonMessageCoordinator.create(this.notebookIdentity, this.serviceContainer);
+        this.messageCoordinator = CommonMessageCoordinator.create(this.notebookIdentity, this.serviceContainer);
         this.messageCoordinatorEvent = this.messageCoordinator.postMessage((e) => {
             // Special case a specific message. It must be posted to the internal class, not the webview
             if (e.message === InteractiveWindowMessages.ConvertUriForUseInWebViewRequest) {
@@ -71,5 +71,6 @@ export class WebviewIPyWidgetCoordinator implements IInteractiveWindowListener {
                 this.postEmitter.fire(e);
             }
         });
+        this.messageCoordinator.initialize();
     }
 }