vite-plugin-uni-inject 0.4.0 → 0.6.0

package/README.md

@@ -1,41 +1,19 @@
-# vite-plugin-uni-inject
+# Vite-plugin-uni-inject
 
-利用 Vite 插件机制，实现自动注入代码，解放你的双手。
+一个轻量级的 uniapp 注入增强插件，旨在为 uniapp 项目提供更舒适的开发体验。
 
-## 功能特点
+## 文档
 
-- 干净的注入任何代码，没有多余的结构。
-- 支持注入 `<page-meta/>` 这种只能放在页面第一个位置的标签节点。
-- 支持基于文件路由的 `page.json` 自动生成并注入，无需手动维护页面表。
+请阅读 [文档](https://kriac.github.io/vite-plugin-uni-inject) 了解更多使用细节。
 
-## 如何使用
+## 使用
 
-### 安装依赖
+如果你在开发过程中遇到了什么问题，或者有更好的建议，欢迎提交 issue 与我们讨论。
 
-```bash
-pnpm i -D vite-plugin-uni-inject
-```
+## 贡献
 
-### vite.config.ts
+首先感谢你考虑为本项目做出贡献！我们欢迎社区成员的贡献，以帮助改进和扩展本项目。
 
-```ts
-import { defineConfig } from "vite";
-import { uniAutoPages, uniInject } from "vite-plugin-uni-inject";
-import uni from "@dcloudio/vite-plugin-uni";
+## 执照
 
-// 如果有重写 page.json 文件的插件，请确保写在 uniInject 之前
-export default defineConfig(() => {
-  return {
-    plugins: [uniAutoPages(), uniInject(), uni()],
-  };
-});
-```
-
-### 独立插件说明
-
-- `uniAutoPages(options)`：负责扫描文件路由并补全 `src/pages.json` 。
-- `uniInject(options)`：负责注入 `App.inject.vue`（或自定义文件）到页面文件。
-
-## 报告错误
-
-欢迎提交 issue 与我们讨论。
+本项目采用 MIT 许可证，详细内容请见 [LICENSE](LICENSE) 文件。

package/dist/index.cjs

@@ -94,14 +94,18 @@ function uniInject(opts) {
 				});
 				let nextScriptCode = scriptCode;
 				if (imports.length) {
-					const ranges = [...imports].sort((a, b) => b.start - a.start);
+					const ranges = [...imports].sort((a, b) => {
+						return b.start - a.start;
+					});
 					for (const { start: rangeStart, end: rangeEnd } of ranges) {
 						let removeEnd = rangeEnd;
 						if (nextScriptCode.startsWith("\r\n", removeEnd)) removeEnd += 2;
 						else if (nextScriptCode[removeEnd] === "\n") removeEnd += 1;
 						nextScriptCode = nextScriptCode.slice(0, rangeStart) + nextScriptCode.slice(removeEnd);
 					}
-					nextScriptCode = `\n${imports.map((item) => item.code).join("\n")}${nextScriptCode}`;
+					nextScriptCode = `\n${imports.map((item) => {
+						return item.code;
+					}).join("\n")}${nextScriptCode}`;
 				}
 				newCode = newCode.slice(0, start) + nextScriptCode + newCode.slice(end);
 			}
@@ -111,6 +115,10 @@ function uniInject(opts) {
 }
 //#endregion
 //#region src/modules/auto-pages/index.ts
+function readPagesJson(pagesJsonPath) {
+	if (!fs.default.existsSync(pagesJsonPath)) return null;
+	return JSON.parse(fs.default.readFileSync(pagesJsonPath, "utf-8"));
+}
 function toPosixPath(value) {
 	return value.replace(/\\/g, "/");
 }
@@ -130,55 +138,97 @@ function walkFiles(dirPath, filePaths = []) {
 	}
 	return filePaths;
 }
