one
Version:
One is a new React Framework that makes Vite serve both native and web.
102 lines (101 loc) • 4.37 kB
JavaScript
import BabelGenerate from "@babel/generator";
import { parse } from "@babel/parser";
import BabelTraverse from "@babel/traverse";
import * as t from "@babel/types";
import { deadCodeElimination, findReferencedIdentifiers } from "babel-dead-code-elimination";
function _instanceof(left, right) {
if (right != null && typeof Symbol !== "undefined" && right[Symbol.hasInstance]) {
return !!right[Symbol.hasInstance](left);
} else {
return left instanceof right;
}
}
var generate = BabelGenerate["default"] || BabelGenerate;
var traverse = BabelTraverse["default"] || BabelTraverse;
function removeServerCodePlugin(_, options) {
var {
routerRoot = "app"
} = options;
return {
name: "one-remove-server-code",
visitor: {
Program: {
exit(path, state) {
var filename = state.filename;
if (!filename) return;
var routerRootPattern = new RegExp(`[/\\\\]${routerRoot}[/\\\\]`);
if (!routerRootPattern.test(filename)) return;
if (filename.includes("node_modules")) return;
var code = path.toString();
if (!/generateStaticParams|loader/.test(code)) return;
try {
var freshAst = parse(code, {
sourceType: "module",
plugins: ["typescript", "jsx"]
});
var referenced = findReferencedIdentifiers(freshAst);
var removed = {
loader: false,
generateStaticParams: false
};
traverse(freshAst, {
ExportNamedDeclaration(expPath) {
var declaration = expPath.node.declaration;
if (!declaration) return;
if (declaration.type === "FunctionDeclaration" && declaration.id) {
var name = declaration.id.name;
if (name === "loader" || name === "generateStaticParams") {
expPath.remove();
removed[name] = true;
}
} else if (declaration.type === "VariableDeclaration") {
var decl = expPath.get("declaration");
var declarators = decl.get("declarations");
for (var i = declarators.length - 1; i >= 0; i--) {
var declarator = declarators[i];
var id = declarator.node.id;
if (id.type === "Identifier" && (id.name === "loader" || id.name === "generateStaticParams")) {
declarator.remove();
removed[id.name] = true;
}
}
if (decl.node && decl.node.declarations.length === 0) {
expPath.remove();
}
}
}
});
var removedFunctions = Object.keys(removed).filter(function (key) {
return removed[key];
});
if (removedFunctions.length === 0) return;
deadCodeElimination(freshAst, referenced);
if (removed.loader) {
freshAst.program.body.push(t.exportNamedDeclaration(t.functionDeclaration(t.identifier("loader"), [], t.blockStatement([t.returnStatement(t.stringLiteral("__vxrn__loader__"))]))));
}
if (removed.generateStaticParams) {
freshAst.program.body.push(t.exportNamedDeclaration(t.functionDeclaration(t.identifier("generateStaticParams"), [], t.blockStatement([]))));
}
var out = generate(freshAst, {
retainLines: true
});
var finalAst = parse(out.code, {
sourceType: "module",
plugins: ["typescript", "jsx"]
});
path.node.body = finalAst.program.body;
path.node.directives = finalAst.program.directives;
if (process.env.DEBUG) {
console.info(` \u{1F9F9} [one/metro] ${filename} removed ${removedFunctions.length} server-only exports`);
}
} catch (error) {
console.warn(`[one/metro] Tree shaking failed for ${filename}:`, _instanceof(error, Error) ? error.message : String(error));
}
}
}
}
};
}
var remove_server_code_default = removeServerCodePlugin;
export { remove_server_code_default as default };
//# sourceMappingURL=remove-server-code.native.js.map