@react.componentWithProps with use memo (#7300)

* Exploring problem space

* Copy attributes from function

* Remove moved attributes from expr

* Update test

* Add changelog entry

CHANGELOG.md

@@ -36,6 +36,7 @@
 - Fix runtime type check for Object in untagged variants when one variant case is `null`. https://github.com/rescript-lang/rescript/pull/7303
 - Fix files that were being truncated when sent to the CDN over FTP. https://github.com/rescript-lang/rescript/pull/7306
 - Fix better editor completion for applications. https://github.com/rescript-lang/rescript/pull/7291
+- Fix @react.componentWithProps no longer works with @directive("'use memo'"). https://github.com/rescript-lang/rescript/pull/7300
 
 #### :house: Internal
 

compiler/syntax/src/jsx_v4.ml

@@ -1244,6 +1244,7 @@ let map_binding ~config ~empty_loc ~pstr_loc ~file_name ~rec_flag binding =
 
       let wrapper_expr =
         Exp.fun_ ~arity:None Nolabel None props_pattern
+          ~attrs:binding.pvb_expr.pexp_attributes
           (Jsx_common.async_component ~async:is_async
              (Exp.apply
                 (Exp.ident
@@ -1282,6 +1283,12 @@ let map_binding ~config ~empty_loc ~pstr_loc ~file_name ~rec_flag binding =
       {
         binding with
         pvb_attributes = binding.pvb_attributes |> List.filter other_attrs_pure;
+        pvb_expr =
+          {
+            binding.pvb_expr with
+            (* moved to wrapper_expr *)
+            pexp_attributes = [];
+          };
       },
       new_binding )
   else (None, binding, None)

tests/syntax_tests/data/ppx/react/expected/sharedPropsWithProps.res.txt

@@ -131,3 +131,15 @@ module V4A6 = {
     \"SharedPropsWithProps$V4A6"
   }
 }
+
+module V4A7 = {
+  type props = {count: int}
+
+  let make = props => {
+    React.int(props.count)
+  }
+  let make = {
+    let \"SharedPropsWithProps$V4A7" = @directive("'use memo'") props => make(props)
+    \"SharedPropsWithProps$V4A7"
+  }
+}

tests/syntax_tests/data/ppx/react/sharedPropsWithProps.res

@@ -91,3 +91,13 @@ module V4A6 = {
     }
   }
 }
+
+module V4A7 = {
+  type props = {count: int}
+
+  @react.componentWithProps
+  let make = @directive("'use memo'")
+  props => {
+    React.int(props.count)
+  }
+}

tests/tests/src/alias_default_value_test.mjs

@@ -82,6 +82,15 @@ let C7 = {
   make: Alias_default_value_test$C7
 };
 
+function Alias_default_value_test$C8(props) {
+  'use memo';
+  return props.count;
+}
+
+let C8 = {
+  make: Alias_default_value_test$C8
+};
+
 export {
   C0,
   C1,
@@ -90,5 +99,6 @@ export {
   C4,
   C6,
   C7,
+  C8,
 }
 /* No side effect */

tests/tests/src/alias_default_value_test.res

@@ -64,3 +64,14 @@ module C7 = {
       React.string(`Hello ${name}, you clicked me ` ++ times)
     }
 }
+
+module C8 = {
+  type props = {count: int}
+
+  @react.componentWithProps
+  let make =
+    @directive("'use memo'")
+    props => {
+      React.int(props.count)
+    }
+}