Enable AWS SSO support

awslabs · Jan 3, 2024 · 79cdc4f · 79cdc4f
1 parent b851e2c
commit 79cdc4f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmd/aws-sigv4-proxy/main.go b/cmd/aws-sigv4-proxy/main.go
@@ -67,14 +67,19 @@ func main() {
 		log.SetLevel(log.DebugLevel)
 	}
 
+	sessionOptions := session.Options{
+		SharedConfigState: session.SharedConfigEnable,
+	}
+
 	sessionConfig := aws.Config{}
 	if v := os.Getenv("AWS_STS_REGIONAL_ENDPOINTS"); len(v) == 0 {
 		sessionConfig.STSRegionalEndpoint = endpoints.RegionalSTSEndpoint
 	}
 
 	sessionConfig.CredentialsChainVerboseErrors = aws.Bool(shouldLogSigning())
 
-	session, err := session.NewSession(&sessionConfig)
+	sessionOptions.Config.MergeIn(&sessionConfig)
+	session, err := session.NewSessionWithOptions(sessionOptions)
 	if err != nil {
 		log.Fatal(err)
 	}