fix: Keep the data frame in view when you reduce its size. (#251)

Author: Artmann

src/test/datascience/mockQuickPick.ts

@@ -6,6 +6,7 @@ import { Event, EventEmitter, QuickInputButton, QuickPick, QuickPickItem, QuickP
 export class MockQuickPick implements QuickPick<QuickPickItem> {
     public value: string = '';
     public placeholder: string | undefined;
+    public prompt: string | undefined;
     public title: string | undefined = 'foo';
     public step: number | undefined;
     public totalSteps: number | undefined;

src/test/datascience/notebook/helper.ts

@@ -1421,6 +1421,7 @@ export async function hijackSavePrompt(
 export class MockQuickPick implements QuickPick<QuickPickItem> {
     value: string;
     placeholder: string | undefined;
+    prompt: string | undefined;
     get onDidChangeValue(): Event<string> {
         return this._onDidChangeValueEmitter.event;
     }

src/webviews/extension-side/dataframe/dataframeController.ts

@@ -4,6 +4,8 @@ import {
     env,
     l10n,
     NotebookEdit,
+    NotebookEditorRevealType,
+    NotebookRange,
     type NotebookCellOutput,
     type NotebookEditor,
     type NotebookRendererMessaging,
@@ -293,6 +295,11 @@ export class DataframeController implements IExtensionSyncActivationService {
             ranges: [{ start: cellIndex, end: cellIndex + 1 }],
             document: editor.notebook.uri
         });
+
+        // Reveal the cell to keep it visible after output size changes
+        const notebookRange = new NotebookRange(cellIndex, cellIndex + 1);
+
+        editor.revealRange(notebookRange, NotebookEditorRevealType.InCenterIfOutsideViewport);
     }
 
     private async handleSelectPageSize(editor: NotebookEditor, message: SelectPageSizeCommand) {
@@ -342,6 +349,11 @@ export class DataframeController implements IExtensionSyncActivationService {
             ranges: [{ start: cellIndex, end: cellIndex + 1 }],
             document: editor.notebook.uri
         });
+
+        // Reveal the cell to keep it visible after output size changes
+        const notebookRange = new NotebookRange(cellIndex, cellIndex + 1);
+
+        editor.revealRange(notebookRange, NotebookEditorRevealType.InCenterIfOutsideViewport);
     }
 
     private async onDidReceiveMessage(