org.springframework.security.oauth.consumer.client.OAuthRestTemplate.java Source code

Introduction

Here is the source code for org.springframework.security.oauth.consumer.client.OAuthRestTemplate.java

Source

package org.springframework.security.oauth.consumer.client;

import org.springframework.http.client.ClientHttpRequestFactory;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.security.oauth.consumer.OAuthConsumerSupport;
import org.springframework.security.oauth.consumer.ProtectedResourceDetails;
import org.springframework.web.client.RestTemplate;

/**
 * Rest template that is able to make OAuth-authenticated REST requests with the credentials of the provided resource.
 *
 * @author Ryan Heaton
 */
public class OAuthRestTemplate extends RestTemplate {

    private final ProtectedResourceDetails resource;
    private OAuthConsumerSupport support = new CoreOAuthConsumerSupport();

    public OAuthRestTemplate(ProtectedResourceDetails resource) {
        this(new SimpleClientHttpRequestFactory(), resource);
    }

    public OAuthRestTemplate(ClientHttpRequestFactory requestFactory, ProtectedResourceDetails resource) {
        super();
        if (resource == null) {
            throw new IllegalArgumentException("An OAuth resource must be supplied.");
        }
        if (support == null) {
            throw new IllegalArgumentException("OAuth support must be supplied.");
        }

        this.resource = resource;
        setRequestFactory(requestFactory);
    }

    @Override
    public void setRequestFactory(ClientHttpRequestFactory requestFactory) {
        if (!(requestFactory instanceof OAuthClientHttpRequestFactory)) {
            requestFactory = new OAuthClientHttpRequestFactory(requestFactory, getResource(), getSupport());
        }
        super.setRequestFactory(requestFactory);
    }

    public ProtectedResourceDetails getResource() {
        return resource;
    }

    /**
     * The support logic to use.
     *
     * @return The support logic to use.
     */
    public OAuthConsumerSupport getSupport() {
        return support;
    }

    /**
     * The support logic to use.
     *
     * @param support The support logic to use.
     */
    public void setSupport(OAuthConsumerSupport support) {
        this.support = support;
    }

}