vite-plugin-uni-inject 0.1.0 → 0.3.0

package/README.md

@@ -1,17 +1,15 @@
 # vite-plugin-uni-inject
 
-利用 Vite 插件机制，实现 uniapp 项目中无法使用公共组件的问题。
+利用 Vite 插件机制，实现自动注入代码，解放你的双手。
 
 ## 功能特点
 
 - 干净的注入任何代码，没有多余的结构。
 - 支持注入 `<page-meta/>` 这种只能放在页面第一个位置的标签节点。
-- 自动识别所有 `page.json` 中的 vue 文件并注入，无需维护注入页面表。
+- 支持基于文件路由的 `page.json` 自动生成并注入，无需手动维护页面表。
 
 ## 如何使用
 
-原理就是通过 vite 插件机制，在构建过程中自动注入 inject.vue 中的代码。
-
 ### 安装依赖
 
 ```bash
@@ -22,17 +20,22 @@ pnpm i -D vite-plugin-uni-inject
 
 ```ts
 import { defineConfig } from "vite";
+import { uniAutoPages, uniInject } from "vite-plugin-uni-inject";
 import uni from "@dcloudio/vite-plugin-uni";
-import uniInject from "vite-plugin-uni-inject";
 
 // 如果有重写 page.json 文件的插件，请确保写在 uniInject 之前
 export default defineConfig(() => {
   return {
-    plugins: [uniInject(), uni()],
+    plugins: [uniAutoPages(), uniInject(), uni()],
   };
 });
 ```
 
+### 独立插件说明
+
+- `uniAutoPages(options)`：负责扫描文件路由并补全 `src/pages.json` 。
+- `uniInject(options)`：负责注入 `App.inject.vue`（或自定义文件）到页面文件。
+
 ## 报告错误
 
 欢迎提交 issue 与我们讨论。

package/dist/index.cjs

@@ -1,6 +1,248 @@
-"use strict";var U=Object.create;var u=Object.defineProperty;var k=Object.getOwnPropertyDescriptor;var x=Object.getOwnPropertyNames;var z=Object.getPrototypeOf,C=Object.prototype.hasOwnProperty;var O=(t,e)=>{for(var s in e)u(t,s,{get:e[s],enumerable:!0})},h=(t,e,s,a)=>{if(e&&typeof e=="object"||typeof e=="function")for(let o of x(e))!C.call(t,o)&&o!==s&&u(t,o,{get:()=>e[o],enumerable:!(a=k(e,o))||a.enumerable});return t};var v=(t,e,s)=>(s=t!=null?U(z(t)):{},h(e||!t||!t.__esModule?u(s,"default",{value:t,enumerable:!0}):s,t)),b=t=>h(u({},"__esModule",{value:!0}),t);var E={};O(E,{default:()=>w});module.exports=b(E);var P=require("@vue/compiler-sfc"),l=v(require("fs"),1),c=v(require("path"),1);function w(t){let{injectPath:e="App.inject.vue"}=t||{},s=new Set,a="",o="";return{name:"vite-plugin-uni-inject",enforce:"pre",configResolved(f){let p=c.default.resolve(f.root,"src"),g=c.default.join(p,e);if(!l.default.existsSync(g))return;let i=l.default.readFileSync(g,"utf-8"),{descriptor:n}=(0,P.parse)(i);n.template&&(a=n.template.content.trim()),n.scriptSetup&&(o=n.scriptSetup.content.trim());let r=c.default.join(p,"pages.json");if(!l.default.existsSync(r))return;let d=l.default.readFileSync(r,"utf-8"),S=JSON.parse(d),j=(m,y="")=>{m.forEach(J=>{let $=c.default.join(p,y,`${J.path}.vue`);s.add(c.default.normalize($))})};j(S.pages),(S.subPackages??[]).forEach(m=>{j(m.pages,m.root)})},transform(f,p){if(!p.endsWith(".vue"))return;let g=c.default.normalize(p);if(!s.has(g))return;let{descriptor:i}=(0,P.parse)(f),n=f;if(a&&i.template){let r=i.template.content.trim(),d=`${a}
-${r}`.trim();n=n.slice(0,i.template.loc.start.offset)+`
-${d}
-`+n.slice(i.template.loc.end.offset)}if(o&&i.scriptSetup){let r=i.scriptSetup.loc.start.offset;n=n.slice(0,r)+`
-${o}
-`+n.slice(r)}return{code:n,map:null}}}}
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+//#region \0rolldown/runtime.js
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+	if (from && typeof from === "object" || typeof from === "function") for (var keys = __getOwnPropNames(from), i = 0, n = keys.length, key; i < n; i++) {
+		key = keys[i];
+		if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, {
+			get: ((k) => from[k]).bind(null, key),
+			enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable
+		});
+	}
+	return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", {
+	value: mod,
+	enumerable: true
+}) : target, mod));
+//#endregion
+let _babel_parser = require("@babel/parser");
+let _vue_compiler_sfc = require("@vue/compiler-sfc");
+let fs = require("fs");
+fs = __toESM(fs);
+let path = require("path");
+path = __toESM(path);
+let magic_string = require("magic-string");
+magic_string = __toESM(magic_string);
+//#region src/inject.ts
+/**
+* 注入代码插件
+*/
+function uniInject(opts) {
+	const { path: injectPath = "App.inject.vue" } = opts || {};
+	const pageSet = /* @__PURE__ */ new Set();
+	let injectTemplate = "";
+	let injectScriptSetup = "";
+	return {
+		name: "vite-plugin-uni-inject",
+		enforce: "pre",
+		configResolved(config) {
+			const srcRoot = path.default.resolve(config.root, "src");
+			const injPath = path.default.join(srcRoot, injectPath);
+			if (!fs.default.existsSync(injPath)) return;
+			const pagesJsonPath = path.default.join(srcRoot, "pages.json");
+			if (!fs.default.existsSync(pagesJsonPath)) return;
+			const { descriptor } = (0, _vue_compiler_sfc.parse)(fs.default.readFileSync(injPath, "utf-8"));
+			if (descriptor.template) injectTemplate = descriptor.template.content.trim();
+			if (descriptor.scriptSetup) injectScriptSetup = descriptor.scriptSetup.content.trim();
+			const pagesJsonContent = fs.default.readFileSync(pagesJsonPath, "utf-8");
+			const pagesJson = JSON.parse(pagesJsonContent);
+			const subPackages = pagesJson.subPackages ?? [];
+			const collectPages = (pages, rootDir = "") => {
+				pages.forEach((page) => {
+					const pageVuePath = path.default.join(srcRoot, rootDir, `${page.path}.vue`);
+					pageSet.add(path.default.normalize(pageVuePath));
+				});
+			};
+			collectPages(pagesJson.pages);
+			subPackages.forEach((sub) => {
+				collectPages(sub.pages, sub.root);
+			});
+		},
+		transform(code, id) {
+			if (!id.endsWith(".vue")) return;
+			const normalizedId = path.default.normalize(id);
+			if (!pageSet.has(normalizedId)) return;
+			let newCode = code;
+			const { descriptor } = (0, _vue_compiler_sfc.parse)(code);
+			if (injectTemplate && descriptor.template) {
+				const tplContent = descriptor.template.content.trim();
+				const injectedTemplate = `${injectTemplate}\n${tplContent}`.trim();
+				newCode = newCode.slice(0, descriptor.template.loc.start.offset) + `\n${injectedTemplate}\n` + newCode.slice(descriptor.template.loc.end.offset);
+			}
+			if (injectScriptSetup && descriptor.scriptSetup) {
+				const start = descriptor.scriptSetup.loc.start.offset;
+				newCode = newCode.slice(0, start) + `\n${injectScriptSetup}` + newCode.slice(start);
+			}
+			const { descriptor: newDescriptor } = (0, _vue_compiler_sfc.parse)(newCode);
+			if (newDescriptor.scriptSetup) {
+				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);
+				}
+				if (imports.length) s.prepend("\n" + imports.join("\n"));
+				newCode = newCode.slice(0, start) + s.toString() + newCode.slice(end);
+			}
+			return { code: newCode };
+		}
+	};
+}
+//#endregion
+//#region src/auto-pages.ts
+function toPosixPath(value) {
+	return value.replace(/\\/g, "/");
+}
+function normalizeDir(value) {
+	return toPosixPath(value).replace(/^\/+|\/+$/g, "");
+}
+function walkFiles(dirPath, filePaths = []) {
+	if (!fs.default.existsSync(dirPath)) return filePaths;
+	const entries = fs.default.readdirSync(dirPath, { withFileTypes: true });
+	for (const entry of entries) {
+		const fullPath = path.default.join(dirPath, entry.name);
+		if (entry.isDirectory()) {
+			walkFiles(fullPath, filePaths);
+			continue;
+		}
+		filePaths.push(fullPath);
+	}
+	return filePaths;
+}
+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();
+}
+function getScanDirs(dir, 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);
+	});
+	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]));
+	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 ?? []
+	});
+}
+function getPagesByFileRoute(routes, 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 + "/");
+		});
+		if (matchedRoot) {
+			const subPath = route.slice(matchedRoot.length + 1);
+			const list = subRouteMap.get(matchedRoot) ?? [];
+			list.push(subPath);
+			subRouteMap.set(matchedRoot, list);
+		} else mainRoutes.push(route);
+	}
+	const pages = mergePages(mainRoutes, pagesJson.pages ?? []);
+	const existSubs = pagesJson.subPackages ?? [];
+	const subPackages = normalizedSubRoots.map((root) => {
+		return {
+			root,
+			pages: mergePages(subRouteMap.get(root) ?? [], existSubs.find((s) => {
+				return normalizeDir(s.root) === root;
+			})?.pages ?? [])
+		};
+	});
+	const merged = {
+		...pagesJson,
+		pages,
+		subPackages
+	};
+	return {
+		merged,
+		changed: hasPagesJsonChanged(pagesJson, merged)
+	};
+}
+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");
+}
+/**
+* 自动补全 pages.json 插件
+*/
+function uniAutoPages(opts) {
+	const { dir = "pages", subPackages = [], dts = "uni-pages.d.ts" } = opts || {};
+	return {
+		name: "vite-plugin-uni-auto-pages",
+		enforce: "pre",
+		configResolved(config) {
+			const srcRoot = path.default.resolve(config.root, "src");
+			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");
+			if (dts) {
+				const dtsFilePath = resolveDtsFilePath(srcRoot, dts);
+				const dtsContent = buildRouteDts(routes);
+				fs.default.writeFileSync(dtsFilePath, dtsContent);
+			}
+		}
+	};
+}
+//#endregion
+exports.uniAutoPages = uniAutoPages;
+exports.uniInject = uniInject;

package/dist/index.d.cts

@@ -1,19 +1,56 @@
-/** 插件配置 */
-interface PluginOptions {
-    /** 要注入的文件路径 - 基于项目 src 目录 */
-    injectPath?: string;
+//#region src/types.d.ts
+/** 注入插件配置 */
+interface InjectPluginOptions {
+  /**
+   * 注入的文件路径
+   * @default 'src/App.inject.vue'
+   */
+  path?: string;
 }
-
-declare function export_default(opts?: PluginOptions): {
-    name: string;
-    enforce: "pre";
-    configResolved(config: {
-        root: string;
-    }): void;
-    transform(code: string, id: string): {
-        code: string;
-        map: null;
-    } | undefined;
+/** 自动补全 pages 插件配置 */
+interface AutoPagesPluginOptions {
+  /**
+   * 扫描目录
+   * @default 'pages'
+   */
+  dir?: string;
+  /**
+   * 分包目录
+   * @default []
+   */
+  subPackages?: string[];
+  /**
+   * 生成类型声明
+   * @default 'src/uni-pages.d.ts'
+   */
+  dts?: string;
+}
+//#endregion
+//#region src/inject.d.ts
+/**
+ * 注入代码插件
+ */
+declare function uniInject(opts?: InjectPluginOptions): {
+  name: string;
+  enforce: "pre";
+  configResolved(config: {
+    root: string;
+  }): void;
+  transform(code: string, id: string): {
+    code: string;
+  } | undefined;
+};
+//#endregion
+//#region src/auto-pages.d.ts
+/**
+ * 自动补全 pages.json 插件
+ */
+declare function uniAutoPages(opts?: AutoPagesPluginOptions): {
+  name: string;
+  enforce: "pre";
+  configResolved(config: {
+    root: string;
+  }): void;
 };
-
-export { export_default as default };
+//#endregion
+export { uniAutoPages, uniInject };

package/dist/index.d.mts

@@ -0,0 +1,56 @@
+//#region src/types.d.ts
+/** 注入插件配置 */
+interface InjectPluginOptions {
+  /**
+   * 注入的文件路径
+   * @default 'src/App.inject.vue'
+   */
+  path?: string;
+}
+/** 自动补全 pages 插件配置 */
+interface AutoPagesPluginOptions {
+  /**
+   * 扫描目录
+   * @default 'pages'
+   */
+  dir?: string;
+  /**
+   * 分包目录
+   * @default []
+   */
+  subPackages?: string[];
+  /**
+   * 生成类型声明
+   * @default 'src/uni-pages.d.ts'
+   */
+  dts?: string;
+}
+//#endregion
+//#region src/inject.d.ts
+/**
+ * 注入代码插件
+ */
+declare function uniInject(opts?: InjectPluginOptions): {
+  name: string;
+  enforce: "pre";
+  configResolved(config: {
+    root: string;
+  }): void;
+  transform(code: string, id: string): {
+    code: string;
+  } | undefined;
+};
+//#endregion
+//#region src/auto-pages.d.ts
+/**
+ * 自动补全 pages.json 插件
+ */
+declare function uniAutoPages(opts?: AutoPagesPluginOptions): {
+  name: string;
+  enforce: "pre";
+  configResolved(config: {
+    root: string;
+  }): void;
+};
+//#endregion
+export { uniAutoPages, uniInject };

package/dist/index.mjs

@@ -0,0 +1,221 @@
+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
+/**
+* 注入代码插件
+*/
+function uniInject(opts) {
+	const { path: injectPath = "App.inject.vue" } = opts || {};
+	const pageSet = /* @__PURE__ */ new Set();
+	let injectTemplate = "";
+	let injectScriptSetup = "";
+	return {
+		name: "vite-plugin-uni-inject",
+		enforce: "pre",
+		configResolved(config) {
+			const srcRoot = path.resolve(config.root, "src");
+			const injPath = path.join(srcRoot, injectPath);
+			if (!fs.existsSync(injPath)) return;
+			const pagesJsonPath = path.join(srcRoot, "pages.json");
+			if (!fs.existsSync(pagesJsonPath)) return;
+			const { descriptor } = parse$1(fs.readFileSync(injPath, "utf-8"));
+			if (descriptor.template) injectTemplate = descriptor.template.content.trim();
+			if (descriptor.scriptSetup) injectScriptSetup = descriptor.scriptSetup.content.trim();
+			const pagesJsonContent = fs.readFileSync(pagesJsonPath, "utf-8");
+			const pagesJson = JSON.parse(pagesJsonContent);
+			const subPackages = pagesJson.subPackages ?? [];
+			const collectPages = (pages, rootDir = "") => {
+				pages.forEach((page) => {
+					const pageVuePath = path.join(srcRoot, rootDir, `${page.path}.vue`);
+					pageSet.add(path.normalize(pageVuePath));
+				});
+			};
+			collectPages(pagesJson.pages);
+			subPackages.forEach((sub) => {
+				collectPages(sub.pages, sub.root);
+			});
+		},
+		transform(code, id) {
+			if (!id.endsWith(".vue")) return;
+			const normalizedId = path.normalize(id);
+			if (!pageSet.has(normalizedId)) return;
+			let newCode = code;
+			const { descriptor } = parse$1(code);
+			if (injectTemplate && descriptor.template) {
+				const tplContent = descriptor.template.content.trim();
+				const injectedTemplate = `${injectTemplate}\n${tplContent}`.trim();
+				newCode = newCode.slice(0, descriptor.template.loc.start.offset) + `\n${injectedTemplate}\n` + newCode.slice(descriptor.template.loc.end.offset);
+			}
+			if (injectScriptSetup && descriptor.scriptSetup) {
+				const start = descriptor.scriptSetup.loc.start.offset;
+				newCode = newCode.slice(0, start) + `\n${injectScriptSetup}` + newCode.slice(start);
+			}
+			const { descriptor: newDescriptor } = parse$1(newCode);
+			if (newDescriptor.scriptSetup) {
+				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);
+				}
+				if (imports.length) s.prepend("\n" + imports.join("\n"));
+				newCode = newCode.slice(0, start) + s.toString() + newCode.slice(end);
+			}
+			return { code: newCode };
+		}
+	};
+}
+//#endregion
+//#region src/auto-pages.ts
+function toPosixPath(value) {
+	return value.replace(/\\/g, "/");
+}
+function normalizeDir(value) {
+	return toPosixPath(value).replace(/^\/+|\/+$/g, "");
+}
+function walkFiles(dirPath, filePaths = []) {
+	if (!fs.existsSync(dirPath)) return filePaths;
+	const entries = fs.readdirSync(dirPath, { withFileTypes: true });
+	for (const entry of entries) {
+		const fullPath = path.join(dirPath, entry.name);
+		if (entry.isDirectory()) {
+			walkFiles(fullPath, filePaths);
+			continue;
+		}
+		filePaths.push(fullPath);
+	}
+	return filePaths;
+}
+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();
+}
+function getScanDirs(dir, 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);
+	});
+	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]));
+	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 ?? []
+	});
+}
+function getPagesByFileRoute(routes, 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 + "/");
+		});
+		if (matchedRoot) {
+			const subPath = route.slice(matchedRoot.length + 1);
+			const list = subRouteMap.get(matchedRoot) ?? [];
+			list.push(subPath);
+			subRouteMap.set(matchedRoot, list);
+		} else mainRoutes.push(route);
+	}
+	const pages = mergePages(mainRoutes, pagesJson.pages ?? []);
+	const existSubs = pagesJson.subPackages ?? [];
+	const subPackages = normalizedSubRoots.map((root) => {
+		return {
+			root,
+			pages: mergePages(subRouteMap.get(root) ?? [], existSubs.find((s) => {
+				return normalizeDir(s.root) === root;
+			})?.pages ?? [])
+		};
+	});
+	const merged = {
+		...pagesJson,
+		pages,
+		subPackages
+	};
+	return {
+		merged,
+		changed: hasPagesJsonChanged(pagesJson, merged)
+	};
+}
+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");
+}
+/**
+* 自动补全 pages.json 插件
+*/
+function uniAutoPages(opts) {
+	const { dir = "pages", subPackages = [], dts = "uni-pages.d.ts" } = opts || {};
+	return {
+		name: "vite-plugin-uni-auto-pages",
+		enforce: "pre",
+		configResolved(config) {
+			const srcRoot = path.resolve(config.root, "src");
+			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");
+			if (dts) {
+				const dtsFilePath = resolveDtsFilePath(srcRoot, dts);
+				const dtsContent = buildRouteDts(routes);
+				fs.writeFileSync(dtsFilePath, dtsContent);
+			}
+		}
+	};
+}
+//#endregion
+export { uniAutoPages, uniInject };

package/package.json

@@ -1,39 +1,40 @@
 {
   "name": "vite-plugin-uni-inject",
-  "type": "module",
-  "version": "0.1.0",
   "description": "vite-plugin-uni-inject",
   "author": "Kriac",
+  "license": "MIT",
+  "type": "module",
+  "version": "0.3.0",
   "homepage": "https://github.com/Kriac/vite-plugin-uni-inject",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/Kriac/vite-plugin-uni-inject.git"
   },
   "keywords": [
-    "vite-plugin",
-    "uni-inject"
+    "vite-plugin-uni-inject"
   ],
   "scripts": {
     "lint": "vue-tsc -b --noEmit",
-    "build": "tsup",
+    "build": "tsdown",
     "release": "npm publish --access public"
   },
   "dependencies": {
-    "@vue/compiler-sfc": "3.5.27"
+    "@vue/compiler-sfc": "3.5.27",
+    "@babel/parser": "7.29.0",
+    "magic-string": "0.30.21"
   },
   "devDependencies": {
     "@types/node": "25.0.10",
-    "tsup": "8.5.1",
+    "tsdown": "0.21.0",
     "vue-tsc": "3.2.1"
   },
-  "license": "MIT",
   "files": [
     "dist"
   ],
   "exports": {
     ".": {
       "types": "./dist/index.d.ts",
-      "import": "./dist/index.js",
+      "import": "./dist/index.mjs",
       "require": "./dist/index.cjs"
     }
   }

package/dist/index.d.ts

@@ -1,19 +0,0 @@
-/** 插件配置 */
-interface PluginOptions {
-    /** 要注入的文件路径 - 基于项目 src 目录 */
-    injectPath?: string;
-}
-
-declare function export_default(opts?: PluginOptions): {
-    name: string;
-    enforce: "pre";
-    configResolved(config: {
-        root: string;
-    }): void;
-    transform(code: string, id: string): {
-        code: string;
-        map: null;
-    } | undefined;
-};
-
-export { export_default as default };

package/dist/index.js

@@ -1,6 +0,0 @@
-import{parse as d}from"@vue/compiler-sfc";import a from"fs";import o from"path";function k(P){let{injectPath:S="App.inject.vue"}=P||{},g=new Set,p="",l="";return{name:"vite-plugin-uni-inject",enforce:"pre",configResolved(i){let s=o.resolve(i.root,"src"),r=o.join(s,S);if(!a.existsSync(r))return;let e=a.readFileSync(r,"utf-8"),{descriptor:t}=d(e);t.template&&(p=t.template.content.trim()),t.scriptSetup&&(l=t.scriptSetup.content.trim());let n=o.join(s,"pages.json");if(!a.existsSync(n))return;let f=a.readFileSync(n,"utf-8"),m=JSON.parse(f),u=(c,j="")=>{c.forEach(h=>{let v=o.join(s,j,`${h.path}.vue`);g.add(o.normalize(v))})};u(m.pages),(m.subPackages??[]).forEach(c=>{u(c.pages,c.root)})},transform(i,s){if(!s.endsWith(".vue"))return;let r=o.normalize(s);if(!g.has(r))return;let{descriptor:e}=d(i),t=i;if(p&&e.template){let n=e.template.content.trim(),f=`${p}
-${n}`.trim();t=t.slice(0,e.template.loc.start.offset)+`
-${f}
-`+t.slice(e.template.loc.end.offset)}if(l&&e.scriptSetup){let n=e.scriptSetup.loc.start.offset;t=t.slice(0,n)+`
-${l}
-`+t.slice(n)}return{code:t,map:null}}}}export{k as default};