Use utilities from importModule (#791)

Author: trivikr

src/transforms/v2-to-v3/apis/getNodesWithDocClientNamedImportFromDeepPath.ts

@@ -7,6 +7,7 @@ import {
 } from "jscodeshift";
 import { DOCUMENT_CLIENT, DYNAMODB, OBJECT_PROPERTY_TYPE_LIST } from "../config";
 import { ImportType } from "../modules";
+import { getImportDeclarations } from "../modules/importModule";
 import { getRequireDeclarators } from "../modules/requireModule";
 import { getClientDeepImportPath } from "../utils";
 
@@ -35,16 +36,11 @@ export const getNodesWithDocClientNamedImportFromDeepPath = (
       ) as Collection<VariableDeclaration>;
   }
 
-  return source
-    .find(j.ImportDeclaration, {
-      type: "ImportDeclaration",
-      source: { value: deepImportPath },
-    })
-    .filter((importDeclaration) =>
-      (importDeclaration.value.specifiers || []).some(
-        (importDeclaration) =>
-          importDeclaration.type === "ImportSpecifier" &&
-          importDeclaration.imported.name === DOCUMENT_CLIENT
-      )
-    );
+  return getImportDeclarations(j, source, deepImportPath).filter((importDeclaration) =>
+    (importDeclaration.value.specifiers || []).some(
+      (importSpecifier) =>
+        importSpecifier.type === "ImportSpecifier" &&
+        importSpecifier.imported.name === DOCUMENT_CLIENT
+    )
+  );
 };

src/transforms/v2-to-v3/modules/hasImport.ts

@@ -1,11 +1,5 @@
 import { Collection, JSCodeshift } from "jscodeshift";
-import { PACKAGE_NAME } from "../config";
+import { getImportDeclarations } from "./importModule";
 
 export const hasImport = (j: JSCodeshift, source: Collection<unknown>) =>
-  source
-    .find(j.ImportDeclaration)
-    .filter((importDeclaration) => {
-      const { value: sourceValue } = importDeclaration.value.source;
-      return typeof sourceValue === "string" && sourceValue.startsWith(PACKAGE_NAME);
-    })
-    .size() > 0;
+  getImportDeclarations(j, source).size() > 0;