From a17eea0742261fa99be2997b572764a3eb29dfff Mon Sep 17 00:00:00 2001
From: sai6855 <saichand685@gmail.com>
Date: Thu, 29 Feb 2024 11:34:55 +0530
Subject: [PATCH] use regex

---
 .../button-group-classes/button-group-classes.js         | 9 ++++++---
 1 file changed, 6 insertions(+), 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
index b64f856fdc07cb..440370cb47b623 100644
--- 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}`,
             ),
           ),