Add import for credential provider

aws · Oct 20, 2023 · ea7f92d · ea7f92d
1 parent 56f6f4f
commit ea7f92d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/transforms/v2-to-v3/apis/replaceAwsCredentials.ts b/src/transforms/v2-to-v3/apis/replaceAwsCredentials.ts
@@ -1,6 +1,6 @@
 import { Collection, JSCodeshift } from "jscodeshift";
 import { AWS_CREDENTIALS_MAP } from "../config";
-import { ImportType } from "../modules";
+import { ImportType, addNamedModule } from "../modules";
 import { getAwsCredentialsNewExpressions } from "./getAwsCredentialsNewExpressions";
 
 export interface ReplaceAwsCredentialsOptions {
@@ -11,7 +11,7 @@ export interface ReplaceAwsCredentialsOptions {
 export const replaceAwsCredentials = (
   j: JSCodeshift,
   source: Collection<unknown>,
-  { v2GlobalName }: ReplaceAwsCredentialsOptions
+  { v2GlobalName, importType }: ReplaceAwsCredentialsOptions
 ) => {
   if (!v2GlobalName) return;
 
@@ -23,11 +23,11 @@ export const replaceAwsCredentials = (
     const credsNewExpressionCount = credsNewExpressions.size();
 
     if (credsNewExpressionCount > 0) {
-      // ToDo: Call importType agnostic addNamedModule
-      // addNamedModule(j, source, {
-      //   importedName: v3ProviderName,
-      //   v3ClientPackageName: "@aws-sdk/credential-provider",
-      // });
+      addNamedModule(j, source, {
+        importType,
+        importedName: v3ProviderName,
+        packageName: "@aws-sdk/credential-providers",
+      });
       credsNewExpressions.replaceWith(({ node }) =>
         j.callExpression(j.identifier(v3ProviderName), node.arguments)
       );

diff --git a/src/transforms/v2-to-v3/modules/index.ts b/src/transforms/v2-to-v3/modules/index.ts
@@ -1,4 +1,5 @@
 export * from "./addClientModules";
+export * from "./addNamedModule";
 export * from "./getGlobalNameFromModule";
 export * from "./getImportEqualsDeclarationType";
 export * from "./getImportSpecifiers";