reveal search result in center in markdown cell

src/vs/workbench/contrib/notebook/browser/view/notebookCellList.ts

@@ -106,7 +106,7 @@ export class NotebookCellList extends WorkbenchList<CellViewModel> {
 		}
 
 		if (revealType === CellRevealType.Range) {
-			element.revealRange(range);
+			element.revealRangeInCenter(range);
 		}
 	}
 
@@ -162,7 +162,7 @@ export class NotebookCellList extends WorkbenchList<CellViewModel> {
 			this.view.setScrollTop(lineOffsetInView - this.view.renderHeight / 2);
 
 			if (revealType === CellRevealType.Range) {
-				element.revealRange(range);
+				element.revealRangeInCenter(range);
 			}
 		};
 
@@ -194,7 +194,7 @@ export class NotebookCellList extends WorkbenchList<CellViewModel> {
 			this.view.setScrollTop(lineOffsetInView - this.view.renderHeight / 2);
 
 			if (revealType === CellRevealType.Range) {
-				element.revealRange(range);
+				element.revealRangeInCenter(range);
 			}
 		};
 
@@ -202,15 +202,17 @@ export class NotebookCellList extends WorkbenchList<CellViewModel> {
 		const wrapperBottom = scrollTop + this.view.renderHeight;
 		const elementTop = this.view.elementTop(index);
 		const viewItemOffset = elementTop;
+		const element = this.view.element(index);
 
 		if (viewItemOffset < scrollTop || viewItemOffset > wrapperBottom) {
 			this.view.setScrollTop(viewItemOffset - this.view.renderHeight / 2);
-			const element = this.view.element(index);
 			if (!element.editorAttached) {
 				getEditorAttachedPromise(element).then(() => reveal(index, range, revealType));
 			} else {
 				// should not happen
 			}
+		} else {
+			element.revealRangeInCenter(range);
 		}
 	}
 

src/vs/workbench/contrib/notebook/browser/view/renderers/markdownCell.ts

@@ -68,6 +68,7 @@ export class StatefullMarkdownCell extends Disposable {
 						}
 					}, {});
 
+					viewCell.attachTextEditor(this.editor);
 
 					const cts = new CancellationTokenSource();
 					this._register({ dispose() { cts.dispose(true); } });

src/vs/workbench/contrib/notebook/browser/viewModel/notebookCellViewModel.ts

@@ -291,8 +291,8 @@ export class CellViewModel extends Disposable {
 		this._onDidChangeEditorAttachState.fire(false);
 	}
 
-	revealRange(range: Range) {
-		this._textEditor?.revealRange(range);
+	revealRangeInCenter(range: Range) {
+		this._textEditor?.revealRangeInCenter(range);
 	}
 
 	getLineScrollTopOffset(line: number): number {

src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModel.ts

@@ -190,7 +190,9 @@ export class NotebookViewModel extends Disposable {
 
 			if (!mapping.has(ownerId)) {
 				const cell = this.viewCells.find(cell => cell.handle === ownerId);
-				mapping.set(ownerId, { cell: cell!, oldDecorations: [], newDecorations: [] });
+				if (cell) {
+					mapping.set(ownerId, { cell: cell, oldDecorations: [], newDecorations: [] });
+				}
 			}
 
 			const data = mapping.get(ownerId)!;
@@ -202,7 +204,10 @@ export class NotebookViewModel extends Disposable {
 
 			if (!mapping.has(ownerId)) {
 				const cell = this.viewCells.find(cell => cell.handle === ownerId);
-				mapping.set(ownerId, { cell: cell!, oldDecorations: [], newDecorations: [] });
+
+				if (cell) {
+					mapping.set(ownerId, { cell: cell, oldDecorations: [], newDecorations: [] });
+				}
 			}
 
 			const data = mapping.get(ownerId)!;