Merge pull request #68 from sljuka/replace-default-props-implicit-return

feat(replace-default-props): make it work for function components with implicit return

Author: amirabbas-gh

codemods/react/19/replace-default-props/.codemodrc.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://codemod-utils.s3.us-west-1.amazonaws.com/configuration_schema.json",
   "name": "react/19/replace-default-props",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "engine": "jscodeshift",
   "private": false,
   "arguments": [],

codemods/react/19/replace-default-props/__testfixtures__/implicit-return.input.jsx

@@ -0,0 +1,13 @@
+const Card = ({ user: { name, age } }) => (
+  <div>
+    <p>{name}</p>
+    <p>{age}</p>
+  </div>
+);
+
+Card.defaultProps = {
+  user: {
+    name: "Unknown",
+    age: 0,
+  },
+};

codemods/react/19/replace-default-props/__testfixtures__/implicit-return.output.jsx

@@ -0,0 +1,11 @@
+const cardDefaultPropUser = {
+  name: "Unknown",
+  age: 0,
+};
+
+const Card = ({ user: { name, age } = cardDefaultPropUser }) => (
+  <div>
+    <p>{name}</p>
+    <p>{age}</p>
+  </div>
+);

codemods/react/19/replace-default-props/src/index.ts

@@ -71,14 +71,15 @@ export default function transform(
       return;
     }
 
-    const componentFunction = j.functionDeclaration(
-      j.identifier(componentName),
-      path.value.params,
-      path.value.body
-    );
-
-    if (componentFunction === null) {
-      return;
+    let componentFunction = null;
+    const isImplicitReturnComponent = path.value.body.type === "JSXElement";
+
+    if (!isImplicitReturnComponent) {
+      componentFunction = j.functionDeclaration(
+        j.identifier(componentName),
+        path.value.params,
+        path.value.body
+      );
     }
 
     const defaultProps = getComponentStaticPropValue(
@@ -168,7 +169,7 @@ export default function transform(
       );
     }
 
-    if (propsArgName && inlineDefaultProps.length) {
+    if (componentFunction && propsArgName && inlineDefaultProps.length) {
       componentFunction.body.body.unshift(
         j.expressionStatement(
           j.assignmentExpression(

codemods/react/19/replace-default-props/test/test.ts

@@ -262,4 +262,28 @@ describe("react/19/replace-default-props", () => {
       OUTPUT.replace(/W/gm, "")
     );
   });
+
+  it("transforms functional components defined via implicit return", async () => {
+    const INPUT = await readFile(
+      join(__dirname, "..", "__testfixtures__/implicit-return.input.jsx"),
+      "utf-8"
+    );
+    const OUTPUT = await readFile(
+      join(__dirname, "..", "__testfixtures__/implicit-return.output.jsx"),
+      "utf-8"
+    );
+
+    const actualOutput = transform(
+      {
+        path: "index.js",
+        source: INPUT,
+      },
+      buildApi("jsx")
+    );
+
+    assert.deepEqual(
+      actualOutput?.replace(/W/gm, ""),
+      OUTPUT.replace(/W/gm, "")
+    );
+  });
 });