notebookEditor.ts 28.5 KB
Newer Older
P
Peng Lyu 已提交
1 2 3 4 5
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

6
import * as nls from 'vs/nls';
R
rebornix 已提交
7
import { getZoomLevel } from 'vs/base/browser/browser';
P
Peng Lyu 已提交
8
import * as DOM from 'vs/base/browser/dom';
P
Peng Lyu 已提交
9
import { IMouseWheelEvent } from 'vs/base/browser/mouseEvent';
R
rebornix 已提交
10
import { CancellationToken } from 'vs/base/common/cancellation';
R
rebornix 已提交
11
import { DisposableStore, MutableDisposable } from 'vs/base/common/lifecycle';
R
rebornix 已提交
12
import 'vs/css!./notebook';
13
import { IEditorOptions } from 'vs/editor/common/config/editorOptions';
R
rebornix 已提交
14
import { BareFontInfo } from 'vs/editor/common/config/fontInfo';
15
import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
R
Rob Lourens 已提交
16
import { IContextKey, IContextKeyService } from 'vs/platform/contextkey/common/contextkey';
R
rebornix 已提交
17 18 19
import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
import { IStorageService } from 'vs/platform/storage/common/storage';
import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
20
import { contrastBorder, editorBackground, focusBorder, foreground, textBlockQuoteBackground, textBlockQuoteBorder, textLinkActiveForeground, textLinkForeground, textPreformatForeground, registerColor } from 'vs/platform/theme/common/colorRegistry';
R
rebornix 已提交
21 22
import { IThemeService, registerThemingParticipant } from 'vs/platform/theme/common/themeService';
import { BaseEditor } from 'vs/workbench/browser/parts/editor/baseEditor';
23
import { EditorOptions, IEditorMemento, IEditorCloseEvent } from 'vs/workbench/common/editor';
R
rebornix 已提交
24
import { INotebookEditor, NotebookLayoutInfo, CellEditState, NOTEBOOK_EDITOR_FOCUSED, CellFocusMode, ICellViewModel, CellRunState, NOTEBOOK_EDITOR_EDITABLE } from 'vs/workbench/contrib/notebook/browser/notebookBrowser';
R
rebornix 已提交
25
import { NotebookEditorInput, NotebookEditorModel } from 'vs/workbench/contrib/notebook/browser/notebookEditorInput';
R
rebornix 已提交
26
import { INotebookService } from 'vs/workbench/contrib/notebook/browser/notebookService';
R
rebornix 已提交
27 28 29
import { OutputRenderer } from 'vs/workbench/contrib/notebook/browser/view/output/outputRenderer';
import { BackLayerWebView } from 'vs/workbench/contrib/notebook/browser/view/renderers/backLayerWebView';
import { CodeCellRenderer, MarkdownCellRenderer, NotebookCellListDelegate } from 'vs/workbench/contrib/notebook/browser/view/renderers/cellRenderer';
30
import { IOutput, CellKind, CellUri } from 'vs/workbench/contrib/notebook/common/notebookCommon';
R
rebornix 已提交
31 32
import { getExtraColor } from 'vs/workbench/contrib/welcome/walkThrough/common/walkThroughUtils';
import { IEditorGroup, IEditorGroupsService } from 'vs/workbench/services/editor/common/editorGroupsService';
J
Johannes Rieken 已提交
33
import { ICodeEditor } from 'vs/editor/browser/editorBrowser';
34
import { IEditor, ICompositeCodeEditor } from 'vs/editor/common/editorCommon';
35
import { IResourceEditorInput } from 'vs/platform/editor/common/editor';
36
import { Emitter, Event } from 'vs/base/common/event';
R
rebornix 已提交
37 38
import { NotebookCellList } from 'vs/workbench/contrib/notebook/browser/view/notebookCellList';
import { NotebookFindWidget } from 'vs/workbench/contrib/notebook/browser/contrib/notebookFindWidget';
R
rebornix 已提交
39
import { NotebookViewModel, INotebookEditorViewState, IModelDecorationsChangeAccessor, CellViewModel } from 'vs/workbench/contrib/notebook/browser/viewModel/notebookViewModel';
R
rebornix 已提交
40
import { IEditorGroupView } from 'vs/workbench/browser/parts/editor/editor';
R
rebornix 已提交
41
import { CodeCellViewModel } from 'vs/workbench/contrib/notebook/browser/viewModel/codeCellViewModel';
R
rebornix 已提交
42
import { Range } from 'vs/editor/common/core/range';
R
Rob Lourens 已提交
43
import { CELL_MARGIN, CELL_RUN_GUTTER } from 'vs/workbench/contrib/notebook/browser/constants';
44
import { Color, RGBA } from 'vs/base/common/color';
R
rebornix 已提交
45
import { NotebookEventDispatcher, NotebookLayoutChangedEvent } from 'vs/workbench/contrib/notebook/browser/viewModel/eventDispatcher';
P
Peng Lyu 已提交
46 47

const $ = DOM.$;
R
rebornix 已提交
48 49
const NOTEBOOK_EDITOR_VIEW_STATE_PREFERENCE_KEY = 'NotebookEditorViewState';

