Update core import to NEW_CORE_IMPORT_PATH

mondaycom · Oct 16, 2024 · 1541e5e · 1541e5e
1 parent 2b5bee5
commit 1541e5e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/packages/codemod/src/utils/import-utils.ts b/packages/codemod/src/utils/import-utils.ts
@@ -15,8 +15,8 @@ export function getImports(root: Collection, path: string): Collection<ImportDec
 /**
  * Retrieves all "monday-ui-react-core" import declarations for a root.
  */
-export function getCoreImportsForFile(root: Collection): Collection<ImportDeclaration> {
-  return getImports(root, CORE_IMPORT_PATH);
+export function getCoreImportsForFile(root: Collection, path = CORE_IMPORT_PATH): Collection<ImportDeclaration> {
+  return getImports(root, path);
 }
 
 /**

diff --git a/packages/codemod/transformations/core/v2-to-v3/optional/Enums-migration.ts b/packages/codemod/transformations/core/v2-to-v3/optional/Enums-migration.ts
@@ -1,14 +1,16 @@
 import { TransformationContext } from "../../../../types";
 import { getCoreImportsForFile, getPropValue, setPropValue, wrap } from "../../../../src/utils";
 import enumToStringMapping from "./enumMappings.json";
+import { NEW_CORE_IMPORT_PATH } from "../../../../src/consts";
 
 const enumToString: Record<string, string> = enumToStringMapping;
 
 /**
  * Replace enums with string equivalent
  */
 function transform({ j, root }: TransformationContext) {
-  const coreImports = getCoreImportsForFile(root);
+  // Since it runs after the imports are updated need to get the new import path
+  const coreImports = getCoreImportsForFile(root, NEW_CORE_IMPORT_PATH);
 
   const importedComponents = coreImports
     .find(j.ImportSpecifier)
@@ -21,7 +23,7 @@ function transform({ j, root }: TransformationContext) {
     })
     .filter(Boolean);
 
-  const allElements = root.find(j.JSXElement).nodes(); // Get the nodes array
+  const allElements = root.find(j.JSXElement).nodes();
 
   const elements = allElements.filter(path => {
     const openingElement = path.openingElement;