cerxes/babel
3
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Requeue generators for re-traversal so we pick up regeneratorRuntime - fixes T6676

Browse Source
This commit is contained in:
Logan Smyth 2015-12-13 16:15:02 -08:00
parent 9c88b8797e
commit ee6f470d77
3 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
packages/babel-plugin-transform-regenerator/src/visit.js
View File

@ -129,6 +129,11 @@ exports.visitor = {
if (wasGeneratorFunction && t.isExpression(node)) { if (wasGeneratorFunction && t.isExpression(node)) {
path.replaceWith(t.callExpression(util.runtimeProperty("mark"), [node])); path.replaceWith(t.callExpression(util.runtimeProperty("mark"), [node]));
} }
// Generators are processed in 'exit' handlers so that regenerator only has to run on
// an ES5 AST, but that means traversal will not pick up newly inserted references
// to things like 'regeneratorRuntime'. To avoid this, we explicitly re-queue.
path.requeue();
} }
} }
}; };

4
packages/babel-plugin-transform-regenerator/test/fixtures/regression/6733/expected.js vendored
View File

@ -10,7 +10,8 @@ var _marked = [_callee].map(regeneratorRuntime.mark);
function _callee() { function _callee() {
var x; var x;
return regeneratorRuntime.wrap(function _callee$(_context) { return regeneratorRuntime.wrap(function _callee$(_context) {
while (1) switch (_context.prev = _context.next) { while (1) {
switch (_context.prev = _context.next) {
case 0: case 0:
_context.next = 2; _context.next = 2;
return 5; return 5;
@ -23,5 +24,6 @@ function _callee() {
case "end": case "end":
return _context.stop(); return _context.stop();
} }
}
}, _marked[0], this); }, _marked[0], this);
} }

2
packages/babel-plugin-transform-runtime/test/fixtures/runtime/full/expected.js vendored
View File

@ -7,7 +7,7 @@ import foo, * as bar from "someModule";
export const myWord = _Symbol("abc"); export const myWord = _Symbol("abc");
export function giveWord() { export function giveWord() {
return regeneratorRuntime.wrap(function giveWord$(_context) { return _regeneratorRuntime.wrap(function giveWord$(_context) {
while (1) switch (_context.prev = _context.next) { while (1) switch (_context.prev = _context.next) {
case 0: case 0:
_context.next = 2; _context.next = 2;