Support spread attribute

mui · Apr 21, 2023 · 1861be3 · 1861be3
1 parent 3d92241
commit 1861be3
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 5 deletions.
diff --git a/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.js b/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.js
@@ -23,7 +23,14 @@ export default function transformer(file, api, options) {
           let slotPropNodeInserted = false;
           let slotsPropNode;
           elementPath.node.openingElement.attributes.forEach((attributeNode) => {
-            if (attributeNode.type !== 'JSXAttribute') {
+            if (
+              attributeNode.type !== 'JSXAttribute' &&
+              attributeNode.type !== 'JSXSpreadAttribute'
+            ) {
+              return;
+            }
+            if (attributeNode.type === 'JSXSpreadAttribute') {
+              attributeNodes.push(attributeNode);
               return;
             }
             const attributeName = attributeNode.name.name;

diff --git a/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/actual.jsx b/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/actual.jsx
@@ -7,7 +7,7 @@ import Badge from '@mui/base/Badge';
 
 <Input component={CustomRoot} />;
 
-<Input component={CustomRoot}></Input>;
+<Input component={CustomRoot} {...others}></Input>;
 
 <Switch
   component={CustomRoot}

diff --git a/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/actual.tsx b/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/actual.tsx
@@ -8,7 +8,7 @@ import Badge from '@mui/base/Badge';
 
 <Input component={CustomRoot} />;
 
-<Input component={CustomRoot}></Input>;
+<Input component={CustomRoot} {...others}></Input>;
 
 <Switch
   component={CustomRoot}

diff --git a/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/expected.jsx b/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/expected.jsx
@@ -11,7 +11,7 @@ import Badge from '@mui/base/Badge';
 
 <Input slots={{
   root: CustomRoot
-}}></Input>;
+}} {...others}></Input>;
 
 <Switch
   slots={{

diff --git a/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/expected.tsx b/packages/mui-codemod/src/v5.0.0/base-remove-component-prop.test/expected.tsx
@@ -12,7 +12,7 @@ import Badge from '@mui/base/Badge';
 
 <Input<typeof CustomRoot> slots={{
   root: CustomRoot
-}}></Input>;
+}} {...others}></Input>;
 
 <Switch<typeof CustomRoot>
   slots={{