ha-bridge/src/main/java/com/bwssystems/HABridge/plugins/http/HTTPHandler.java

package com.bwssystems.HABridge.plugins.http;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.charset.Charset;

import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.config.CookieSpecs;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.entity.ContentType;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.bwssystems.HABridge.api.NameValue;

public class HTTPHandler {
	private static final Logger log = LoggerFactory.getLogger(HTTPHandler.class);
	private CloseableHttpClient httpClient;
	private RequestConfig globalConfig;
	
	
	public HTTPHandler() {
		globalConfig = RequestConfig.custom().setCookieSpec(CookieSpecs.STANDARD).build();
		httpClient = HttpClients.custom().setDefaultRequestConfig(globalConfig).build();
	}


	// This function executes the url from the device repository against the
	// target as http or https as defined
	public String doHttpRequest(String url, String httpVerb, String contentType, String body, NameValue[] headers) {
		log.debug("doHttpRequest with url: " + url + " with http command: " + httpVerb + " with body: " + body);
		HttpUriRequest request = null;
		String theContent = null;
		URI theURI = null;
		ContentType parsedContentType = null;
		StringEntity requestBody = null;
		if (contentType != null && !contentType.trim().isEmpty()) {
			parsedContentType = ContentType.parse(contentType);
			if (body != null && body.length() > 0)
				requestBody = new StringEntity(body, parsedContentType);
		}
		try {
			theURI = new URI(url);
		} catch (URISyntaxException e1) {
			log.warn("Error creating URI http request: " + url + " with message: " + e1.getMessage());
			return null;
		}
		try {
			if (httpVerb == null || httpVerb.trim().isEmpty() || HttpGet.METHOD_NAME.equalsIgnoreCase(httpVerb)) {
				request = new HttpGet(theURI);
			} else if (HttpPost.METHOD_NAME.equalsIgnoreCase(httpVerb)) {
				HttpPost postRequest = new HttpPost(theURI);
				if (requestBody != null)
					postRequest.setEntity(requestBody);
				request = postRequest;
			} else if (HttpPut.METHOD_NAME.equalsIgnoreCase(httpVerb)) {
				HttpPut putRequest = new HttpPut(theURI);
				if (requestBody != null)
					putRequest.setEntity(requestBody);
				request = putRequest;
			} else
				request = new HttpGet(theURI);

		} catch (IllegalArgumentException e) {
			log.warn("Error creating outbound http request: IllegalArgumentException in log", e);
			return null;
		}
		log.debug("Making outbound call in doHttpRequest: <<<" + request.toString() + ">>>");
		if (headers != null && headers.length > 0) {
			for (int i = 0; i < headers.length; i++) {
				request.setHeader(headers[i].getName(), headers[i].getValue());
			}
		}
		HttpResponse response = null;
		for (int retryCount = 0; retryCount < 2; retryCount++) {
			try {
				response = httpClient.execute(request);
			} catch (ClientProtocolException e) {
				log.warn("Client Protocol Exception received, retyring....");
			} catch (IOException e) {
				log.warn("Error calling out to HA gateway: IOException in log", e);
				retryCount = 2;
			}
			log.debug((httpVerb == null ? "GET" : httpVerb) + " execute (" + retryCount + ") on URL responded: "
					+ response.getStatusLine().getStatusCode());
			if (response != null && response.getStatusLine().getStatusCode() >= 200 && response.getStatusLine().getStatusCode() < 300) {
				log.debug("Successfull response - The http response is <<<" + theContent + ">>>");
				retryCount = 2;
			} else if (response != null) {
				log.warn("HTTP response code was not an expected successful response of between 200 - 299, the code was: "
								+ response.getStatusLine());
				if (response.getStatusLine().getStatusCode() == 504) {
					log.warn("HTTP response code was 504, retrying...");
				} else
					retryCount = 2;
			}
			
			if (response != null && response.getEntity() != null) {
				try {
					theContent = EntityUtils.toString(response.getEntity(), Charset.forName("UTF-8")); // read
																										// content
																										// for
																										// data
					EntityUtils.consume(response.getEntity()); // close out
																// inputstream
																// ignore
																// content
				} catch (Exception e) {
					log.debug("Error ocurred in handling response entity after successful call, still responding success. "
									+ e.getMessage(), e);
				}
			}

			if(retryCount < 2) {
				theContent = null;
				try {
					Thread.sleep(1000);
				} catch (InterruptedException e1) {
					// noop
				}
			}
		}
		return theContent;
	}
	
//	public HttpClient getHttpClient() {
//		return httpClient;
//	}


	public CloseableHttpClient getHttpClient() {
		return httpClient;
	}


	public void closeHandler() {
		try {
			httpClient.close();
		} catch (IOException e) {
			// noop
		}
		httpClient = null;
	}
}