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

40 lines
1.1 KiB
JavaScript

import * as t from "babel-types";
import nameFunction from "babel-helper-function-name";
export default function () {
return {
visitor: {
"ArrowFunctionExpression|FunctionExpression": {
exit(path) {
if (path.key !== "value" && !path.parentPath.isObjectProperty()) {
let replacement = nameFunction(path);
if (replacement) path.replaceWith(replacement);
}
}
},
ObjectExpression(path) {
let props: Array<Object> = path.get("properties");
for (let prop of props) {
if (prop.isObjectMethod({ kind: "method", computed: false })) {
let node = prop.node;
prop.replaceWith(t.objectProperty(
node.key,
t.functionExpression(null, node.params, node.body, node.generator, node.async)
));
}
if (prop.isObjectProperty()) {
let value = prop.get("value");
if (value.isFunction()) {
let newNode = nameFunction(value);
if (newNode) value.replaceWith(newNode);
}
}
}
}
}
};
}