diff --git a/core/trino-main/src/main/java/io/trino/server/security/oauth2/NimbusOAuth2Client.java b/core/trino-main/src/main/java/io/trino/server/security/oauth2/NimbusOAuth2Client.java index 019ceb7affbd..9860703bdf59 100644 --- a/core/trino-main/src/main/java/io/trino/server/security/oauth2/NimbusOAuth2Client.java +++ b/core/trino-main/src/main/java/io/trino/server/security/oauth2/NimbusOAuth2Client.java @@ -30,7 +30,6 @@ import com.nimbusds.oauth2.sdk.AccessTokenResponse; import com.nimbusds.oauth2.sdk.AuthorizationCode; import com.nimbusds.oauth2.sdk.AuthorizationCodeGrant; -import com.nimbusds.oauth2.sdk.AuthorizationGrant; import com.nimbusds.oauth2.sdk.AuthorizationRequest; import com.nimbusds.oauth2.sdk.ParseException; import com.nimbusds.oauth2.sdk.RefreshTokenGrant; @@ -347,19 +346,19 @@ private String hashNonce(String nonce) private T getTokenResponse(String code, URI callbackUri, NimbusAirliftHttpClient.Parser parser) throws ChallengeFailedException { - return getTokenResponse(new AuthorizationCodeGrant(new AuthorizationCode(code), callbackUri), parser); + return getTokenResponse(new TokenRequest(tokenUrl, clientAuth, new AuthorizationCodeGrant(new AuthorizationCode(code), callbackUri)), parser); } private T getTokenResponse(String refreshToken, NimbusAirliftHttpClient.Parser parser) throws ChallengeFailedException { - return getTokenResponse(new RefreshTokenGrant(new RefreshToken(refreshToken)), parser); + return getTokenResponse(new TokenRequest(tokenUrl, clientAuth, new RefreshTokenGrant(new RefreshToken(refreshToken)), scope), parser); } - private T getTokenResponse(AuthorizationGrant authorizationGrant, NimbusAirliftHttpClient.Parser parser) + private T getTokenResponse(TokenRequest tokenRequest, NimbusAirliftHttpClient.Parser parser) throws ChallengeFailedException { - T tokenResponse = httpClient.execute(new TokenRequest(tokenUrl, clientAuth, authorizationGrant, scope), parser); + T tokenResponse = httpClient.execute(tokenRequest, parser); if (!tokenResponse.indicatesSuccess()) { throw new ChallengeFailedException("Error while fetching access token: " + tokenResponse.toErrorResponse().toJSONObject()); }