vite-plugin-uni-inject 0.3.1 → 0.5.0

package/dist/index.cjs

@@ -24,12 +24,10 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 let _babel_parser = require("@babel/parser");
 let _vue_compiler_sfc = require("@vue/compiler-sfc");
 let fs = require("fs");
-fs = __toESM(fs);
+fs = __toESM(fs, 1);
 let path = require("path");
-path = __toESM(path);
-let magic_string = require("magic-string");
-magic_string = __toESM(magic_string);
-//#region src/inject.ts
+path = __toESM(path, 1);
+//#region src/modules/inject/index.ts
 /**
 * 注入代码插件
 */
@@ -84,25 +82,39 @@ function uniInject(opts) {
 				const start = newDescriptor.scriptSetup.loc.start.offset;
 				const end = newDescriptor.scriptSetup.loc.end.offset;
 				const scriptCode = newDescriptor.scriptSetup.content;
-				const s = new magic_string.default(scriptCode);
 				const ast = (0, _babel_parser.parse)(scriptCode, {
 					sourceType: "module",
 					plugins: ["typescript"]
 				});
 				const imports = [];
-				for (const node of ast.program.body) if (node.type === "ImportDeclaration") {
-					imports.push(scriptCode.slice(node.start, node.end));
-					s.remove(node.start, node.end + 1);
+				for (const node of ast.program.body) if (node.type === "ImportDeclaration" && typeof node.start === "number" && typeof node.end === "number") imports.push({
+					code: scriptCode.slice(node.start, node.end),
+					start: node.start,
+					end: node.end
+				});
+				let nextScriptCode = scriptCode;
+				if (imports.length) {
+					const ranges = [...imports].sort((a, b) => 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}`;
 				}
-				if (imports.length) s.prepend("\n" + imports.join("\n"));
-				newCode = newCode.slice(0, start) + s.toString() + newCode.slice(end);
+				newCode = newCode.slice(0, start) + nextScriptCode + newCode.slice(end);
 			}
 			return { code: newCode };
 		}
 	};
 }
 //#endregion
-//#region src/auto-pages.ts
+//#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, "/");
 }
@@ -122,115 +134,152 @@ 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) => 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) => 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);
 	for (const route of routes) {
-		const matchedRoot = normalizedSubRoots.find((root) => {
-			return route.startsWith(root + "/");
-		});
+		const matchedRoot = normalizedSubRoots.find((root) => 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) => r),
+		subPackages: normalizedSubRoots.map((root) => ({
+			root,
+			pages: buildPages(subRouteMap.get(root) ?? [], analysisMap, (r) => `${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) => `"/${r}"`).join(` |\n`)};
+
+declare global {
+  /**
+   * Vue \`<script setup>\` 宏定义 uniapp 页面级配置。
+   */
+  function definePage(config: DefinePageConfig): void;
+}
+`;
+}
+function writeIfTextChanged(targetPath, nextContent) {
+	const normalize = (s) => {
+		return s.replace(/\s+/g, "").replace(/,(?=[)\]}>])/g, "").replace(/([=(<,|&?:])\|/g, "$1");
+	};
+	if (fs.default.existsSync(targetPath) && normalize(fs.default.readFileSync(targetPath, "utf-8")) === normalize(nextContent)) return;
+	fs.default.mkdirSync(path.default.dirname(targetPath), { recursive: true });
+	fs.default.writeFileSync(targetPath, 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",
@@ -239,14 +288,31 @@ 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) 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);
+			writeIfTextChanged(pagesJsonPath, JSON.stringify(merged, null, 2) + "\n");
 			if (dts) {
-				const dtsFilePath = resolveDtsFilePath(srcRoot, dts);
-				const dtsContent = buildRouteDts(routes);
-				fs.default.writeFileSync(dtsFilePath, dtsContent);
+				const newDts = buildRouteDts(routes);
+				writeIfTextChanged(path.default.join(srcRoot, dts), newDts);
+			}
+		},
+		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) => 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,6 +7,35 @@ 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 {
   /**
@@ -15,18 +44,18 @@ interface AutoPagesPluginOptions {
    */
   dts?: string;
   /**
-   * 扫描目录
+   * 主包目录
    * @default 'pages'
    */
-  dir?: string;
+  mainPackage?: string;
   /**
-   * 分包目录
+   * 分包列表
    * @default []
    */
   subPackages?: string[];
 }
 //#endregion
-//#region src/inject.d.ts
+//#region src/modules/inject/index.d.ts
 /**
  * 注入代码插件
  */
@@ -41,7 +70,7 @@ declare function uniInject(opts?: InjectPluginOptions): {
   } | undefined;
 };
 //#endregion
-//#region src/auto-pages.d.ts
+//#region src/modules/auto-pages/index.d.ts
 /**
  * 自动补全 pages.json 插件
  */
@@ -51,6 +80,10 @@ declare function uniAutoPages(opts?: AutoPagesPluginOptions): {
   configResolved(config: {
     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,6 +7,35 @@ 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 {
   /**
@@ -15,18 +44,18 @@ interface AutoPagesPluginOptions {
    */
   dts?: string;
   /**
-   * 扫描目录
+   * 主包目录
    * @default 'pages'
    */
-  dir?: string;
+  mainPackage?: string;
   /**
-   * 分包目录
+   * 分包列表
    * @default []
    */
   subPackages?: string[];
 }
 //#endregion
-//#region src/inject.d.ts
+//#region src/modules/inject/index.d.ts
 /**
  * 注入代码插件
  */
@@ -41,7 +70,7 @@ declare function uniInject(opts?: InjectPluginOptions): {
   } | undefined;
 };
 //#endregion
-//#region src/auto-pages.d.ts
+//#region src/modules/auto-pages/index.d.ts
 /**
  * 自动补全 pages.json 插件
  */
@@ -51,6 +80,10 @@ declare function uniAutoPages(opts?: AutoPagesPluginOptions): {
   configResolved(config: {
     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

@@ -2,8 +2,7 @@ import { parse } from "@babel/parser";
 import { parse as parse$1 } from "@vue/compiler-sfc";
 import fs from "fs";
 import path from "path";
-import MagicString from "magic-string";
-//#region src/inject.ts
+//#region src/modules/inject/index.ts
 /**
 * 注入代码插件
 */
@@ -58,25 +57,39 @@ function uniInject(opts) {
 				const start = newDescriptor.scriptSetup.loc.start.offset;
 				const end = newDescriptor.scriptSetup.loc.end.offset;
 				const scriptCode = newDescriptor.scriptSetup.content;
-				const s = new MagicString(scriptCode);
 				const ast = parse(scriptCode, {
 					sourceType: "module",
 					plugins: ["typescript"]
 				});
 				const imports = [];
-				for (const node of ast.program.body) if (node.type === "ImportDeclaration") {
-					imports.push(scriptCode.slice(node.start, node.end));
-					s.remove(node.start, node.end + 1);
+				for (const node of ast.program.body) if (node.type === "ImportDeclaration" && typeof node.start === "number" && typeof node.end === "number") imports.push({
+					code: scriptCode.slice(node.start, node.end),
+					start: node.start,
+					end: node.end
+				});
+				let nextScriptCode = scriptCode;
+				if (imports.length) {
+					const ranges = [...imports].sort((a, b) => 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}`;
 				}
