one
Version:
One is a new React Framework that makes Vite serve both native and web.
117 lines (116 loc) • 6.2 kB
JavaScript
import * as t from "@babel/types";
import { deadCodeElimination, findReferencedIdentifiers } from "babel-dead-code-elimination";
function _instanceof(left, right) {
return right != null && typeof Symbol < "u" && right[Symbol.hasInstance] ? !!right[Symbol.hasInstance](left) : left instanceof right;
}
var SERVER_EXPORTS = ["loader", "generateStaticParams"];
function removeServerCodePlugin(_, options) {
var {
routerRoot = "app"
} = options;
return {
name: "one-remove-server-code",
visitor: {
Program: {
enter(path, state) {
var filename = state.filename;
if (filename) {
var routerRootPattern = new RegExp(`[/\\\\]${routerRoot}[/\\\\]`);
if (routerRootPattern.test(filename) && !filename.includes("node_modules")) {
var code = path.toString();
if (/generateStaticParams|loader/.test(code)) try {
state.referenced = findReferencedIdentifiers(path.node);
} catch (error) {
console.warn(`[one/metro] Skipping tree shaking for ${filename} due to identifier analysis error:`, _instanceof(error, Error) ? error.message : String(error));
}
}
}
},
exit(path, state) {
var filename = state.filename;
if (filename) {
var routerRootPattern = new RegExp(`[/\\\\]${routerRoot}[/\\\\]`);
if (routerRootPattern.test(filename) && !filename.includes("node_modules") && state.referenced) {
var removedExports = /* @__PURE__ */new Set();
if (path.traverse({
ExportNamedDeclaration(exportPath) {
var declaration = exportPath.node.declaration;
if (t.isFunctionDeclaration(declaration) && declaration.id) {
var name = declaration.id.name;
SERVER_EXPORTS.includes(name) && (removedExports.add(name), exportPath.remove());
} else if (t.isVariableDeclaration(declaration)) for (var i = declaration.declarations.length - 1; i >= 0; i--) {
var declarator = declaration.declarations[i];
if (t.isIdentifier(declarator.id)) {
var name1 = declarator.id.name;
SERVER_EXPORTS.includes(name1) && (removedExports.add(name1), declaration.declarations.length === 1 ? exportPath.remove() : declaration.declarations.splice(i, 1));
}
}
},
// Also remove helper functions created by babel's async-to-generator transform
// These are named _loader, _generateStaticParams, etc.
FunctionDeclaration(funcPath) {
if (funcPath.node.id) {
var name = funcPath.node.id.name,
_iteratorNormalCompletion2 = !0,
_didIteratorError2 = !1,
_iteratorError2 = void 0;
try {
for (var _loop = function () {
var serverExport = _step2.value;
if (name === `_${serverExport}`) {
var isAsyncHelper = !1;
funcPath.traverse({
Identifier(idPath) {
idPath.node.name === "_asyncToGenerator" && (isAsyncHelper = !0, idPath.stop());
}
}), isAsyncHelper && (removedExports.add(serverExport), funcPath.remove());
}
}, _iterator2 = SERVER_EXPORTS[Symbol.iterator](), _step2; !(_iteratorNormalCompletion2 = (_step2 = _iterator2.next()).done); _iteratorNormalCompletion2 = !0) _loop();
} catch (err) {
_didIteratorError2 = !0, _iteratorError2 = err;
} finally {
try {
!_iteratorNormalCompletion2 && _iterator2.return != null && _iterator2.return();
} finally {
if (_didIteratorError2) throw _iteratorError2;
}
}
}
}
}), removedExports.size !== 0) {
try {
deadCodeElimination(path.node, state.referenced);
} catch (error) {
console.warn(`[one/metro] Dead code elimination failed for ${filename}:`, _instanceof(error, Error) ? error.message : String(error));
}
var stubs = [];
removedExports.has("loader") && stubs.push(t.exportNamedDeclaration(t.functionDeclaration(t.identifier("loader"), [], t.blockStatement([t.returnStatement(t.stringLiteral("__vxrn__loader__"))])))), removedExports.has("generateStaticParams") && stubs.push(t.exportNamedDeclaration(t.functionDeclaration(t.identifier("generateStaticParams"), [], t.blockStatement([]))));
var _iteratorNormalCompletion = !0,
_didIteratorError = !1,
_iteratorError = void 0;
try {
for (var _iterator = stubs[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = !0) {
var stub = _step.value;
path.pushContainer("body", stub);
}
} catch (err) {
_didIteratorError = !0, _iteratorError = err;
} finally {
try {
!_iteratorNormalCompletion && _iterator.return != null && _iterator.return();
} finally {
if (_didIteratorError) throw _iteratorError;
}
}
console.info(` \u{1F9F9} [one/metro] ${filename} removed ${removedExports.size} server-only exports`);
}
}
}
}
}
}
};
}
var remove_server_code_default = removeServerCodePlugin;
export { remove_server_code_default as default };
//# sourceMappingURL=remove-server-code.native.js.map