Sebastian McKenzie ae7d5367f1 6.0.0
I'm extremely stupid and didn't commit as I go. To anyone reading this
I'm extremely sorry. A lot of these changes are very broad and I plan on
releasing Babel 6.0.0 today live on stage at Ember Camp London so I'm
afraid I couldn't wait. If you're ever in London I'll buy you a beer
(or assorted beverage!) to make up for it, also I'll kiss your feet and
give you a back massage, maybe.
2015-10-29 17:51:24 +00:00

47 lines
1.0 KiB
JavaScript

export default function ({ types: t }) {
function hasSpread(node) {
for (let prop of (node.properties: Array<Object>)) {
if (t.isSpreadProperty(prop)) {
return true;
}
}
return false;
}
return {
inherits: require("babel-plugin-syntax-object-rest-spread"),
visitor: {
ObjectExpression(path, file) {
if (!hasSpread(path.node)) return;
let args = [];
let props = [];
function push() {
if (!props.length) return;
args.push(t.objectExpression(props));
props = [];
}
for (let prop of (path.node.properties: Array)) {
if (t.isSpreadProperty(prop)) {
push();
args.push(prop.argument);
} else {
props.push(prop);
}
}
push();
if (!t.isObjectExpression(args[0])) {
args.unshift(t.objectExpression([]));
}
path.replaceWith(t.callExpression(file.addHelper("extends"), args));
}
}
};
}