fix: migrate nextui provider change to use regex (#133)

heroui-inc · Jan 13, 2025 · af02fbe · af02fbe
1 parent b3bf4a8
commit af02fbe
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 8 deletions.
diff --git a/packages/codemod/src/helpers/actions/migrate/migrate-common.ts b/packages/codemod/src/helpers/actions/migrate/migrate-common.ts
@@ -100,6 +100,20 @@ export function migrateJSXElementName(
   return dirtyFlag;
 }
 
+export function migrateByRegex(rawContent: string, match: string, replace: string) {
+  const regex = new RegExp(match, 'g');
+  const dirtyFlag = regex.test(rawContent);
+
+  if (dirtyFlag) {
+    rawContent = rawContent.replace(regex, replace);
+  }
+
+  return {
+    dirtyFlag,
+    rawContent
+  };
+}
+
 /**
  * Migrate the name of the CallExpression
  * @example

diff --git a/packages/codemod/src/helpers/actions/migrate/migrate-nextui-provider.ts b/packages/codemod/src/helpers/actions/migrate/migrate-nextui-provider.ts
@@ -1,7 +1,7 @@
 import {HEROUI_PROVIDER, NEXTUI_PROVIDER} from '../../../constants/prefix';
 import {getStore, writeFileAndUpdateStore} from '../../store';
 
-import {migrateImportName, migrateJSXElementName} from './migrate-common';
+import {migrateByRegex} from './migrate-common';
 
 /**
  * Migrate the NextUIProvider to HeroUIProvider will directly write the file
@@ -12,22 +12,20 @@ import {migrateImportName, migrateJSXElementName} from './migrate-common';
 export function migrateNextuiProvider(paths: string[]) {
   for (const path of paths) {
     try {
-      const parsedContent = getStore(path, 'parsedContent');
+      let rawContent = getStore(path, 'rawContent');
       let dirtyFlag = false;
 
-      if (!parsedContent) {
+      if (!rawContent) {
         continue;
       }
 
       // Replace JSX element NextUIProvider with HeroUIProvider
-      dirtyFlag = migrateJSXElementName(parsedContent, NEXTUI_PROVIDER, HEROUI_PROVIDER);
-
       // Replace NextUIProvider with HeroUIProvider in import statements
-      if (dirtyFlag) {
-        migrateImportName(parsedContent, NEXTUI_PROVIDER, HEROUI_PROVIDER);
+      ({dirtyFlag, rawContent} = migrateByRegex(rawContent, NEXTUI_PROVIDER, HEROUI_PROVIDER));
 
+      if (dirtyFlag) {
         // Write the modified content back to the file
-        writeFileAndUpdateStore(path, 'parsedContent', parsedContent);
+        writeFileAndUpdateStore(path, 'rawContent', rawContent);
       }
       // eslint-disable-next-line no-empty
     } catch {}