50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
export class NotebookEditorOptions extends EditorOptions {

	readonly cellOptions?: IResourceEditorInput;

	constructor(options: Partial<NotebookEditorOptions>) {
		super();
		this.overwrite(options);
		this.cellOptions = options.cellOptions;
	}

	with(options: Partial<NotebookEditorOptions>): NotebookEditorOptions {
		return new NotebookEditorOptions({ ...this, ...options });
	}
}

R
rebornix 已提交
65
export class NotebookCodeEditors implements ICompositeCodeEditor {
66

67
	private readonly _disposables = new DisposableStore();
68 69 70 71
	private readonly _onDidChangeActiveEditor = new Emitter<this>();
	readonly onDidChangeActiveEditor: Event<this> = this._onDidChangeActiveEditor.event;

	constructor(
R
rebornix 已提交
72
		private _list: NotebookCellList,
R
rebornix 已提交
73
		private _renderedEditors: Map<ICellViewModel, ICodeEditor | undefined>
74
	) {
75
		_list.onDidChangeFocus(_e => this._onDidChangeActiveEditor.fire(this), undefined, this._disposables);
76 77 78 79 80 81
	}

	dispose(): void {
		this._onDidChangeActiveEditor.dispose();
		this._disposables.dispose();
	}
82 83 84

	get activeCodeEditor(): IEditor | undefined {
		const [focused] = this._list.getFocusedElements();
R
rebornix 已提交
85
		return this._renderedEditors.get(focused);
86 87 88
	}
}

R
rebornix 已提交
89
export class NotebookEditor extends BaseEditor implements INotebookEditor {
P
Peng Lyu 已提交
90 91 92
	static readonly ID: string = 'workbench.editor.notebook';
	private rootElement!: HTMLElement;
	private body!: HTMLElement;
P
Peng Lyu 已提交
93
	private webview: BackLayerWebView | null = null;
R
rebornix 已提交
94
	private list: NotebookCellList | undefined;
95
	private control: ICompositeCodeEditor | undefined;
R
rebornix 已提交
96
	private renderedEditors: Map<ICellViewModel, ICodeEditor | undefined> = new Map();
R
rebornix 已提交
97
	private eventDispatcher: NotebookEventDispatcher | undefined;
R
rebornix 已提交
98
	private notebookViewModel: NotebookViewModel | undefined;
99
	private localStore: DisposableStore = this._register(new DisposableStore());
R
rebornix 已提交
100
	private editorMemento: IEditorMemento<INotebookEditorViewState>;
R
rebornix 已提交
101
	private readonly groupListener = this._register(new MutableDisposable());
102
	private fontInfo: BareFontInfo | undefined;
103
	private dimension: DOM.Dimension | null = null;
R
rebornix 已提交
104
	private editorFocus: IContextKey<boolean> | null = null;
R
rebornix 已提交
105
	private editorEditable: IContextKey<boolean> | null = null;
R
rebornix 已提交
106
	private outputRenderer: OutputRenderer;
R
rebornix 已提交
107
	private findWidget: NotebookFindWidget;
P
Peng Lyu 已提交
108 109 110 111 112

	constructor(
		@ITelemetryService telemetryService: ITelemetryService,
		@IThemeService themeService: IThemeService,
		@IInstantiationService private readonly instantiationService: IInstantiationService,
P
Peng Lyu 已提交
113
		@IStorageService storageService: IStorageService,
R
rebornix 已提交
114
		@INotebookService private notebookService: INotebookService,
115
		@IEditorGroupsService editorGroupService: IEditorGroupsService,
R
rebornix 已提交
116
		@IConfigurationService private readonly configurationService: IConfigurationService,
R
rebornix 已提交
117
		@IContextKeyService private readonly contextKeyService: IContextKeyService,
P
Peng Lyu 已提交
118 119
	) {
		super(NotebookEditor.ID, telemetryService, themeService, storageService);
R
rebornix 已提交
120 121

		this.editorMemento = this.getEditorMemento<INotebookEditorViewState>(editorGroupService, NOTEBOOK_EDITOR_VIEW_STATE_PREFERENCE_KEY);
R
rebornix 已提交
122
		this.outputRenderer = new OutputRenderer(this, this.instantiationService);
R
rebornix 已提交
123
		this.findWidget = this.instantiationService.createInstance(NotebookFindWidget, this);
124
		this.findWidget.updateTheme(this.themeService.getColorTheme());
R
rebornix 已提交
125 126
	}

R
rebornix 已提交
127 128 129 130
	get viewModel() {
		return this.notebookViewModel;
	}

P
Peng Lyu 已提交
131 132 133 134 135 136 137 138
	get minimumWidth(): number { return 375; }
	get maximumWidth(): number { return Number.POSITIVE_INFINITY; }

	// these setters need to exist because this extends from BaseEditor
	set minimumWidth(value: number) { /*noop*/ }
	set maximumWidth(value: number) { /*noop*/ }


R
rebornix 已提交
139
	//#region Editor Core
R
rebornix 已提交
140

R
rebornix 已提交
141 142 143 144 145

