Support transformation for global require (#62)

Author: trivikr

.changeset/lemon-mangos-remember.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": minor
+---
+
+Support transformation for global require

src/transforms/v2-to-v3/__fixtures__/global-require.input.ts

@@ -0,0 +1,3 @@
+const AWS = require("aws-sdk");
+
+const client = new AWS.DynamoDB();

src/transforms/v2-to-v3/__fixtures__/global-require.output.ts

@@ -0,0 +1,5 @@
+const {
+  DynamoDB
+} = require("@aws-sdk/client-dynamodb");
+
+const client = new DynamoDB();

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

@@ -1,12 +1,12 @@
 import { API, FileInfo } from "jscodeshift";
 
 import {
-  addV3ClientImport,
+  addV3ClientModule,
   getClientMetadata,
   getV2ClientImportNames,
   getV2ClientNames,
   getV2DefaultImportName,
-  removeDefaultImportIfNotUsed,
+  removeDefaultModuleIfNotUsed,
   removePromiseCalls,
   removeV2ClientImport,
   replaceClientCreation,
@@ -27,13 +27,13 @@ export default function transformer(file: FileInfo, api: API) {
 
   for (const [v2ClientName, v3ClientMetadata] of Object.entries(clientMetadata).reverse()) {
     const { v3ClientName, v3ClientPackageName } = v3ClientMetadata;
-    addV3ClientImport(j, source, { v2ClientName, v3ClientName, v3ClientPackageName });
+    addV3ClientModule(j, source, { v2ClientName, v3ClientName, v3ClientPackageName });
     removeV2ClientImport(j, source, v2ClientName);
     removePromiseCalls(j, source, { v2DefaultImportName, v2ClientName });
     replaceClientCreation(j, source, { v2DefaultImportName, v2ClientName, v3ClientName });
   }
 
-  removeDefaultImportIfNotUsed(j, source, v2DefaultImportName);
+  removeDefaultModuleIfNotUsed(j, source, v2DefaultImportName);
 
   return source.toSource();
 }

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

@@ -1,15 +1,11 @@
 import { Collection, JSCodeshift } from "jscodeshift";
 
-export interface AddV3ClientImportOptions {
-  v2ClientName: string;
-  v3ClientName: string;
-  v3ClientPackageName: string;
-}
+import { AddV3ClientModuleOptions } from "./addV3ClientModule";
 
 export const addV3ClientImport = (
   j: JSCodeshift,
   source: Collection<any>,
-  { v2ClientName, v3ClientName, v3ClientPackageName }: AddV3ClientImportOptions
+  { v2ClientName, v3ClientName, v3ClientPackageName }: AddV3ClientModuleOptions
 ): void => {
   const existingImports = source.find(j.ImportDeclaration, {
     source: { value: v3ClientPackageName },

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

@@ -0,0 +1,20 @@
+import { Collection, JSCodeshift } from "jscodeshift";
+
+import { addV3ClientImport } from "./addV3ClientImport";
+import { addV3ClientRequire } from "./addV3ClientRequire";
+import { containsRequire } from "./containsRequire";
+
+export interface AddV3ClientModuleOptions {
+  v2ClientName: string;
+  v3ClientName: string;
+  v3ClientPackageName: string;
+}
+
+export const addV3ClientModule = (
+  j: JSCodeshift,
+  source: Collection<any>,
+  { v2ClientName, v3ClientName, v3ClientPackageName }: AddV3ClientModuleOptions
+): void =>
+  containsRequire(j, source)
+    ? addV3ClientRequire(j, source, { v2ClientName, v3ClientName, v3ClientPackageName })
+    : addV3ClientImport(j, source, { v2ClientName, v3ClientName, v3ClientPackageName });

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

@@ -0,0 +1,60 @@
+import { Collection, Identifier, JSCodeshift, ObjectPattern, Property } from "jscodeshift";
+
+import { AddV3ClientModuleOptions } from "./addV3ClientModule";
+import { getRequireVariableDeclaration } from "./getRequireVariableDeclaration";
+
+export const addV3ClientRequire = (
+  j: JSCodeshift,
+  source: Collection<any>,
+  { v2ClientName, v3ClientName, v3ClientPackageName }: AddV3ClientModuleOptions
+): void => {
+  const v3ClientNameIdentifier = j.identifier(v3ClientName);
+  const v3ClientNameProperty = j.property.from({
+    kind: "init",
+    key: v3ClientNameIdentifier,
+    shorthand: true,
+    value: v3ClientNameIdentifier,
+  });
+
+  const existingRequires = source.find(j.VariableDeclarator, {
+    id: { type: "ObjectPattern" },
+    init: {
+      type: "CallExpression",
+      callee: { type: "Identifier", name: "require" },
+      arguments: [{ type: "Literal", value: v3ClientPackageName }],
+    },
+  });
+
+  // Require decleration already exists.
+  if (existingRequires.size()) {
+    existingRequires.forEach((nodePath) => {
+      // Append to existing require if property not present.
+      if (
+        !(nodePath.value.id as ObjectPattern).properties.find(
+          (property) => ((property as Property).key as Identifier).name === v3ClientName
+        )
+      ) {
+        (nodePath.value.id as ObjectPattern).properties.push(v3ClientNameProperty);
+      }
+    });
+    return;
+  }
+
+  // Insert after default require if present. If not, insert after client require.
+  const clientRequireValue = `aws-sdk/clients/${v2ClientName.toLowerCase()}`;
+  const defaultRequireVarDeclaration = getRequireVariableDeclaration(j, source, "aws-sdk");
+  const clientRequireVarDeclaration = getRequireVariableDeclaration(j, source, clientRequireValue);
+
+  const requireVarDeclaration =
+    defaultRequireVarDeclaration.size() > 0
+      ? defaultRequireVarDeclaration
+      : clientRequireVarDeclaration;
+  requireVarDeclaration.insertAfter(
+    j.variableDeclaration("const", [
+      j.variableDeclarator(
+        j.objectPattern([v3ClientNameProperty]),
+        j.callExpression(j.identifier("require"), [j.literal(v3ClientPackageName)])
+      ),
+    ])
+  );
+};

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

@@ -0,0 +1,8 @@
+import { Collection, JSCodeshift } from "jscodeshift";
+
+export const containsRequire = (j: JSCodeshift, source: Collection<any>) =>
+  source
+    .find(j.CallExpression, {
+      callee: { type: "Identifier", name: "require" },
+    })
+    .size() > 0;

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

@@ -0,0 +1,18 @@
+import { Collection, JSCodeshift } from "jscodeshift";
+
+export const getRequireVariableDeclaration = (
+  j: JSCodeshift,
+  source: Collection<any>,
+  literalValue: string
+) =>
+  source.find(j.VariableDeclaration, {
+    declarations: [
+      {
+        init: {
+          type: "CallExpression",
+          callee: { type: "Identifier", name: "require" },
+          arguments: [{ type: "Literal", value: literalValue }],
+        },
+      },
+    ],
+  });

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

@@ -1,4 +1,4 @@
-import { Collection, JSCodeshift } from "jscodeshift";
+import { Collection, Identifier, JSCodeshift } from "jscodeshift";
 
 export const getV2DefaultImportName = (
   j: JSCodeshift,
@@ -22,5 +22,19 @@ export const getV2DefaultImportName = (
       });
     });
 
+  // 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) => {
+      v2DefaultImportName = (declerationPath.value.id as Identifier).name;
+    });
+
   return v2DefaultImportName;
 };