TwoLeggedOAuthHelper.java

package org.fluxtream.core.connectors;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Map;

import org.fluxtream.core.domain.ApiKey;
import oauth.signpost.OAuthConsumer;
import oauth.signpost.basic.DefaultOAuthConsumer;
import oauth.signpost.signature.HmacSha1MessageSigner;

import org.apache.commons.io.IOUtils;
import org.springframework.stereotype.Component;

import org.fluxtream.core.connectors.updaters.RateLimitReachedException;

@Component
public class TwoLeggedOAuthHelper extends ApiClientSupport {

    public final String makeRestCall(ApiKey apiKey,
                                     String accessToken, String tokenSecret, Map<String, String> additionalParameters,
                                     int objectTypes, String urlString) throws RateLimitReachedException {
		if (hasReachedRateLimit(apiKey.getConnector(), apiKey.getGuestId()))
			throw new RateLimitReachedException();
		try {
			long then = System.currentTimeMillis();
			URL url = new URL(urlString);
			HttpURLConnection request = (HttpURLConnection) url
					.openConnection();

			OAuthConsumer consumer = new DefaultOAuthConsumer(accessToken,
					tokenSecret);
			consumer.setTokenWithSecret("", "");
			consumer.setMessageSigner(new HmacSha1MessageSigner());
			consumer.setSendEmptyTokens(true);

			// sign the request (consumer is a Signpost DefaultOAuthConsumer)
			try {
				consumer.sign(request);
			} catch (Exception e) {
				throw new RuntimeException("OAuth exception: " + e.getMessage());
			}
			request.connect();
            final int httpResponseCode = request.getResponseCode();
            final String httpResponseMessage = request.getResponseMessage();
            if (httpResponseCode == 200) {
				String response = IOUtils.toString(request.getInputStream());
				connectorUpdateService.addApiUpdate(apiKey,
						objectTypes, then, System.currentTimeMillis() - then,
						urlString, true, null, null);
				// logger.info(apiKey.getGuestId(), "REST call success: " +
				// urlString);
				return response;
			} else {
				connectorUpdateService.addApiUpdate(apiKey,
						objectTypes, then, System.currentTimeMillis() - then,
						urlString, false, httpResponseCode, httpResponseMessage);
				throw new RuntimeException(
						"Could not make REST call, got response code: "
								+ httpResponseCode + ", message: "
								+ request.getResponseMessage()
								+ "\n+REST url: " + urlString);
			}
		} catch (IOException e) {
			throw new RuntimeException("IOException trying to make rest call: "
					+ e.getMessage());
		}
	}

}