+function analyzeVueContent(content, filePath) {
+	if (!content.includes("definePage")) return null;
+	const stripRanges = [];
+	const { descriptor } = (0, _vue_compiler_sfc.parse)(content);
+	const blocks = [descriptor.scriptSetup, descriptor.script].filter((b) => {
+		return Boolean(b);
+	});
+	let config = null;
+	for (const block of blocks) {
+		const src = block.content;
+		if (!src.includes("definePage")) continue;
+		let ast;
+		try {
+			ast = (0, _babel_parser.parse)(src, {
+				sourceType: "module",
+				plugins: ["typescript"]
+			});
+		} catch {
+			continue;
+		}
+		const base = block.loc.start.offset;
+		for (const node of ast.program.body) {
+			if (typeof node.start !== "number" || typeof node.end !== "number") continue;
+			if (node.type === "ImportDeclaration") {
+				if (node.specifiers.some((s) => {
+					return s.type === "ImportSpecifier" && s.imported.type === "Identifier" && s.imported.name === "definePage";
+				})) stripRanges.push({
+					start: base + node.start,
+					end: base + node.end
+				});
+				continue;
+			}
+			if (node.type !== "ExpressionStatement" || node.expression.type !== "CallExpression" || node.expression.callee.type !== "Identifier" || node.expression.callee.name !== "definePage") continue;
+			stripRanges.push({
+				start: base + node.start,
+				end: base + node.end
+			});
+			const arg = node.expression.arguments[0];
+			if (config || !arg || typeof arg.start !== "number" || typeof arg.end !== "number") continue;
+			const argSrc = src.slice(arg.start, arg.end);
+			try {
+				config = new Function(`return (${argSrc})`)();
+			} catch (err) {
+				console.warn(`解析失败(${filePath}): ${err.message}`);
+			}
+		}
+	}
+	if (!stripRanges.length && !config) return null;
+	return {
+		config,
+		stripRanges
+	};
+}
 function collectFileRoutes(srcRoot, routeDirs) {
 	const routeSet = /* @__PURE__ */ new Set();
-	routeDirs.forEach((dir) => {
-		walkFiles(path.default.resolve(srcRoot, dir)).forEach((filePath) => {
-			if (!filePath.endsWith(".vue")) return;
-			const noExt = path.default.relative(srcRoot, filePath).slice(0, -4);
-			routeSet.add(toPosixPath(noExt));
-		});
-	});
-	return Array.from(routeSet).sort();
+	const analysisMap = /* @__PURE__ */ new Map();
+	for (const dir of routeDirs) for (const filePath of walkFiles(path.default.resolve(srcRoot, dir))) {
+		if (!filePath.endsWith(".vue")) continue;
+		const route = toPosixPath(path.default.relative(srcRoot, filePath).slice(0, -4));
+		routeSet.add(route);
+		const result = analyzeVueContent(fs.default.readFileSync(filePath, "utf-8"), filePath);
+		if (result) analysisMap.set(route, result);
+	}
+	return {
+		routes: Array.from(routeSet).sort(),
+		analysisMap
+	};
 }
