91 lines
2.6 KiB
JavaScript
91 lines
2.6 KiB
JavaScript
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);
|
||
};
|