54 lines
1.4 KiB
JavaScript
54 lines
1.4 KiB
JavaScript
import { types as t } from "@babel/core";
|
|
|
|
const minimalVisitor = {
|
|
BinaryExpression(path) {
|
|
const { scope } = path;
|
|
const { node } = path;
|
|
const { operator, left } = node;
|
|
let { right } = node;
|
|
if (operator !== "|>") return;
|
|
|
|
let optimizeArrow =
|
|
t.isArrowFunctionExpression(right) &&
|
|
t.isExpression(right.body) &&
|
|
!right.async &&
|
|
!right.generator;
|
|
let param;
|
|
|
|
if (optimizeArrow) {
|
|
const { params } = right;
|
|
if (params.length === 1 && t.isIdentifier(params[0])) {
|
|
param = params[0];
|
|
} else if (params.length > 0) {
|
|
optimizeArrow = false;
|
|
}
|
|
} else if (t.isIdentifier(right, { name: "eval" })) {
|
|
right = t.sequenceExpression([t.numericLiteral(0), right]);
|
|
}
|
|
|
|
if (optimizeArrow && !param) {
|
|
// Arrow function with 0 arguments
|
|
path.replaceWith(t.sequenceExpression([left, right.body]));
|
|
return;
|
|
}
|
|
|
|
const placeholder = scope.generateUidIdentifierBasedOnNode(param || left);
|
|
scope.push({ id: placeholder });
|
|
if (param) {
|
|
path.get("right").scope.rename(param.name, placeholder.name);
|
|
}
|
|
|
|
const call = optimizeArrow
|
|
? right.body
|
|
: t.callExpression(right, [t.cloneNode(placeholder)]);
|
|
path.replaceWith(
|
|
t.sequenceExpression([
|
|
t.assignmentExpression("=", t.cloneNode(placeholder), left),
|
|
call,
|
|
]),
|
|
);
|
|
},
|
|
};
|
|
|
|
export default minimalVisitor;
|