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;
  }

}