Pass screen variable for the mfa modal. Make tokenFn anonymous

aleios-cloud · May 12, 2020 · 7bac9d1 · 7bac9d1
1 parent 85dd0ea
commit 7bac9d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/src/CLIMain.js b/src/CLIMain.js
@@ -188,7 +188,7 @@ class Main {
   }
 
   init() {
-    const creds = getAWSCredentials(this.program.profile);
+    const creds = getAWSCredentials(this.program.profile, this.screen);
 
     return creds
       .getPromise()

diff --git a/src/services/awsCredentials.js b/src/services/awsCredentials.js
@@ -2,20 +2,15 @@ import AWS from "aws-sdk";
 
 import { promptMfaModal } from "../modals";
 
-function getMfaToken(serial, callback) {
-  promptMfaModal(callback, this.screen);
-}
-
-function getAWSCredentials(profile) {
+function getAWSCredentials(profile, screen) {
   if (profile) {
     process.env.AWS_SDK_LOAD_CONFIG = 1;
     return new AWS.SharedIniFileCredentials({
       profile,
-      tokenCodeFn: getMfaToken,
+      tokenCodeFn: (serial, callback) => promptMfaModal(callback, screen),
       callback: (err) => {
         if (err) {
           console.error(`SharedIniFileCreds Error: ${err}`);
-          process.exit(0);
         }
       },
     });
@@ -30,11 +25,10 @@ function getAWSCredentials(profile) {
   if (process.env.AWS_PROFILE) {
     return new AWS.SharedIniFileCredentials({
       profile: process.env.AWS_PROFILE,
-      tokenCodeFn: getMfaToken,
+      tokenCodeFn: (serial, callback) => promptMfaModal(callback, screen),
       callback: (err) => {
         if (err) {
           console.error(`SharedIniFileCreds Error: ${err}`);
-          process.exit(0);
         }
       },
     });