完善生产查询与登录权限配置

2026-09-03 19:56:19 +08:00
parent 793f140a45
commit 02c8fb7021
438 changed files with 1595 additions and 573 deletions
@@ -0,0 +1,78 @@
+import { resolve, join } from "node:path";
+import process from "node:process";
+import { watch, readFileSync } from "node:fs";
+import { normalizePath } from "vite";
+import { registerRootApp, rebuildRootApp } from "./root.mjs";
+import { transformPage } from "./page.mjs";
+import { loadPagesJson, normalizePlatformPath, debounce } from "./utils.mjs";
+function normalizeId(id) {
+  return normalizePath(id).replace(/^[A-Z]:/, (match) => match.toLowerCase());
+}
+function UniRoot(options) {
+  options = {
+    rootFileName: "App.root",
+    ...options
+  };
+  const rootPath = process.env.UNI_INPUT_DIR || join(process.cwd(), "src");
+  const appRootPath = resolve(rootPath, `${options.rootFileName}.vue`);
+  const pagesPath = resolve(rootPath, "pages.json");
+  let pagesJson = loadPagesJson(pagesPath, rootPath);
+  let watcher = null;
+  let hasPlatformPlugin = false;
+  const normalizedAppRoot = normalizeId(appRootPath);
+  return {
+    name: "vite-plugin-uni-root",
+    enforce: "pre",
+    // 在其他插件之前执行，确保页面模板先被包裹
+    configResolved({ plugins }) {
+      hasPlatformPlugin = plugins.some((v) => v.name === "vite-plugin-uni-platform");
+    },
+    configureServer(server) {
+      const reload = debounce(() => {
+        try {
+          pagesJson = loadPagesJson(pagesPath, rootPath);
+          server.ws.send({ type: "full-reload" });
+        } catch (e) {
+          console.error("[vite-plugin-uni-root] pages.json reload failed:", e);
+        }
+      }, 100);
+      watcher = watch(pagesPath, (eventType) => {
+        if (eventType === "change")
+          reload();
+      });
+    },
+    load(id) {
+      const normalizedId = normalizeId(id);
+      if (normalizedId === normalizedAppRoot) {
+        const code = readFileSync(appRootPath, "utf-8");
+        return rebuildRootApp(code).code;
+      }
+    },
+    async transform(code, id) {
+      const normalizedId = normalizeId(id);
+      const normalizedRootPath = normalizeId(rootPath);
+      if (normalizedId === `${normalizedRootPath}/main.ts` || normalizedId === `${normalizedRootPath}/main.js`) {
+        return registerRootApp(code, options.rootFileName);
+      }
+      if (normalizedId === normalizedAppRoot) {
+        return rebuildRootApp(code);
+      }
+      const pageId = hasPlatformPlugin ? normalizePlatformPath(normalizedId) : normalizedId;
+      const isPage = pagesJson.some((p) => pageId === p || pageId.startsWith(`${p}?`));
+      if (isPage) {
+        return transformPage(code);
+      }
+    },
+    buildEnd() {
+      if (watcher) {
+        watcher.close();
+        watcher = null;
+      }
+    }
+  };
+}
+var root_default = UniRoot;
+export {
+  UniRoot,
+  root_default as default
+};