Add basic support for await with the pipeline operator

packages/babel-plugin-proposal-pipeline-operator/src/index.js

@@ -15,6 +15,8 @@ export default function() {
 
         let optimizeArrow =
           t.isArrowFunctionExpression(right) && t.isExpression(right.body);
+        const isAwait = t.isAwaitExpression(right);
+        const isYield = t.isYieldExpression(right);
         let param;
 
         if (optimizeArrow) {
@@ -24,6 +26,11 @@ export default function() {
           } else if (params.length > 0) {
             optimizeArrow = false;
           }
+        } else if (isAwait || isYield) {
+          right.argument = t.sequenceExpression([
+            t.numericLiteral(0),
+            right.argument,
+          ]);
         } else if (t.isIdentifier(right, { name: "eval" })) {
           right = t.sequenceExpression([t.numericLiteral(0), right]);
         }
@@ -44,7 +51,11 @@ export default function() {
 
         const call = optimizeArrow
           ? right.body
-          : t.callExpression(right, [placeholder]);
+          : isAwait || isYield
+            ? t[isAwait ? "awaitExpression" : "yieldExpression"](
+                t.callExpression(right.argument, [placeholder]),
+              )
+            : t.callExpression(right, [placeholder]);
         path.replaceWith(
           t.sequenceExpression([
             t.assignmentExpression("=", placeholder, left),

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async-with-transform-async/actual.js

@@ -0,0 +1,5 @@
+var incPromise = (x) => Promise.resolve(x + 1);
+var double = (x) => x * 2;
+
+var result = async () => 10 |> await incPromise;
+var result2 = async () => 10 |> await incPromise |> double;

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async-with-transform-async/exec.js

@@ -0,0 +1,10 @@
+var incPromise = x => Promise.resolve(x + 1);
+var double = x => x * 2;
+
+var result = async () => 10 |> await incPromise;
+var result2 = async () => 10 |> await incPromise |> double;
+
+return Promise.all([result(), result2()]).then(([r, r2]) => {
+  assert.equal(r, 11);
+  assert.equal(r2, 22);
+});

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async-with-transform-async/expected.js

@@ -0,0 +1,31 @@
+var incPromise = x => Promise.resolve(x + 1);
+
+var double = x => x * 2;
+
+var result =
+/*#__PURE__*/
+function () {
+  var _ref = babelHelpers.asyncToGenerator(function* () {
+    var _;
+
+    return _ = 10, yield (0, incPromise)(_);
+  });
+
+  return function result() {
+    return _ref.apply(this, arguments);
+  };
+}();
+
+var result2 =
+/*#__PURE__*/
+function () {
+  var _ref2 = babelHelpers.asyncToGenerator(function* () {
+    var _ref3, _2;
+
+    return _ref3 = (_2 = 10, yield (0, incPromise)(_2)), double(_ref3);
+  });
+
+  return function result2() {
+    return _ref2.apply(this, arguments);
+  };
+}();

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async-with-transform-async/options.json

@@ -0,0 +1,6 @@
+{
+  "plugins": ["proposal-pipeline-operator", "external-helpers", "transform-async-to-generator"],
+  "parserOpts": {
+    "allowReturnOutsideFunction": true
+  }
+}

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async/actual.js

@@ -0,0 +1,5 @@
+var incPromise = (x) => Promise.resolve(x + 1);
+var double = (x) => x * 2;
+
+var result = async () => 10 |> await incPromise;
+var result2 = async () => 10 |> await incPromise |> double;

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async/exec.js

@@ -0,0 +1,10 @@
+var incPromise = x => Promise.resolve(x + 1);
+var double = x => x * 2;
+
+var result = async () => 10 |> await incPromise;
+var result2 = async () => 10 |> await incPromise |> double;
+
+return Promise.all([result(), result2()]).then(([r, r2]) => {
+  assert.equal(r, 11);
+  assert.equal(r2, 22);
+});

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async/expected.js

@@ -0,0 +1,15 @@
+var incPromise = x => Promise.resolve(x + 1);
+
+var double = x => x * 2;
+
+var result = async () => {
+  var _;
+
+  return _ = 10, await (0, incPromise)(_);
+};
+
+var result2 = async () => {
+  var _ref, _2;
+
+  return _ref = (_2 = 10, await (0, incPromise)(_2)), double(_ref);
+};

packages/babel-plugin-proposal-pipeline-operator/test/fixtures/pipeline-operator/async/options.json

@@ -0,0 +1,6 @@
+{
+  "minNodeVersion": "7.6.0",
+  "parserOpts": {
+    "allowReturnOutsideFunction": true
+  }
+}