55 lines
1.6 KiB
TypeScript
55 lines
1.6 KiB
TypeScript
import { declare } from "@babel/helper-plugin-utils";
|
|
import syntaxLogicalAssignmentOperators from "@babel/plugin-syntax-logical-assignment-operators";
|
|
import { types as t } from "@babel/core";
|
|
|
|
export default declare(api => {
|
|
api.assertVersion(7);
|
|
|
|
return {
|
|
name: "proposal-logical-assignment-operators",
|
|
inherits: syntaxLogicalAssignmentOperators.default,
|
|
|
|
visitor: {
|
|
AssignmentExpression(path) {
|
|
const { node, scope } = path;
|
|
const { operator, left, right } = node;
|
|
const operatorTrunc = operator.slice(0, -1);
|
|
if (!t.LOGICAL_OPERATORS.includes(operatorTrunc)) {
|
|
return;
|
|
}
|
|
|
|
const lhs = t.cloneNode(left);
|
|
if (t.isMemberExpression(left)) {
|
|
const { object, property, computed } = left;
|
|
const memo = scope.maybeGenerateMemoised(object);
|
|
if (memo) {
|
|
left.object = memo;
|
|
lhs.object = t.assignmentExpression("=", t.cloneNode(memo), object);
|
|
}
|
|
|
|
if (computed) {
|
|
const memo = scope.maybeGenerateMemoised(property);
|
|
if (memo) {
|
|
left.property = memo;
|
|
lhs.property = t.assignmentExpression(
|
|
"=",
|
|
t.cloneNode(memo),
|
|
// @ts-expect-error todo(flow->ts): property can be t.PrivateName
|
|
property,
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
path.replaceWith(
|
|
t.logicalExpression(
|
|
operatorTrunc,
|
|
lhs,
|
|
t.assignmentExpression("=", left, right),
|
|
),
|
|
);
|
|
},
|
|
},
|
|
};
|
|
});
|