fix regenerator bug

Author: guybedford

…transform-regenerator@6.16.1/lib/util.js …/regenerator-transform@0.9.8/lib/util.jsbuild-babel/jspm_packages_override/npm/babel-plugin-transform-regenerator@6.16.1/lib/util.js renamed to build-babel/jspm_packages/npm/regenerator-transform@0.9.8/lib/util.js build-babel/jspm_packages_override/npm/babel-plugin-transform-regenerator@6.16.1/lib/util.js renamed to build-babel/jspm_packages/npm/regenerator-transform@0.9.8/lib/util.js

@@ -24,4 +24,4 @@ function runtimeProperty(name, regeneratorIdentifier) {
 
 function isReference(path) {
   return path.isReferenced() || path.parentPath.isAssignmentExpression({ left: path.node });
-}
+}

…ransform-regenerator@6.16.1/lib/visit.js …regenerator-transform@0.9.8/lib/visit.jsbuild-babel/jspm_packages_override/npm/babel-plugin-transform-regenerator@6.16.1/lib/visit.js renamed to build-babel/jspm_packages/npm/regenerator-transform@0.9.8/lib/visit.js build-babel/jspm_packages_override/npm/babel-plugin-transform-regenerator@6.16.1/lib/visit.js renamed to build-babel/jspm_packages/npm/regenerator-transform@0.9.8/lib/visit.js

@@ -70,7 +70,14 @@ exports.visitor = {
 
       bodyBlockPath.get("body").forEach(function (childPath) {
         var node = childPath.node;
-        if (node && node._blockHoist != null) {
+        if (t.isExpressionStatement(node) && t.isStringLiteral(node.expression)) {
+          // Babylon represents directives like "use strict" as elements
+          // of a bodyBlockPath.node.directives array, but they could just
+          // as easily be represented (by other parsers) as traditional
+          // string-literal-valued expression statements, so we need to
+          // handle that here. (#248)
+          outerBody.push(node);
+        } else if (node && node._blockHoist != null) {
           outerBody.push(node);
         } else {
           innerBody.push(node);
@@ -123,6 +130,13 @@ exports.visitor = {
       outerBody.push(t.returnStatement(wrapCall));
       node.body = t.blockStatement(outerBody);
 
+      var oldDirectives = bodyBlockPath.node.directives;
+      if (oldDirectives) {
+        // Babylon represents directives like "use strict" as elements of
+        // a bodyBlockPath.node.directives array. (#248)
+        node.body.directives = oldDirectives;
+      }
+
       var wasGeneratorFunction = node.generator;
       if (wasGeneratorFunction) {
         node.generator = false;
@@ -251,4 +265,4 @@ var awaitVisitor = {
     // can distinguish between awaited and merely yielded values.
     path.replaceWith(t.yieldExpression(t.callExpression(util.runtimeProperty("awrap", state.file.get("regeneratorIdentifier")), [argument]), false));
   }
-};
+};

test/testing-code.js

@@ -1,3 +1,14 @@
 export async function p() {
   return (await import('./testing-dep.js')).p;
 }
+
+function * responseTime (next) {
+  let start = new Date()
+  yield next
+  var ms = new Date() - start
+  this.set('X-Response-Time', ms + 'ms')
+}
+
+export default function init (app) {
+  app.use(responseTime)
+}