	public get isNotebookEditor() {
		return true;
	}

P
Peng Lyu 已提交
146 147 148
	protected createEditor(parent: HTMLElement): void {
		this.rootElement = DOM.append(parent, $('.notebook-editor'));
		this.createBody(this.rootElement);
149
		this.generateFontInfo();
R
rebornix 已提交
150 151 152 153 154 155 156 157
		this.editorFocus = NOTEBOOK_EDITOR_FOCUSED.bindTo(this.contextKeyService);
		this._register(this.onDidFocus(() => {
			this.editorFocus?.set(true);
		}));

		this._register(this.onDidBlur(() => {
			this.editorFocus?.set(false);
		}));
R
rebornix 已提交
158 159 160

		this.editorEditable = NOTEBOOK_EDITOR_EDITABLE.bindTo(this.contextKeyService);
		this.editorEditable.set(true);
161 162 163 164 165
	}

	private generateFontInfo(): void {
		const editorOptions = this.configurationService.getValue<IEditorOptions>('editor');
		this.fontInfo = BareFontInfo.createFromRawSettings(editorOptions, getZoomLevel());
P
Peng Lyu 已提交
166 167 168
	}

	private createBody(parent: HTMLElement): void {
P
Peng Lyu 已提交
169
		this.body = document.createElement('div');
P
Peng Lyu 已提交
170 171 172
		DOM.addClass(this.body, 'cell-list-container');
		this.createCellList();
		DOM.append(parent, this.body);
R
rebornix 已提交
173
		DOM.append(parent, this.findWidget.getDomNode());
P
Peng Lyu 已提交
174 175
	}

P
Peng Lyu 已提交
176 177
	private createCellList(): void {
		DOM.addClass(this.body, 'cell-list-container');
P
Peng Lyu 已提交
178

P
Peng Lyu 已提交
179
		const renders = [
R
rebornix 已提交
180 181
			this.instantiationService.createInstance(CodeCellRenderer, this, this.contextKeyService, this.renderedEditors),
			this.instantiationService.createInstance(MarkdownCellRenderer, this.contextKeyService, this),
P
Peng Lyu 已提交
182 183
		];

R
rebornix 已提交
184
		this.list = <NotebookCellList>this.instantiationService.createInstance(
R
rebornix 已提交
185
			NotebookCellList,
P
Peng Lyu 已提交
186 187 188 189
			'NotebookCellList',
			this.body,
			this.instantiationService.createInstance(NotebookCellListDelegate),
			renders,
R
rebornix 已提交
190
			this.contextKeyService,
P
Peng Lyu 已提交
191 192
			{
				setRowLineHeight: false,
R
rebornix 已提交
193
				setRowHeight: false,
P
Peng Lyu 已提交
194 195 196
				supportDynamicHeights: true,
				horizontalScrolling: false,
				keyboardSupport: false,
R
rebornix 已提交
197
				mouseSupport: true,
P
Peng Lyu 已提交
198
				multipleSelectionSupport: false,
R
rebornix 已提交
199
				enableKeyboardNavigation: true,
200
				additionalScrollHeight: 0,
201
				styleController: (_suffix: string) => { return this.list!; },
P
Peng Lyu 已提交
202 203 204 205 206 207 208 209 210 211
				overrideStyles: {
					listBackground: editorBackground,
					listActiveSelectionBackground: editorBackground,
					listActiveSelectionForeground: foreground,
					listFocusAndSelectionBackground: editorBackground,
					listFocusAndSelectionForeground: foreground,
					listFocusBackground: editorBackground,
					listFocusForeground: foreground,
					listHoverForeground: foreground,
					listHoverBackground: editorBackground,
212 213
					listHoverOutline: focusBorder,
					listFocusOutline: focusBorder,
214 215 216 217
					listInactiveSelectionBackground: editorBackground,
					listInactiveSelectionForeground: foreground,
					listInactiveFocusBackground: editorBackground,
					listInactiveFocusOutline: editorBackground,
P
Peng Lyu 已提交
218
				}
R
rebornix 已提交
219
			},
P
Peng Lyu 已提交
220
		);
P
Peng Lyu 已提交
221

222
		this.control = new NotebookCodeEditors(this.list, this.renderedEditors);
223
		this.webview = this.instantiationService.createInstance(BackLayerWebView, this);
224 225 226 227 228
		this._register(this.webview.onMessage(message => {
			if (this.viewModel) {
				this.notebookService.onDidReceiveMessage(this.viewModel.viewType, this.viewModel.uri, message);
			}
		}));
R
rebornix 已提交
229
		this.list.rowsContainer.appendChild(this.webview.element);
P
Peng Lyu 已提交
230 231 232
		this._register(this.list);
	}

233 234 235 236
	getControl() {
		return this.control;
	}

P
Peng Lyu 已提交
237
	onHide() {
R
rebornix 已提交
238
		this.editorFocus?.set(false);
239 240
		if (this.webview) {
			this.localStore.clear();
R
rebornix 已提交
241
			this.list?.rowsContainer.removeChild(this.webview?.element);
242 243 244 245 246 247
			this.webview?.dispose();
			this.webview = null;
		}

		this.list?.splice(0, this.list?.length);
		super.onHide();
P
Peng Lyu 已提交
248 249
	}

R
rebornix 已提交
250 251 252 253 254 255 256 257 258 259 260 261 262
	setEditorVisible(visible: boolean, group: IEditorGroup | undefined): void {
		super.setEditorVisible(visible, group);
		this.groupListener.value = ((group as IEditorGroupView).onWillCloseEditor(e => this.onWillCloseEditorInGroup(e)));
	}

