Dan Harper 682e9658c9 Fix bodyless async functions (#4600)
* fix bodyless async functions (#4599)

* Do the same in the async-generator-functions transform
2016-09-29 12:01:37 +02:00

37 lines
1.1 KiB
JavaScript

import remapAsyncToGenerator from "babel-helper-remap-async-to-generator";
export default function ({ types: t }) {
let yieldStarVisitor = {
Function(path) {
path.skip();
},
YieldExpression({ node }, state) {
if (!node.delegate) return;
let callee = state.addHelper("asyncGeneratorDelegate");
node.argument = t.callExpression(callee, [
t.callExpression(state.addHelper("asyncIterator"), [node.argument]),
t.memberExpression(state.addHelper("asyncGenerator"), t.identifier("await"))
]);
}
};
return {
inherits: require("babel-plugin-syntax-async-generators"),
visitor: {
Function(path, state) {
if (!path.node.async || !path.node.generator) return;
path.traverse(yieldStarVisitor, state);
remapAsyncToGenerator(path, state.file, {
wrapAsync: t.memberExpression(
state.addHelper("asyncGenerator"), t.identifier("wrap")),
wrapAwait: t.memberExpression(
state.addHelper("asyncGenerator"), t.identifier("await"))
});
}
}
};
}