fix #117792.

microsoft · Feb 26, 2021 · e590188 · e590188
1 parent fc7005d
commit e590188
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/vs/workbench/contrib/notebook/browser/view/notebookCellList.ts b/src/vs/workbench/contrib/notebook/browser/view/notebookCellList.ts
@@ -595,11 +595,25 @@ export class NotebookCellList extends WorkbenchList<CellViewModel> implements ID
 	}
 
 	focusNext(n: number | undefined, loop: boolean | undefined, browserEvent?: UIEvent, filter?: (element: CellViewModel) => boolean): void {
-		this._focusNextPreviousDelegate.onFocusNext(() => super.focusNext(n, loop, browserEvent, filter));
+		this._focusNextPreviousDelegate.onFocusNext(() => {
+			super.focusNext(n, loop, browserEvent, filter);
+			const focus = this.getFocus();
+			if (focus.length) {
+				const focusedElementHandle = this.element(focus[0]).handle;
+				this._viewModel?.updateSelectionsFromView(focusedElementHandle, [focusedElementHandle]);
+			}
+		});
 	}
 
 	focusPrevious(n: number | undefined, loop: boolean | undefined, browserEvent?: UIEvent, filter?: (element: CellViewModel) => boolean): void {
-		this._focusNextPreviousDelegate.onFocusPrevious(() => super.focusPrevious(n, loop, browserEvent, filter));
+		this._focusNextPreviousDelegate.onFocusPrevious(() => {
+			super.focusPrevious(n, loop, browserEvent, filter);
+			const focus = this.getFocus();
+			if (focus.length) {
+				const focusedElementHandle = this.element(focus[0]).handle;
+				this._viewModel?.updateSelectionsFromView(focusedElementHandle, [focusedElementHandle]);
+			}
+		});
 	}
 
 	setFocus(indexes: number[], browserEvent?: UIEvent, ignoreTextModelUpdate?: boolean): void {