com.github.seratch.signedrequest4j.SignedRequestHTTPilotImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of signedrequest4j Show documentation
The newest version!
/* 
 * Copyright 2011 Kazuhiro Sera
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); 
 * you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at 
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0 
 * 
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
 * either express or implied. See the License for the specific language 
 * governing permissions and limitations under the License. 
 */
package com.github.seratch.signedrequest4j;

import httpilot.HTTP;
import httpilot.HTTPIOException;
import httpilot.Method;
import httpilot.Request;
import httpilot.Response;

import java.io.IOException;
import java.security.SecureRandom;
import java.util.Map;

/**
 * implementation of {@link SignedRequest} with HTTPilot
 */
public class SignedRequestHTTPilotImpl extends SignedRequestBaseImpl implements SignedRequest {

	/**
	 * 2 Legged OAuth Request
	 */
	public SignedRequestHTTPilotImpl(
			OAuthRealm realm, OAuthConsumer consumer, SignatureMethod signatureMethod) {
		this(realm, consumer, null, signatureMethod);
	}

	/**
	 * 2 Legged OAuth Request
	 */
	public SignedRequestHTTPilotImpl(
			OAuthRealm realm, OAuthConsumer consumer, SignatureMethod signatureMethod,
			Map additionalParameters) {
		this(realm, consumer, null, signatureMethod, additionalParameters);
	}

	public SignedRequestHTTPilotImpl(
			OAuthRealm realm, OAuthConsumer consumer, OAuthAccessToken accessToken, SignatureMethod signatureMethod) {
		this.realm = realm;
		this.consumer = consumer;
		this.accessToken = accessToken;
		this.signatureMethod = signatureMethod;
	}

	public SignedRequestHTTPilotImpl(
			OAuthRealm realm, OAuthConsumer consumer, OAuthAccessToken accessToken, SignatureMethod signatureMethod,
			Map additionalParameters) {
		this.realm = realm;
		this.consumer = consumer;
		this.accessToken = accessToken;
		this.signatureMethod = signatureMethod;
		this.additionalParameters = additionalParameters;
	}

	@Override
	public HttpResponse doRequest(String url, HttpMethod method, RequestBody body, String charset) throws IOException {

		// create http request
		Request request = new Request(url);
		// read get parameters for signature base string
		readQueryStringAndAddToSignatureBaseString(url);

		request.setEnableThrowingIOException(true);
		request.setUserAgent(USER_AGENT);
		request.setConnectTimeoutMillis(connectTimeoutMillis);
		request.setReadTimeoutMillis(readTimeoutMillis);
		request.setCharset(charset);
		for (String name : headersToOverwrite.keySet()) {
			request.setHeader(name, headersToOverwrite.get(name));
		}

		String oAuthNonce = String.valueOf(new SecureRandom().nextLong());
		Long oAuthTimestamp = System.currentTimeMillis() / 1000;
		String signature = getSignature(url, method, oAuthNonce, oAuthTimestamp);
		String authorizationHeader = getAuthorizationHeader(signature, oAuthNonce, oAuthTimestamp);
		request.setHeader("Authorization", authorizationHeader);

		request.setBody(body.getBody(), body.getContentType());

		try {
			Response response = HTTP.request(new Method(method.name()), request);
			return toReturnValue(response);
		} catch (HTTPIOException ex) {
			HttpResponse httpResponse = toReturnValue(ex.getResponse());
			throw new HttpException(ex.getMessage(), httpResponse);
		}

	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public HttpResponse doRequest(String url, HttpMethod method, Map requestParameters, String charset)
			throws IOException {

		// create http request
		Request request = new Request(url);
		request.setEnableThrowingIOException(true);
		request.setUserAgent(USER_AGENT);
		request.setConnectTimeoutMillis(connectTimeoutMillis);
		request.setReadTimeoutMillis(readTimeoutMillis);
		request.setCharset(charset);
		for (String name : headersToOverwrite.keySet()) {
			request.setHeader(name, headersToOverwrite.get(name));
		}


		if (method == HttpMethod.GET) {
			request.setQueryParams(requestParameters);
		} else {
			// message body
			request.setFormParams(requestParameters);
		}
		// read get parameters for signature base string
		readQueryStringAndAddToSignatureBaseString(request.getUrl());
		readQueryStringAndAddToSignatureBaseString(request.getQueryParams().toString());

		String oAuthNonce = String.valueOf(new SecureRandom().nextLong());
		Long oAuthTimestamp = System.currentTimeMillis() / 1000;
		String signature = getSignature(url, method, oAuthNonce, oAuthTimestamp);
		String authorizationHeader = getAuthorizationHeader(signature, oAuthNonce, oAuthTimestamp);
		request.setHeader("Authorization", authorizationHeader);

		try {
			Response response = HTTP.request(new Method(method.name()), request);
			return toReturnValue(response);
		} catch (HTTPIOException ex) {
			HttpResponse httpResponse = toReturnValue(ex.getResponse());
			throw new HttpException(ex.getMessage(), httpResponse);
		}

	}

	static HttpResponse toReturnValue(Response response) {
		HttpResponse httpResponse = new HttpResponse();
		httpResponse.setStatusCode(response.getStatus());
		httpResponse.setHeaders(response.getHeaders());
		httpResponse.setBody(response.getBody());
		httpResponse.setCharset(response.getCharset());
		return httpResponse;
	}

}