	private onWillCloseEditorInGroup(e: IEditorCloseEvent): void {
		const editor = e.editor;
		if (!(editor instanceof NotebookEditorInput)) {
			return; // only handle files
		}

		if (editor === this.input) {
			this.saveTextEditorViewState(editor);
263 264 265
		}
	}

R
rebornix 已提交
266 267 268 269 270
	focus() {
		super.focus();
		this.editorFocus?.set(true);
	}

R
rebornix 已提交
271
	async setInput(input: NotebookEditorInput, options: EditorOptions | undefined, token: CancellationToken): Promise<void> {
R
rebornix 已提交
272 273 274 275
		if (this.input instanceof NotebookEditorInput) {
			this.saveTextEditorViewState(this.input);
		}

R
rebornix 已提交
276 277
		await super.setInput(input, options, token);
		const model = await input.resolve();
P
Peng Lyu 已提交
278

279 280 281
		if (this.notebookViewModel === undefined || !this.notebookViewModel.equal(model) || this.webview === null) {
			this.detachModel();
			await this.attachModel(input, model);
R
rebornix 已提交
282
		}
P
Peng Lyu 已提交
283

284 285 286
		// reveal cell if editor options tell to do so
		if (options instanceof NotebookEditorOptions && options.cellOptions) {
			const cellOptions = options.cellOptions;
R
rebornix 已提交
287
			const cell = this.notebookViewModel!.viewCells.find(cell => cell.uri.toString() === cellOptions.resource.toString());
288
			if (cell) {
289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
				this.revealInCenterIfOutsideViewport(cell);
				const editor = this.renderedEditors.get(cell)!;
				if (editor) {
					if (cellOptions.options?.selection) {
						const { selection } = cellOptions.options;
						editor.setSelection({
							...selection,
							endLineNumber: selection.endLineNumber || selection.startLineNumber,
							endColumn: selection.endColumn || selection.startColumn
						});
					}
					if (!cellOptions.options?.preserveFocus) {
						editor.focus();
					}
				}
304 305
			}
		}
R
rebornix 已提交
306
	}
307

R
rebornix 已提交
308 309 310 311 312 313 314 315
	clearInput(): void {
		if (this.input && this.input instanceof NotebookEditorInput && !this.input.isDisposed()) {
			this.saveTextEditorViewState(this.input);
		}

		super.clearInput();
	}

R
rebornix 已提交
316 317 318 319 320 321
	private detachModel() {
		this.localStore.clear();
		this.notebookViewModel?.dispose();
		this.notebookViewModel = undefined;
		this.webview?.clearInsets();
		this.webview?.clearPreloadsCache();
322
		this.findWidget.clear();
R
rebornix 已提交
323
	}
R
rebornix 已提交
324

R
rebornix 已提交
325 326
	private async attachModel(input: NotebookEditorInput, model: NotebookEditorModel) {
		if (!this.webview) {
327
			this.webview = this.instantiationService.createInstance(BackLayerWebView, this);
R
rebornix 已提交
328 329
			this.list?.rowsContainer.insertAdjacentElement('afterbegin', this.webview!.element);
		}
330

R
rebornix 已提交
331
		this.eventDispatcher = new NotebookEventDispatcher();
332
		this.notebookViewModel = this.instantiationService.createInstance(NotebookViewModel, input.viewType!, model, this.eventDispatcher, this.getLayoutInfo());
333
		this.editorEditable?.set(!!this.notebookViewModel.metadata?.editable);
R
rebornix 已提交
334
		this.eventDispatcher.emit([new NotebookLayoutChangedEvent({ width: true, fontInfo: true }, this.getLayoutInfo())]);
R
rebornix 已提交
335
		const viewState = this.loadTextEditorViewState(input);
R
rebornix 已提交
336
		this.notebookViewModel.restoreEditorViewState(viewState);
337

R
rebornix 已提交
338 339 340 341
		this.localStore.add(this.eventDispatcher.onDidChangeMetadata((e) => {
			this.editorEditable?.set(e.source.editable);
		}));

R
rebornix 已提交
342 343 344 345 346 347 348 349 350 351 352 353
		this.localStore.add(this.notebookViewModel.onDidChangeViewCells((e) => {
			if (e.synchronous) {
				e.splices.reverse().forEach((diff) => {
					this.list?.splice(diff[0], diff[1], diff[2]);
				});
			} else {
				DOM.scheduleAtNextAnimationFrame(() => {
					e.splices.reverse().forEach((diff) => {
						this.list?.splice(diff[0], diff[1], diff[2]);
					});
				});
			}
R
rebornix 已提交
354 355 356 357 358 359 360 361 362 363 364 365
		}));

		this.webview?.updateRendererPreloads(this.notebookViewModel.renderers);

		this.localStore.add(this.list!.onWillScroll(e => {
			this.webview!.updateViewScrollTop(-e.scrollTop, []);
		}));

		this.localStore.add(this.list!.onDidChangeContentHeight(() => {
			const scrollTop = this.list?.scrollTop || 0;
			const scrollHeight = this.list?.scrollHeight || 0;
			this.webview!.element.style.height = `${scrollHeight}px`;
R
rebornix 已提交
366
			let updateItems: { cell: CodeCellViewModel, output: IOutput, cellTop: number }[] = [];
R
rebornix 已提交
367 368 369 370 371 372 373 374 375 376 377 378

			if (this.webview?.insetMapping) {
				this.webview?.insetMapping.forEach((value, key) => {
					let cell = value.cell;
					let index = this.notebookViewModel!.getViewCellIndex(cell);
					let cellTop = this.list?.getAbsoluteTop(index) || 0;
					if (this.webview!.shouldUpdateInset(cell, key, cellTop)) {
						updateItems.push({
							cell: cell,
							output: key,
							cellTop: cellTop
						});
R
rebornix 已提交
379
					}
R
rebornix 已提交
380
				});
381

R
rebornix 已提交
382 383 384 385 386 387
				if (updateItems.length) {
					this.webview?.updateViewScrollTop(-scrollTop, updateItems);
				}
			}
		}));

R
rebornix 已提交
388
		this.list?.splice(0, this.list?.length || 0);
389
		this.list?.splice(0, 0, this.notebookViewModel!.viewCells as CellViewModel[]);
R
rebornix 已提交
390
		this.list?.layout();
P
Peng Lyu 已提交
391 392
	}

R
rebornix 已提交
393
	private saveTextEditorViewState(input: NotebookEditorInput): void {
R
npe  
rebornix 已提交
394
		if (this.group && this.notebookViewModel) {
R
rebornix 已提交
395
			const state = this.notebookViewModel.saveEditorViewState();
R
rebornix 已提交
396
			this.editorMemento.saveEditorState(this.group, input.resource, state);
R
rebornix 已提交
397 398 399 400 401
		}
	}

