add few test for function components

Author: pawk3k

community/react/remove-default-props/motions/moveDefaultPropsToFunctionDeclaration.ts

@@ -6,14 +6,12 @@ export function moveDefaultPropsToFunctionDeclaration(
   j: JSCodeshift,
 ) {
   source
-    .find(j.FunctionDeclaration, {
-      id: { name: 'Component' },
-    })
+    .find(j.FunctionDeclaration)
     .forEach(component => {
       // Find the defaultProps assignment expression
       const defaultProps = source.find(j.AssignmentExpression, {
         left: {
-          object: { name: 'Component' },
+          object: { name: component.node.id?.name },
           property: { name: 'defaultProps' },
         },
       });
@@ -24,33 +22,48 @@ export function moveDefaultPropsToFunctionDeclaration(
       const defaultValues = defaultProps.get('right').value.properties;
 
       // Generate a new function parameter for each default prop
-      const params = defaultValues.map((prop: any) => {
+      const defaultParams = defaultValues.map((prop: any) => {
         const key = prop.key.name;
         const value = prop.value.value;
         // return j.objectPattern(`${key}=${JSON.stringify(value)}`);
-        return j.objectPattern([
-          j.objectProperty(
-            j.identifier(key),
-            j.assignmentPattern(j.identifier(key), j.literal(value)),
-          ),
-        ]);
+        return j.objectProperty(
+          j.identifier(key),
+          j.assignmentPattern(j.identifier(key), j.literal(value)),
+        );
       });
 
+      // Find the existing destructured props parameter, if any
+      const existingPropsParam = component.node.params.find(
+        param => param.type === 'ObjectPattern',
+      );
+
+      // If an existing props parameter was found, extract its properties
+      const existingPropsProperties =
+        existingPropsParam && 'properties' in existingPropsParam
+          ? existingPropsParam.properties.map(prop =>
+              j.property('init', (prop as any).key, (prop as any).value),
+            )
+          : [];
+
+      // Generate the new params array by concatenating the existing props properties with the default params
+      const newParams = [
+        j.objectPattern(existingPropsProperties.concat(defaultParams)),
+      ];
+
       // Replace the original function declaration with a new one
       j(component).replaceWith(nodePath =>
         j.functionDeclaration(
           nodePath.node.id,
-          params,
+          newParams,
           nodePath.node.body,
           nodePath.node.generator,
           nodePath.node.async,
         ),
       );
-      // Remove the
     })
     .find(j.AssignmentExpression, {
       left: {
-        object: { name: 'Component' },
+        object: { type: 'Identifier' },
         property: { name: 'defaultProps' },
       },
     })

community/react/remove-default-props/motions/removeDefaultPropsAssignment.ts

@@ -5,12 +5,14 @@ export function removeDefaultPropsAssignment(
   source: Collection<any>,
   j: JSCodeshift,
 ) {
-  source
-    .find(j.AssignmentExpression, {
-      left: {
-        object: { name: 'Component' },
-        property: { name: 'defaultProps' },
-      },
-    })
-    .remove();
+  const removePath = (path: any) => j(path).remove();
+  const isAssigningDefaultProps = (e: any) =>
+    e.node.left &&
+    e.node.left.property &&
+    e.node.left.property.name === 'defaultProps';
+
+  return source
+    .find(j.AssignmentExpression)
+    .filter(isAssigningDefaultProps)
+    .forEach(removePath);
 }