amis-rpc-design/libs/amis/scripts/transform-node-env-inline.js

91 lines
2.6 KiB
JavaScript
Raw Normal View History

2023-10-07 19:42:30 +08:00
const parse = require('@babel/parser').parse;
const traverse = require('@babel/traverse').default;
const generate = require('@babel/generator').default;
const types = require('@babel/types');
module.exports = function transformNodeEnvInline(file) {
// if (file.subpath !== '/node_modules/mobx/lib/index.js') {
// return;
// }
if (!file.isJsLike || !file.isMod) {
return;
}
let contents = file.getContent();
const idx = contents.indexOf('process.env.NODE_ENV');
if (idx === -1) {
return;
}
const idx2 = contents.indexOf('require(', idx);
if (idx2 === -1) {
return;
}
const env = file.optimizer ? 'production' : 'development';
contents = contents.replace(/typeof\s+process\b/g, () =>
JSON.stringify('object')
);
const ast = parse(contents, {
sourceType: 'module'
});
traverse(ast, {
// 参考: https://www.trickster.dev/post/javascript-ast-manipulation-with-babel-removing-unreachable-code/
// 参考https://github.com/babel/minify/blob/master/packages/babel-plugin-transform-node-env-inline/src/index.js
MemberExpression(path) {
if (path.matchesPattern('process.env.NODE_ENV')) {
path.replaceWith(types.valueToNode(env));
let parentPath = path.parentPath;
if (parentPath.isBinaryExpression()) {
const evaluated = parentPath.evaluate();
if (evaluated.confident) {
parentPath.replaceWith(types.valueToNode(evaluated.value));
}
}
let ifStatement = null;
while (parentPath) {
if (
parentPath.isIfStatement() ||
parentPath.isConditionalExpression()
) {
ifStatement = parentPath;
}
parentPath = parentPath.parentPath;
}
if (ifStatement) {
let isTruthy = ifStatement.get('test').evaluateTruthy();
const node = ifStatement.node;
if (isTruthy) {
if (types.isBlockStatement(node.consequent)) {
ifStatement.replaceWithMultiple(node.consequent.body);
} else {
ifStatement.replaceWith(node.consequent);
}
} else if (node.alternate != null) {
if (types.isBlockStatement(node.alternate)) {
ifStatement.replaceWithMultiple(node.alternate.body);
} else {
ifStatement.replaceWith(node.alternate);
}
} else {
ifStatement.remove();
}
}
}
}
});
contents = generate(ast, {}).code;
file.setContent(contents);
// console.log('\n', file.subpath, '\n', file.getContent());
// process.exit(1);
};