	private loadTextEditorViewState(input: NotebookEditorInput): INotebookEditorViewState | undefined {
		if (this.group) {
R
rebornix 已提交
402
			return this.editorMemento.loadEditorState(this.group, input.resource);
R
rebornix 已提交
403 404 405 406 407 408 409 410 411 412
		}

		return;
	}

	layout(dimension: DOM.Dimension): void {
		this.dimension = new DOM.Dimension(dimension.width, dimension.height);
		DOM.toggleClass(this.rootElement, 'mid-width', dimension.width < 1000 && dimension.width >= 600);
		DOM.toggleClass(this.rootElement, 'narrow-width', dimension.width < 600);
		DOM.size(this.body, dimension.width, dimension.height);
413
		this.list?.updateOptions({ additionalScrollHeight: dimension.height });
R
rebornix 已提交
414
		this.list?.layout(dimension.height, dimension.width);
R
rebornix 已提交
415
		this.eventDispatcher?.emit([new NotebookLayoutChangedEvent({ width: true, fontInfo: true }, this.getLayoutInfo())]);
R
rebornix 已提交
416 417 418 419 420 421 422 423 424 425 426 427
	}

	protected saveState(): void {
		if (this.input instanceof NotebookEditorInput) {
			this.saveTextEditorViewState(this.input);
		}

		super.saveState();
	}

	//#endregion

R
rebornix 已提交
428 429
	//#region Editor Features

R
rebornix 已提交
430
	selectElement(cell: ICellViewModel) {
R
rebornix 已提交
431 432 433 434 435 436 437 438
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.setSelection([index]);
			this.list?.setFocus([index]);
		}
	}

R
rebornix 已提交
439
	revealInView(cell: ICellViewModel) {
R
rebornix 已提交
440 441 442
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
R
rebornix 已提交
443
			this.list?.revealInView(index);
R
rebornix 已提交
444 445 446
		}
	}

R
rebornix 已提交
447
	revealInCenterIfOutsideViewport(cell: ICellViewModel) {
R
rebornix 已提交
448 449 450
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
R
rebornix 已提交
451
			this.list?.revealInCenterIfOutsideViewport(index);
R
rebornix 已提交
452 453 454
		}
	}

R
rebornix 已提交
455
	revealInCenter(cell: ICellViewModel) {
R
rebornix 已提交
456 457 458
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
R
rebornix 已提交
459 460 461 462
			this.list?.revealInCenter(index);
		}
	}

R
rebornix 已提交
463
	revealLineInView(cell: ICellViewModel, line: number): void {
R
rebornix 已提交
464 465 466 467
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.revealLineInView(index, line);
R
rebornix 已提交
468 469
		}
	}
