Functional tests for native interactive window

.vscode/launch.json

@@ -370,4 +370,4 @@
             }
         }
     ]
-}
+}

src/client/datascience/editor-integration/cellhashprovider.ts

@@ -23,7 +23,7 @@ import { IConfigurationService } from '../../common/types';
 import { getCellResource } from '../cellFactory';
 import { CellMatcher } from '../cellMatcher';
 import { Identifiers } from '../constants';
-import { getInteractiveCellMetadata } from '../interactive-window/nativeInteractiveWindow';
+import { getInteractiveCellMetadata } from '../interactive-window/interactiveWindow';
 import { ICellHash, ICellHashListener, ICellHashProvider, IFileHashes } from '../types';
 
 // eslint-disable-next-line @typescript-eslint/no-require-imports, @typescript-eslint/no-var-requires

src/client/datascience/editor-integration/codeLensFactory.ts

@@ -22,7 +22,7 @@ import { IConfigurationService, IDisposableRegistry, Resource } from '../../comm
 import * as localize from '../../common/utils/localize';
 import { generateCellRangesFromDocument } from '../cellFactory';
 import { CodeLensCommands, Commands } from '../constants';
-import { getInteractiveCellMetadata } from '../interactive-window/nativeInteractiveWindow';
+import { getInteractiveCellMetadata } from '../interactive-window/interactiveWindow';
 import { IKernelProvider } from '../jupyter/kernels/types';
 import { InteractiveWindowView } from '../notebook/constants';
 import { ICellHashProvider, ICellRange, ICodeLensFactory, IFileHashes } from '../types';
@@ -184,7 +184,7 @@ export class CodeLensFactory implements ICodeLensFactory {
             };
             this.notebookData.set(e.cell.notebook.uri.toString(), data);
         }
