/*
 * Copyright 2016-2017, Nokia Corporation
 *
 * 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 org.onap.vfc.nfvo.driver.vnfm.svnfm.http.client;

import java.io.IOException;
import java.net.URI;

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.methods.HttpPatch;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpRequestBase;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.util.EntityUtils;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.constant.CommonConstants;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.RequestMethod;

public class HttpRequestProcessor {
	private static final Logger logger = LoggerFactory.getLogger(HttpRequestProcessor.class);
	private CloseableHttpClient httpClient;
	private HttpRequestBase httpRequest;
	
	public HttpRequestProcessor(HttpClientBuilder httpClientBuilder, RequestMethod requestMethod)
	{
		httpClient = httpClientBuilder.build();
		httpRequest = HttpClientUtils.getHttpRequest(requestMethod);
	}
	
	public HttpResult process(String url) throws ClientProtocolException, IOException
	{
		httpRequest.setURI(URI.create(url));
		
		HttpResponse response = httpClient.execute(httpRequest);
		HttpResult httpResult = buildHttpResult(response);
		httpRequest.releaseConnection();
//		httpClient.close();
		
		return httpResult;
	}

	private HttpResult buildHttpResult(HttpResponse response) throws IOException {
		HttpEntity resEntity = response.getEntity();
		String responseContent = "";
		if (resEntity != null) {
			responseContent = EntityUtils.toString(resEntity, CommonConstants.UTF_8);
			EntityUtils.consume(resEntity);
		}
		
		HttpResult httpResult = new HttpResult();
		httpResult.setStatusCode(response.getStatusLine().getStatusCode());
		httpResult.setStatusCause(response.getStatusLine().getReasonPhrase());
		httpResult.setHeaders(response.getAllHeaders());
		httpResult.setContent(responseContent);
		
		return httpResult;
	}

	public void addHdeader(String key, String value) {
		httpRequest.setHeader(key, value);
		
	}

	public void addPostEntity(String bodyStr) {
		if(httpRequest instanceof HttpPost) {
			((HttpPost)httpRequest).setEntity(new StringEntity(bodyStr, CommonConstants.UTF_8));
		} else if(httpRequest instanceof HttpPatch) {
			((HttpPatch)httpRequest).setEntity(new StringEntity(bodyStr, CommonConstants.UTF_8));
		}
	}
	
	public void addBytesPostEntity(byte[] byteArray) {
		((HttpPost)httpRequest).setEntity(new ByteArrayEntity(byteArray));
	}
}