Book Editing: Support adding new sections to a book (#17074)

* add a new section command
2026-02-11 02:32:35 -05:00 · 2021-09-14 11:31:43 -07:00
parent f92e6d24d4
commit ce13d01efb
10 changed files with 65 additions and 19 deletions
--- a/extensions/notebook/src/dialog/addTocEntryDialog.ts
+++ b/extensions/notebook/src/dialog/addTocEntryDialog.ts
@@ -0,0 +1,115 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+import * as azdata from 'azdata';
+import * as path from 'path';
+import { pathExists } from 'fs-extra';
+import * as loc from '../common/localizedConstants';
+import { IBookTocManager } from '../book/bookTocManager';
+import { confirmMessageDialog, FileExtension } from '../common/utils';
+import { IPrompter } from '../prompts/question';
+import CodeAdapter from '../prompts/adapter';
+import { BookTreeItem, BookTreeItemType } from '../book/bookTreeItem';
+import { TocEntryPathHandler } from '../book/tocEntryPathHandler';
+
+export class AddTocEntryDialog {
+	private _dialog: azdata.window.Dialog;
+	private readonly _dialogName = 'addNewFileBookTreeViewDialog';
+	public view: azdata.ModelView;
+	private _formModel: azdata.FormContainer;
+	private _fileNameInputBox: azdata.InputBoxComponent;
+	private _titleInputBox: azdata.InputBoxComponent;
+	private _saveLocationInputBox: azdata.TextComponent;
+	private _prompter: IPrompter;
+
+	constructor(private _tocManager: IBookTocManager, private _bookItem: BookTreeItem, private _extension: FileExtension, private _isSection?: boolean) {
+		this._prompter = new CodeAdapter();
+	}
+
+	public get dialog(): azdata.window.Dialog | undefined {
+		return this._dialog;
+	}
+
+	public async validatePath(folderPath: string, fileBasename: string): Promise<void> {
+		const destinationUri = path.join(folderPath, fileBasename);
+		if (await pathExists(destinationUri)) {
+			const doOverwrite = await confirmMessageDialog(this._prompter, loc.confirmOverwrite);
+			if (!doOverwrite) {
+				throw (new Error(loc.msgDuplicateFileName(destinationUri)));
+			}
+		}
+		if (!(await pathExists(folderPath))) {
+			throw (new Error(loc.msgSaveFolderError));
+		}
+	}
+
+	public async createDialog(): Promise<void> {
+		const dialogTitle = this._isSection ? loc.newSection : this._extension === FileExtension.Notebook ? loc.newNotebook : loc.newMarkdown;
+		this._dialog = azdata.window.createModelViewDialog(dialogTitle, this._dialogName);
+		this._dialog.registerContent(async view => {
+			this.view = view;
+			this._fileNameInputBox = this.view.modelBuilder.inputBox()
+				.withProps({
+					enabled: true,
+					width: '400px'
+				}).component();
+
+			this._titleInputBox = this.view.modelBuilder.inputBox()
+				.withProps({
+					enabled: true,
+					width: '400px'
+				}).component();
+
+			this._saveLocationInputBox = this.view.modelBuilder.inputBox()
+				.withProps({
+					value: this._bookItem.contextValue === BookTreeItemType.savedBook ? this._bookItem.rootContentPath : path.dirname(this._bookItem.resourceUri.fsPath),
+					enabled: false,
+					width: '400px'
+				}).component();
+
+			this._formModel = this.view.modelBuilder.formContainer()
+				.withFormItems([{
+					components: [
+						{
+							title: loc.title,
+							required: true,
+							component: this._titleInputBox
+						},
+						{
+							component: this._fileNameInputBox,
+							title: loc.fileName,
+							required: true
+						},
+						{
+							component: this._saveLocationInputBox,
+							title: loc.saveLocation,
+							required: false
+						}
+					],
+					title: ''
+				}]).component();
+			await this.view.initializeModel(this._formModel);
+		});
+		this._dialog.okButton.label = loc.add;
+		this._dialog.registerCloseValidator(async () => await this.createFile(this._fileNameInputBox.value, this._titleInputBox.value));
+		azdata.window.openDialog(this._dialog);
+	}
+
+	public async createFile(fileName: string, titleName: string): Promise<boolean> {
+		try {
+			const dirPath = this._bookItem.contextValue === BookTreeItemType.savedBook ? this._bookItem.rootContentPath : path.dirname(this._bookItem.book.contentPath);
+			const filePath = path.posix.join(dirPath, fileName).concat(this._extension);
+			await this.validatePath(dirPath, fileName.concat(this._extension));
+			const pathDetails = new TocEntryPathHandler(filePath, this._bookItem.rootContentPath, titleName);
+			await this._tocManager.addNewTocEntry(pathDetails, this._bookItem, this._isSection);
+			return true;
+		} catch (error) {
+			this._dialog.message = {
+				text: error.message,
+				level: azdata.window.MessageLevel.Error
+			};
+			return false;
+		}
+	}
+}