From ce68bd3616e12fe3dacef6936c9210f6e810d582 Mon Sep 17 00:00:00 2001
From: Raymond Mak <raymond.mak@chargepoint.com>
Date: Wed, 3 Jan 2024 12:49:25 -0500
Subject: [PATCH] Enable AWS SSO support

---
 cmd/aws-sigv4-proxy/main.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cmd/aws-sigv4-proxy/main.go b/cmd/aws-sigv4-proxy/main.go
index 05efffb..92a25cd 100644
--- a/cmd/aws-sigv4-proxy/main.go
+++ b/cmd/aws-sigv4-proxy/main.go
@@ -92,6 +92,10 @@ func main() {
 		}
 	}
 
+	sessionOptions := session.Options{
+		SharedConfigState: session.SharedConfigEnable,
+	}
+
 	sessionConfig := aws.Config{}
 	if v := os.Getenv("AWS_STS_REGIONAL_ENDPOINTS"); len(v) == 0 {
 		sessionConfig.STSRegionalEndpoint = endpoints.RegionalSTSEndpoint
@@ -99,7 +103,8 @@ func main() {
 
 	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)
 	}