fix: make removeGlobalModule importType independent

Author: trivikr

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

@@ -4,18 +4,12 @@ import { PACKAGE_NAME } from "../config";
 import { removeImportDefault } from "./removeImportDefault";
 import { removeImportEquals } from "./removeImportEquals";
 import { removeRequireIdentifier } from "./removeRequireIdentifier";
-import { ImportType } from "./types";
-
-export interface RemoveGlobalModuleOptions {
-  importType: ImportType;
-  v2GlobalName?: string;
-}
 
 // Removes the import of "aws-sdk" if it's not used.
 export const removeGlobalModule = (
   j: JSCodeshift,
   source: Collection<unknown>,
-  { importType, v2GlobalName }: RemoveGlobalModuleOptions
+  v2GlobalName?: string
 ) => {
   if (!v2GlobalName) return;
 
@@ -24,12 +18,8 @@ export const removeGlobalModule = (
   // Only usage is import/require.
   if (identifierUsages.size() === 1) {
     const defaultOptions = { localName: v2GlobalName, sourceValue: PACKAGE_NAME };
-    if (importType === ImportType.REQUIRE) {
-      removeRequireIdentifier(j, source, defaultOptions);
-    } else if (importType === ImportType.IMPORT_EQUALS) {
-      removeImportEquals(j, source, defaultOptions);
-    } else {
-      removeImportDefault(j, source, defaultOptions);
-    }
+    removeRequireIdentifier(j, source, defaultOptions);
+    removeImportEquals(j, source, defaultOptions);
+    removeImportDefault(j, source, defaultOptions);
   }
 };

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

@@ -99,7 +99,7 @@ const transformer = async (file: FileInfo, api: API) => {
   replaceAwsConfig(j, source, v2GlobalName);
   replaceAwsIdentity(j, source, { v2GlobalName, importType });
   replaceAwsUtilFunctions(j, source, v2GlobalName);
-  removeGlobalModule(j, source, { v2GlobalName, importType });
+  removeGlobalModule(j, source, v2GlobalName);
 
   return source.toSource();
 };