-				if (imports.length) s.prepend("\n" + imports.join("\n"));
-				newCode = newCode.slice(0, start) + s.toString() + newCode.slice(end);
+				newCode = newCode.slice(0, start) + nextScriptCode + newCode.slice(end);
 			}
 			return { code: newCode };
 		}
 	};
 }
 //#endregion
-//#region src/auto-pages.ts
+//#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, "/");
 }
@@ -96,115 +109,152 @@ 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) => 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) => 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);
 	for (const route of routes) {
-		const matchedRoot = normalizedSubRoots.find((root) => {
-			return route.startsWith(root + "/");
-		});
+		const matchedRoot = normalizedSubRoots.find((root) => 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) => r),
+		subPackages: normalizedSubRoots.map((root) => ({
+			root,
+			pages: buildPages(subRouteMap.get(root) ?? [], analysisMap, (r) => `${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) => `"/${r}"`).join(` |\n`)};
+
+declare global {
+  /**
+   * Vue \`<script setup>\` 宏定义 uniapp 页面级配置。
+   */
+  function definePage(config: DefinePageConfig): void;
+}
+`;
+}
+function writeIfTextChanged(targetPath, nextContent) {
+	const normalize = (s) => {
+		return s.replace(/\s+/g, "").replace(/,(?=[)\]}>])/g, "").replace(/([=(<,|&?:])\|/g, "$1");
+	};
+	if (fs.existsSync(targetPath) && normalize(fs.readFileSync(targetPath, "utf-8")) === normalize(nextContent)) return;
+	fs.mkdirSync(path.dirname(targetPath), { recursive: true });
+	fs.writeFileSync(targetPath, 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",
@@ -213,14 +263,31 @@ 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) fs.writeFileSync(pagesJsonPath, JSON.stringify(merged, null, 2) + "\n");
+			const { routes, analysisMap } = collectFileRoutes(srcRoot, getScanDirs(mainPackage, subPackages));
+			const merged = getPagesByFileRoute(routes, analysisMap, pagesJson, subPackages);
+			writeIfTextChanged(pagesJsonPath, JSON.stringify(merged, null, 2) + "\n");
 			if (dts) {
-				const dtsFilePath = resolveDtsFilePath(srcRoot, dts);
-				const dtsContent = buildRouteDts(routes);
-				fs.writeFileSync(dtsFilePath, dtsContent);
+				const newDts = buildRouteDts(routes);
+				writeIfTextChanged(path.join(srcRoot, dts), newDts);
+			}
+		},
+		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) => 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.3.1",
+  "version": "0.5.0",
   "homepage": "https://github.com/Kriac/vite-plugin-uni-inject",
   "repository": {
     "type": "git",
@@ -20,12 +20,11 @@
   },
   "dependencies": {
     "@vue/compiler-sfc": "3.5.27",
-    "@babel/parser": "7.29.0",
-    "magic-string": "0.30.21"
+    "@babel/parser": "7.29.0"
   },
   "devDependencies": {
     "@types/node": "25.0.10",
-    "tsdown": "0.21.0",
+    "tsdown": "0.22.0",
     "vue-tsc": "3.2.1"
   },
   "files": [