@weapp-vite/volar 2.0.8 → 2.1.1

package/README.md

@@ -1,58 +1,72 @@
 # @weapp-vite/volar
 
-> 为 weapp-vite 提供 VSCode 智能提示和类型检查的 Volar 插件
+## 1. 简介
 
-## 安装
+`@weapp-vite/volar` 是 `weapp-vite` 的 Volar 语言服务插件，用来给小程序 Vue SFC 中的自定义块和相关宏提供更准确的类型、补全和诊断能力。
 
-此包已作为 `weapp-vite` 的依赖自动安装，**无需单独安装**。
+它主要覆盖：
 
-如果你确实需要单独安装：
+- `<json>` 自定义块的 schema 与类型提示
+- `definePageJson` / `defineComponentJson` 等配置宏配套体验
+- `wxs` 模块声明补全
+- `script setup` 中为 weapp-vite / wevu 注入的额外声明
+
+## 2. 安装方式
+
+大多数情况下不需要单独安装。
+
+`weapp-vite` 已经依赖并自动启用这个插件，项目里通常只需要按文档配置 `tsconfig.json` / `jsconfig.json` 即可。
+
+如果你要在非标准工程里单独集成，可以安装：
 
 ```bash
-npm install -D @weapp-vite/volar
-# or
-pnpm add -D @weapp-vite/volar
-# or
-yarn add -D @weapp-vite/volar
+pnpm add -D @weapp-vite/volar @weapp-core/schematics
 ```
 
-## 使用方法
-
-**推荐方式：** 通过 weapp-vite 使用，插件会自动启用。
+## 3. 常见配置
 
-详细使用文档请参考：[weapp-vite Volar 文档](../weapp-vite/docs/volar.md)
+```json
+{
+  "vueCompilerOptions": {
+    "plugins": ["weapp-vite/volar"]
+  }
+}
+```
 
-## 功能特性
+> **注意**：更完整的编辑器接入方式，请优先参考 `weapp-vite` 主包文档中的 Volar 章节。
 
-- ✅ **配置文件智能提示** - 为 `<json>` 代码块提供完整的类型检查和自动补全
-- ✅ **JSON Schema 支持** - 支持 JSON Schema 验证和自动补全
-- ✅ **TypeScript 类型检查** - 利用 TypeScript 类型系统确保配置正确性
-- ✅ **自动推断配置类型** - 根据文件路径自动推断是 App/Page/Component 配置
-- ✅ **多模式支持** - 支持 JSON 模式和 TypeScript 模式
+## 4. 提供的能力
 
-## 开发
+| 能力                    | 说明                                                   |
+| ----------------------- | ------------------------------------------------------ |
+| `<json>` 块诊断         | 根据 App / Page / Component 语义提供 schema 与类型校验 |
+| `script setup` 声明增强 | 自动补齐 weapp 相关额外声明                            |
+| `wxs` 模块名收集        | 让模板与脚本中的 `wxs` 使用获得更稳定的类型体验        |
+| 自定义块嵌入代码        | 让 Volar 能识别并处理额外代码片段                      |
 
-```bash
-# 安装依赖
-pnpm install
+## 5. 适用场景
 
-# 开发模式
-pnpm dev
+- 在 `.vue` 文件里直接编写小程序页面 / 组件配置
+- 使用 `definePageJson`、`defineComponentJson` 等宏
+- 需要让 VS Code / Volar 正确理解 weapp-vite 的自定义块与扩展语义
 
-# 构建
-pnpm build
+## 6. 本地开发
 
-# 测试
-pnpm test
+```bash
+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。对比修改前构建时可传入基线插件：
 