-function getScanDirs(dir, subPackages) {
+function getScanDirs(mainPackage, subPackages) {
 	const dirSet = /* @__PURE__ */ new Set();
-	const normalizedMainDir = normalizeDir(dir);
-	if (normalizedMainDir) dirSet.add(normalizedMainDir);
-	subPackages.forEach((subRoot) => {
-		const normalizedSubRoot = normalizeDir(subRoot);
-		if (!normalizedSubRoot) return;
-		if (normalizedMainDir) {
-			dirSet.add(`${normalizedSubRoot}/${normalizedMainDir}`);
-			return;
-		}
-		dirSet.add(normalizedSubRoot);
-	});
+	const mainDir = normalizeDir(mainPackage);
+	if (mainDir) dirSet.add(mainDir);
+	for (const subRoot of subPackages) {
+		const sub = normalizeDir(subRoot);
+		if (!sub) continue;
+		dirSet.add(mainDir ? `${sub}/${mainDir}` : sub);
+	}
 	return Array.from(dirSet);
 }
-function readPagesJson(pagesJsonPath) {
-	if (!fs.default.existsSync(pagesJsonPath)) return null;
-	const content = fs.default.readFileSync(pagesJsonPath, "utf-8");
-	return JSON.parse(content);
-}
-function mergePages(routes, existingPages) {
-	const existingMap = new Map(existingPages.map((p) => [p.path, p]));
+function buildPages(routes, analysisMap, toFullRoute) {
 	return routes.map((route) => {
-		return existingMap.get(route) ?? { path: route };
-	});
-}
-function hasPagesJsonChanged(current, next) {
-	return JSON.stringify({
-		pages: next.pages,
-		subPackages: next.subPackages ?? []
-	}) !== JSON.stringify({
-		pages: current.pages ?? [],
-		subPackages: current.subPackages ?? []
+		const cfg = analysisMap.get(toFullRoute(route))?.config;
+		if (!cfg) return { path: route };
+		return {
+			path: route,
+			...cfg,
+			home: void 0
+		};
 	});
 }
-function getPagesByFileRoute(routes, pagesJson, subPackageRoots) {
-	const originPages = pagesJson.pages ?? [];
-	const originSubPackages = pagesJson.subPackages ?? [];
+function getPagesByFileRoute(routes, analysisMap, pagesJson, subPackageRoots) {
 	const mainRoutes = [];
 	const subRouteMap = /* @__PURE__ */ new Map();
 	const normalizedSubRoots = subPackageRoots.map(normalizeDir);
@@ -187,58 +237,71 @@ function getPagesByFileRoute(routes, pagesJson, subPackageRoots) {
 			return route.startsWith(root + "/");
 		});
 		if (matchedRoot) {
-			const subPath = route.slice(matchedRoot.length + 1);
 			const list = subRouteMap.get(matchedRoot) ?? [];
-			list.push(subPath);
+			list.push(route.slice(matchedRoot.length + 1));
 			subRouteMap.set(matchedRoot, list);
 		} else mainRoutes.push(route);
 	}
-	const homePage = originPages[0]?.path;
-	if (mainRoutes.includes(homePage)) {
-		const idx = mainRoutes.indexOf(homePage);
-		mainRoutes.splice(idx, 1);
-		mainRoutes.unshift(homePage);
-	}
-	const pages = mergePages(mainRoutes, originPages);
-	const subPackages = normalizedSubRoots.map((root) => {
-		return {
-			root,
-			pages: mergePages(subRouteMap.get(root) ?? [], originSubPackages.find((s) => {
-				return normalizeDir(s.root) === root;
-			})?.pages ?? [])
-		};
+	const homeRoute = mainRoutes.find((r) => {
+		return analysisMap.get(r)?.config?.home === true;
 	});
-	const merged = {
-		...pagesJson,
-		pages,
-		subPackages
-	};
+	if (homeRoute) {
+		mainRoutes.splice(mainRoutes.indexOf(homeRoute), 1);
+		mainRoutes.unshift(homeRoute);
+	}
 	return {
-		merged,
-		changed: hasPagesJsonChanged(pagesJson, merged)
+		...pagesJson,
+		pages: buildPages(mainRoutes, analysisMap, (r) => {
+			return r;
+		}),
+		subPackages: normalizedSubRoots.map((root) => {
+			return {
+				root,
+				pages: buildPages(subRouteMap.get(root) ?? [], analysisMap, (r) => {
+					return `${root}/${r}`;
+				})
+			};
+		})
 	};
 }
-function resolveDtsFilePath(srcRoot, dts) {
-	return path.default.isAbsolute(dts) ? dts : path.default.resolve(srcRoot, dts);
-}
 function buildRouteDts(routes) {
-	return [
-		"// Auto-generated by vite-plugin-uni-inject. Do not edit.",
-		"",
-		"/** 提取路径 */",
-		"export type ExtractPath<T extends string, Prefix extends string> = T extends `${Prefix}${infer P}` ? P : never;",
-		"",
-		"/** 路由路径 */",
-		"export type RoutePath =",
-		routes.map((route) => `  | "/${route}"`).join("\n") + ";",
-		""
-	].join("\n");
+	return `import type { DefinePageConfig } from "vite-plugin-uni-inject";
+
+// Auto-generated by vite-plugin-uni-inject. Do not edit.
+
+/** 提取路径 */
+export type ExtractPath<T extends string, Prefix extends string> = T extends \`\${Prefix}\${infer P}\` ? P : never;
+
+/** 路由路径 */
+export type RoutePath = ${routes.map((r) => {
+		return `"/${r}"`;
+	}).join(` |\n`)};
+
+declare global {
+  /**
+   * Vue \`<script setup>\` 宏定义 uniapp 页面级配置。
+   */
+  function definePage(config: DefinePageConfig): void;
+}
+`;
+}
+function getCachePath(cacheDir, targetPath) {
+	const cacheName = Buffer.from(path.default.resolve(targetPath)).toString("base64url");
+	return path.default.join(cacheDir, "vite-plugin-uni-inject", "auto-pages", cacheName);
+}
+function writeFileWithCache(targetPath, nextContent, cacheDir, forceWrite = false) {
+	const cachePath = getCachePath(cacheDir, targetPath);
+	if (!forceWrite && fs.default.existsSync(targetPath) && fs.default.existsSync(cachePath) && fs.default.readFileSync(cachePath, "utf-8") === nextContent) return;
+	fs.default.mkdirSync(path.default.dirname(targetPath), { recursive: true });
+	fs.default.writeFileSync(targetPath, nextContent);
+	fs.default.mkdirSync(path.default.dirname(cachePath), { recursive: true });
+	fs.default.writeFileSync(cachePath, nextContent);
 }
 /**
 * 自动补全 pages.json 插件
 */
 function uniAutoPages(opts) {
-	const { dir = "pages", subPackages = [], dts = "uni-pages.d.ts" } = opts || {};
+	const { mainPackage = "pages", subPackages = [], dts = "uni-pages.d.ts" } = opts || {};
 	return {
 		name: "vite-plugin-uni-auto-pages",
 		enforce: "pre",
@@ -247,16 +310,33 @@ function uniAutoPages(opts) {
 			const pagesJsonPath = path.default.join(srcRoot, "pages.json");
 			const pagesJson = readPagesJson(pagesJsonPath);
 			if (!pagesJson) return;
-			const routes = collectFileRoutes(srcRoot, getScanDirs(dir, subPackages));
-			const { merged, changed } = getPagesByFileRoute(routes, pagesJson, subPackages);
-			if (changed) {
-				if (dts) {
-					const dtsFilePath = resolveDtsFilePath(srcRoot, dts);
-					const dtsContent = buildRouteDts(routes);
-					fs.default.writeFileSync(dtsFilePath, dtsContent);
-				}
-				fs.default.writeFileSync(pagesJsonPath, JSON.stringify(merged, null, 2) + "\n");
+			const { routes, analysisMap } = collectFileRoutes(srcRoot, getScanDirs(mainPackage, subPackages));
+			const merged = getPagesByFileRoute(routes, analysisMap, pagesJson, subPackages);
+			const newPagesJson = JSON.stringify(merged, null, 2) + "\n";
+			const dtsPath = dts ? path.default.join(srcRoot, dts) : null;
+			const shouldRefreshPagesJson = dtsPath ? !fs.default.existsSync(dtsPath) : false;
+			writeFileWithCache(pagesJsonPath, newPagesJson, config.cacheDir, shouldRefreshPagesJson);
+			if (dtsPath) writeFileWithCache(dtsPath, buildRouteDts(routes), config.cacheDir);
+		},
+		transform(code, id) {
+			const pure = id.split("?")[0];
+			if (!pure.endsWith(".vue")) return;
+			const analysis = analyzeVueContent(code, pure);
+			if (!analysis?.stripRanges.length) return;
+			let newCode = code;
+			const ranges = [...analysis.stripRanges].sort((a, b) => {
+				return b.start - a.start;
+			});
+			for (const { start, end } of ranges) {
+				let removeEnd = end;
+				if (newCode.startsWith("\r\n", removeEnd)) removeEnd += 2;
+				else if (newCode[removeEnd] === "\n") removeEnd += 1;
+				newCode = newCode.slice(0, start) + newCode.slice(removeEnd);
 			}
+			return {
+				code: newCode,
+				map: null
+			};
 		}
 	};
 }

package/dist/index.d.cts

@@ -7,20 +7,49 @@ interface InjectPluginOptions {
    */
   path?: string;
 }
+/** 页面 style 配置 */
+interface UniPageStyle {
+  [key: string]: unknown;
+  /** 窗口背景色 */
+  backgroundColor?: string;
+  /** 下拉加载样式 */
+  backgroundTextStyle?: "dark" | "light";
+  /** 是否禁止页面滚动 */
+  disableScroll?: boolean;
+  /** 是否开启下拉刷新 */
+  enablePullDownRefresh?: boolean;
+  /** 导航栏背景颜色 */
+  navigationBarBackgroundColor?: string;
+  /** 导航栏标题颜色 */
+  navigationBarTextStyle?: "black" | "white";
+  /** 导航栏标题文字 */
+  navigationBarTitleText?: string;
+  /** 触底距离 */
+  onReachBottomDistance?: number;
+  /** 自定义组件配置 */
+  usingComponents?: Record<string, string>;
+}
+/** definePage 宏接收的页面配置 */
+interface DefinePageConfig {
+  /** 标记为首页，仅主包页面生效。 */
+  home?: boolean;
+  /** 页面样式 */
+  style?: UniPageStyle;
+}
 /** 自动补全 pages 插件配置 */
 interface AutoPagesPluginOptions {
-  /**
-   * 扫描目录
-   * @default 'pages'
-   */
-  dir?: string;
   /**
    * 输出类型
    * @default './uni-pages.d.ts'
    */
   dts?: string;
   /**
-   * 分包目录
+   * 主包目录
+   * @default 'pages'
+   */
+  mainPackage?: string;
+  /**
+   * 分包列表
    * @default []
    */
   subPackages?: string[];
@@ -49,8 +78,13 @@ declare function uniAutoPages(opts?: AutoPagesPluginOptions): {
   name: string;
   enforce: "pre";
   configResolved(config: {
+    cacheDir: string;
     root: string;
   }): void;
+  transform(code: string, id: string): {
+    code: string;
+    map: null;
+  } | undefined;
 };
 //#endregion
-export { uniAutoPages, uniInject };
+export { type DefinePageConfig, uniAutoPages, uniInject };

package/dist/index.d.mts

@@ -7,20 +7,49 @@ interface InjectPluginOptions {
    */
   path?: string;
 }
+/** 页面 style 配置 */
+interface UniPageStyle {
+  [key: string]: unknown;
+  /** 窗口背景色 */
+  backgroundColor?: string;
+  /** 下拉加载样式 */
+  backgroundTextStyle?: "dark" | "light";
+  /** 是否禁止页面滚动 */
+  disableScroll?: boolean;
+  /** 是否开启下拉刷新 */
+  enablePullDownRefresh?: boolean;
+  /** 导航栏背景颜色 */
+  navigationBarBackgroundColor?: string;
+  /** 导航栏标题颜色 */
+  navigationBarTextStyle?: "black" | "white";
+  /** 导航栏标题文字 */
+  navigationBarTitleText?: string;
+  /** 触底距离 */
+  onReachBottomDistance?: number;
+  /** 自定义组件配置 */
+  usingComponents?: Record<string, string>;
+}
+/** definePage 宏接收的页面配置 */
+interface DefinePageConfig {
+  /** 标记为首页，仅主包页面生效。 */
+  home?: boolean;
+  /** 页面样式 */
+  style?: UniPageStyle;
+}
 /** 自动补全 pages 插件配置 */
 interface AutoPagesPluginOptions {
-  /**
-   * 扫描目录
-   * @default 'pages'
-   */
-  dir?: string;
   /**
    * 输出类型
    * @default './uni-pages.d.ts'
    */
   dts?: string;
   /**
-   * 分包目录
+   * 主包目录
+   * @default 'pages'
+   */
+  mainPackage?: string;
+  /**
+   * 分包列表
    * @default []
    */
   subPackages?: string[];
@@ -49,8 +78,13 @@ declare function uniAutoPages(opts?: AutoPagesPluginOptions): {
   name: string;
   enforce: "pre";
   configResolved(config: {
+    cacheDir: string;
     root: string;
   }): void;
+  transform(code: string, id: string): {
+    code: string;
+    map: null;
+  } | undefined;
 };
 //#endregion
-export { uniAutoPages, uniInject };
+export { type DefinePageConfig, uniAutoPages, uniInject };

package/dist/index.mjs

@@ -69,14 +69,18 @@ function uniInject(opts) {
 				});
 				let nextScriptCode = scriptCode;
 				if (imports.length) {
-					const ranges = [...imports].sort((a, b) => b.start - a.start);
+					const ranges = [...imports].sort((a, b) => {
+						return b.start - a.start;
+					});
 					for (const { start: rangeStart, end: rangeEnd } of ranges) {
 						let removeEnd = rangeEnd;
 						if (nextScriptCode.startsWith("\r\n", removeEnd)) removeEnd += 2;
 						else if (nextScriptCode[removeEnd] === "\n") removeEnd += 1;
 						nextScriptCode = nextScriptCode.slice(0, rangeStart) + nextScriptCode.slice(removeEnd);
 					}
-					nextScriptCode = `\n${imports.map((item) => item.code).join("\n")}${nextScriptCode}`;
+					nextScriptCode = `\n${imports.map((item) => {
+						return item.code;
+					}).join("\n")}${nextScriptCode}`;
 				}
 				newCode = newCode.slice(0, start) + nextScriptCode + newCode.slice(end);
 			}
@@ -86,6 +90,10 @@ function uniInject(opts) {
 }
 //#endregion
 //#region src/modules/auto-pages/index.ts
+function readPagesJson(pagesJsonPath) {
+	if (!fs.existsSync(pagesJsonPath)) return null;
+	return JSON.parse(fs.readFileSync(pagesJsonPath, "utf-8"));
+}
 function toPosixPath(value) {
 	return value.replace(/\\/g, "/");
 }
@@ -105,55 +113,97 @@ function walkFiles(dirPath, filePaths = []) {
 	}
 	return filePaths;
 }
+function analyzeVueContent(content, filePath) {
+	if (!content.includes("definePage")) return null;
+	const stripRanges = [];
+	const { descriptor } = parse$1(content);
+	const blocks = [descriptor.scriptSetup, descriptor.script].filter((b) => {
+		return Boolean(b);
+	});
+	let config = null;
+	for (const block of blocks) {
+		const src = block.content;
+		if (!src.includes("definePage")) continue;
+		let ast;
+		try {
+			ast = parse(src, {
+				sourceType: "module",
+				plugins: ["typescript"]
+			});
+		} catch {
+			continue;
+		}
+		const base = block.loc.start.offset;
+		for (const node of ast.program.body) {
+			if (typeof node.start !== "number" || typeof node.end !== "number") continue;
+			if (node.type === "ImportDeclaration") {
+				if (node.specifiers.some((s) => {
+					return s.type === "ImportSpecifier" && s.imported.type === "Identifier" && s.imported.name === "definePage";
+				})) stripRanges.push({
+					start: base + node.start,
+					end: base + node.end
+				});
+				continue;
+			}
+			if (node.type !== "ExpressionStatement" || node.expression.type !== "CallExpression" || node.expression.callee.type !== "Identifier" || node.expression.callee.name !== "definePage") continue;
+			stripRanges.push({
+				start: base + node.start,
+				end: base + node.end
+			});
+			const arg = node.expression.arguments[0];
+			if (config || !arg || typeof arg.start !== "number" || typeof arg.end !== "number") continue;
+			const argSrc = src.slice(arg.start, arg.end);
+			try {
+				config = new Function(`return (${argSrc})`)();
+			} catch (err) {
+				console.warn(`解析失败(${filePath}): ${err.message}`);
+			}
+		}
+	}
+	if (!stripRanges.length && !config) return null;
+	return {
+		config,
+		stripRanges
+	};
+}
 function collectFileRoutes(srcRoot, routeDirs) {
 	const routeSet = /* @__PURE__ */ new Set();
-	routeDirs.forEach((dir) => {
-		walkFiles(path.resolve(srcRoot, dir)).forEach((filePath) => {
-			if (!filePath.endsWith(".vue")) return;
-			const noExt = path.relative(srcRoot, filePath).slice(0, -4);
-			routeSet.add(toPosixPath(noExt));
-		});
-	});
-	return Array.from(routeSet).sort();
+	const analysisMap = /* @__PURE__ */ new Map();
+	for (const dir of routeDirs) for (const filePath of walkFiles(path.resolve(srcRoot, dir))) {
+		if (!filePath.endsWith(".vue")) continue;
+		const route = toPosixPath(path.relative(srcRoot, filePath).slice(0, -4));
+		routeSet.add(route);
+		const result = analyzeVueContent(fs.readFileSync(filePath, "utf-8"), filePath);
+		if (result) analysisMap.set(route, result);
+	}
+	return {
+		routes: Array.from(routeSet).sort(),
+		analysisMap
+	};
 }
-function getScanDirs(dir, subPackages) {
+function getScanDirs(mainPackage, subPackages) {
 	const dirSet = /* @__PURE__ */ new Set();
-	const normalizedMainDir = normalizeDir(dir);
-	if (normalizedMainDir) dirSet.add(normalizedMainDir);
-	subPackages.forEach((subRoot) => {
-		const normalizedSubRoot = normalizeDir(subRoot);
-		if (!normalizedSubRoot) return;
-		if (normalizedMainDir) {
-			dirSet.add(`${normalizedSubRoot}/${normalizedMainDir}`);
-			return;
-		}
-		dirSet.add(normalizedSubRoot);
-	});
+	const mainDir = normalizeDir(mainPackage);
+	if (mainDir) dirSet.add(mainDir);
+	for (const subRoot of subPackages) {
+		const sub = normalizeDir(subRoot);
+		if (!sub) continue;
+		dirSet.add(mainDir ? `${sub}/${mainDir}` : sub);
+	}
 	return Array.from(dirSet);
 }
-function readPagesJson(pagesJsonPath) {
-	if (!fs.existsSync(pagesJsonPath)) return null;
-	const content = fs.readFileSync(pagesJsonPath, "utf-8");
-	return JSON.parse(content);
-}
-function mergePages(routes, existingPages) {
-	const existingMap = new Map(existingPages.map((p) => [p.path, p]));
+function buildPages(routes, analysisMap, toFullRoute) {
 	return routes.map((route) => {
-		return existingMap.get(route) ?? { path: route };
-	});
-}
-function hasPagesJsonChanged(current, next) {
-	return JSON.stringify({
-		pages: next.pages,
-		subPackages: next.subPackages ?? []
-	}) !== JSON.stringify({
-		pages: current.pages ?? [],
-		subPackages: current.subPackages ?? []
+		const cfg = analysisMap.get(toFullRoute(route))?.config;
+		if (!cfg) return { path: route };
+		return {
+			path: route,
+			...cfg,
+			home: void 0
+		};
 	});
 }
-function getPagesByFileRoute(routes, pagesJson, subPackageRoots) {
-	const originPages = pagesJson.pages ?? [];
-	const originSubPackages = pagesJson.subPackages ?? [];
+function getPagesByFileRoute(routes, analysisMap, pagesJson, subPackageRoots) {
 	const mainRoutes = [];
 	const subRouteMap = /* @__PURE__ */ new Map();
 	const normalizedSubRoots = subPackageRoots.map(normalizeDir);
@@ -162,58 +212,71 @@ function getPagesByFileRoute(routes, pagesJson, subPackageRoots) {
 			return route.startsWith(root + "/");
 		});
 		if (matchedRoot) {
-			const subPath = route.slice(matchedRoot.length + 1);
 			const list = subRouteMap.get(matchedRoot) ?? [];
-			list.push(subPath);
+			list.push(route.slice(matchedRoot.length + 1));
 			subRouteMap.set(matchedRoot, list);
 		} else mainRoutes.push(route);
 	}
-	const homePage = originPages[0]?.path;
-	if (mainRoutes.includes(homePage)) {
-		const idx = mainRoutes.indexOf(homePage);
-		mainRoutes.splice(idx, 1);
-		mainRoutes.unshift(homePage);
-	}
-	const pages = mergePages(mainRoutes, originPages);
-	const subPackages = normalizedSubRoots.map((root) => {
-		return {
-			root,
-			pages: mergePages(subRouteMap.get(root) ?? [], originSubPackages.find((s) => {
-				return normalizeDir(s.root) === root;
-			})?.pages ?? [])
-		};
+	const homeRoute = mainRoutes.find((r) => {
+		return analysisMap.get(r)?.config?.home === true;
 	});
-	const merged = {
-		...pagesJson,
-		pages,
-		subPackages
-	};
+	if (homeRoute) {
+		mainRoutes.splice(mainRoutes.indexOf(homeRoute), 1);
+		mainRoutes.unshift(homeRoute);
+	}
 	return {
-		merged,
-		changed: hasPagesJsonChanged(pagesJson, merged)
+		...pagesJson,
+		pages: buildPages(mainRoutes, analysisMap, (r) => {
+			return r;
+		}),
+		subPackages: normalizedSubRoots.map((root) => {
+			return {
+				root,
+				pages: buildPages(subRouteMap.get(root) ?? [], analysisMap, (r) => {
+					return `${root}/${r}`;
+				})
+			};
+		})
 	};
 }
-function resolveDtsFilePath(srcRoot, dts) {
-	return path.isAbsolute(dts) ? dts : path.resolve(srcRoot, dts);
-}
 function buildRouteDts(routes) {
-	return [
-		"// Auto-generated by vite-plugin-uni-inject. Do not edit.",
-		"",
-		"/** 提取路径 */",
-		"export type ExtractPath<T extends string, Prefix extends string> = T extends `${Prefix}${infer P}` ? P : never;",
-		"",
-		"/** 路由路径 */",
-		"export type RoutePath =",
-		routes.map((route) => `  | "/${route}"`).join("\n") + ";",
-		""
-	].join("\n");
+	return `import type { DefinePageConfig } from "vite-plugin-uni-inject";
+
+// Auto-generated by vite-plugin-uni-inject. Do not edit.
+
+/** 提取路径 */
+export type ExtractPath<T extends string, Prefix extends string> = T extends \`\${Prefix}\${infer P}\` ? P : never;
+
+/** 路由路径 */
+export type RoutePath = ${routes.map((r) => {
+		return `"/${r}"`;
+	}).join(` |\n`)};
+
+declare global {
+  /**
+   * Vue \`<script setup>\` 宏定义 uniapp 页面级配置。
+   */
+  function definePage(config: DefinePageConfig): void;
+}
+`;
+}
+function getCachePath(cacheDir, targetPath) {
+	const cacheName = Buffer.from(path.resolve(targetPath)).toString("base64url");
+	return path.join(cacheDir, "vite-plugin-uni-inject", "auto-pages", cacheName);
+}
+function writeFileWithCache(targetPath, nextContent, cacheDir, forceWrite = false) {
+	const cachePath = getCachePath(cacheDir, targetPath);
+	if (!forceWrite && fs.existsSync(targetPath) && fs.existsSync(cachePath) && fs.readFileSync(cachePath, "utf-8") === nextContent) return;
+	fs.mkdirSync(path.dirname(targetPath), { recursive: true });
+	fs.writeFileSync(targetPath, nextContent);
+	fs.mkdirSync(path.dirname(cachePath), { recursive: true });
+	fs.writeFileSync(cachePath, nextContent);
 }
 /**
 * 自动补全 pages.json 插件
 */
 function uniAutoPages(opts) {
-	const { dir = "pages", subPackages = [], dts = "uni-pages.d.ts" } = opts || {};
+	const { mainPackage = "pages", subPackages = [], dts = "uni-pages.d.ts" } = opts || {};
 	return {
 		name: "vite-plugin-uni-auto-pages",
 		enforce: "pre",
@@ -222,16 +285,33 @@ function uniAutoPages(opts) {
 			const pagesJsonPath = path.join(srcRoot, "pages.json");
 			const pagesJson = readPagesJson(pagesJsonPath);
 			if (!pagesJson) return;
-			const routes = collectFileRoutes(srcRoot, getScanDirs(dir, subPackages));
-			const { merged, changed } = getPagesByFileRoute(routes, pagesJson, subPackages);
-			if (changed) {
-				if (dts) {
-					const dtsFilePath = resolveDtsFilePath(srcRoot, dts);
-					const dtsContent = buildRouteDts(routes);
-					fs.writeFileSync(dtsFilePath, dtsContent);
-				}
-				fs.writeFileSync(pagesJsonPath, JSON.stringify(merged, null, 2) + "\n");
+			const { routes, analysisMap } = collectFileRoutes(srcRoot, getScanDirs(mainPackage, subPackages));
+			const merged = getPagesByFileRoute(routes, analysisMap, pagesJson, subPackages);
+			const newPagesJson = JSON.stringify(merged, null, 2) + "\n";
+			const dtsPath = dts ? path.join(srcRoot, dts) : null;
+			const shouldRefreshPagesJson = dtsPath ? !fs.existsSync(dtsPath) : false;
+			writeFileWithCache(pagesJsonPath, newPagesJson, config.cacheDir, shouldRefreshPagesJson);
+			if (dtsPath) writeFileWithCache(dtsPath, buildRouteDts(routes), config.cacheDir);
+		},
+		transform(code, id) {
+			const pure = id.split("?")[0];
+			if (!pure.endsWith(".vue")) return;
+			const analysis = analyzeVueContent(code, pure);
+			if (!analysis?.stripRanges.length) return;
+			let newCode = code;
+			const ranges = [...analysis.stripRanges].sort((a, b) => {
+				return b.start - a.start;
+			});
+			for (const { start, end } of ranges) {
+				let removeEnd = end;
+				if (newCode.startsWith("\r\n", removeEnd)) removeEnd += 2;
+				else if (newCode[removeEnd] === "\n") removeEnd += 1;
+				newCode = newCode.slice(0, start) + newCode.slice(removeEnd);
 			}
+			return {
+				code: newCode,
+				map: null
+			};
 		}
 	};
 }

package/package.json

@@ -4,7 +4,7 @@
   "author": "Kriac",
   "license": "MIT",
   "type": "module",
-  "version": "0.4.0",
+  "version": "0.6.0",
   "homepage": "https://github.com/Kriac/vite-plugin-uni-inject",
   "repository": {
     "type": "git",