com.youcruit.billogram.client.http.HttpURLConnectionBillogramClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of billogram-v2-api-java-lib Show documentation
Library for connecting to the Billogram v2 API
There is a newer version: 1.0.3
Show newest version
package com.youcruit.billogram.client.http;

import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintStream;
import java.io.Reader;
import java.io.StringReader;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.URI;
import java.util.Scanner;

import javax.xml.bind.DatatypeConverter;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import com.google.gson.Gson;
import com.youcruit.billogram.client.BillogramCallback;
import com.youcruit.billogram.exception.ApiException;
import com.youcruit.billogram.objects.response.error.ApiError;
import com.youcruit.billogram.objects.response.error.ErrorData;

public class HttpURLConnectionBillogramClient extends AbstractHttpClient {
    private static final String APPLICATION_JSON = "application/json";
	private static final String UTF_8 = "UTF-8";
	private static final String ISO_8859_1 = "ISO-8859-1";
	private final String basicAuth;
	private int connectionTimeout = 10000;
	private int requestTimeout = 10000;

	private static final Logger LOGGER = LogManager.getLogger(HttpURLConnectionBillogramClient.class);

	public HttpURLConnectionBillogramClient(String username, String password) {
		this(username, password, null, null);
	}

	public HttpURLConnectionBillogramClient(String username, String password, String apiUrl) {
		this(username, password, null, apiUrl);
	}

	public HttpURLConnectionBillogramClient(String username, String password, boolean sandbox) {
		this(username, password, null, getApiUrl(sandbox));
	}

	public HttpURLConnectionBillogramClient(String username, String password, Gson gson, String apiUrl) {
		super(gson, apiUrl);
		try {
			// Using ISO8859-1 to have it work the same as OkHttp Credentials class
			basicAuth = "Basic " + DatatypeConverter.printBase64Binary((username + ":" + password).getBytes(ISO_8859_1));
		} catch (UnsupportedEncodingException e) {
			throw new RuntimeException("JVM without ISO-8859-1. SERIOUSLY broken.", e);
		}
	}

	@Override
	public  void async(URI uri, Object requestBody, Method method, BillogramCallback callback, Class responseClass) {
		throw new IllegalArgumentException("Asynchronous is not implemented in " + getClass().getName());
	}

	@Override
	public  V sync(URI uri, Object requestBody, Method method, Class responseClass) throws IOException {
		HttpURLConnection connection = (HttpURLConnection) uri.toURL().openConnection();
		connection.setRequestMethod(method.name());
		if (LOGGER.isDebugEnabled()) {
			StringBuilder sb = new StringBuilder(method.name() + " Request for ").append(uri);
			LOGGER.debug(sb);
			if (LOGGER.isTraceEnabled()) {
				sb.append(" : ").append(requestBody);
				LOGGER.trace(sb);
			}
		}
		connection.setConnectTimeout(connectionTimeout);
		connection.setReadTimeout(requestTimeout);
		connection.setUseCaches(false);
		connection.setDoInput(true);
		setHeaders(connection);
		setRequestBody(connection, requestBody);
		connection.connect();
		int responseCode = connection.getResponseCode();
		if (responseCode == 202 || responseCode == 204) {
			return null;
		} else {
			if (responseCode >= 200 && responseCode < 300) {
				if (Void.class.getName().equals(responseClass.getName())) {
					return null;
				}
				try (InputStream is = connection.getInputStream()) {
					final Reader reader;
					reader = logAndGetReader(uri, is);
					return gson.fromJson(reader, responseClass);
				}
			} else {
				try (InputStream is = connection.getErrorStream()) {
					ApiError error;
					try {
						final Reader reader = logAndGetReader(uri, is);
						error = gson.fromJson(reader, ApiError.class);
					} catch (RuntimeException e) {
						LOGGER.warn(e.getMessage(), e);
						error = new ApiError();
						error.setData(new ErrorData());
						error.getData().setMessage("Unpalatable response from request");
					}
					error.setHttpStatusCode(responseCode);
					throw new ApiException(error);
				}
			}
		}
	}

	private Reader logAndGetReader(URI uri, InputStream is) throws UnsupportedEncodingException {
		Reader reader;
		if (LOGGER.isTraceEnabled()) {
            String responseJson = toString(is);
            LOGGER.trace("Response json for " + uri + " : " + responseJson);
            reader = new StringReader(responseJson);
        } else {
            reader = new InputStreamReader(is, UTF_8);
        }
		return reader;
	}

	private String toString(InputStream is) {
		Scanner s = new Scanner(is, UTF_8).useDelimiter("\\A");
		return s.hasNext() ? s.next() : "";
	}

	private void setRequestBody(HttpURLConnection connection, Object body) throws IOException {
		if (body != null) {
			connection.setDoOutput(true);
			try (OutputStream os = connection.getOutputStream()) {
				gson.toJson(body, new PrintStream(os));
			}
		}
	}

	private void setHeaders(HttpURLConnection connection) {
		connection.setRequestProperty("User-Agent", USER_AGENT);
		connection.setRequestProperty("Accept-Charset", UTF_8);
		connection.setRequestProperty("Accept", APPLICATION_JSON);
		connection.setRequestProperty("Content-Type", APPLICATION_JSON);
		connection.setRequestProperty("Authorization", basicAuth);
	}

	public int getConnectionTimeout() {
		return connectionTimeout;
	}

	public void setConnectionTimeout(int connectionTimeout) {
		this.connectionTimeout = connectionTimeout;
	}

	public int getRequestTimeout() {
		return requestTimeout;
	}

	public void setRequestTimeout(int requestTimeout) {
		this.requestTimeout = requestTimeout;
	}
}