use regex

mui · Feb 29, 2024 · 0a6182b · 0a6182b
1 parent 78b8474
commit 0a6182b
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 21 deletions.
diff --git a/packages/mui-codemod/src/deprecations/button-group-classes/button-group-classes.js b/packages/mui-codemod/src/deprecations/button-group-classes/button-group-classes.js
@@ -106,14 +106,17 @@ export default function transformer(file, api, options) {
         });
       });
 
-    const selector = `${replacementSelectorPrefix}${deprecatedClass}`;
+    const selectorRegex = new RegExp(`${replacementSelectorPrefix}${deprecatedClass}($)`);
     root
-      .find(j.Literal, (literal) => typeof literal.value === 'string' && literal.value === selector)
+      .find(
+        j.Literal,
+        (literal) => typeof literal.value === 'string' && literal.value.match(selectorRegex),
+      )
       .forEach((path) => {
         path.replace(
           j.literal(
             path.value.value.replace(
-              selector,
+              selectorRegex,
               `${replacementSelectorPrefix}${replacementSelector}`,
             ),
           ),

diff --git a/packages/mui-codemod/src/deprecations/button-group-classes/button-group-classes.test.js b/packages/mui-codemod/src/deprecations/button-group-classes/button-group-classes.test.js
@@ -20,7 +20,7 @@ describe('@mui/codemod', () => {
           const actual = jsTransform(
             { source: read('./test-cases/actual.js') },
             { jscodeshift },
-            { printOptions: { quote: 'double', trailingComma: true } },
+            { printOptions: { quote: 'single', trailingComma: true } },
           );
 
           const expected = read('./test-cases/expected.js');

diff --git a/packages/mui-codemod/src/deprecations/button-group-classes/test-cases/expected.js b/packages/mui-codemod/src/deprecations/button-group-classes/test-cases/expected.js
@@ -1,22 +1,22 @@
 import { buttonGroupClasses } from '@mui/material/ButtonGroup';
 
-("&.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-vertical > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-text > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-text.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-text.MuiButtonGroup-vertical > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-text.MuiButtonGroup-colorPrimary > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-text.MuiButtonGroup-colorSecondary > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-outlined > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-outlined.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-outlined.MuiButtonGroup-vertical > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-outlined.MuiButtonGroup-colorPrimary > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-outlined.MuiButtonGroup-colorSecondary > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-contained > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-contained.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-contained.MuiButtonGroup-vertical > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-contained.MuiButtonGroup-colorPrimary > .MuiButtonGroup-grouped");
-("&.MuiButtonGroup-contained.MuiButtonGroup-colorSecondary > .MuiButtonGroup-grouped");
+('&.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-vertical > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-text > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-text.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-text.MuiButtonGroup-vertical > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-text.MuiButtonGroup-colorPrimary > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-text.MuiButtonGroup-colorSecondary > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-outlined > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-outlined.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-outlined.MuiButtonGroup-vertical > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-outlined.MuiButtonGroup-colorPrimary > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-outlined.MuiButtonGroup-colorSecondary > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-contained > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-contained.MuiButtonGroup-horizontal > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-contained.MuiButtonGroup-vertical > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-contained.MuiButtonGroup-colorPrimary > .MuiButtonGroup-grouped');
+('&.MuiButtonGroup-contained.MuiButtonGroup-colorSecondary > .MuiButtonGroup-grouped');
 `&.${buttonGroupClasses.horizontal} > .${buttonGroupClasses.grouped}`;
 `&.${buttonGroupClasses.vertical} > .${buttonGroupClasses.grouped}`;
 `&.${buttonGroupClasses.text} > .${buttonGroupClasses.grouped}`;