Split getV2DefaultModuleName into import/require components (#71)

Author: trivikr

.changeset/tasty-monkeys-turn.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Split getV2DefaultModuleName into import/require components

src/transforms/v2-to-v3/utils/getV2DefaultImportName.ts

@@ -0,0 +1,11 @@
+import { Collection, JSCodeshift } from "jscodeshift";
+
+export const getV2DefaultImportName = (
+  j: JSCodeshift,
+  source: Collection<any>
+): string | undefined =>
+  source
+    .find(j.ImportDeclaration, {
+      source: { value: "aws-sdk" },
+    })
+    .nodes()[0]?.specifiers[0]?.local.name;

src/transforms/v2-to-v3/utils/getV2DefaultModuleName.ts

@@ -1,40 +1,13 @@
-import { Collection, Identifier, JSCodeshift } from "jscodeshift";
+import { Collection, JSCodeshift } from "jscodeshift";
+
+import { containsRequire } from "./containsRequire";
+import { getV2DefaultImportName } from "./getV2DefaultImportName";
+import { getV2DefaultRequireName } from "./getV2DefaultRequireName";
 
 export const getV2DefaultModuleName = (
   j: JSCodeshift,
   source: Collection<any>
-): string | undefined => {
-  let v2DefaultModuleName = undefined;
-
-  // Set specifier name in v2DefaultImportName if it is imported in the source.
-  source
-    .find(j.ImportDeclaration, {
-      source: { value: "aws-sdk" },
-    })
-    .forEach((declerationPath) => {
-      declerationPath.value.specifiers.forEach((specifier) => {
-        if (
-          specifier.type === "ImportDefaultSpecifier" ||
-          specifier.type === "ImportNamespaceSpecifier"
-        ) {
-          v2DefaultModuleName = specifier.local.name;
-        }
-      });
-    });
-
-  // Set specifier name in v2DefaultImportName if it is required in the source.
-  source
-    .find(j.VariableDeclarator, {
-      id: { type: "Identifier" },
-      init: {
-        type: "CallExpression",
-        callee: { type: "Identifier", name: "require" },
-        arguments: [{ type: "Literal", value: "aws-sdk" }],
-      },
-    })
-    .forEach((declerationPath) => {
-      v2DefaultModuleName = (declerationPath.value.id as Identifier).name;
-    });
-
-  return v2DefaultModuleName;
-};
+): string | undefined =>
+  containsRequire(j, source)
+    ? getV2DefaultRequireName(j, source)
+    : getV2DefaultImportName(j, source);

src/transforms/v2-to-v3/utils/getV2DefaultRequireName.ts

@@ -0,0 +1,18 @@
+import { Collection, Identifier, JSCodeshift } from "jscodeshift";
+
+export const getV2DefaultRequireName = (
+  j: JSCodeshift,
+  source: Collection<any>
+): string | undefined =>
+  (
+    source
+      .find(j.VariableDeclarator, {
+        id: { type: "Identifier" },
+        init: {
+          type: "CallExpression",
+          callee: { type: "Identifier", name: "require" },
+          arguments: [{ type: "Literal", value: "aws-sdk" }],
+        },
+      })
+      .nodes()[0]?.id as Identifier
+  )?.name;