From 34162acb238c3c7a168d1e8cef4c53b29dc6e636 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Vin=C3=ADcius=20Louren=C3=A7o?= <contact@viniciusl.com.br>
Date: Fri, 12 Jan 2024 23:19:21 -0300
Subject: [PATCH] feat(formatGenerator): added support for format using
 generators

---
 src/impl/format.ts | 33 +++++++++++++++++++++------------
 src/main.ts        | 18 ++++++++++++++++++
 2 files changed, 39 insertions(+), 12 deletions(-)

diff --git a/src/impl/format.ts b/src/impl/format.ts
index 383f2fb..41a0244 100644
--- a/src/impl/format.ts
+++ b/src/impl/format.ts
@@ -9,6 +9,16 @@ import { createScanner } from './scanner';
 import { cachedSpaces, cachedBreakLinesWithSpaces, supportedEols, SupportedEOL } from './string-intern';
 
 export function format(documentText: string, range: Range | undefined, options: FormattingOptions): Edit[] {
+	const operations: Edit[] = [];
+
+	for (const edit of formatGenerator(documentText, range, options)) {
+		operations.push(edit);
+	}
+
+	return operations;
+}
+
+export function* formatGenerator(documentText: string, range: Range | undefined, options: FormattingOptions): Generator<Edit, void, void> {
 	let initialIndentLevel: number;
 	let formatText: string;
 	let formatTextStart: number;
@@ -85,16 +95,12 @@ export function format(documentText: string, range: Range | undefined, options:
 		hasError = token === SyntaxKind.Unknown || scanner.getTokenError() !== ScanError.None;
 		return token;
 	}
-	const editOperations: Edit[] = [];
-	function addEdit(text: string, startOffset: number, endOffset: number) {
-		if (!hasError && (!range || (startOffset < rangeEnd && endOffset > rangeStart)) && documentText.substring(startOffset, endOffset) !== text) {
-			editOperations.push({ offset: startOffset, length: endOffset - startOffset, content: text });
-		}
-	}
-
+	const shouldAddEdit = (text: string, startOffset: number, endOffset: number) => !hasError && (!range || (startOffset < rangeEnd && endOffset > rangeStart)) && documentText.substring(startOffset, endOffset) !== text
 	let firstToken = scanNext();
 	if (options.keepLines && numberLineBreaks > 0) {
-		addEdit(repeat(eol, numberLineBreaks), 0, 0);
+		const text = repeat(eol, numberLineBreaks);
+		if (shouldAddEdit(text, 0, 0))
+			yield { offset: 0, length: 0, content: text };
 	}
 
 	if (firstToken !== SyntaxKind.EOF) {
@@ -102,7 +108,8 @@ export function format(documentText: string, range: Range | undefined, options:
 		let initialIndent = (indentValue.length * initialIndentLevel < 20) && options.insertSpaces 
 			? cachedSpaces[indentValue.length * initialIndentLevel] 
 			: repeat(indentValue, initialIndentLevel);
-		addEdit(initialIndent, formatTextStart, firstTokenStart);
+		if (shouldAddEdit(initialIndent, formatTextStart, firstTokenStart))
+			yield { offset: formatTextStart, length: firstTokenStart - formatTextStart, content: initialIndent };
 	}
 
 	while (firstToken !== SyntaxKind.EOF) {
@@ -114,7 +121,9 @@ export function format(documentText: string, range: Range | undefined, options:
 
 		while (numberLineBreaks === 0 && (secondToken === SyntaxKind.LineCommentTrivia || secondToken === SyntaxKind.BlockCommentTrivia)) {
 			let commentTokenStart = scanner.getTokenOffset() + formatTextStart;
-			addEdit(cachedSpaces[1], firstTokenEnd, commentTokenStart);
+			if (shouldAddEdit(cachedSpaces[1], firstTokenEnd, commentTokenStart))
+				yield { offset: firstTokenEnd, length: commentTokenStart - firstTokenEnd, content: cachedSpaces[1] };
+
 			firstTokenEnd = scanner.getTokenOffset() + scanner.getTokenLength() + formatTextStart;
 			needsLineBreak = secondToken === SyntaxKind.LineCommentTrivia;
 			replaceContent = needsLineBreak ? newLinesAndIndent() : '';
@@ -211,10 +220,10 @@ export function format(documentText: string, range: Range | undefined, options:
 			}
 		}
 		const secondTokenStart = scanner.getTokenOffset() + formatTextStart;
-		addEdit(replaceContent, firstTokenEnd, secondTokenStart);
+		if (shouldAddEdit(replaceContent, firstTokenEnd, secondTokenStart))
+			yield { offset: firstTokenEnd, length: secondTokenStart - firstTokenEnd, content: replaceContent };
 		firstToken = secondToken;
 	}
-	return editOperations;
 }
 
 function repeat(s: string, count: number): string {
diff --git a/src/main.ts b/src/main.ts
index 9852b67..33059ce 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -305,6 +305,11 @@ export interface JSONVisitor {
  */
 export type EditResult = Edit[];
 
+/**
+ * See more {@link EditResult}
+ */
+export type EditGeneratorResult = Generator<Edit, void, void>;
+
 /**
  * Represents a text modification
  */
@@ -376,6 +381,19 @@ export function format(documentText: string, range: Range | undefined, options:
 	return formatter.format(documentText, range, options);
 }
 
+/**
+ * Computes the edit operations needed to format a JSON document.
+ * 
+ * @param documentText The input text 
+ * @param range The range to format or `undefined` to format the full content
+ * @param options The formatting options
+ * @returns The edit operations describing the formatting changes to the original document following the format described in {@linkcode EditResult}.
+ * To apply the edit operations to the input, use {@linkcode applyEdits}.
+ */
+export function formatGenerator(documentText: string, range: Range | undefined, options: FormattingOptions): EditGeneratorResult {
+	return formatter.formatGenerator(documentText, range, options);
+}
+
 /** 
  * Options used by {@linkcode modify} when computing the modification edit operations
  */