Merge commit 'be3e8236086165e5e45a5a10783823874b3f3ebd' as 'lib/vscode'

lib/vscode/extensions/html-language-features/server/src/modes/formatting.ts

@@ -0,0 +1,92 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { LanguageModes, Settings, LanguageModeRange, TextDocument, Range, TextEdit, FormattingOptions, Position } from './languageModes';
+import { pushAll } from '../utils/arrays';
+import { isEOL } from '../utils/strings';
+
+export async function format(languageModes: LanguageModes, document: TextDocument, formatRange: Range, formattingOptions: FormattingOptions, settings: Settings | undefined, enabledModes: { [mode: string]: boolean }) {
+	let result: TextEdit[] = [];
+
+	let endPos = formatRange.end;
+	let endOffset = document.offsetAt(endPos);
+	let content = document.getText();
+	if (endPos.character === 0 && endPos.line > 0 && endOffset !== content.length) {
+		// if selection ends after a new line, exclude that new line
+		let prevLineStart = document.offsetAt(Position.create(endPos.line - 1, 0));
+		while (isEOL(content, endOffset - 1) && endOffset > prevLineStart) {
+			endOffset--;
+		}
+		formatRange = Range.create(formatRange.start, document.positionAt(endOffset));
+	}
+
+
+	// run the html formatter on the full range and pass the result content to the embedded formatters.
+	// from the final content create a single edit
+	// advantages of this approach are
+	//  - correct indents in the html document
+	//  - correct initial indent for embedded formatters
+	//  - no worrying of overlapping edits
+
+	// make sure we start in html
+	let allRanges = languageModes.getModesInRange(document, formatRange);
+	let i = 0;
+	let startPos = formatRange.start;
+	let isHTML = (range: LanguageModeRange) => range.mode && range.mode.getId() === 'html';
+
+	while (i < allRanges.length && !isHTML(allRanges[i])) {
+		let range = allRanges[i];
+		if (!range.attributeValue && range.mode && range.mode.format) {
+			let edits = await range.mode.format(document, Range.create(startPos, range.end), formattingOptions, settings);
+			pushAll(result, edits);
+		}
+		startPos = range.end;
+		i++;
+	}
+	if (i === allRanges.length) {
+		return result;
+	}
+	// modify the range
+	formatRange = Range.create(startPos, formatRange.end);
+
+	// perform a html format and apply changes to a new document
+	let htmlMode = languageModes.getMode('html')!;
+	let htmlEdits = await htmlMode.format!(document, formatRange, formattingOptions, settings);
+	let htmlFormattedContent = TextDocument.applyEdits(document, htmlEdits);
+	let newDocument = TextDocument.create(document.uri + '.tmp', document.languageId, document.version, htmlFormattedContent);
+	try {
+		// run embedded formatters on html formatted content: - formatters see correct initial indent
+		let afterFormatRangeLength = document.getText().length - document.offsetAt(formatRange.end); // length of unchanged content after replace range
+		let newFormatRange = Range.create(formatRange.start, newDocument.positionAt(htmlFormattedContent.length - afterFormatRangeLength));
+		let embeddedRanges = languageModes.getModesInRange(newDocument, newFormatRange);
+
+		let embeddedEdits: TextEdit[] = [];
+
+		for (let r of embeddedRanges) {
+			let mode = r.mode;
+			if (mode && mode.format && enabledModes[mode.getId()] && !r.attributeValue) {
+				let edits = await mode.format(newDocument, r, formattingOptions, settings);
+				for (let edit of edits) {
+					embeddedEdits.push(edit);
+				}
+			}
+		}
+
+		if (embeddedEdits.length === 0) {
+			pushAll(result, htmlEdits);
+			return result;
+		}
+
+		// apply all embedded format edits and create a single edit for all changes
+		let resultContent = TextDocument.applyEdits(newDocument, embeddedEdits);
+		let resultReplaceText = resultContent.substring(document.offsetAt(formatRange.start), resultContent.length - afterFormatRangeLength);
+
+		result.push(TextEdit.replace(formatRange, resultReplaceText));
+		return result;
+	} finally {
+		languageModes.onDocumentRemoved(newDocument);
+	}
+
+}