Create utility getV2DefaultImportName (#21)

Author: trivikr

src/transforms/v2-to-v3/transformer.ts

@@ -4,28 +4,32 @@ import findImports from "jscodeshift-find-imports";
 import {
   addV3ClientImport,
   getV2ClientNames,
+  getV2DefaultImportName,
   getV3ClientName,
   getV3ClientPackageName,
   replaceClientCreation,
 } from "./utils";
 
 export default function transformer(file: FileInfo, api: API) {
   const j = api.jscodeshift;
-  const { statement } = j.template;
   const source = j(file.source);
 
-  const imports = findImports(source, statement`import AWS from 'aws-sdk'`);
-  for (const importObj of Object.values(imports)) {
-    if (importObj.type === "Identifier") {
-      const v2ClientNames = getV2ClientNames(j, source, importObj);
+  const v2DefaultImportName = getV2DefaultImportName(j, source);
+  if (!v2DefaultImportName) {
+    return source.toSource();
+  }
+
+  const v2ClientNames = getV2ClientNames(j, source, v2DefaultImportName);
 
-      for (const v2ClientName of v2ClientNames) {
-        const v3ClientName = getV3ClientName(v2ClientName);
-        const v3ClientPackageName = getV3ClientPackageName(v2ClientName);
-        addV3ClientImport(j, source, { v3ClientName, v3ClientPackageName });
-        replaceClientCreation(j, source, { importObj, v2ClientName, v3ClientName });
-      }
-    }
+  for (const v2ClientName of v2ClientNames) {
+    const v3ClientName = getV3ClientName(v2ClientName);
+    const v3ClientPackageName = getV3ClientPackageName(v2ClientName);
+    addV3ClientImport(j, source, { v3ClientName, v3ClientPackageName });
+    replaceClientCreation(j, source, {
+      v2DefaultImportName,
+      v2ClientName,
+      v3ClientName,
+    });
   }
 
   return source.toSource();

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

@@ -3,13 +3,13 @@ import { Collection, Identifier, JSCodeshift, MemberExpression } from "jscodeshi
 export const getV2ClientNames = (
   j: JSCodeshift,
   source: Collection<any>,
-  importObj: Identifier
+  v2DefaultImportName: string
 ): Array<string> =>
   source
     .find(j.NewExpression, {
       callee: {
         type: "MemberExpression",
-        object: { type: "Identifier", name: importObj.name },
+        object: { type: "Identifier", name: v2DefaultImportName },
         property: { type: "Identifier" },
       },
     })

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

@@ -0,0 +1,16 @@
+import { Collection, JSCodeshift } from "jscodeshift";
+import findImports from "jscodeshift-find-imports";
+
+export const getV2DefaultImportName = (
+  j: JSCodeshift,
+  source: Collection<any>
+): string | undefined => {
+  const { statement } = j.template;
+  const imports = findImports(source, statement`import AWS from 'aws-sdk'`);
+
+  for (const importObj of Object.values(imports)) {
+    if (importObj.type === "Identifier") return importObj.name;
+  }
+
+  return undefined;
+};

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

@@ -1,5 +1,6 @@
 export * from "./addV3ClientImport";
 export * from "./getV2ClientNames";
+export * from "./getV2DefaultImportName";
 export * from "./getV3ClientName";
 export * from "./getV3ClientPackageName";
 export * from "./replaceClientCreation";

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

@@ -1,21 +1,21 @@
-import { Collection, Identifier, JSCodeshift } from "jscodeshift";
+import { Collection, JSCodeshift } from "jscodeshift";
 
 export interface ReplaceClientCreationOptions {
-  importObj: Identifier;
   v2ClientName: string;
   v3ClientName: string;
+  v2DefaultImportName: string;
 }
 
 // Replace v2 client creation with v3 client creation.
 export const replaceClientCreation = (
   j: JSCodeshift,
   source: Collection<any>,
-  { importObj, v2ClientName, v3ClientName }: ReplaceClientCreationOptions
+  { v2DefaultImportName, v2ClientName, v3ClientName }: ReplaceClientCreationOptions
 ): void => {
   source
     .find(j.NewExpression, {
       callee: {
-        object: { type: "Identifier", name: importObj.name },
+        object: { type: "Identifier", name: v2DefaultImportName },
         property: { type: "Identifier", name: v2ClientName },
       },
     })