@weapp-vite/volar 2.1.0 → 2.1.1

package/README.md

@@ -56,6 +56,13 @@ pnpm add -D @weapp-vite/volar @weapp-core/schematics
 pnpm --filter @weapp-vite/volar build
 pnpm --filter @weapp-vite/volar test
 pnpm --filter @weapp-vite/volar typecheck
+pnpm --filter @weapp-vite/volar benchmark
+```
+
+benchmark 会为普通 SFC 和 `defineOptions` SFC 创建全新的 content 与 snapshot，并强制生成 service script 和全部 embedded code。对比修改前构建时可传入基线插件：
+
+```bash
+pnpm --filter @weapp-vite/volar benchmark -- --baseline /path/to/baseline/dist/index.mjs
 ```
 
 ## 7. 相关链接

package/dist/constants.d.ts

@@ -1,4 +1,5 @@
 export declare const BLOCK_TYPE = "json";
+export declare const DEFINE_OPTIONS_CACHE_MAX = 128;
 export declare const JS_LANG = "js";
 export declare const JSONC_LANG = "jsonc";
 export declare const JSON_LANG = "json";

package/dist/index.cjs

@@ -232,6 +232,157 @@ function resolveEmbeddedJsonBlock(fileName, sfc, embeddedCode, tsModule, hasSche
 	]);
 }
 //#endregion
+//#region src/parseSfc.ts
+function parseAttr(property, node) {
+	if (!property.value) return true;
+	const { source, start } = property.value.loc;
+	const quote = source[0];
+	const quoted = (quote === "\"" || quote === "'") && source.endsWith(quote);
+	return {
+		text: quoted ? source.slice(1, -1) : source,
+		offset: (quoted ? start.offset + 1 : start.offset) - node.loc.start.offset
+	};
+}
+function repairIncompleteTemplate(descriptor, errors) {
+	const template = descriptor.template;
+	if (!template) return;
+	for (const error of errors) {
+		if (!("code" in error) || error.code !== 24 || error.loc?.start.line !== template.loc.start.line) continue;
+		const endTagOffset = template.content.lastIndexOf("<");
+		const endTagText = template.content.slice(endTagOffset).trimEnd();
+		if ("</template>".startsWith(endTagText)) {
+			template.loc.end.offset = template.loc.start.offset + endTagOffset;
+			template.content = template.content.slice(0, endTagOffset);
+			template.loc.source = template.content;
+		}
+	}
+}
+function createBlock(compilerDom, node, source) {
+	const type = node.tag;
+	let { start, end } = node.loc;
+	let content = "";
+	if (node.children.length) {
+		start = node.children[0].loc.start;
+		end = node.children[node.children.length - 1].loc.end;
+		content = source.slice(start.offset, end.offset);
+	} else {
+		const offset = node.loc.source.indexOf("</");
+		if (offset > -1) start = {
+			line: start.line,
+			column: start.column + offset,
+			offset: start.offset + offset
+		};
+		end = { ...start };
+	}
+	const attrs = {};
+	const block = {
+		type,
+		content,
+		loc: {
+			source: content,
+			start,
+			end
+		},
+		attrs
+	};
+	for (const property of node.props) {
+		if (property.type !== compilerDom.NodeTypes.ATTRIBUTE) continue;
+		attrs[property.name] = property.value ? property.value.content || true : true;
+		if (property.name === "lang") block.lang = property.value?.content;
+		else if (property.name === "src") block.__src = parseAttr(property, node);
+		else if (type === "script") {
+			const scriptBlock = block;
+			if (property.name === "vapor") {
+				scriptBlock.setup ??= attrs[property.name];
+				scriptBlock.__generic ??= true;
+			} else if (property.name === "setup") scriptBlock.setup = attrs[property.name];
+			else if (property.name === "generic") scriptBlock.__generic = parseAttr(property, node);
+		} else if (type === "style") {
+			const styleBlock = block;
+			if (property.name === "scoped") styleBlock.scoped = true;
+			else if (property.name === "module") styleBlock.__module = parseAttr(property, node);
+		}
+	}
+	return block;
+}
+function parseSfc(compilerDom, source, filename) {
+	const errors = [];
+	const ast = compilerDom.parse(source, {
+		isNativeTag: () => true,
+		isPreTag: () => true,
+		parseMode: "sfc",
+		onError: (error) => errors.push(error),
+		comments: true
+	});
+	const descriptor = {
+		filename,
+		source,
+		comments: [],
+		template: null,
+		script: null,
+		scriptSetup: null,
+		styles: [],
+		customBlocks: [],
+		cssVars: [],
+		slotted: false,
+		shouldForceReload: () => false
+	};
+	for (const node of ast.children) {
+		if (node.type === compilerDom.NodeTypes.COMMENT) {
+			descriptor.comments.push(node.content);
+			continue;
+		}
+		if (node.type !== compilerDom.NodeTypes.ELEMENT) continue;
+		if (node.tag === "template") {
+			descriptor.template = createBlock(compilerDom, node, source);
+			continue;
+		}
+		if (node.tag === "script") {
+			const scriptBlock = createBlock(compilerDom, node, source);
+			if (scriptBlock.setup && !descriptor.scriptSetup) descriptor.scriptSetup = scriptBlock;
+			else if (!scriptBlock.setup && !descriptor.script) descriptor.script = scriptBlock;
+			continue;
+		}
+		if (node.tag === "style") {
+			descriptor.styles.push(createBlock(compilerDom, node, source));
+			continue;
+		}
+		descriptor.customBlocks.push(createBlock(compilerDom, node, source));
+	}
+	repairIncompleteTemplate(descriptor, errors);
+	return {
+		descriptor,
+		errors
+	};
+}
+function updateSfc(sfc, change) {
+	const blocks = [
+		sfc.descriptor.template,
+		sfc.descriptor.script,
+		sfc.descriptor.scriptSetup,
+		...sfc.descriptor.styles,
+		...sfc.descriptor.customBlocks
+	].filter((block) => block !== null);
+	const hitBlock = blocks.find((block) => change.start >= block.loc.start.offset && change.end <= block.loc.end.offset);
+	if (!hitBlock) return;
+	const oldContent = hitBlock.content;
+	const relativeStart = change.start - hitBlock.loc.start.offset;
+	const relativeEnd = change.end - hitBlock.loc.start.offset;
+	const newContent = oldContent.slice(0, relativeStart) + change.newText + oldContent.slice(relativeEnd);
+	const endTagRE = new RegExp(`</\\s*${hitBlock.type}\\s*>`);
+	if (endTagRE.test(oldContent) !== endTagRE.test(newContent)) return;
+	hitBlock.content = newContent;
+	hitBlock.loc.source = newContent;
+	const lengthDiff = change.newText.length - (change.end - change.start);
+	for (const block of blocks) {
+		if (block.loc.start.offset > change.end) block.loc.start.offset += lengthDiff;
+		if (block.loc.end.offset >= change.end) block.loc.end.offset += lengthDiff;
+	}
+	const source = sfc.descriptor.source;
+	sfc.descriptor.source = source.slice(0, change.start) + change.newText + source.slice(change.end);
+	return sfc;
+}
+//#endregion
 //#region src/scriptSetup.ts
 function collectWxsModuleNames(templateContent) {
 	if (!templateContent) return [];
@@ -302,9 +453,7 @@ function collectBindingNames(name, tsModule, bindings) {
 	}
 	for (const element of name.elements) if (tsModule.isBindingElement(element)) collectBindingNames(element.name, tsModule, bindings);
 }
-function collectTopLevelBindingNames(code, tsModule, lang) {
-	const scriptKind = lang === "ts" ? 3 : 1;
-	const sourceFile = tsModule.createSourceFile(`bindings.${lang}`, code, 99, true, scriptKind);
+function collectTopLevelBindingNames(sourceFile, tsModule) {
 	const bindings = /* @__PURE__ */ new Set();
 	for (const statement of sourceFile.statements) {
 		if (tsModule.isVariableStatement(statement)) {
@@ -428,9 +577,7 @@ function getFunctionLikeReturnTypeText(node, tsModule, sourceFile) {
 	const returnedExpression = tsModule.isBlock(node.body) ? findReturnedExpression(node.body, tsModule) : findReturnedExpression(node.body, tsModule);
 	return returnedExpression ? getExpressionTypeText(returnedExpression, tsModule, sourceFile) : "any";
 }
-function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
-	const scriptKind = lang === "ts" ? 3 : 1;
-	const sourceFile = tsModule.createSourceFile(`script-setup.${lang}`, code, 99, true, scriptKind);
+function collectDefineOptionsTemplateBindings(sourceFile, tsModule) {
 	const valueBindings = /* @__PURE__ */ new Map();
 	const functionBindings = /* @__PURE__ */ new Map();
 	const visit = (node) => {
@@ -445,7 +592,8 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 						if (!tsModule.isPropertyAssignment(entry) && !tsModule.isMethodDeclaration(entry) && !tsModule.isShorthandPropertyAssignment(entry)) continue;
 						const name = getPropertyNameText(entry.name, tsModule);
 						if (!name || !isIdentifierName(name)) continue;
-						setBindingType(functionBindings, name, tsModule.isPropertyAssignment(entry) && (tsModule.isArrowFunction(entry.initializer) || tsModule.isFunctionExpression(entry.initializer)) ? getFunctionSignatureTypeText(entry.initializer, sourceFile) : tsModule.isMethodDeclaration(entry) ? getFunctionSignatureTypeText(entry, sourceFile) : "(...args: any[]) => any");
+						const typeText = tsModule.isPropertyAssignment(entry) && (tsModule.isArrowFunction(entry.initializer) || tsModule.isFunctionExpression(entry.initializer)) ? getFunctionSignatureTypeText(entry.initializer, sourceFile) : tsModule.isMethodDeclaration(entry) ? getFunctionSignatureTypeText(entry, sourceFile) : "(...args: any[]) => any";
+						setBindingType(functionBindings, name, typeText);
 					}
 					continue;
 				}
@@ -479,7 +627,8 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 						if (!tsModule.isPropertyAssignment(property) && !tsModule.isMethodDeclaration(property) && !tsModule.isShorthandPropertyAssignment(property) && !tsModule.isGetAccessorDeclaration(property)) continue;
 						const name = getPropertyNameText(property.name, tsModule);
 						if (!name || !isIdentifierName(name)) continue;
-						setBindingType(valueBindings, name, tsModule.isPropertyAssignment(property) ? getExpressionTypeText(property.initializer, tsModule, sourceFile) : "(...args: any[]) => any");
+						const typeText = tsModule.isPropertyAssignment(property) ? getExpressionTypeText(property.initializer, tsModule, sourceFile) : "(...args: any[]) => any";
+						setBindingType(valueBindings, name, typeText);
 					}
 				}
 			}
@@ -493,8 +642,10 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 	};
 }
 function createDefineOptionsTemplateDeclarations(code, tsModule, lang) {
-	const topLevelBindings = collectTopLevelBindingNames(code, tsModule, lang);
-	const bindings = collectDefineOptionsTemplateBindings(code, tsModule, lang);
+	const scriptKind = lang === "ts" ? 3 : 1;
+	const sourceFile = tsModule.createSourceFile(`script-setup.${lang}`, code, 99, true, scriptKind);
+	const topLevelBindings = collectTopLevelBindingNames(sourceFile, tsModule);
+	const bindings = collectDefineOptionsTemplateBindings(sourceFile, tsModule);
 	const declarations = [];
 	for (const [name, typeText] of bindings.values) {
 		if (topLevelBindings.has(name)) continue;
@@ -519,43 +670,74 @@ try {
 } catch {
 	hasSchematicsTypes = false;
 }
-function parseVueSfc(content, filename = "component.vue") {
-	try {
-		return require$1("@vue/compiler-sfc").parse(content, { filename });
-	} catch {
-		return;
-	}
-}
 /**
 * Volar 语言插件：为 weapp 配置块提供类型与 schema 提示。
 */
 const plugin = (ctx) => {
-	let tsModule = ctx?.modules?.typescript;
-	if (!tsModule) try {
-		tsModule = require$1("typescript");
-	} catch {
-		tsModule = void 0;
+	const tsModule = ctx?.modules?.typescript;
+	const compilerDom = ctx?.modules?.["@vue/compiler-dom"];
+	const rawSfcStates = /* @__PURE__ */ new WeakMap();
+	const defineOptionsDeclarationsCache = /* @__PURE__ */ new Map();
+	function getDefineOptionsDeclarations(code, lang) {
+		if (!tsModule) return "";
+		const key = `${lang}\0${code}`;
+		const cached = defineOptionsDeclarationsCache.get(key);
+		if (cached !== void 0) {
+			defineOptionsDeclarationsCache.delete(key);
+			defineOptionsDeclarationsCache.set(key, cached);
+			return cached;
+		}
+		const declarations = createDefineOptionsTemplateDeclarations(code, tsModule, lang);
+		if (defineOptionsDeclarationsCache.size >= 128) {
+			const oldestKey = defineOptionsDeclarationsCache.keys().next().value;
+			if (oldestKey !== void 0) defineOptionsDeclarationsCache.delete(oldestKey);
+		}
+		defineOptionsDeclarationsCache.set(key, declarations);
+		return declarations;
+	}
+	function restoreRawSfc(parsed) {
+		const state = rawSfcStates.get(parsed);
+		const scriptSetup = parsed.descriptor.scriptSetup;
+		if (state && !state.hadScriptSetup) parsed.descriptor.scriptSetup = null;
+		else if (state && scriptSetup && state.scriptSetupContent !== void 0) {
+			scriptSetup.content = state.scriptSetupContent;
+			scriptSetup.loc.source = state.scriptSetupSource ?? state.scriptSetupContent;
+		}
+	}
+	function enhanceSfc(parsed) {
+		const descriptor = parsed.descriptor;
+		const scriptSetup = descriptor.scriptSetup;
+		rawSfcStates.set(parsed, {
+			hadScriptSetup: Boolean(scriptSetup),
+			scriptSetupContent: scriptSetup?.content,
+			scriptSetupSource: scriptSetup?.loc.source
+		});
+		const wxsModuleNames = collectWxsModuleNames(descriptor.template?.content);
+		const scriptSetupLang = resolveScriptSetupLang(scriptSetup?.lang);
+		const defineOptionsDeclarations = scriptSetup?.content.includes("defineOptions") ? getDefineOptionsDeclarations(scriptSetup.content, scriptSetupLang) : "";
+		if (!wxsModuleNames.length && !defineOptionsDeclarations) return;
+		if (scriptSetup) {
+			scriptSetup.content = appendScriptSetupDeclarations(appendWxsDeclarations(scriptSetup.content, wxsModuleNames), defineOptionsDeclarations);
+			syncScriptBlockSource(scriptSetup);
+		} else descriptor.scriptSetup = createSyntheticScriptSetup(wxsModuleNames);
 	}
 	return {
 		name,
 		version: PLUGIN_VERSION,
 		order: -1,
 		parseSFC2(fileName, languageId, content) {
-			if (languageId !== "vue") return;
-			const parsed = parseVueSfc(content, fileName);
-			if (!parsed) return;
-			const descriptor = parsed.descriptor;
-			const wxsModuleNames = collectWxsModuleNames(descriptor.template?.content);
-			const scriptSetup = descriptor.scriptSetup;
-			const scriptSetupLang = resolveScriptSetupLang(scriptSetup?.lang);
-			const defineOptionsDeclarations = scriptSetup?.content && tsModule ? createDefineOptionsTemplateDeclarations(scriptSetup.content, tsModule, scriptSetupLang) : "";
-			if (!wxsModuleNames.length && !defineOptionsDeclarations) return parsed;
-			if (descriptor.scriptSetup) {
-				descriptor.scriptSetup.content = appendScriptSetupDeclarations(appendWxsDeclarations(descriptor.scriptSetup.content, wxsModuleNames), defineOptionsDeclarations);
-				syncScriptBlockSource(descriptor.scriptSetup);
-			} else descriptor.scriptSetup = createSyntheticScriptSetup(wxsModuleNames);
+			if (languageId !== "vue" || !compilerDom) return;
+			const parsed = parseSfc(compilerDom, content, fileName);
+			enhanceSfc(parsed);
 			return parsed;
 		},
+		updateSFC(oldResult, change) {
+			restoreRawSfc(oldResult);
+			const updated = updateSfc(oldResult, change);
+			if (!updated) return;
+			enhanceSfc(updated);
+			return updated;
+		},
 		getEmbeddedCodes(_, sfc) {
 			return getEmbeddedCodesFromCustomBlocks(sfc);
 		},

package/dist/index.mjs

@@ -208,6 +208,157 @@ function resolveEmbeddedJsonBlock(fileName, sfc, embeddedCode, tsModule, hasSche
 	]);
 }
 //#endregion
+//#region src/parseSfc.ts
+function parseAttr(property, node) {
+	if (!property.value) return true;
+	const { source, start } = property.value.loc;
+	const quote = source[0];
+	const quoted = (quote === "\"" || quote === "'") && source.endsWith(quote);
+	return {
+		text: quoted ? source.slice(1, -1) : source,
+		offset: (quoted ? start.offset + 1 : start.offset) - node.loc.start.offset
+	};
+}
+function repairIncompleteTemplate(descriptor, errors) {
+	const template = descriptor.template;
+	if (!template) return;
+	for (const error of errors) {
+		if (!("code" in error) || error.code !== 24 || error.loc?.start.line !== template.loc.start.line) continue;
+		const endTagOffset = template.content.lastIndexOf("<");
+		const endTagText = template.content.slice(endTagOffset).trimEnd();
+		if ("</template>".startsWith(endTagText)) {
+			template.loc.end.offset = template.loc.start.offset + endTagOffset;
+			template.content = template.content.slice(0, endTagOffset);
+			template.loc.source = template.content;
+		}
+	}
+}
+function createBlock(compilerDom, node, source) {
+	const type = node.tag;
+	let { start, end } = node.loc;
+	let content = "";
+	if (node.children.length) {
+		start = node.children[0].loc.start;
+		end = node.children[node.children.length - 1].loc.end;
+		content = source.slice(start.offset, end.offset);
+	} else {
+		const offset = node.loc.source.indexOf("</");
+		if (offset > -1) start = {
+			line: start.line,
+			column: start.column + offset,
+			offset: start.offset + offset
+		};
+		end = { ...start };
+	}
+	const attrs = {};
+	const block = {
+		type,
+		content,
+		loc: {
+			source: content,
+			start,
+			end
+		},
+		attrs
+	};
+	for (const property of node.props) {
+		if (property.type !== compilerDom.NodeTypes.ATTRIBUTE) continue;
+		attrs[property.name] = property.value ? property.value.content || true : true;
+		if (property.name === "lang") block.lang = property.value?.content;
+		else if (property.name === "src") block.__src = parseAttr(property, node);
+		else if (type === "script") {
+			const scriptBlock = block;
+			if (property.name === "vapor") {
+				scriptBlock.setup ??= attrs[property.name];
+				scriptBlock.__generic ??= true;
+			} else if (property.name === "setup") scriptBlock.setup = attrs[property.name];
+			else if (property.name === "generic") scriptBlock.__generic = parseAttr(property, node);
+		} else if (type === "style") {
+			const styleBlock = block;
+			if (property.name === "scoped") styleBlock.scoped = true;
+			else if (property.name === "module") styleBlock.__module = parseAttr(property, node);
+		}
+	}
+	return block;
+}
+function parseSfc(compilerDom, source, filename) {
+	const errors = [];
+	const ast = compilerDom.parse(source, {
+		isNativeTag: () => true,
+		isPreTag: () => true,
+		parseMode: "sfc",
+		onError: (error) => errors.push(error),
+		comments: true
+	});
+	const descriptor = {
+		filename,
+		source,
+		comments: [],
+		template: null,
+		script: null,
+		scriptSetup: null,
+		styles: [],
+		customBlocks: [],
+		cssVars: [],
+		slotted: false,
+		shouldForceReload: () => false
+	};
+	for (const node of ast.children) {
+		if (node.type === compilerDom.NodeTypes.COMMENT) {
+			descriptor.comments.push(node.content);
+			continue;
+		}
+		if (node.type !== compilerDom.NodeTypes.ELEMENT) continue;
+		if (node.tag === "template") {
+			descriptor.template = createBlock(compilerDom, node, source);
+			continue;
+		}
+		if (node.tag === "script") {
+			const scriptBlock = createBlock(compilerDom, node, source);
+			if (scriptBlock.setup && !descriptor.scriptSetup) descriptor.scriptSetup = scriptBlock;
+			else if (!scriptBlock.setup && !descriptor.script) descriptor.script = scriptBlock;
+			continue;
+		}
+		if (node.tag === "style") {
+			descriptor.styles.push(createBlock(compilerDom, node, source));
+			continue;
+		}
+		descriptor.customBlocks.push(createBlock(compilerDom, node, source));
+	}
+	repairIncompleteTemplate(descriptor, errors);
+	return {
+		descriptor,
+		errors
+	};
+}
+function updateSfc(sfc, change) {
+	const blocks = [
+		sfc.descriptor.template,
+		sfc.descriptor.script,
+		sfc.descriptor.scriptSetup,
+		...sfc.descriptor.styles,
+		...sfc.descriptor.customBlocks
+	].filter((block) => block !== null);
+	const hitBlock = blocks.find((block) => change.start >= block.loc.start.offset && change.end <= block.loc.end.offset);
+	if (!hitBlock) return;
+	const oldContent = hitBlock.content;
+	const relativeStart = change.start - hitBlock.loc.start.offset;
+	const relativeEnd = change.end - hitBlock.loc.start.offset;
+	const newContent = oldContent.slice(0, relativeStart) + change.newText + oldContent.slice(relativeEnd);
+	const endTagRE = new RegExp(`</\\s*${hitBlock.type}\\s*>`);
+	if (endTagRE.test(oldContent) !== endTagRE.test(newContent)) return;
+	hitBlock.content = newContent;
+	hitBlock.loc.source = newContent;
+	const lengthDiff = change.newText.length - (change.end - change.start);
+	for (const block of blocks) {
+		if (block.loc.start.offset > change.end) block.loc.start.offset += lengthDiff;
+		if (block.loc.end.offset >= change.end) block.loc.end.offset += lengthDiff;
+	}
+	const source = sfc.descriptor.source;
+	sfc.descriptor.source = source.slice(0, change.start) + change.newText + source.slice(change.end);
+	return sfc;
+}
+//#endregion
 //#region src/scriptSetup.ts
 function collectWxsModuleNames(templateContent) {
 	if (!templateContent) return [];
@@ -278,9 +429,7 @@ function collectBindingNames(name, tsModule, bindings) {
 	}
 	for (const element of name.elements) if (tsModule.isBindingElement(element)) collectBindingNames(element.name, tsModule, bindings);
 }
-function collectTopLevelBindingNames(code, tsModule, lang) {
-	const scriptKind = lang === "ts" ? 3 : 1;
-	const sourceFile = tsModule.createSourceFile(`bindings.${lang}`, code, 99, true, scriptKind);
+function collectTopLevelBindingNames(sourceFile, tsModule) {
 	const bindings = /* @__PURE__ */ new Set();
 	for (const statement of sourceFile.statements) {
 		if (tsModule.isVariableStatement(statement)) {
@@ -404,9 +553,7 @@ function getFunctionLikeReturnTypeText(node, tsModule, sourceFile) {
 	const returnedExpression = tsModule.isBlock(node.body) ? findReturnedExpression(node.body, tsModule) : findReturnedExpression(node.body, tsModule);
 	return returnedExpression ? getExpressionTypeText(returnedExpression, tsModule, sourceFile) : "any";
 }
-function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
-	const scriptKind = lang === "ts" ? 3 : 1;
-	const sourceFile = tsModule.createSourceFile(`script-setup.${lang}`, code, 99, true, scriptKind);
+function collectDefineOptionsTemplateBindings(sourceFile, tsModule) {
 	const valueBindings = /* @__PURE__ */ new Map();
 	const functionBindings = /* @__PURE__ */ new Map();
 	const visit = (node) => {
@@ -421,7 +568,8 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 						if (!tsModule.isPropertyAssignment(entry) && !tsModule.isMethodDeclaration(entry) && !tsModule.isShorthandPropertyAssignment(entry)) continue;
 						const name = getPropertyNameText(entry.name, tsModule);
 						if (!name || !isIdentifierName(name)) continue;
-						setBindingType(functionBindings, name, tsModule.isPropertyAssignment(entry) && (tsModule.isArrowFunction(entry.initializer) || tsModule.isFunctionExpression(entry.initializer)) ? getFunctionSignatureTypeText(entry.initializer, sourceFile) : tsModule.isMethodDeclaration(entry) ? getFunctionSignatureTypeText(entry, sourceFile) : "(...args: any[]) => any");
+						const typeText = tsModule.isPropertyAssignment(entry) && (tsModule.isArrowFunction(entry.initializer) || tsModule.isFunctionExpression(entry.initializer)) ? getFunctionSignatureTypeText(entry.initializer, sourceFile) : tsModule.isMethodDeclaration(entry) ? getFunctionSignatureTypeText(entry, sourceFile) : "(...args: any[]) => any";
+						setBindingType(functionBindings, name, typeText);
 					}
 					continue;
 				}
@@ -455,7 +603,8 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 						if (!tsModule.isPropertyAssignment(property) && !tsModule.isMethodDeclaration(property) && !tsModule.isShorthandPropertyAssignment(property) && !tsModule.isGetAccessorDeclaration(property)) continue;
 						const name = getPropertyNameText(property.name, tsModule);
 						if (!name || !isIdentifierName(name)) continue;
-						setBindingType(valueBindings, name, tsModule.isPropertyAssignment(property) ? getExpressionTypeText(property.initializer, tsModule, sourceFile) : "(...args: any[]) => any");
+						const typeText = tsModule.isPropertyAssignment(property) ? getExpressionTypeText(property.initializer, tsModule, sourceFile) : "(...args: any[]) => any";
+						setBindingType(valueBindings, name, typeText);
 					}
 				}
 			}
@@ -469,8 +618,10 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 	};
 }
 function createDefineOptionsTemplateDeclarations(code, tsModule, lang) {
-	const topLevelBindings = collectTopLevelBindingNames(code, tsModule, lang);
-	const bindings = collectDefineOptionsTemplateBindings(code, tsModule, lang);
+	const scriptKind = lang === "ts" ? 3 : 1;
+	const sourceFile = tsModule.createSourceFile(`script-setup.${lang}`, code, 99, true, scriptKind);
+	const topLevelBindings = collectTopLevelBindingNames(sourceFile, tsModule);
+	const bindings = collectDefineOptionsTemplateBindings(sourceFile, tsModule);
 	const declarations = [];
 	for (const [name, typeText] of bindings.values) {
 		if (topLevelBindings.has(name)) continue;
@@ -495,43 +646,74 @@ try {
 } catch {
 	hasSchematicsTypes = false;
 }
-function parseVueSfc(content, filename = "component.vue") {
-	try {
-		return require("@vue/compiler-sfc").parse(content, { filename });
-	} catch {
-		return;
-	}
-}
 /**
 * Volar 语言插件：为 weapp 配置块提供类型与 schema 提示。
 */
 const plugin = (ctx) => {
-	let tsModule = ctx?.modules?.typescript;
-	if (!tsModule) try {
-		tsModule = require("typescript");
-	} catch {
-		tsModule = void 0;
+	const tsModule = ctx?.modules?.typescript;
+	const compilerDom = ctx?.modules?.["@vue/compiler-dom"];
+	const rawSfcStates = /* @__PURE__ */ new WeakMap();
+	const defineOptionsDeclarationsCache = /* @__PURE__ */ new Map();
+	function getDefineOptionsDeclarations(code, lang) {
+		if (!tsModule) return "";
+		const key = `${lang}\0${code}`;
+		const cached = defineOptionsDeclarationsCache.get(key);
+		if (cached !== void 0) {
+			defineOptionsDeclarationsCache.delete(key);
+			defineOptionsDeclarationsCache.set(key, cached);
+			return cached;
+		}
+		const declarations = createDefineOptionsTemplateDeclarations(code, tsModule, lang);
+		if (defineOptionsDeclarationsCache.size >= 128) {
+			const oldestKey = defineOptionsDeclarationsCache.keys().next().value;
+			if (oldestKey !== void 0) defineOptionsDeclarationsCache.delete(oldestKey);
+		}
+		defineOptionsDeclarationsCache.set(key, declarations);
+		return declarations;
+	}
+	function restoreRawSfc(parsed) {
+		const state = rawSfcStates.get(parsed);
+		const scriptSetup = parsed.descriptor.scriptSetup;
+		if (state && !state.hadScriptSetup) parsed.descriptor.scriptSetup = null;
+		else if (state && scriptSetup && state.scriptSetupContent !== void 0) {
+			scriptSetup.content = state.scriptSetupContent;
+			scriptSetup.loc.source = state.scriptSetupSource ?? state.scriptSetupContent;
+		}
+	}
+	function enhanceSfc(parsed) {
+		const descriptor = parsed.descriptor;
+		const scriptSetup = descriptor.scriptSetup;
+		rawSfcStates.set(parsed, {
+			hadScriptSetup: Boolean(scriptSetup),
+			scriptSetupContent: scriptSetup?.content,
+			scriptSetupSource: scriptSetup?.loc.source
+		});
+		const wxsModuleNames = collectWxsModuleNames(descriptor.template?.content);
+		const scriptSetupLang = resolveScriptSetupLang(scriptSetup?.lang);
+		const defineOptionsDeclarations = scriptSetup?.content.includes("defineOptions") ? getDefineOptionsDeclarations(scriptSetup.content, scriptSetupLang) : "";
+		if (!wxsModuleNames.length && !defineOptionsDeclarations) return;
+		if (scriptSetup) {
+			scriptSetup.content = appendScriptSetupDeclarations(appendWxsDeclarations(scriptSetup.content, wxsModuleNames), defineOptionsDeclarations);
+			syncScriptBlockSource(scriptSetup);
+		} else descriptor.scriptSetup = createSyntheticScriptSetup(wxsModuleNames);
 	}
 	return {
 		name,
 		version: PLUGIN_VERSION,
 		order: -1,
 		parseSFC2(fileName, languageId, content) {
-			if (languageId !== "vue") return;
-			const parsed = parseVueSfc(content, fileName);
-			if (!parsed) return;
-			const descriptor = parsed.descriptor;
-			const wxsModuleNames = collectWxsModuleNames(descriptor.template?.content);
-			const scriptSetup = descriptor.scriptSetup;
-			const scriptSetupLang = resolveScriptSetupLang(scriptSetup?.lang);
-			const defineOptionsDeclarations = scriptSetup?.content && tsModule ? createDefineOptionsTemplateDeclarations(scriptSetup.content, tsModule, scriptSetupLang) : "";
-			if (!wxsModuleNames.length && !defineOptionsDeclarations) return parsed;
-			if (descriptor.scriptSetup) {
-				descriptor.scriptSetup.content = appendScriptSetupDeclarations(appendWxsDeclarations(descriptor.scriptSetup.content, wxsModuleNames), defineOptionsDeclarations);
-				syncScriptBlockSource(descriptor.scriptSetup);
-			} else descriptor.scriptSetup = createSyntheticScriptSetup(wxsModuleNames);
+			if (languageId !== "vue" || !compilerDom) return;
+			const parsed = parseSfc(compilerDom, content, fileName);
+			enhanceSfc(parsed);
 			return parsed;
 		},
+		updateSFC(oldResult, change) {
+			restoreRawSfc(oldResult);
+			const updated = updateSfc(oldResult, change);
+			if (!updated) return;
+			enhanceSfc(updated);
+			return updated;
+		},
 		getEmbeddedCodes(_, sfc) {
 			return getEmbeddedCodesFromCustomBlocks(sfc);
 		},

package/dist/parseSfc.d.ts

@@ -0,0 +1,9 @@
+import type * as CompilerDOM from '@vue/compiler-dom';
+import type { SFCParseResult } from '@vue/language-core';
+export interface SfcTextChange {
+    start: number;
+    end: number;
+    newText: string;
+}
+export declare function parseSfc(compilerDom: typeof CompilerDOM, source: string, filename: string): SFCParseResult;
+export declare function updateSfc(sfc: SFCParseResult, change: SfcTextChange): SFCParseResult | undefined;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@weapp-vite/volar",
-  "version": "2.1.0",
+  "version": "2.1.1",
   "description": "Volar plugin for weapp-vite - Provides IntelliSense and type checking for WeChat mini-program config blocks",
   "author": "ice breaker <1324318532@qq.com>",
   "license": "MIT",
@@ -43,6 +43,7 @@
     "@weapp-core/schematics": "6.0.4"
   },
   "scripts": {
+    "benchmark": "node --import tsx benchmark/sfc.ts",
     "dev": "tsdown -w --sourcemap",
     "build": "tsdown --no-dts && tsc -p tsconfig.build.json",
     "test": "vitest run",