support React stateless components

rubys · rubys · commit 87603af79b66 · 2021-06-19T10:36:28.000-04:00
Fixes #130
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,7 @@
 # master
 
 * Add chomp, delete_prefix, and delete_suffix support via ActiveFunctions
+* Support React stateless components
 
 # 4.1.4 / 2021-05-08
 
diff --git a/lib/ruby2js/filter/react.rb b/lib/ruby2js/filter/react.rb
@@ -1492,7 +1492,13 @@ def on_xstr(node)
        source = Ruby2JS.jsx2_rb(source)
        ast =  Ruby2JS.parse(source).first
        ast = s(:block, s(:send, nil, :_), s(:args), ast) if ast.type == :begin
-       process ast
+
+       begin
+         react, @react = @react, @react || :react
+         process ast
+       ensure
+         @react = react
+       end
       end
     end
 
diff --git a/spec/react_spec.rb b/spec/react_spec.rb
@@ -247,6 +247,11 @@ def to_esm(string)
         must_include 'React.createElement(React.Fragment, null, ' + 
           'React.createElement("p"), React.createElement("p"))'
     end
+
+    it "should handle stateless components" do
+      to_js( 'Button = ->(x) { %x(<button>{x}</button>)}' ).
+        must_equal 'var Button = function(x) {return React.createElement("button", null, x)}'
+    end
   end
 
   describe "render method" do
