add build comprehension helper

lib/6to5/transformation/helpers/build-comprehension.js

@@ -0,0 +1,23 @@
+var t = require("../../types");
+
+module.exports = function build(node, buildBody) {
+  var self = node.blocks.shift();
+  if (!self) return;
+
+  var child = build(node, buildBody);
+  if (!child) {
+    // last item
+    child = buildBody();
+
+    // add a filter as this is our final stop
+    if (node.filter) {
+      child = t.ifStatement(node.filter, t.blockStatement([child]));
+    }
+  }
+
+  return t.forOfStatement(
+    t.variableDeclaration("let", [t.variableDeclarator(self.left)]),
+    self.right,
+    t.blockStatement([child])
+  );
+};

lib/6to5/transformation/transformers/es7-array-comprehension.js

@@ -1,6 +1,7 @@
-var traverse = require("../../traverse");
-var util     = require("../../util");
-var t        = require("../../types");
+var buildComprehension = require("../helpers/build-comprehension");
+var traverse           = require("../../traverse");
+var util               = require("../../util");
+var t                  = require("../../types");
 
 exports.experimental = true;
 
@@ -22,7 +23,7 @@ var build = function (node, parent, file, scope) {
 
   var returnStatement = body.pop();
 
-  body.push(exports._build(node, function () {
+  body.push(buildComprehension(node, function () {
     return util.template("array-push", {
       STATEMENT: node.body,
       KEY:       uid
@@ -33,28 +34,6 @@ var build = function (node, parent, file, scope) {
   return container;
 };
 
-exports._build = function (node, buildBody) {
-  var self = node.blocks.shift();
-  if (!self) return;
-
-  var child = exports._build(node, buildBody);
-  if (!child) {
-    // last item
-    child = buildBody();
-
-    // add a filter as this is our final stop
-    if (node.filter) {
-      child = t.ifStatement(node.filter, t.blockStatement([child]));
-    }
-  }
-
-  return t.forOfStatement(
-    t.variableDeclaration("let", [t.variableDeclarator(self.left)]),
-    self.right,
-    t.blockStatement([child])
-  );
-};
-
 exports.ComprehensionExpression = function (node, parent, file, scope) {
   if (node.generator) return;
 

lib/6to5/transformation/transformers/es7-generator-comprehension.js

@@ -1,4 +1,4 @@
-var arrayComprehension = require("./es7-array-comprehension");
+var buildComprehension = require("../helpers/build-comprehension");
 var t                  = require("../../types");
 
 exports.experimental = true;
@@ -10,7 +10,7 @@ exports.ComprehensionExpression = function (node) {
   var container = t.functionExpression(null, [], t.blockStatement(body), true);
   container._aliasFunction = true;
 
-  body.push(arrayComprehension._build(node, function () {
+  body.push(buildComprehension(node, function () {
     return t.expressionStatement(t.yieldExpression(node.body));
   }));