Fixes for L5

Signed-off-by: Anand Swaminathan <aswaminathan@lyft.com>
flyteorg · Mar 3, 2021 · 0d1cc1d · 0d1cc1d
1 parent d65105c
commit 0d1cc1d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pkg/auth/auth_context.go b/pkg/auth/auth_context.go
@@ -196,7 +196,7 @@ func GetL5Oauth2Config(mainConfig *oauth2.Config) oauth2.Config {
 		RedirectURL:  "https://flyte-rs.av.lyft.net/callback",
 		ClientID:     mainConfig.ClientID,
 		ClientSecret: mainConfig.ClientSecret,
-		Scopes:       []string{OidcScope, OfflineAccessType, ProfileScope},
+		Scopes:       mainConfig.Scopes,
 		Endpoint: oauth2.Endpoint{
 			AuthURL:  mainConfig.Endpoint.AuthURL,
 			TokenURL: mainConfig.Endpoint.TokenURL,

diff --git a/pkg/auth/handlers.go b/pkg/auth/handlers.go
@@ -18,6 +18,7 @@ import (
 	"github.com/lyft/flytestdlib/contextutils"
 	"github.com/lyft/flytestdlib/errors"
 	"github.com/lyft/flytestdlib/logger"
+	"golang.org/x/oauth2"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"