notebookEditorInput.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import { EditorInput, EditorModel, IEditorInput } from 'vs/workbench/common/editor';
import { ITextModelService } from 'vs/editor/common/services/resolverService';
import { ITextModel } from 'vs/editor/common/model';

export interface IStreamOutput {
	output_type: 'stream';
	text: string;
}

export interface IErrorOutput {
	output_type: 'error';
	evalue: string;
	traceback: string[];
}

export interface IDisplayOutput {
	output_type: 'display_data';
	data: { string: string };
}

export interface IGenericOutput {
	output_type: string;
}

export type IOutput = IStreamOutput | any;

export interface ICell {
	source: string[];
	cell_type: 'markdown' | 'code';
	outputs: IOutput[];
}

export interface LanguageInfo {
	file_extension: string;
}
export interface IMetadata {
	language_info: LanguageInfo;
}
export interface INotebook {
	metadata: IMetadata;
	cells: ICell[];
}


export class NotebookEditorModel extends EditorModel {
	private _notebook: INotebook | undefined;

	constructor(
		public readonly textModel: ITextModel
	) {
		super();
	}

	public getNookbook(): INotebook {
		if (this._notebook) {
			return this._notebook;
		}

		let content = this.textModel.getValue();
		this._notebook = JSON.parse(content);
		return this._notebook!;
	}
}

export class NotebookEditorInput extends EditorInput {
	static readonly ID: string = 'workbench.input.notebook';
	private promise: Promise<NotebookEditorModel> | null = null;

	constructor(
		public readonly editorInput: IEditorInput,
		@ITextModelService private readonly textModelResolverService: ITextModelService
	) {
		super();
	}

	getTypeId(): string {
		return NotebookEditorInput.ID;
	}

	getName(): string {
		return this.editorInput.getName();
	}

	resolve(): Promise<NotebookEditorModel> {
		if (!this.promise) {
			this.promise = this.textModelResolverService.createModelReference(this.editorInput.getResource()!)
				.then(ref => {
					const textModel = ref.object.textEditorModel;

					return new NotebookEditorModel(textModel);
				});
		}

		return this.promise;
	}
}