Fix #5046: Adjacent JSX (#5049)

* Fix #5046: Adjacent JSX

* check CSX only when wrapped in parentheses

* Fix indentation

* Add test for unlikely, but valid, JSX syntax

Author: zdenko

lib/coffeescript/nodes.js

@@ -478,16 +478,21 @@ exports.Block = class Block extends Base
   # ensures that the final expression is returned.
   makeReturn: (res) ->
     len = @expressions.length
+    [..., lastExp] = @expressions
+    lastExp = lastExp?.unwrap() or no
+    # We also need to check that we’re not returning a CSX tag if there’s an
+    # adjacent one at the same level; JSX doesn’t allow that.
+    if lastExp and lastExp instanceof Parens and lastExp.body.expressions.length > 1
+      {body:{expressions}} = lastExp
+      [..., penult, last] = expressions
+      penult = penult.unwrap()
+      last = last.unwrap()
+      if penult instanceof Call and penult.csx and last instanceof Call and last.csx
+        expressions[expressions.length - 1].error 'Adjacent JSX elements must be wrapped in an enclosing tag'
     while len--
       expr = @expressions[len]
       @expressions[len] = expr.makeReturn res
       @expressions.splice(len, 1) if expr instanceof Return and not expr.expression
-      # We also need to check that we’re not returning a CSX tag if there’s an
-      # adjacent one at the same level; JSX doesn’t allow that.
-      if expr.unwrapAll().csx
-        for csxCheckIndex in [len..0]
-          if @expressions[csxCheckIndex].unwrapAll().csx
-            expr.error 'Adjacent JSX elements must be wrapped in an enclosing tag'
       break
     this
 

src/nodes.coffee

@@ -771,11 +771,11 @@ test 'JSX fragments: fragment with text nodes', ->
 
 test 'JSX fragments: fragment with component nodes', ->
   eqJS '''
-   Component = (props) =>
-     <Fragment>
-       <OtherComponent />
-       <OtherComponent />
-     </Fragment>
+    Component = (props) =>
+      <Fragment>
+        <OtherComponent />
+        <OtherComponent />
+      </Fragment>
   ''', '''
     var Component;
 
@@ -821,3 +821,20 @@ test '#5055: JSX expression indentation bug', ->
       {typeof a !== "undefined" && a !== null ? a : <span />}
     </div>;
   '''
+
+# JSX is like XML, in that there needs to be a root element; but
+# technically, adjacent top-level elements where only the last one
+# is returned (as opposed to a fragment or root element) is permissible
+# syntax. It’s almost certainly an error, but it’s valid, so need to leave it
+# to linters to catch. https://github.com/jashkenas/coffeescript/pull/5049
+test '“Adjacent” tags on separate lines should still compile', ->
+  eqJS '''
+    ->
+      <a />
+      <b />
+  ''', '''
+    (function() {
+      <a />;
+      return <b />;
+    });
+  '''

test/csx.coffee

@@ -1664,25 +1664,26 @@ test 'CSX error: invalid attributes', ->
 
 test '#5034: CSX error: Adjacent JSX elements must be wrapped in an enclosing tag', ->
   assertErrorFormat '''
-    render = ->
+    render = -> (
       <Row>a</Row>
       <Row>b</Row>
+    )
   ''', '''
     [stdin]:3:4: error: Adjacent JSX elements must be wrapped in an enclosing tag
       <Row>b</Row>
        ^^^^^^^^^^^
   '''
   assertErrorFormat '''
     render = -> (
+      a = "foo"
       <Row>a</Row>
       <Row>b</Row>
     )
   ''', '''
-    [stdin]:3:4: error: Adjacent JSX elements must be wrapped in an enclosing tag
+    [stdin]:4:4: error: Adjacent JSX elements must be wrapped in an enclosing tag
       <Row>b</Row>
        ^^^^^^^^^^^
   '''
-
 test 'Bound method called as callback before binding throws runtime error', ->
   class Base
     constructor: ->