fix: check for deep imports in import equals

aws · Oct 20, 2023 · 5381967 · 5381967
1 parent 35db270
commit 5381967
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/transforms/v2-to-v3/modules/importEqualsModule/addDefaultModule.ts b/src/transforms/v2-to-v3/modules/importEqualsModule/addDefaultModule.ts
@@ -31,10 +31,16 @@ export const addDefaultModule = (
     j.tsExternalModuleReference(j.stringLiteral(packageName))
   );
 
-  const v2ImportEquals = source.find(
-    j.TSImportEqualsDeclaration,
-    getImportEqualsDeclarationType(PACKAGE_NAME)
-  );
+  const v2ImportEquals = source
+    .find(j.TSImportEqualsDeclaration, getImportEqualsDeclarationType())
+    .filter((importEqualsDeclaration) => {
+      const { moduleReference } = importEqualsDeclaration.value;
+      if (moduleReference.type !== "TSExternalModuleReference") return false;
+      const { expression } = moduleReference;
+      if (expression.type !== "StringLiteral") return false;
+      const { value } = expression;
+      return value.startsWith(PACKAGE_NAME);
+    });
 
   if (v2ImportEquals.size()) {
     // Insert it after the first import equals declaration.