Merge pull request #18 from koooge/mfa_profile

feat: Support assuming a role with MFA token
tsub · Oct 23, 2019 · 6a92c1a · 6a92c1a
2 parents e5cb48f + b4994ec
commit 6a92c1a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/config/aws_config.go b/config/aws_config.go
@@ -1,6 +1,7 @@
 package config
 
 import (
+	"github.com/aws/aws-sdk-go/aws/credentials/stscreds"
 	"github.com/aws/aws-sdk-go/aws/session"
 )
 
@@ -12,8 +13,9 @@ type AWSParams struct {
 // NewAWSParams creates a new AWSParams object
 func NewAWSParams(awsProfile string) (*AWSParams, error) {
 	sess := session.Must(session.NewSessionWithOptions(session.Options{
-		Profile:           awsProfile,
-		SharedConfigState: session.SharedConfigEnable,
+		Profile:                 awsProfile,
+		SharedConfigState:       session.SharedConfigEnable,
+		AssumeRoleTokenProvider: stscreds.StdinTokenProvider,
 	}))
 
 	return &AWSParams{