one
Version:
One is a new React Framework that makes Vite serve both native and web.
87 lines (85 loc) • 3.69 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;
let ast;
try {
ast = parse(code, {
sourceType: "module",
plugins: ["typescript", "jsx"]
});
} catch (error) {
const errorMessage = error instanceof Error ? error.message : String(error);
console.warn(`[one] Skipping tree shaking for ${id} due to syntax error:`, errorMessage);
return;
}
let referenced;
try {
referenced = findReferencedIdentifiers(ast);
} catch (error) {
const errorMessage = error instanceof Error ? error.message : String(error);
console.warn(`[one] Skipping tree shaking for ${id} due to identifier analysis error:`, errorMessage);
return;
}
const removed = {
loader: !1,
generateStaticParams: !1
};
try {
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);
}
});
}
});
} catch (error) {
const errorMessage = error instanceof Error ? error.message : String(error);
console.warn(`[one] Skipping tree shaking for ${id} due to traversal error:`, errorMessage);
return;
}
const removedFunctions = Object.keys(removed).filter(key => removed[key]);
if (removedFunctions.length) try {
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
};
} catch (error) {
const errorMessage = error instanceof Error ? error.message : String(error);
console.warn(`[one] Skipping tree shaking for ${id} due to code generation error:`, errorMessage);
return;
}
}
export { clientTreeShakePlugin, transformTreeShakeClient };
//# sourceMappingURL=clientTreeShakePlugin.mjs.map