不使用HttpSecurity,使用Spring Boot访问OAuth2受保护资源的方法是什么?
创始人
2024-12-28 20:01:38
0

使用Spring Boot访问OAuth2受保护资源的方法是通过使用OAuth2RestTemplate来发送OAuth2认证请求和访问受保护资源。

下面是一个使用OAuth2RestTemplate的代码示例:

import org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties;
import org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties.Provider;
import org.springframework.boot.autoconfigure.security.oauth2.client.OAuth2ClientProperties.Registration;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.RequestEntity;
import org.springframework.http.ResponseEntity;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.oauth2.client.OAuth2RestTemplate;
import org.springframework.security.oauth2.common.OAuth2AccessToken;
import org.springframework.security.oauth2.provider.authentication.OAuth2AuthenticationDetails;
import org.springframework.security.oauth2.provider.authentication.OAuth2AuthenticationToken;
import org.springframework.security.oauth2.provider.token.ResourceServerTokenServices;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestClientException;

import java.net.URI;
import java.util.Base64;

public class OAuth2ResourceClient {

    private final OAuth2RestTemplate restTemplate;
    private final ResourceServerTokenServices tokenServices;
    private final OAuth2ClientProperties clientProperties;

    public OAuth2ResourceClient(OAuth2RestTemplate restTemplate, ResourceServerTokenServices tokenServices, OAuth2ClientProperties clientProperties) {
        this.restTemplate = restTemplate;
        this.tokenServices = tokenServices;
        this.clientProperties = clientProperties;
    }

    public ResponseEntity getResource(String url) {
        HttpHeaders headers = new HttpHeaders();
        headers.add("Authorization", "Bearer " + getAccessToken());

        RequestEntity requestEntity = new RequestEntity<>(headers, HttpMethod.GET, URI.create(url));

        try {
            return restTemplate.exchange(requestEntity, String.class);
        } catch (RestClientException e) {
            throw new RuntimeException("Error accessing resource", e);
        }
    }

    private String getAccessToken() {
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        if (authentication instanceof OAuth2AuthenticationToken) {
            OAuth2AuthenticationToken oauthToken = (OAuth2AuthenticationToken) authentication;
            OAuth2AccessToken token = tokenServices.readAccessToken(oauthToken.getAccessToken().getTokenValue());
            if (token.isExpired()) {
                token = tokenServices.refreshAccessToken(token.getRefreshToken().getValue());
            }
            return token.getValue();
        }
        throw new IllegalStateException("No OAuth2 authentication found");
    }

    public static OAuth2RestTemplate createOAuth2RestTemplate(OAuth2ClientProperties clientProperties, ResourceServerTokenServices tokenServices) {
        Registration registration = clientProperties.getRegistration().values().iterator().next();
        Provider provider = clientProperties.getProvider().get(registration.getProvider());

        String clientId = registration.getClientId();
        String clientSecret = registration.getClientSecret();
        String accessTokenUri = provider.getTokenUri();

        OAuth2RestTemplate restTemplate = new OAuth2RestTemplate(clientId, clientSecret);
        restTemplate.setAccessTokenProvider(new CustomAccessTokenProvider(accessTokenUri));
        restTemplate.setAccessTokenConverter(new CustomAccessTokenConverter());
        restTemplate.setAuthenticator(new CustomOAuth2RequestAuthenticator());

        return restTemplate;
    }

    private static class CustomAccessTokenProvider extends OAuth2AccessTokenProviderChain {
        public CustomAccessTokenProvider(String accessTokenUri) {
            super(Arrays.asList(new CustomAuthorizationCodeAccessTokenProvider(accessTokenUri),
                    new CustomImplicitAccessTokenProvider(accessTokenUri),
                    new CustomClientCredentialsAccessTokenProvider(accessTokenUri)));
        }
    }

    private static class CustomAuthorizationCodeAccessTokenProvider extends AuthorizationCodeAccessTokenProvider {
        public CustomAuthorizationCodeAccessTokenProvider(String accessTokenUri) {
            this.setTokenRequestEnhancer(new CustomTokenRequestEnhancer(accessTokenUri));
        }
    }

    private static class CustomImplicitAccessTokenProvider extends ImplicitAccessTokenProvider {
        public CustomImplicitAccessTokenProvider(String accessTokenUri) {
            this.setTokenRequestEnhancer(new CustomTokenRequestEnhancer(accessTokenUri));
        }
    }

    private static class CustomClientCredentialsAccessTokenProvider extends ClientCredentialsAccessTokenProvider {
        public CustomClientCredentialsAccessTokenProvider(String accessTokenUri) {
            this.setTokenRequestEnhancer(new CustomTokenRequestEnhancer(accessTokenUri));
        }
    }

    private static class CustomTokenRequestEnhancer implements RequestEnhancer {

        private final String accessTokenUri;

        public CustomTokenRequestEnhancer(String accessTokenUri) {
            this.accessTokenUri = accessTokenUri;
        }

        @Override
        public void enhance(AccessTokenRequest request, OAuth2ProtectedResourceDetails resource,
                            MultiValueMap form, HttpHeaders headers) {
            headers.setBasicAuth(resource.getClientId(), resource.getClientSecret());
            form.set("grant_type", "client_credentials");
            form.set("scope", "read");
        }

相关内容

热门资讯

安卓换鸿蒙系统会卡吗,体验流畅... 最近手机圈可是热闹非凡呢!不少安卓用户都在议论纷纷,说鸿蒙系统要来啦!那么,安卓手机换上鸿蒙系统后,...
安卓系统拦截短信在哪,安卓系统... 你是不是也遇到了这种情况:手机里突然冒出了很多垃圾短信,烦不胜烦?别急,今天就来教你怎么在安卓系统里...
app安卓系统登录不了,解锁登... 最近是不是你也遇到了这样的烦恼:手机里那个心爱的APP,突然就登录不上了?别急,让我来帮你一步步排查...
安卓系统要维护多久,安卓系统维... 你有没有想过,你的安卓手机里那个陪伴你度过了无数日夜的安卓系统,它究竟要陪伴你多久呢?这个问题,估计...
windows官网系统多少钱 Windows官网系统价格一览:了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌... 手机里的App越来越多,是不是感觉内存不够用了?别急,今天就来教你怎么轻松卸载安卓系统里的App,让...
怎么复制照片安卓系统,操作步骤... 亲爱的手机控们,是不是有时候想把自己的手机照片分享给朋友,或者备份到电脑上呢?别急,今天就来教你怎么...
安卓系统应用怎么重装,安卓应用... 手机里的安卓应用突然罢工了,是不是让你头疼不已?别急,今天就来手把手教你如何重装安卓系统应用,让你的...
iwatch怎么连接安卓系统,... 你有没有想过,那款时尚又实用的iWatch,竟然只能和iPhone好上好?别急,今天就来给你揭秘,怎...
安装了Anaconda之后找不... 在安装Anaconda后,如果找不到Jupyter Notebook,可以尝试以下解决方法:检查环境...