one
Version:
One is a new React Framework that makes Vite serve both native and web.
63 lines (61 loc) • 2.74 kB
JavaScript
import BabelGenerate from "@babel/generator";
import { parse } from "@babel/parser";
import BabelTraverse from "@babel/traverse";
import { deadCodeElimination, findReferencedIdentifiers } from "babel-dead-code-elimination";
import { extname, relative } from "node:path";
import { EMPTY_LOADER_STRING } from "../constants.mjs";
const traverse = BabelTraverse.default,
generate = BabelGenerate.default,
clientTreeShakePlugin = () => ({
name: "one-client-tree-shake",
enforce: "pre",
applyToEnvironment(env) {
return env.name === "client" || env.name === "ios" || env.name === "android";
},
transform: {
order: "pre",
async handler(code, id, settings) {
return this.environment.name === "ssr" || !/\.(js|jsx|ts|tsx)/.test(extname(id)) || /node_modules/.test(id) ? void 0 : await transformTreeShakeClient(code, id);
}
}
});
async function transformTreeShakeClient(code, id) {
if (!/generateStaticParams|loader/.test(code)) return;
const ast = parse(code, {
sourceType: "module",
plugins: ["typescript", "jsx"]
}),
referenced = findReferencedIdentifiers(ast),
removed = {
loader: !1,
generateStaticParams: !1
};
traverse(ast, {
ExportNamedDeclaration(path) {
if (path.node.declaration && path.node.declaration.type === "FunctionDeclaration") {
if (!path.node.declaration.id) return;
const functionName = path.node.declaration.id.name;
(functionName === "loader" || functionName === "generateStaticParams") && (path.remove(), removed[functionName] = !0);
} else path.node.declaration && path.node.declaration.type === "VariableDeclaration" && path.node.declaration.declarations.forEach((declarator, index) => {
if (declarator.id.type === "Identifier" && (declarator.id.name === "loader" || declarator.id.name === "generateStaticParams")) {
const declaration = path.get("declaration.declarations." + index);
Array.isArray(declaration) || (declaration.remove(), removed[declarator.id.name] = !0);
}
});
}
});
const removedFunctions = Object.keys(removed).filter(key => removed[key]);
if (removedFunctions.length) {
deadCodeElimination(ast, referenced);
const out = generate(ast),
codeOut = out.code + `
` + removedFunctions.map(key => key === "loader" ? EMPTY_LOADER_STRING : "export function generateStaticParams() {};").join(`
`);
return console.info(` \u{1F9F9} [one] ${relative(process.cwd(), id)} removed ${removedFunctions.length} server-only exports`), {
code: codeOut,
map: out.map
};
}
}
export { clientTreeShakePlugin, transformTreeShakeClient };
//# sourceMappingURL=clientTreeShakePlugin.mjs.map