R
rebornix 已提交
470

R
rebornix 已提交
471
	revealLineInCenter(cell: ICellViewModel, line: number) {
R
rebornix 已提交
472 473 474
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
R
rebornix 已提交
475
			this.list?.revealLineInCenter(index, line);
R
rebornix 已提交
476 477 478
		}
	}

R
rebornix 已提交
479
	revealLineInCenterIfOutsideViewport(cell: ICellViewModel, line: number) {
R
rebornix 已提交
480 481 482 483 484 485 486
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.revealLineInCenterIfOutsideViewport(index, line);
		}
	}

R
rebornix 已提交
487
	revealRangeInView(cell: ICellViewModel, range: Range): void {
R
rebornix 已提交
488 489 490 491 492 493 494
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.revealRangeInView(index, range);
		}
	}

R
rebornix 已提交
495
	revealRangeInCenter(cell: ICellViewModel, range: Range): void {
R
rebornix 已提交
496 497 498 499 500 501 502
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.revealRangeInCenter(index, range);
		}
	}

R
rebornix 已提交
503
	revealRangeInCenterIfOutsideViewport(cell: ICellViewModel, range: Range): void {
R
rebornix 已提交
504 505 506 507 508 509 510
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.revealRangeInCenterIfOutsideViewport(index, range);
		}
	}

R
rebornix 已提交
511
	setCellSelection(cell: ICellViewModel, range: Range): void {
R
rebornix 已提交
512 513 514 515 516 517 518
		const index = this.notebookViewModel?.getViewCellIndex(cell);

		if (index !== undefined) {
			this.list?.setCellSelection(index, range);
		}
	}

R
rebornix 已提交
519 520 521 522 523 524
	changeDecorations(callback: (changeAccessor: IModelDecorationsChangeAccessor) => any): any {
		return this.notebookViewModel?.changeDecorations(callback);
	}

	//#endregion

R
rebornix 已提交
525 526 527 528 529 530 531 532
	//#region Find Delegate

	public showFind() {
		this.findWidget.reveal();
	}

	public hideFind() {
		this.findWidget.hide();
R
rebornix 已提交
533
		this.focus();
R
rebornix 已提交
534 535 536 537
	}

	//#endregion

R
rebornix 已提交
538
	//#region Cell operations
R
rebornix 已提交
539 540
	layoutNotebookCell(cell: ICellViewModel, height: number) {
		let relayout = (cell: ICellViewModel, height: number) => {
R
rebornix 已提交
541
			let index = this.notebookViewModel!.getViewCellIndex(cell);
R
rebornix 已提交
542
			if (index >= 0) {
R
rebornix 已提交
543
				this.list?.updateElementHeight(index, height);
R
rebornix 已提交
544
			}
R
rebornix 已提交
545 546
		};

R
rebornix 已提交
547
		DOM.scheduleAtNextAnimationFrame(() => {
R
rebornix 已提交
548
			relayout(cell, height);
R
rebornix 已提交
549
		});
550 551
	}

R
rebornix 已提交
552
	async insertNotebookCell(cell: ICellViewModel, type: CellKind, direction: 'above' | 'below', initialText: string = ''): Promise<void> {
R
rebornix 已提交
553 554 555
		const newLanguages = this.notebookViewModel!.languages;
		const language = newLanguages && newLanguages.length ? newLanguages[0] : 'markdown';
		const index = this.notebookViewModel!.getViewCellIndex(cell);
P
Peng Lyu 已提交
556
		const insertIndex = direction === 'above' ? index : index + 1;
R
rebornix 已提交
557
		const newCell = this.notebookViewModel!.createCell(insertIndex, initialText.split(/\r?\n/g), language, type, true);
R
rebornix 已提交
558
		this.list?.setFocus([insertIndex]);
P
Peng Lyu 已提交
559

R
rebornix 已提交
560
		if (type === CellKind.Markdown) {
561
			newCell.editState = CellEditState.Editing;
P
Peng Lyu 已提交
562
		}
R
rebornix 已提交
563 564

		DOM.scheduleAtNextAnimationFrame(() => {
R
rebornix 已提交
565
			this.list?.revealInCenterIfOutsideViewport(insertIndex);
R
rebornix 已提交
566
		});
P
Peng Lyu 已提交
567 568
	}

R
rebornix 已提交
569
	async deleteNotebookCell(cell: ICellViewModel): Promise<void> {
570
		(cell as CellViewModel).save();
R
rebornix 已提交
571
		const index = this.notebookViewModel!.getViewCellIndex(cell);
R
rebornix 已提交
572
		this.notebookViewModel!.deleteCell(index, true);
R
rebornix 已提交
573 574
	}

575
	moveCellDown(cell: ICellViewModel): void {
576 577
		const index = this.notebookViewModel!.getViewCellIndex(cell);
		const newIdx = index + 1;
578
		this.moveCellToIndex(index, newIdx);
579 580
	}

581
	moveCellUp(cell: ICellViewModel): void {
582 583
		const index = this.notebookViewModel!.getViewCellIndex(cell);
		const newIdx = index - 1;
584
		this.moveCellToIndex(index, newIdx);
585 586
	}