-- [weapp-vite 文档](../weapp-vite)
-- [weapp-vite Volar 使用指南](../weapp-vite/docs/volar.md)
-- [Vue 3 文档](https://vuejs.org/)
-- [微信小程序官方文档](https://developers.weixin.qq.com/miniprogram/dev/framework/)
+```bash
+pnpm --filter @weapp-vite/volar benchmark -- --baseline /path/to/baseline/dist/index.mjs
+```
 
-## License
+## 7. 相关链接
 
-MIT
+- 详细用法：[./USAGE.md](./USAGE.md)
+- 主包文档：[../weapp-vite/docs/volar.md](../weapp-vite/docs/volar.md)
+- `weapp-vite`：[../weapp-vite/README.md](../weapp-vite/README.md)

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)) {
@@ -329,15 +478,18 @@ function unwrapParenthesizedExpression(node, tsModule) {
 	while (tsModule.isParenthesizedExpression(current)) current = current.expression;
 	return current;
 }
-function findReturnedObjectLiteral(node, tsModule) {
+function findReturnedExpression(node, tsModule) {
 	if (tsModule.isObjectLiteralExpression(node)) return node;
-	if (!tsModule.isBlock(node)) return;
+	if (!tsModule.isBlock(node)) return node;
 	for (const statement of node.statements) {
 		if (!tsModule.isReturnStatement(statement) || !statement.expression) continue;
-		const expression = unwrapParenthesizedExpression(statement.expression, tsModule);
-		if (tsModule.isObjectLiteralExpression(expression)) return expression;
+		return unwrapParenthesizedExpression(statement.expression, tsModule);
 	}
 }
+function findReturnedObjectLiteral(node, tsModule) {
+	const expression = findReturnedExpression(node, tsModule);
+	return expression && tsModule.isObjectLiteralExpression(expression) ? expression : void 0;
+}
 function extractDefineOptionsObjectLiteral(node, tsModule) {
 	if (!tsModule.isIdentifier(node.expression) || node.expression.text !== "defineOptions") return;
 	const [firstArg] = node.arguments;
@@ -346,20 +498,88 @@ function extractDefineOptionsObjectLiteral(node, tsModule) {
 	if (tsModule.isObjectLiteralExpression(arg)) return arg;
 	if (tsModule.isArrowFunction(arg) || tsModule.isFunctionExpression(arg)) return findReturnedObjectLiteral(arg.body, tsModule);
 }
-function collectObjectLiteralKeys(node, tsModule) {
-	const keys = [];
+function setBindingType(map, name, typeText) {
+	if (!map.has(name)) map.set(name, typeText);
+}
+function getRuntimeConstructorType(node, tsModule) {
+	if (tsModule.isIdentifier(node)) {
+		if (node.text === "String") return "string";
+		if (node.text === "Number") return "number";
+		if (node.text === "Boolean") return "boolean";
+		if (node.text === "Array") return "any[]";
+		if (node.text === "Object") return "Record<string, any>";
+	}
+	if (tsModule.isArrayLiteralExpression(node)) {
+		const types = [...new Set(node.elements.map((element) => {
+			if (tsModule.isSpreadElement(element)) return null;
+			return getRuntimeConstructorType(element, tsModule);
+		}).filter((typeText) => Boolean(typeText)))];
+		return types.length > 0 ? types.join(" | ") : null;
+	}
+	return null;
+}
+function getFunctionSignatureTypeText(node, sourceFile) {
+	if (!(node.parameters.some((parameter) => Boolean(parameter.type)) || Boolean(node.type))) return "(...args: any[]) => any";
+	return `(${node.parameters.map((parameter) => parameter.getText(sourceFile)).join(", ")}) => ${node.type?.getText(sourceFile) ?? "any"}`;
+}
+function getMiniProgramPropertyType(node, tsModule) {
+	const runtimeType = getRuntimeConstructorType(node, tsModule);
+	if (runtimeType) return runtimeType;
+	if (!tsModule.isObjectLiteralExpression(node)) return "any";
 	for (const property of node.properties) {
-		if (!tsModule.isPropertyAssignment(property) && !tsModule.isMethodDeclaration(property) && !tsModule.isShorthandPropertyAssignment(property) && !tsModule.isGetAccessorDeclaration(property)) continue;
-		const key = getPropertyNameText(property.name, tsModule);
-		if (key && isIdentifierName(key)) keys.push(key);
+		if (!tsModule.isPropertyAssignment(property)) continue;
+		if (getPropertyNameText(property.name, tsModule) !== "type") continue;
+		return getRuntimeConstructorType(unwrapParenthesizedExpression(property.initializer, tsModule), tsModule) ?? "any";
 	}
-	return keys;
+	return "any";
 }
-function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
-	const scriptKind = lang === "ts" ? 3 : 1;
-	const sourceFile = tsModule.createSourceFile(`script-setup.${lang}`, code, 99, true, scriptKind);
-	const valueBindings = /* @__PURE__ */ new Set();
-	const functionBindings = /* @__PURE__ */ new Set();
+function getExpressionTypeText(node, tsModule, sourceFile) {
+	const expression = unwrapParenthesizedExpression(node, tsModule);
+	if (tsModule.isAsExpression(expression) || tsModule.isTypeAssertionExpression(expression)) return expression.type.getText(sourceFile);
+	if (tsModule.isStringLiteral(expression) || tsModule.isNoSubstitutionTemplateLiteral(expression)) return "string";
+	if (tsModule.isNumericLiteral(expression)) return "number";
+	if (expression.kind === tsModule.SyntaxKind.TrueKeyword || expression.kind === tsModule.SyntaxKind.FalseKeyword) return "boolean";
+	if (tsModule.isPrefixUnaryExpression(expression) && tsModule.isNumericLiteral(expression.operand)) return "number";
+	if (tsModule.isArrayLiteralExpression(expression)) {
+		const itemTypes = [...new Set(expression.elements.map((element) => {
+			if (tsModule.isSpreadElement(element)) return null;
+			return getExpressionTypeText(element, tsModule, sourceFile);
+		}).filter((typeText) => Boolean(typeText) && typeText !== "undefined"))];
+		if (itemTypes.length === 0) return "any[]";
+		return itemTypes.length === 1 ? `${itemTypes[0]}[]` : `(${itemTypes.join(" | ")})[]`;
+	}
+	if (tsModule.isObjectLiteralExpression(expression)) {
+		const fields = [];
+		for (const property of expression.properties) {
+			if (tsModule.isPropertyAssignment(property)) {
+				const name = getPropertyNameText(property.name, tsModule);
+				if (name) fields.push(`${name}: ${getExpressionTypeText(property.initializer, tsModule, sourceFile)}`);
+				continue;
+			}
+			if (tsModule.isShorthandPropertyAssignment(property)) {
+				fields.push(`${property.name.text}: any`);
+				continue;
+			}
+			if (tsModule.isMethodDeclaration(property) || tsModule.isGetAccessorDeclaration(property)) {
+				const name = getPropertyNameText(property.name, tsModule);
+				if (name) fields.push(`${name}: (...args: any[]) => any`);
+			}
+		}
+		return fields.length > 0 ? `{ ${fields.join("; ")} }` : "Record<string, any>";
+	}
+	if (tsModule.isArrowFunction(expression) || tsModule.isFunctionExpression(expression)) return "(...args: any[]) => any";
+	if (tsModule.isIdentifier(expression) && expression.text === "undefined") return "undefined";
+	return "any";
+}
+function getFunctionLikeReturnTypeText(node, tsModule, sourceFile) {
+	if (node.type) return node.type.getText(sourceFile);
+	if (!node.body) return "any";
+	const returnedExpression = tsModule.isBlock(node.body) ? findReturnedExpression(node.body, tsModule) : findReturnedExpression(node.body, tsModule);
+	return returnedExpression ? getExpressionTypeText(returnedExpression, tsModule, sourceFile) : "any";
+}
+function collectDefineOptionsTemplateBindings(sourceFile, tsModule) {
+	const valueBindings = /* @__PURE__ */ new Map();
+	const functionBindings = /* @__PURE__ */ new Map();
 	const visit = (node) => {
 		if (tsModule.isCallExpression(node)) {
 			const optionsObject = extractDefineOptionsObjectLiteral(node, tsModule);
@@ -368,12 +588,31 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 				if (!sectionName) continue;
 				if (sectionName === "methods") {
 					const methodsObject = tsModule.isPropertyAssignment(property) ? unwrapParenthesizedExpression(property.initializer, tsModule) : void 0;
-					if (methodsObject && tsModule.isObjectLiteralExpression(methodsObject)) for (const name of collectObjectLiteralKeys(methodsObject, tsModule)) functionBindings.add(name);
+					if (methodsObject && tsModule.isObjectLiteralExpression(methodsObject)) for (const entry of methodsObject.properties) {
+						if (!tsModule.isPropertyAssignment(entry) && !tsModule.isMethodDeclaration(entry) && !tsModule.isShorthandPropertyAssignment(entry)) continue;
+						const name = getPropertyNameText(entry.name, tsModule);
+						if (!name || !isIdentifierName(name)) continue;
+						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;
 				}
 				if (sectionName === "properties" || sectionName === "computed") {
 					const objectValue = tsModule.isPropertyAssignment(property) ? unwrapParenthesizedExpression(property.initializer, tsModule) : void 0;
-					if (objectValue && tsModule.isObjectLiteralExpression(objectValue)) for (const name of collectObjectLiteralKeys(objectValue, tsModule)) valueBindings.add(name);
+					if (objectValue && tsModule.isObjectLiteralExpression(objectValue)) for (const entry of objectValue.properties) {
+						if (!tsModule.isPropertyAssignment(entry) && !tsModule.isMethodDeclaration(entry) && !tsModule.isShorthandPropertyAssignment(entry) && !tsModule.isGetAccessorDeclaration(entry)) continue;
+						const name = getPropertyNameText(entry.name, tsModule);
+						if (!name || !isIdentifierName(name)) continue;
+						let typeText = "any";
+						if (sectionName === "properties" && tsModule.isPropertyAssignment(entry)) typeText = getMiniProgramPropertyType(unwrapParenthesizedExpression(entry.initializer, tsModule), tsModule);
+						else if (sectionName === "computed") {
+							if (tsModule.isPropertyAssignment(entry)) {
+								const initializer = unwrapParenthesizedExpression(entry.initializer, tsModule);
+								if (tsModule.isArrowFunction(initializer) || tsModule.isFunctionExpression(initializer)) typeText = getFunctionLikeReturnTypeText(initializer, tsModule, sourceFile);
+							} else if (tsModule.isMethodDeclaration(entry) || tsModule.isGetAccessorDeclaration(entry)) typeText = getFunctionLikeReturnTypeText(entry, tsModule, sourceFile);
+						}
+						setBindingType(valueBindings, name, typeText);
+					}
 					continue;
 				}
 				if (sectionName === "data") {
@@ -381,9 +620,16 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 					if (tsModule.isMethodDeclaration(property)) dataObject = findReturnedObjectLiteral(property.body ?? tsModule.factory.createBlock([], false), tsModule);
 					else if (tsModule.isPropertyAssignment(property)) {
 						const initializer = unwrapParenthesizedExpression(property.initializer, tsModule);
-						if (tsModule.isArrowFunction(initializer) || tsModule.isFunctionExpression(initializer)) dataObject = findReturnedObjectLiteral(initializer.body, tsModule);
+						if (tsModule.isObjectLiteralExpression(initializer)) dataObject = initializer;
+						else if (tsModule.isArrowFunction(initializer) || tsModule.isFunctionExpression(initializer)) dataObject = findReturnedObjectLiteral(initializer.body, tsModule);
+					}
+					if (dataObject) for (const property of dataObject.properties) {
+						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;
+						const typeText = tsModule.isPropertyAssignment(property) ? getExpressionTypeText(property.initializer, tsModule, sourceFile) : "(...args: any[]) => any";
+						setBindingType(valueBindings, name, typeText);
 					}
-					if (dataObject) for (const name of collectObjectLiteralKeys(dataObject, tsModule)) valueBindings.add(name);
 				}
 			}
 		}
@@ -396,16 +642,18 @@ 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 of bindings.values) {
+	for (const [name, typeText] of bindings.values) {
 		if (topLevelBindings.has(name)) continue;
-		declarations.push(`const ${name}: any = null as any`);
+		declarations.push(`const ${name}: ${typeText} = null as any`);
 	}
-	for (const name of bindings.functions) {
+	for (const [name, typeText] of bindings.functions) {
 		if (topLevelBindings.has(name)) continue;
-		declarations.push(`const ${name}: (...args: any[]) => any = null as any`);
+		declarations.push(`const ${name}: ${typeText} = null as any`);
 	}
 	return declarations.join("\n");
 }
@@ -422,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)) {
@@ -305,15 +454,18 @@ function unwrapParenthesizedExpression(node, tsModule) {
 	while (tsModule.isParenthesizedExpression(current)) current = current.expression;
 	return current;
 }
-function findReturnedObjectLiteral(node, tsModule) {
+function findReturnedExpression(node, tsModule) {
 	if (tsModule.isObjectLiteralExpression(node)) return node;
-	if (!tsModule.isBlock(node)) return;
+	if (!tsModule.isBlock(node)) return node;
 	for (const statement of node.statements) {
 		if (!tsModule.isReturnStatement(statement) || !statement.expression) continue;
-		const expression = unwrapParenthesizedExpression(statement.expression, tsModule);
-		if (tsModule.isObjectLiteralExpression(expression)) return expression;
+		return unwrapParenthesizedExpression(statement.expression, tsModule);
 	}
 }
+function findReturnedObjectLiteral(node, tsModule) {
+	const expression = findReturnedExpression(node, tsModule);
+	return expression && tsModule.isObjectLiteralExpression(expression) ? expression : void 0;
+}
 function extractDefineOptionsObjectLiteral(node, tsModule) {
 	if (!tsModule.isIdentifier(node.expression) || node.expression.text !== "defineOptions") return;
 	const [firstArg] = node.arguments;
@@ -322,20 +474,88 @@ function extractDefineOptionsObjectLiteral(node, tsModule) {
 	if (tsModule.isObjectLiteralExpression(arg)) return arg;
 	if (tsModule.isArrowFunction(arg) || tsModule.isFunctionExpression(arg)) return findReturnedObjectLiteral(arg.body, tsModule);
 }
-function collectObjectLiteralKeys(node, tsModule) {
-	const keys = [];
+function setBindingType(map, name, typeText) {
+	if (!map.has(name)) map.set(name, typeText);
+}
+function getRuntimeConstructorType(node, tsModule) {
+	if (tsModule.isIdentifier(node)) {
+		if (node.text === "String") return "string";
+		if (node.text === "Number") return "number";
+		if (node.text === "Boolean") return "boolean";
+		if (node.text === "Array") return "any[]";
+		if (node.text === "Object") return "Record<string, any>";
+	}
+	if (tsModule.isArrayLiteralExpression(node)) {
+		const types = [...new Set(node.elements.map((element) => {
+			if (tsModule.isSpreadElement(element)) return null;
+			return getRuntimeConstructorType(element, tsModule);
+		}).filter((typeText) => Boolean(typeText)))];
+		return types.length > 0 ? types.join(" | ") : null;
+	}
+	return null;
+}
+function getFunctionSignatureTypeText(node, sourceFile) {
+	if (!(node.parameters.some((parameter) => Boolean(parameter.type)) || Boolean(node.type))) return "(...args: any[]) => any";
+	return `(${node.parameters.map((parameter) => parameter.getText(sourceFile)).join(", ")}) => ${node.type?.getText(sourceFile) ?? "any"}`;
+}
+function getMiniProgramPropertyType(node, tsModule) {
+	const runtimeType = getRuntimeConstructorType(node, tsModule);
+	if (runtimeType) return runtimeType;
+	if (!tsModule.isObjectLiteralExpression(node)) return "any";
 	for (const property of node.properties) {
-		if (!tsModule.isPropertyAssignment(property) && !tsModule.isMethodDeclaration(property) && !tsModule.isShorthandPropertyAssignment(property) && !tsModule.isGetAccessorDeclaration(property)) continue;
-		const key = getPropertyNameText(property.name, tsModule);
-		if (key && isIdentifierName(key)) keys.push(key);
+		if (!tsModule.isPropertyAssignment(property)) continue;
+		if (getPropertyNameText(property.name, tsModule) !== "type") continue;
+		return getRuntimeConstructorType(unwrapParenthesizedExpression(property.initializer, tsModule), tsModule) ?? "any";
 	}
-	return keys;
+	return "any";
 }
-function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
-	const scriptKind = lang === "ts" ? 3 : 1;
-	const sourceFile = tsModule.createSourceFile(`script-setup.${lang}`, code, 99, true, scriptKind);
-	const valueBindings = /* @__PURE__ */ new Set();
-	const functionBindings = /* @__PURE__ */ new Set();
+function getExpressionTypeText(node, tsModule, sourceFile) {
+	const expression = unwrapParenthesizedExpression(node, tsModule);
+	if (tsModule.isAsExpression(expression) || tsModule.isTypeAssertionExpression(expression)) return expression.type.getText(sourceFile);
+	if (tsModule.isStringLiteral(expression) || tsModule.isNoSubstitutionTemplateLiteral(expression)) return "string";
+	if (tsModule.isNumericLiteral(expression)) return "number";
+	if (expression.kind === tsModule.SyntaxKind.TrueKeyword || expression.kind === tsModule.SyntaxKind.FalseKeyword) return "boolean";
+	if (tsModule.isPrefixUnaryExpression(expression) && tsModule.isNumericLiteral(expression.operand)) return "number";
+	if (tsModule.isArrayLiteralExpression(expression)) {
+		const itemTypes = [...new Set(expression.elements.map((element) => {
+			if (tsModule.isSpreadElement(element)) return null;
+			return getExpressionTypeText(element, tsModule, sourceFile);
+		}).filter((typeText) => Boolean(typeText) && typeText !== "undefined"))];
+		if (itemTypes.length === 0) return "any[]";
+		return itemTypes.length === 1 ? `${itemTypes[0]}[]` : `(${itemTypes.join(" | ")})[]`;
+	}
+	if (tsModule.isObjectLiteralExpression(expression)) {
+		const fields = [];
+		for (const property of expression.properties) {
+			if (tsModule.isPropertyAssignment(property)) {
+				const name = getPropertyNameText(property.name, tsModule);
+				if (name) fields.push(`${name}: ${getExpressionTypeText(property.initializer, tsModule, sourceFile)}`);
+				continue;
+			}
+			if (tsModule.isShorthandPropertyAssignment(property)) {
+				fields.push(`${property.name.text}: any`);
+				continue;
+			}
+			if (tsModule.isMethodDeclaration(property) || tsModule.isGetAccessorDeclaration(property)) {
+				const name = getPropertyNameText(property.name, tsModule);
+				if (name) fields.push(`${name}: (...args: any[]) => any`);
+			}
+		}
+		return fields.length > 0 ? `{ ${fields.join("; ")} }` : "Record<string, any>";
+	}
+	if (tsModule.isArrowFunction(expression) || tsModule.isFunctionExpression(expression)) return "(...args: any[]) => any";
+	if (tsModule.isIdentifier(expression) && expression.text === "undefined") return "undefined";
+	return "any";
+}
+function getFunctionLikeReturnTypeText(node, tsModule, sourceFile) {
+	if (node.type) return node.type.getText(sourceFile);
+	if (!node.body) return "any";
+	const returnedExpression = tsModule.isBlock(node.body) ? findReturnedExpression(node.body, tsModule) : findReturnedExpression(node.body, tsModule);
+	return returnedExpression ? getExpressionTypeText(returnedExpression, tsModule, sourceFile) : "any";
+}
+function collectDefineOptionsTemplateBindings(sourceFile, tsModule) {
+	const valueBindings = /* @__PURE__ */ new Map();
+	const functionBindings = /* @__PURE__ */ new Map();
 	const visit = (node) => {
 		if (tsModule.isCallExpression(node)) {
 			const optionsObject = extractDefineOptionsObjectLiteral(node, tsModule);
@@ -344,12 +564,31 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 				if (!sectionName) continue;
 				if (sectionName === "methods") {
 					const methodsObject = tsModule.isPropertyAssignment(property) ? unwrapParenthesizedExpression(property.initializer, tsModule) : void 0;
-					if (methodsObject && tsModule.isObjectLiteralExpression(methodsObject)) for (const name of collectObjectLiteralKeys(methodsObject, tsModule)) functionBindings.add(name);
+					if (methodsObject && tsModule.isObjectLiteralExpression(methodsObject)) for (const entry of methodsObject.properties) {
+						if (!tsModule.isPropertyAssignment(entry) && !tsModule.isMethodDeclaration(entry) && !tsModule.isShorthandPropertyAssignment(entry)) continue;
+						const name = getPropertyNameText(entry.name, tsModule);
+						if (!name || !isIdentifierName(name)) continue;
+						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;
 				}
 				if (sectionName === "properties" || sectionName === "computed") {
 					const objectValue = tsModule.isPropertyAssignment(property) ? unwrapParenthesizedExpression(property.initializer, tsModule) : void 0;
-					if (objectValue && tsModule.isObjectLiteralExpression(objectValue)) for (const name of collectObjectLiteralKeys(objectValue, tsModule)) valueBindings.add(name);
+					if (objectValue && tsModule.isObjectLiteralExpression(objectValue)) for (const entry of objectValue.properties) {
+						if (!tsModule.isPropertyAssignment(entry) && !tsModule.isMethodDeclaration(entry) && !tsModule.isShorthandPropertyAssignment(entry) && !tsModule.isGetAccessorDeclaration(entry)) continue;
+						const name = getPropertyNameText(entry.name, tsModule);
+						if (!name || !isIdentifierName(name)) continue;
+						let typeText = "any";
+						if (sectionName === "properties" && tsModule.isPropertyAssignment(entry)) typeText = getMiniProgramPropertyType(unwrapParenthesizedExpression(entry.initializer, tsModule), tsModule);
+						else if (sectionName === "computed") {
+							if (tsModule.isPropertyAssignment(entry)) {
+								const initializer = unwrapParenthesizedExpression(entry.initializer, tsModule);
+								if (tsModule.isArrowFunction(initializer) || tsModule.isFunctionExpression(initializer)) typeText = getFunctionLikeReturnTypeText(initializer, tsModule, sourceFile);
+							} else if (tsModule.isMethodDeclaration(entry) || tsModule.isGetAccessorDeclaration(entry)) typeText = getFunctionLikeReturnTypeText(entry, tsModule, sourceFile);
+						}
+						setBindingType(valueBindings, name, typeText);
+					}
 					continue;
 				}
 				if (sectionName === "data") {
@@ -357,9 +596,16 @@ function collectDefineOptionsTemplateBindings(code, tsModule, lang) {
 					if (tsModule.isMethodDeclaration(property)) dataObject = findReturnedObjectLiteral(property.body ?? tsModule.factory.createBlock([], false), tsModule);
 					else if (tsModule.isPropertyAssignment(property)) {
 						const initializer = unwrapParenthesizedExpression(property.initializer, tsModule);
-						if (tsModule.isArrowFunction(initializer) || tsModule.isFunctionExpression(initializer)) dataObject = findReturnedObjectLiteral(initializer.body, tsModule);
+						if (tsModule.isObjectLiteralExpression(initializer)) dataObject = initializer;
+						else if (tsModule.isArrowFunction(initializer) || tsModule.isFunctionExpression(initializer)) dataObject = findReturnedObjectLiteral(initializer.body, tsModule);
+					}
+					if (dataObject) for (const property of dataObject.properties) {
+						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;
+						const typeText = tsModule.isPropertyAssignment(property) ? getExpressionTypeText(property.initializer, tsModule, sourceFile) : "(...args: any[]) => any";
+						setBindingType(valueBindings, name, typeText);
 					}
-					if (dataObject) for (const name of collectObjectLiteralKeys(dataObject, tsModule)) valueBindings.add(name);
 				}
 			}
 		}
@@ -372,16 +618,18 @@ 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 of bindings.values) {
+	for (const [name, typeText] of bindings.values) {
 		if (topLevelBindings.has(name)) continue;
-		declarations.push(`const ${name}: any = null as any`);
+		declarations.push(`const ${name}: ${typeText} = null as any`);
 	}
-	for (const name of bindings.functions) {
+	for (const [name, typeText] of bindings.functions) {
 		if (topLevelBindings.has(name)) continue;
-		declarations.push(`const ${name}: (...args: any[]) => any = null as any`);
+		declarations.push(`const ${name}: ${typeText} = null as any`);
 	}
 	return declarations.join("\n");
 }
@@ -398,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.0.8",
+  "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",