chore: use getImportEqualsDeclarations in removeImportEquals

aws · Feb 29, 2024 · d41647f · d41647f
1 parent 6a91b13
commit d41647f
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/src/transforms/v2-to-v3/modules/removeImportEquals.ts b/src/transforms/v2-to-v3/modules/removeImportEquals.ts
@@ -1,4 +1,5 @@
 import { Collection, JSCodeshift } from "jscodeshift";
+import { getImportEqualsDeclarations } from "./importEqualsModule";
 import { removeDeclaration } from "./removeDeclaration";
 
 export interface RemoveImportEqualsOptions {
@@ -11,14 +12,9 @@ export const removeImportEquals = (
   source: Collection<unknown>,
   { localName, sourceValue }: RemoveImportEqualsOptions
 ) => {
-  const importEqualsDeclaration = source.find(j.TSImportEqualsDeclaration, {
-    type: "TSImportEqualsDeclaration",
-    id: { name: localName },
-    moduleReference: {
-      type: "TSExternalModuleReference",
-      expression: { type: "StringLiteral", value: sourceValue },
-    },
-  });
+  const importEqualsDeclaration = getImportEqualsDeclarations(j, source, sourceValue).filter(
+    (importEqualsDeclaration) => importEqualsDeclaration.value.id.name === localName
+  );
   if (importEqualsDeclaration.length) {
     removeDeclaration(j, source, importEqualsDeclaration.get());
   }