Enable transformation when a mixture of require and imports is present (#336)

Author: trivikr

.changeset/nasty-zoos-hammer.md

@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Enable transformation when a mixture of require and imports is present

src/transforms/v2-to-v3/__fixtures__/misc/mixed-imports.input.js

@@ -0,0 +1,4 @@
+const debug = require("debug")("http");
+import AWS from "aws-sdk";
+
+const client = new AWS.DynamoDB();

src/transforms/v2-to-v3/__fixtures__/misc/mixed-imports.output.js

@@ -0,0 +1,4 @@
+const debug = require("debug")("http");
+import { DynamoDB } from "@aws-sdk/client-dynamodb";
+
+const client = new DynamoDB();

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

@@ -3,4 +3,16 @@ import { Collection, JSCodeshift } from "jscodeshift";
 import { getImportEqualsDeclaration } from "./getImportEqualsDeclaration";
 
 export const hasImportEquals = (j: JSCodeshift, source: Collection<unknown>) =>
-  source.find(j.TSImportEqualsDeclaration, getImportEqualsDeclaration()).size() > 0;
+  source
+    .find(j.TSImportEqualsDeclaration, getImportEqualsDeclaration())
+    .filter((importEqualsDeclaration) => {
+      const { moduleReference } = importEqualsDeclaration.value;
+      if (moduleReference.type !== "TSExternalModuleReference") return false;
+      const { expression } = moduleReference;
+      return (
+        expression.type === "StringLiteral" &&
+        typeof expression.value === "string" &&
+        expression.value.startsWith("aws-sdk")
+      );
+    })
+    .size() > 0;

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

@@ -5,4 +5,13 @@ export const hasRequire = (j: JSCodeshift, source: Collection<unknown>) =>
     .find(j.CallExpression, {
       callee: { type: "Identifier", name: "require" },
     })
+    .filter((callExpression) => {
+      const { arguments: args } = callExpression.value;
+      return (
+        args.length > 0 &&
+        (args[0].type === "Literal" || args[0].type === "StringLiteral") &&
+        typeof args[0].value === "string" &&
+        args[0].value.startsWith("aws-sdk")
+      );
+    })
     .size() > 0;