-        if (data) {
+        if (data && metadata.interactive !== undefined) {
             data.cellExecutionCounts.set(metadata.id, e.cell.executionSummary.executionOrder);
             data.documentExecutionCounts.set(
                 metadata.interactive.file.toLowerCase(),

src/client/datascience/editor-integration/hoverProvider.ts

@@ -15,7 +15,7 @@ import { sleep } from '../../common/utils/async';
 import { StopWatch } from '../../common/utils/stopWatch';
 import { sendTelemetryEvent } from '../../telemetry';
 import { Identifiers, Telemetry } from '../constants';
-import { getInteractiveCellMetadata } from '../interactive-window/nativeInteractiveWindow';
+import { getInteractiveCellMetadata } from '../interactive-window/interactiveWindow';
 import { IKernelProvider } from '../jupyter/kernels/types';
 import { InteractiveWindowView } from '../notebook/constants';
 import {

src/client/datascience/interactive-window/nativeInteractiveWindow.ts → src/client/datascience/interactive-window/interactiveWindow.ts

@@ -87,7 +87,7 @@ type InteractiveCellMetadata = {
 export function getInteractiveCellMetadata(cell: NotebookCell): InteractiveCellMetadata {
     return cell.metadata as InteractiveCellMetadata;
 }
-export class NativeInteractiveWindow implements IInteractiveWindowLoadable {
+export class InteractiveWindow implements IInteractiveWindowLoadable {
     public get onDidChangeViewState(): Event<void> {
         return this._onDidChangeViewState.event;
     }

src/client/datascience/interactive-window/nativeInteractiveWindowCommandListener.ts → src/client/datascience/interactive-window/interactiveWindowCommandListener.ts

@@ -15,6 +15,7 @@ import {
     TextEditor,
     Uri,
     ViewColumn,
+    window,
     workspace,
     WorkspaceEdit
 } from 'vscode';
@@ -134,14 +135,14 @@ export class NativeInteractiveWindowCommandListener implements IDataScienceComma
         this.disposableRegistry.push(
             commandManager.registerCommand(
                 Commands.ExpandAllCells,
-                (context?: { notebookEditor: { notebookUri: Uri } }) =>
+                async (context?: { notebookEditor: { notebookUri: Uri } }) =>
                     this.expandAllCells(context?.notebookEditor.notebookUri)
             )
         );
         this.disposableRegistry.push(
             commandManager.registerCommand(
                 Commands.CollapseAllCells,
-                (context?: { notebookEditor: { notebookUri: Uri } }) =>
+                async (context?: { notebookEditor: { notebookUri: Uri } }) =>
                     this.collapseAllCells(context?.notebookEditor.notebookUri)
             )
         );
@@ -372,21 +373,19 @@ export class NativeInteractiveWindowCommandListener implements IDataScienceComma
         return this.exportDialog.showDialog(ExportFormat.ipynb, file);
     }
 
-    private expandAllCells(uri?: Uri) {
-        const interactiveWindow = uri
-            ? this.interactiveWindowProvider.windows.find((window) => window.notebookUri?.toString() === uri.toString())
-            : this.interactiveWindowProvider.activeWindow;
+    private async expandAllCells(uri?: Uri) {
+        const interactiveWindow = this.getTargetInteractiveWindow(uri);
+        traceInfo(`Expanding all cells in interactive window with uri ${interactiveWindow?.notebookUri}`);
         if (interactiveWindow) {
-            interactiveWindow.expandAllCells();
+            await interactiveWindow.expandAllCells();
         }
     }
 
-    private collapseAllCells(uri?: Uri) {
-        const interactiveWindow = uri
-            ? this.interactiveWindowProvider.windows.find((window) => window.notebookUri?.toString() === uri.toString())
-            : this.interactiveWindowProvider.activeWindow;
+    private async collapseAllCells(uri?: Uri) {
+        const interactiveWindow = this.getTargetInteractiveWindow(uri);
+        traceInfo(`Collapsing all cells in interactive window with uri ${interactiveWindow?.notebookUri}`);
         if (interactiveWindow) {
-            interactiveWindow.collapseAllCells();
+            await interactiveWindow.collapseAllCells();
         }
     }
 
@@ -398,18 +397,14 @@ export class NativeInteractiveWindowCommandListener implements IDataScienceComma
     }
 
     private exportAs(uri?: Uri) {
-        const interactiveWindow = uri
-            ? this.interactiveWindowProvider.windows.find((window) => window.notebookUri?.toString() === uri.toString())
-            : this.interactiveWindowProvider.activeWindow;
+        const interactiveWindow = this.getTargetInteractiveWindow(uri);
         if (interactiveWindow) {
             interactiveWindow.exportAs();
         }
     }
 
     private export(uri?: Uri) {
-        const interactiveWindow = uri
-            ? this.interactiveWindowProvider.windows.find((window) => window.notebookUri?.toString() === uri.toString())
-            : this.interactiveWindowProvider.activeWindow;
+        const interactiveWindow = this.getTargetInteractiveWindow(uri);
         if (interactiveWindow) {
             interactiveWindow.export();
         }
@@ -488,9 +483,7 @@ export class NativeInteractiveWindowCommandListener implements IDataScienceComma
     }
 
     private async clearAllCellsInInteractiveWindow(context?: { notebookEditor: { notebookUri: Uri } }): Promise<void> {
-        // Use the context if invoked from interactive/toolbar
-        // Then fallback to the active interactive window
-        const uri = context?.notebookEditor.notebookUri ?? this.interactiveWindowProvider.activeWindow?.notebookUri;
+        const uri = this.getTargetInteractiveWindow(context?.notebookEditor.notebookUri)?.notebookUri;
         if (!uri) {
             return;
         }
@@ -551,4 +544,22 @@ export class NativeInteractiveWindowCommandListener implements IDataScienceComma
             await this.clipboard.writeText(source);
         }
     }
+
+    private getTargetInteractiveWindow(notebookUri: Uri | undefined) {
+        let targetInteractiveWindow;
+        if (notebookUri !== undefined) {
+            targetInteractiveWindow = this.interactiveWindowProvider.windows.find(
+                (w) => w.notebookUri?.toString() === notebookUri.toString()
+            );
+        } else if (notebookUri === undefined && this.interactiveWindowProvider.activeWindow !== undefined) {
+            targetInteractiveWindow = this.interactiveWindowProvider.activeWindow;
+        } else if (
+            notebookUri === undefined &&
+            this.interactiveWindowProvider.activeWindow === undefined &&
+            window.activeTextEditor?.document.uri !== undefined
+        ) {
+            targetInteractiveWindow = this.interactiveWindowProvider.get(window.activeTextEditor.document.uri);
+        }
+        return targetInteractiveWindow;
+    }
 }

src/client/datascience/interactive-window/nativeInteractiveWindowProvider.ts → src/client/datascience/interactive-window/interactiveWindowProvider.ts

@@ -31,13 +31,13 @@ import { IExportDialog } from '../export/types';
 import { IKernelProvider } from '../jupyter/kernels/types';
 import { INotebookControllerManager } from '../notebook/types';
 import { IInteractiveWindow, IInteractiveWindowProvider, IJupyterDebugger, INotebookExporter } from '../types';
-import { NativeInteractiveWindow } from './nativeInteractiveWindow';
+import { InteractiveWindow } from './interactiveWindow';
 
 // Export for testing
 export const AskedForPerFileSettingKey = 'ds_asked_per_file_interactive';
 
 @injectable()
-export class NativeInteractiveWindowProvider implements IInteractiveWindowProvider, IAsyncDisposable {
+export class InteractiveWindowProvider implements IInteractiveWindowProvider, IAsyncDisposable {
     public get onDidChangeActiveInteractiveWindow(): Event<IInteractiveWindow | undefined> {
         return this._onDidChangeActiveInteractiveWindow.event;
     }
@@ -57,7 +57,7 @@ export class NativeInteractiveWindowProvider implements IInteractiveWindowProvid
     private readonly _onDidChangeActiveInteractiveWindow = new EventEmitter<IInteractiveWindow | undefined>();
     private readonly _onDidCreateInteractiveWindow = new EventEmitter<IInteractiveWindow>();
     private lastActiveInteractiveWindow: IInteractiveWindow | undefined;
-    private _windows: NativeInteractiveWindow[] = [];
+    private _windows: InteractiveWindow[] = [];
 
     constructor(
         @inject(IServiceContainer) private serviceContainer: IServiceContainer,
@@ -114,7 +114,7 @@ export class NativeInteractiveWindowProvider implements IInteractiveWindowProvid
     private create(resource: Resource, mode: InteractiveWindowMode) {
         // Set it as soon as we create it. The .ctor for the interactive window
         // may cause a subclass to talk to the IInteractiveWindowProvider to get the active interactive window.
-        const result = new NativeInteractiveWindow(
+        const result = new InteractiveWindow(
             this.serviceContainer.get<IApplicationShell>(IApplicationShell),
             this.serviceContainer.get<IDocumentManager>(IDocumentManager),
             this.serviceContainer.get<IFileSystem>(IFileSystem),

src/client/datascience/serviceRegistry.ts

@@ -156,8 +156,8 @@ import { IApplicationEnvironment } from '../common/application/types';
 import { NotebookIPyWidgetCoordinator } from './ipywidgets/notebookIPyWidgetCoordinator';
 import { ExtensionRecommendationService } from './extensionRecommendation';
 import { PythonVariablesRequester } from './jupyter/pythonVariableRequester';
-import { NativeInteractiveWindowCommandListener } from './interactive-window/nativeInteractiveWindowCommandListener';
-import { NativeInteractiveWindowProvider } from './interactive-window/nativeInteractiveWindowProvider';
+import { NativeInteractiveWindowCommandListener } from './interactive-window/interactiveWindowCommandListener';
+import { InteractiveWindowProvider } from './interactive-window/interactiveWindowProvider';
 import { JupyterPaths } from './kernel-launcher/jupyterPaths';
 import { LocalKnownPathKernelSpecFinder } from './kernel-launcher/localKnownPathKernelSpecFinder';
 import { LocalPythonAndRelatedNonPythonKernelSpecFinder } from './kernel-launcher/localPythonAndRelatedNonPythonKernelSpecFinder';
@@ -238,7 +238,7 @@ export function registerTypes(serviceManager: IServiceManager, inNotebookApiExpe
     serviceManager.addSingleton<IExtensionSingleActivationService>(IExtensionSingleActivationService, NotebookUsageTracker);
     serviceManager.addSingleton<IExtensionSingleActivationService>(IExtensionSingleActivationService, MigrateJupyterInterpreterStateService);
     serviceManager.addSingleton<IExtensionSingleActivationService>(IExtensionSingleActivationService, VariableViewActivationService);
-    serviceManager.addSingleton<IInteractiveWindowProvider>(IInteractiveWindowProvider, NativeInteractiveWindowProvider);
+    serviceManager.addSingleton<IInteractiveWindowProvider>(IInteractiveWindowProvider, InteractiveWindowProvider);
     serviceManager.addSingleton<IDataScienceCommandListener>(IDataScienceCommandListener, NativeInteractiveWindowCommandListener);
     serviceManager.addSingleton<IDataScienceCommandListener>(IDataScienceCommandListener, KernelCommandListener);
     serviceManager.addSingleton<IJupyterDebugger>(IJupyterDebugger, JupyterDebugger, undefined, [ICellHashListener]);

src/client/datascience/types.ts

@@ -519,8 +519,8 @@ export interface IInteractiveWindow extends IInteractiveBase {
         editor?: TextEditor,
         runningStopWatch?: StopWatch
     ): Promise<boolean>;
-    expandAllCells(): void;
-    collapseAllCells(): void;
+    expandAllCells(): Promise<void>;
+    collapseAllCells(): Promise<void>;
     exportCells(): void;
     scrollToCell(id: string): void;
     exportAs(cells?: ICell[]): void;