85 lines
2.2 KiB
JavaScript
85 lines
2.2 KiB
JavaScript
import { declare } from "@babel/helper-plugin-utils";
|
|
import pull from "lodash/pull";
|
|
import { types as t } from "@babel/core";
|
|
|
|
export default declare(api => {
|
|
api.assertVersion(7);
|
|
|
|
function isProtoKey(node) {
|
|
return t.isLiteral(t.toComputedKey(node, node.key), { value: "__proto__" });
|
|
}
|
|
|
|
function isProtoAssignmentExpression(node) {
|
|
const left = node.left;
|
|
return (
|
|
t.isMemberExpression(left) &&
|
|
t.isLiteral(t.toComputedKey(left, left.property), { value: "__proto__" })
|
|
);
|
|
}
|
|
|
|
function buildDefaultsCallExpression(expr, ref, file) {
|
|
return t.expressionStatement(
|
|
t.callExpression(file.addHelper("defaults"), [ref, expr.right]),
|
|
);
|
|
}
|
|
|
|
return {
|
|
name: "transform-proto-to-assign",
|
|
|
|
visitor: {
|
|
AssignmentExpression(path, file) {
|
|
if (!isProtoAssignmentExpression(path.node)) return;
|
|
|
|
const nodes = [];
|
|
const left = path.node.left.object;
|
|
const temp = path.scope.maybeGenerateMemoised(left);
|
|
|
|
if (temp) {
|
|
nodes.push(
|
|
t.expressionStatement(t.assignmentExpression("=", temp, left)),
|
|
);
|
|
}
|
|
nodes.push(
|
|
buildDefaultsCallExpression(
|
|
path.node,
|
|
t.cloneNode(temp || left),
|
|
file,
|
|
),
|
|
);
|
|
if (temp) nodes.push(t.cloneNode(temp));
|
|
|
|
path.replaceWithMultiple(nodes);
|
|
},
|
|
|
|
ExpressionStatement(path, file) {
|
|
const expr = path.node.expression;
|
|
if (!t.isAssignmentExpression(expr, { operator: "=" })) return;
|
|
|
|
if (isProtoAssignmentExpression(expr)) {
|
|
path.replaceWith(
|
|
buildDefaultsCallExpression(expr, expr.left.object, file),
|
|
);
|
|
}
|
|
},
|
|
|
|
ObjectExpression(path, file) {
|
|
let proto;
|
|
const { node } = path;
|
|
|
|
for (const prop of (node.properties: Array)) {
|
|
if (isProtoKey(prop)) {
|
|
proto = prop.value;
|
|
pull(node.properties, prop);
|
|
}
|
|
}
|
|
|
|
if (proto) {
|
|
const args = [t.objectExpression([]), proto];
|
|
if (node.properties.length) args.push(node);
|
|
path.replaceWith(t.callExpression(file.addHelper("extends"), args));
|
|
}
|
|
},
|
|
},
|
|
};
|
|
});
|