587
	private moveCellToIndex(index: number, newIdx: number): void {
R
rebornix 已提交
588
		if (!this.notebookViewModel!.moveCellToIdx(index, newIdx, true)) {
589 590 591 592 593 594 595 596
			return;
		}

		DOM.scheduleAtNextAnimationFrame(() => {
			this.list?.revealInCenterIfOutsideViewport(index + 1);
		});
	}

R
rebornix 已提交
597
	editNotebookCell(cell: CellViewModel): void {
598
		cell.editState = CellEditState.Editing;
R
rebornix 已提交
599 600

		this.renderedEditors.get(cell)?.focus();
P
Peng Lyu 已提交
601 602
	}

R
rebornix 已提交
603
	saveNotebookCell(cell: ICellViewModel): void {
604
		cell.editState = CellEditState.Preview;
P
Peng Lyu 已提交
605 606
	}

R
rebornix 已提交
607 608 609 610 611 612 613 614 615 616
	getActiveCell() {
		let elements = this.list?.getFocusedElements();

		if (elements && elements.length) {
			return elements[0];
		}

		return undefined;
	}

617 618 619
	async executeNotebookCell(cell: ICellViewModel): Promise<void> {
		try {
			cell.runState = CellRunState.Running;
R
rebornix 已提交
620
			const provider = this.notebookService.getContributedNotebookProviders(this.viewModel!.uri)[0];
621 622 623 624 625 626 627 628 629 630 631 632
			if (provider) {
				const viewType = provider.id;
				const notebookUri = CellUri.parse(cell.uri)?.notebook;
				if (notebookUri) {
					return await this.notebookService.executeNotebookCell(viewType, notebookUri, cell.handle);
				}
			}
		} finally {
			cell.runState = CellRunState.Idle;
		}
	}

R
rebornix 已提交
633
	focusNotebookCell(cell: ICellViewModel, focusEditor: boolean) {
R
rebornix 已提交
634
		const index = this.notebookViewModel!.getViewCellIndex(cell);
R
rebornix 已提交
635 636

		if (focusEditor) {
R
rebornix 已提交
637 638 639
			this.list?.setFocus([index]);
			this.list?.setSelection([index]);
			this.list?.focusView();
R
rebornix 已提交
640

641
			cell.editState = CellEditState.Editing;
R
rebornix 已提交
642
			cell.focusMode = CellFocusMode.Editor;
643
			this.revealInCenterIfOutsideViewport(cell);
R
rebornix 已提交
644
		} else {
R
rebornix 已提交
645
			let itemDOM = this.list?.domElementAtIndex(index);
R
rebornix 已提交
646 647 648
			if (document.activeElement && itemDOM && itemDOM.contains(document.activeElement)) {
				(document.activeElement as HTMLElement).blur();
			}
649

650
			cell.editState = CellEditState.Preview;
651
			cell.focusMode = CellFocusMode.Editor;
R
rebornix 已提交
652

R
rebornix 已提交
653 654
			this.list?.setFocus([index]);
			this.list?.setSelection([index]);
655
			this.revealInCenterIfOutsideViewport(cell);
R
rebornix 已提交
656
			this.list?.focusView();
R
rebornix 已提交
657 658 659
		}
	}

R
rebornix 已提交
660 661 662 663
	//#endregion

	//#region MISC

R
rebornix 已提交
664 665 666 667 668 669 670 671 672 673 674
	getLayoutInfo(): NotebookLayoutInfo {
		if (!this.list) {
			throw new Error('Editor is not initalized successfully');
		}

		return {
			width: this.dimension!.width,
			height: this.dimension!.height,
			fontInfo: this.fontInfo!
		};
	}
R
rebornix 已提交
675

R
rebornix 已提交
676 677
	triggerScroll(event: IMouseWheelEvent) {
		this.list?.triggerScrollFromMouseWheelEvent(event);
R
rebornix 已提交
678 679
	}

R
rebornix 已提交
680
	createInset(cell: CodeCellViewModel, output: IOutput, shadowContent: string, offset: number) {
R
rebornix 已提交
681 682
		if (!this.webview) {
			return;
R
rebornix 已提交
683 684
		}

R
rebornix 已提交
685
		let preloads = this.notebookViewModel!.renderers;
R
rebornix 已提交
686

687
		if (!this.webview!.insetMapping.has(output)) {
R
rebornix 已提交
688
			let index = this.notebookViewModel!.getViewCellIndex(cell);
689 690 691
			let cellTop = this.list?.getAbsoluteTop(index) || 0;

			this.webview!.createInset(cell, output, cellTop, offset, shadowContent, preloads);
R
rebornix 已提交
692
		} else {
R
rebornix 已提交
693
			let index = this.notebookViewModel!.getViewCellIndex(cell);
694
			let cellTop = this.list?.getAbsoluteTop(index) || 0;
R
rebornix 已提交
695 696
			let scrollTop = this.list?.scrollTop || 0;

697
			this.webview!.updateViewScrollTop(-scrollTop, [{ cell: cell, output: output, cellTop: cellTop }]);
R
rebornix 已提交
698
		}
R
rebornix 已提交
699
	}
