79 lines
2.6 KiB
JavaScript
79 lines
2.6 KiB
JavaScript
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
|
|
};
|