Merge pull request #1670 from DanielXMoore/yield-star-pipe

`yield*` in pipeline
DanielXMoore · Jan 5, 2025 · 3514f86 · 3514f86
2 parents f5a7224 + c07f970
commit 3514f86
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/civet.dev/reference.md b/civet.dev/reference.md
@@ -900,7 +900,7 @@ Use `as T` to cast types in your pipeline:
 data |> JSON.parse |> as MyRecord |> addRecord
 </Playground>
 
-Use `await`, `throw`, `yield`, or `return` in your pipeline:
+Use `await`, `throw`, `yield`, `yield*`, or `return` in your pipeline:
 
 <Playground>
 fetch url |> await

diff --git a/source/parser.hera b/source/parser.hera
@@ -966,7 +966,9 @@ PipelineHeadItem
   OptimizedParenthesizedExpression
 
 PipelineTailItem
-  ( AwaitOp / Yield / Return / Throw ) !AccessStart !MaybeParenNestedExpression -> $1
+  ( AwaitOp / Return / Throw ) !AccessStart !MaybeParenNestedExpression -> $1
+  $( Yield ( _? Star )? ) !AccessStart !MaybeParenNestedExpression ->
+    return { $loc, token: $1, type: "Yield" }
   "import" !AccessStart ->
     return {
       type: "Identifier",

diff --git a/test/pipe.civet b/test/pipe.civet
@@ -159,6 +159,20 @@ describe "pipe", ->
     }
   """
 
+  testCase """
+    yield in pipeline
+    ---
+    [x, y]
+    |> .map (+ 1)
+    |> yield*
+    ---
+    yield*([x, y].map((a => a+ 1)))
+  """, wrapper: """
+    function* f() {
+      CODE
+    }
+  """
+
   testCase """
     throw statement in pipeline
     ---