R
rebornix 已提交
700

R
rebornix 已提交
701 702 703 704 705 706 707 708
	removeInset(output: IOutput) {
		if (!this.webview) {
			return;
		}

		this.webview!.removeInset(output);
	}

R
rebornix 已提交
709 710
	getOutputRenderer(): OutputRenderer {
		return this.outputRenderer;
R
rebornix 已提交
711
	}
712

713 714 715 716
	postMessage(message: any) {
		this.webview?.webview.sendMessage(message);
	}

R
rebornix 已提交
717
	//#endregion
718 719 720 721 722 723

	toJSON(): any {
		return {
			notebookHandle: this.viewModel?.handle
		};
	}
P
Peng Lyu 已提交
724 725 726 727
}

const embeddedEditorBackground = 'walkThrough.embeddedEditorBackground';

728 729 730 731 732 733
export const focusedCellIndicator = registerColor('notebook.focusedCellIndicator', {
	light: new Color(new RGBA(102, 175, 224)),
	dark: new Color(new RGBA(12, 125, 157)),
	hc: new Color(new RGBA(0, 73, 122))
}, nls.localize('notebook.focusedCellIndicator', "The color of the focused notebook cell indicator."));

734 735 736 737 738 739 740
export const notebookOutputContainerColor = registerColor('notebook.outputContainerBackgroundColor', {
	dark: new Color(new RGBA(255, 255, 255, 0.06)),
	light: new Color(new RGBA(228, 230, 241)),
	hc: null
}
	, nls.localize('notebook.outputContainerBackgroundColor', "The Color of the notebook output container background."));

741

P
Peng Lyu 已提交
742 743 744
registerThemingParticipant((theme, collector) => {
	const color = getExtraColor(theme, embeddedEditorBackground, { dark: 'rgba(0, 0, 0, .4)', extra_dark: 'rgba(200, 235, 255, .064)', light: '#f4f4f4', hc: null });
	if (color) {
R
rebornix 已提交
745 746
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .cell .monaco-editor-background,
			.monaco-workbench .part.editor > .content .notebook-editor .cell .margin-view-overlays { background: ${color}; }`);
P
Peng Lyu 已提交
747 748 749
	}
	const link = theme.getColor(textLinkForeground);
	if (link) {
R
Rob Lourens 已提交
750
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .cell .output a { color: ${link}; }`);
P
Peng Lyu 已提交
751 752 753
	}
	const activeLink = theme.getColor(textLinkActiveForeground);
	if (activeLink) {
R
Rob Lourens 已提交
754 755
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .cell .output a:hover,
			.monaco-workbench .part.editor > .content .notebook-editor .cell .output a:active { color: ${activeLink}; }`);
P
Peng Lyu 已提交
756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773
	}
	const shortcut = theme.getColor(textPreformatForeground);
	if (shortcut) {
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor code,
			.monaco-workbench .part.editor > .content .notebook-editor .shortcut { color: ${shortcut}; }`);
	}
	const border = theme.getColor(contrastBorder);
	if (border) {
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .monaco-editor { border-color: ${border}; }`);
	}
	const quoteBackground = theme.getColor(textBlockQuoteBackground);
	if (quoteBackground) {
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor blockquote { background: ${quoteBackground}; }`);
	}
	const quoteBorder = theme.getColor(textBlockQuoteBorder);
	if (quoteBorder) {
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor blockquote { border-color: ${quoteBorder}; }`);
	}
774

775
	const containerBackground = theme.getColor(notebookOutputContainerColor);
776

777 778
	if (containerBackground) {
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .output { background-color: ${containerBackground}; }`);
779
	}
780

781 782 783 784 785 786
	const focusedCellIndicatorColor = theme.getColor(focusedCellIndicator);
	if (focusedCellIndicatorColor) {
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .monaco-list-row.focused .notebook-cell-focus-indicator { border-color: ${focusedCellIndicatorColor}; }`);
		collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .monaco-list-row.selected .notebook-cell-focus-indicator { border-color: ${focusedCellIndicatorColor}; }`);
	}

787
	// Cell Margin
788
	collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .monaco-list-row > div.cell { margin: 8px ${CELL_MARGIN}px 0px ${CELL_MARGIN}px; }`);
789
	collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .output { margin: 0px ${CELL_MARGIN}px 0px ${CELL_MARGIN + CELL_RUN_GUTTER}px }`);
R
Rob Lourens 已提交
790

R
Rob Lourens 已提交
791 792 793
	collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .cell .cell-editor-container { width: calc(100% - ${CELL_RUN_GUTTER}px); }`);
	collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .monaco-list-row > div.cell.markdown { padding-left: ${CELL_RUN_GUTTER}px; }`);
	collector.addRule(`.monaco-workbench .part.editor > .content .notebook-editor .cell .run-button-container { width: ${CELL_RUN_GUTTER}px; }`);
P
Peng Lyu 已提交
794
});