refactor(language-core): extract generateEscaped

Author: KazariEX

packages/language-core/lib/codegen/template/styleScopedClasses.ts

@@ -3,9 +3,12 @@ import type * as ts from 'typescript';
 import { getNodeText } from '../../parsers/scriptSetupRanges';
 import type { Code } from '../../types';
 import { endOfLine, normalizeAttributeValue } from '../utils';
+import { generateEscaped } from '../utils/escaped';
 import type { TemplateCodegenContext } from './context';
 import type { TemplateCodegenOptions } from './index';
 
+const classNameEscapeRegex = /([\\'])/;
+
 export function* generateStyleScopedClassReferences(
 	ctx: TemplateCodegenContext,
 	withDot = false
@@ -31,7 +34,13 @@ export function* generateStyleScopedClassReferences(
 		yield `'`;
 
 		// fix https://github.com/vuejs/language-tools/issues/4537
-		yield* escapeString(source, className, offset, ['\\', '\'']);
+		yield* generateEscaped(
+			className,
+			source,
+			offset,
+			ctx.codeFeatures.navigationAndAdditionalCompletion,
+			classNameEscapeRegex
+		);
 		yield `'`;
 		yield [
 			'',
@@ -41,45 +50,6 @@ export function* generateStyleScopedClassReferences(
 		];
 		yield `]} */${endOfLine}`;
 	}
-
-	function* escapeString(source: string, className: string, offset: number, escapeTargets: string[]): Generator<Code> {
-		let count = 0;
-
-		const currentEscapeTargets = [...escapeTargets];
-		const firstEscapeTarget = currentEscapeTargets.shift()!;
-		const splitted = className.split(firstEscapeTarget);
-
-		for (let i = 0; i < splitted.length; i++) {
-			const part = splitted[i];
-			const partLength = part.length;
-
-			if (escapeTargets.length > 0) {
-				yield* escapeString(source, part, offset + count, [...currentEscapeTargets]);
-			} else {
-				yield [
-					part,
-					source,
-					offset + count,
-					ctx.codeFeatures.navigationAndAdditionalCompletion,
-				];
-			}
-
-			if (i !== splitted.length - 1) {
-				yield '\\';
-
-				yield [
-					firstEscapeTarget,
-					source,
-					offset + count + partLength,
-					ctx.codeFeatures.navigationAndAdditionalCompletion,
-				];
-
-				count += partLength + 1;
-			} else {
-				count += partLength;
-			}
-		}
-	}
 }
 
 export function collectStyleScopedClassReferences(

packages/language-core/lib/codegen/utils/escaped.ts

@@ -0,0 +1,27 @@
+import type { Code, VueCodeInformation } from "../../types";
+
+export function* generateEscaped(
+	text: string,
+	source: string,
+	offset: number,
+	features: VueCodeInformation,
+	escapeTarget: RegExp
+): Generator<Code> {
+	const parts = text.split(escapeTarget);
+	let isEscapeTarget = false;
+
+	for (let i = 0; i < parts.length; i++) {
+		const part = parts[i];
+		if (isEscapeTarget) {
+			yield `\\`;
+		}
+		yield [
+			part,
+			source,
+			offset,
+			i === 0 ? features : { __combineOffset: i },
+		];
+		offset += part.length;
+		isEscapeTarget = !isEscapeTarget;
+	}
+}