Merge pull request #1468 from DanielXMoore/return-if

Forbid next-line braced blocks after `return if` and `yield if`

source/parser.hera

@@ -2650,7 +2650,7 @@ EmptyBareBlock
 
 NoBlock
   # Check that there isn't a block here. Used for empty loop bodies.
-  &EOS !IndentedFurther
+  &EOS !IndentedFurther !( Nested Then )
 
 # A nonempty block that must include braces
 # This version allows same-line postfixes like `while cond`.
@@ -5209,10 +5209,11 @@ KeywordStatement
 
   # https://262.ecma-international.org/#prod-ReturnStatement
   # NOTE: Modified to leave room for `return.value` and `return =`
-  Return !(":" / "." / AfterReturnShorthand) MaybeParenNestedExpression?:expression -> {
+  # NOTE: Stop parsing if there's a postfix if/etc.
+  Return:ret !(":" / "." / AfterReturnShorthand) MaybeParenNestedExpression?:expression -> {
     type: "ReturnStatement",
     expression,
-    children: $0,
+    children: [ ret, expression ],
   }
 
   ThrowStatement
@@ -5274,6 +5275,8 @@ NestedExpression
 # parentheses if indented, so that the expression starts on the same line.
 # (e.g. for `return` or `yield`)
 MaybeParenNestedExpression
+  # Skip if there's a postfix if/etc.
+  &( _? PostfixStatement NoBlock ) -> ""
   # Not nested case
   !EOS Expression -> $2
   # Avoid wrapping array/object return value in parentheses.

test/if.civet

@@ -560,6 +560,40 @@ describe "if", ->
     }
   """
 
+  testCase """
+    return postfix if with braced expression after
+    ---
+    =>
+      return if x
+      {
+        x
+      }
+    ---
+    () => {
+      if (x) { return }
+      return ({
+        x
+      })
+    }
+  """
+
+  testCase """
+    yield postfix if with braced expression after
+    ---
+    ->
+      yield if x
+      {
+        x
+      }
+    ---
+    (function*() {
+      if (x) { yield };
+      ({
+        x
+      })
+    })
+  """
+
   testCase """
     implied parens parethesized expression
     ---
@@ -954,15 +988,15 @@ describe "if", ->
 
   describe "return if expression", ->
     testCase """
-      return if expression
+      one line
       ---
       return if y then 1 else 0
       ---
       return (y? 1 : 0)
     """
 
     testCase """
-      return if expression
+      if/then/else
       ---
       return if y
       then 1
@@ -973,7 +1007,7 @@ describe "if", ->
     """
 
     testCase """
-      return if expression
+      indented if/else
       ---
       return if y
         1