@@ -9,11 +9,11 @@ import (
99 "github.com/stackql/any-sdk/pkg/constants"
1010 "github.com/stackql/any-sdk/pkg/dto"
1111 "github.com/stackql/any-sdk/pkg/logging"
12+ "github.com/stackql/any-sdk/pkg/netutils"
1213 "github.com/stackql/stackql/internal/stackql/discovery"
1314 "github.com/stackql/stackql/internal/stackql/docparser"
1415 "github.com/stackql/stackql/internal/stackql/internal_data_transfer/internaldto"
1516 "github.com/stackql/stackql/internal/stackql/methodselect"
16- "github.com/stackql/stackql/internal/stackql/netutils"
1717 "github.com/stackql/stackql/internal/stackql/parserutil"
1818 "github.com/stackql/stackql/internal/stackql/relational"
1919
@@ -108,25 +108,27 @@ func (gp *GenericProvider) Auth(
108108 at := gp .inferAuthType (* authCtx , authTypeRequested )
109109 switch at {
110110 case dto .AuthAPIKeyStr :
111- return gp .apiTokenFileAuth (authCtx , false )
111+ return gp .authUtil . ApiTokenAuth (authCtx , gp . runtimeCtx , false )
112112 case dto .AuthBearerStr :
113- return gp .apiTokenFileAuth (authCtx , true )
113+ return gp .authUtil . ApiTokenAuth (authCtx , gp . runtimeCtx , true )
114114 case dto .AuthServiceAccountStr :
115- return gp .googleKeyFileAuth (authCtx )
115+ scopes := authCtx .Scopes
116+ return gp .authUtil .GoogleOauthServiceAccount (gp .GetProviderString (), authCtx , scopes , gp .runtimeCtx )
116117 case dto .OAuth2Str :
117118 if authCtx .GrantType == dto .ClientCredentialsStr {
118- return gp .clientCredentialsAuth (authCtx )
119+ scopes := authCtx .Scopes
120+ return gp .authUtil .GenericOauthClientCredentials (authCtx , scopes , gp .runtimeCtx )
119121 }
120122 case dto .AuthBasicStr :
121- return gp .basicAuth (authCtx )
123+ return gp .authUtil . BasicAuth (authCtx , gp . runtimeCtx )
122124 case dto .AuthCustomStr :
123- return gp .customAuth (authCtx )
125+ return gp .authUtil . CustomAuth (authCtx , gp . runtimeCtx )
124126 case dto .AuthAzureDefaultStr :
125- return gp .azureDefaultAuth (authCtx )
127+ return gp .authUtil . AzureDefaultAuth (authCtx , gp . runtimeCtx )
126128 case dto .AuthInteractiveStr :
127- return gp .oAuth ( authCtx , enforceRevokeFirst )
129+ return gp .authUtil . GCloudOAuth ( gp . runtimeCtx , authCtx , enforceRevokeFirst )
128130 case dto .AuthAWSSigningv4Str :
129- return gp .awsSigningAuth (authCtx )
131+ return gp .authUtil . AwsSigningAuth (authCtx , gp . runtimeCtx )
130132 case dto .AuthNullStr :
131133 return netutils .GetHTTPClient (gp .runtimeCtx , http .DefaultClient ), nil
132134 }
@@ -239,40 +241,6 @@ func (gp *GenericProvider) ShowAuth(authCtx *dto.AuthCtx) (*anysdk.AuthMetadata,
239241 return retVal , err
240242}
241243
242- func (gp * GenericProvider ) oAuth (authCtx * dto.AuthCtx , enforceRevokeFirst bool ) (* http.Client , error ) {
243- return gp .authUtil .GCloudOAuth (gp .runtimeCtx , authCtx , enforceRevokeFirst )
244- }
245-
246- func (gp * GenericProvider ) googleKeyFileAuth (authCtx * dto.AuthCtx ) (* http.Client , error ) {
247- scopes := authCtx .Scopes
248- return gp .authUtil .GoogleOauthServiceAccount (gp .GetProviderString (), authCtx , scopes , gp .runtimeCtx )
249- }
250-
251- func (gp * GenericProvider ) clientCredentialsAuth (authCtx * dto.AuthCtx ) (* http.Client , error ) {
252- scopes := authCtx .Scopes
253- return gp .authUtil .GenericOauthClientCredentials (authCtx , scopes , gp .runtimeCtx )
254- }
255-
256- func (gp * GenericProvider ) apiTokenFileAuth (authCtx * dto.AuthCtx , enforceBearer bool ) (* http.Client , error ) {
257- return gp .authUtil .ApiTokenAuth (authCtx , gp .runtimeCtx , enforceBearer )
258- }
259-
260- func (gp * GenericProvider ) awsSigningAuth (authCtx * dto.AuthCtx ) (* http.Client , error ) {
261- return gp .authUtil .AwsSigningAuth (authCtx , gp .runtimeCtx )
262- }
263-
264- func (gp * GenericProvider ) basicAuth (authCtx * dto.AuthCtx ) (* http.Client , error ) {
265- return gp .authUtil .BasicAuth (authCtx , gp .runtimeCtx )
266- }
267-
268- func (gp * GenericProvider ) customAuth (authCtx * dto.AuthCtx ) (* http.Client , error ) {
269- return gp .authUtil .CustomAuth (authCtx , gp .runtimeCtx )
270- }
271-
272- func (gp * GenericProvider ) azureDefaultAuth (authCtx * dto.AuthCtx ) (* http.Client , error ) {
273- return gp .authUtil .AzureDefaultAuth (authCtx , gp .runtimeCtx )
274- }
275-
276244func (gp * GenericProvider ) GetLikeableColumns (tableName string ) []string {
277245 var retVal []string
278246 switch tableName {
0 commit comments