use regex

mui · Feb 29, 2024 · a17eea0 · a17eea0
1 parent 78b8474
commit a17eea0
Showing 1 changed file with 6 additions and 3 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}`,
             ),
           ),