1 files changed, 398 insertions, 0 deletions
diff --git a/sdc-distribution-client/src/main/java/org/onap/sdc/http/SdcConnectorClient.java b/sdc-distribution-client/src/main/java/org/onap/sdc/http/SdcConnectorClient.java
new file mode 100644
index 0000000..41d78ba
--- /dev/null
+++ b/sdc-distribution-client/src/main/java/org/onap/sdc/http/SdcConnectorClient.java
@@ -0,0 +1,398 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * sdc-distribution-client
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * 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.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.sdc.http;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.lang.reflect.Type;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+
+import org.apache.commons.io.IOUtils;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpHeaders;
+import org.apache.http.HttpStatus;
+import org.apache.http.client.methods.CloseableHttpResponse;
+import org.apache.http.entity.ContentType;
+import org.apache.http.entity.StringEntity;
+import org.onap.sdc.api.notification.IArtifactInfo;
+import org.onap.sdc.api.results.IDistributionClientResult;
+import org.onap.sdc.impl.DistributionClientResultImpl;
+import org.onap.sdc.utils.DistributionActionResultEnum;
+import org.onap.sdc.utils.GeneralUtils;
+import org.onap.sdc.api.asdc.RegistrationRequest;
+import org.onap.sdc.api.asdc.ServerListResponse;
+import org.onap.sdc.api.consumer.IConfiguration;
+import org.onap.sdc.impl.DistributionClientDownloadResultImpl;
+import org.onap.sdc.utils.DistributionClientConstants;
+import org.onap.sdc.utils.Pair;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.att.nsa.apiClient.credentials.ApiCredential;
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import com.google.gson.reflect.TypeToken;
+
+import fj.data.Either;
+
+public class SdcConnectorClient {
+	String contentDispositionHeader = "Content-Disposition";
+	private static Logger log = LoggerFactory.getLogger(SdcConnectorClient.class.getName());
+	private IConfiguration configuration;
+	private HttpAsdcClient httpClient = null;
+
+	public void init(IConfiguration configuraion) {
+		this.configuration = configuraion;
+		httpClient = new HttpAsdcClient(configuration);
+	}
+
+	public void close() {
+		if (httpClient != null) {
+			httpClient.closeHttpClient();
+		}
+	}
+
+	public IConfiguration getConfiguration() {
+		return configuration;
+	}
+
+	public void setConfiguration(IConfiguration configuration) {
+		this.configuration = configuration;
+	}
+
+	public HttpAsdcClient getHttpClient() {
+		return httpClient;
+	}
+
+	public void setHttpClient(HttpAsdcClient httpClient) {
+		this.httpClient = httpClient;
+	}
+
+
+	public Either<List<String>, IDistributionClientResult> getValidArtifactTypesList() {
+		Pair<HttpAsdcResponse, CloseableHttpResponse> getServersResponsePair = performAsdcServerRequest(AsdcUrls.GET_VALID_ARTIFACT_TYPES);
+		HttpAsdcResponse getArtifactTypeResponse = getServersResponsePair.getFirst();
+
+		Either<List<String>, IDistributionClientResult> response;
+		if (getArtifactTypeResponse.getStatus() == HttpStatus.SC_OK) {
+			response = parseGetValidArtifactTypesResponse(getArtifactTypeResponse);
+		} else {
+			IDistributionClientResult asdcError = handleAsdcError(getArtifactTypeResponse);
+			response = Either.right(asdcError);
+
+		}
+		handeAsdcConnectionClose(getServersResponsePair);
+		return response;
+
+	}
+
+	private void handeAsdcConnectionClose(Pair<HttpAsdcResponse, CloseableHttpResponse> getServersResponsePair) {
+		if (getServersResponsePair.getSecond() != null) {
+			try {
+				getServersResponsePair.getSecond().close();
+
+			} catch (IOException e) {
+				log.error("failed to close http response");
+			}
+
+		}
+	}
+
+	private Pair<HttpAsdcResponse, CloseableHttpResponse> performAsdcServerRequest(final String url) {
+		String requestId = UUID.randomUUID().toString();
+		Map<String, String> requestHeaders = addHeadersToHttpRequest(requestId);
+		log.debug("about to perform getServerList. requestId= {} url= {}", requestId, url);
+		Pair<HttpAsdcResponse, CloseableHttpResponse> getServersResponsePair = httpClient.getRequest(url, requestHeaders, false);
+		return getServersResponsePair;
+	}
+
+	public Either<TopicRegistrationResponse, DistributionClientResultImpl> registerAsdcTopics(ApiCredential credential) {
+
+		Either<TopicRegistrationResponse, DistributionClientResultImpl> response = null;
+
+		String requestId = UUID.randomUUID().toString();
+		Map<String, String> requestHeaders = addHeadersToHttpRequest(requestId);
+
+		RegistrationRequest registrationRequest = new RegistrationRequest(credential.getApiKey(), configuration.getEnvironmentName(), configuration.isConsumeProduceStatusTopic(), configuration.getMsgBusAddress());
+		Gson gson = new GsonBuilder().setPrettyPrinting().create();
+		String jsonRequest = gson.toJson(registrationRequest);
+		StringEntity body = new StringEntity(jsonRequest, ContentType.APPLICATION_JSON);
+
+		log.debug("about to perform registerAsdcTopics. requestId= " + requestId + " url= " + AsdcUrls.POST_FOR_TOPIC_REGISTRATION);
+		Pair<HttpAsdcResponse, CloseableHttpResponse> registerResponsePair = httpClient.postRequest(AsdcUrls.POST_FOR_TOPIC_REGISTRATION, body, requestHeaders, false);
+		HttpAsdcResponse registerResponse = registerResponsePair.getFirst();
+		int status = registerResponse.getStatus();
+
+		if (status == HttpStatus.SC_OK) {
+			response = parseRegistrationResponse(registerResponse);
+
+		} else {
+			DistributionClientResultImpl asdcError = handleAsdcError(registerResponse);
+			return Either.right(asdcError);
+		}
+		handeAsdcConnectionClose(registerResponsePair);
+
+		log.debug("registerAsdcTopics response= " + status + ". requestId= " + requestId + " url= " + AsdcUrls.POST_FOR_TOPIC_REGISTRATION);
+		return response;
+
+	}
+
+	public IDistributionClientResult unregisterTopics(ApiCredential credential) {
+
+		DistributionClientResultImpl response = null;
+
+		String requestId = UUID.randomUUID().toString();
+		HttpAsdcClient httpClient = new HttpAsdcClient(configuration);
+		Map<String, String> requestHeaders = addHeadersToHttpRequest(requestId);
+
+		RegistrationRequest registrationRequest = new RegistrationRequest(credential.getApiKey(), configuration.getEnvironmentName(), configuration.isConsumeProduceStatusTopic(), configuration.getMsgBusAddress());
+		Gson gson = new GsonBuilder().setPrettyPrinting().create();
+		String jsonRequest = gson.toJson(registrationRequest);
+		StringEntity body = new StringEntity(jsonRequest, ContentType.APPLICATION_JSON);
+
+		log.debug("about to perform unregisterTopics. requestId= " + requestId + " url= " + AsdcUrls.POST_FOR_UNREGISTER);
+		Pair<HttpAsdcResponse, CloseableHttpResponse> unRegisterResponsePair = httpClient.postRequest(AsdcUrls.POST_FOR_UNREGISTER, body, requestHeaders, false);
+		HttpAsdcResponse unRegisterResponse = unRegisterResponsePair.getFirst();
+		int status = unRegisterResponse.getStatus();
+		if (status == HttpStatus.SC_NO_CONTENT || status == HttpStatus.SC_OK) {
+			response = new DistributionClientResultImpl(DistributionActionResultEnum.SUCCESS, "unregistration successful");
+
+		} else {
+			response = handleAsdcError(unRegisterResponse);
+		}
+
+		handeAsdcConnectionClose(unRegisterResponsePair);
+
+		log.debug("unregisterTopics response = " + status + ". requestId= " + requestId + " url= " + AsdcUrls.POST_FOR_UNREGISTER);
+
+		return response;
+
+	}
+
+	public DistributionClientDownloadResultImpl dowloadArtifact(IArtifactInfo artifactInfo) {
+		DistributionClientDownloadResultImpl response = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "failed to download artifact from ASDC");
+
+		String requestId = UUID.randomUUID().toString();
+		Map<String, String> requestHeaders = new HashMap<String, String>();
+		requestHeaders.put(DistributionClientConstants.HEADER_REQUEST_ID, requestId);
+		requestHeaders.put(DistributionClientConstants.HEADER_INSTANCE_ID, configuration.getConsumerID());
+		requestHeaders.put(HttpHeaders.ACCEPT, ContentType.APPLICATION_OCTET_STREAM.toString());
+		String requestUrl = artifactInfo.getArtifactURL();
+		Pair<HttpAsdcResponse, CloseableHttpResponse> downloadPair = httpClient.getRequest(requestUrl, requestHeaders, false);
+		HttpAsdcResponse downloadResponse = downloadPair.getFirst();
+
+		int status = downloadResponse.getStatus();
+		if (status == 200) {
+
+			response = parseDownloadArtifactResponse(artifactInfo, downloadResponse);
+		} else {
+			response = handleAsdcDownloadArtifactError(downloadResponse);
+
+		}
+		handeAsdcConnectionClose(downloadPair);
+		return response;
+	}
+
+	/* **************************** private methods ********************************************/
+
+	private Either<List<String>, IDistributionClientResult> parseGetServersResponse(HttpAsdcResponse getServersResponse) {
+		Either<List<String>, IDistributionClientResult> result;
+		try {
+			String jsonMessage = IOUtils.toString(getServersResponse.getMessage().getContent());
+
+			Gson gson = new GsonBuilder().create();
+			ServerListResponse serverListResponse = gson.fromJson(jsonMessage, ServerListResponse.class);
+			List<String> serverList = serverListResponse.getUebServerList();
+			result = Either.left(serverList);
+
+		} catch (UnsupportedOperationException | IOException e) {
+			result = handleParsingError(e);
+		}
+
+		return result;
+	}
+
+	private Either<List<String>, IDistributionClientResult> parseGetValidArtifactTypesResponse(HttpAsdcResponse getArtifactTypesResponse) {
+		Either<List<String>, IDistributionClientResult> result;
+		try {
+			String jsonMessage = IOUtils.toString(getArtifactTypesResponse.getMessage().getContent());
+			Type listType = new TypeToken<ArrayList<String>>() {
+			}.getType();
+			Gson gson = new GsonBuilder().create();
+			List<String> artifactTypesList = gson.fromJson(jsonMessage, listType);
+			result = Either.left(artifactTypesList);
+
+		} catch (UnsupportedOperationException | IOException e) {
+			result = handleParsingError(e);
+		}
+
+		return result;
+	}
+
+	private Either<List<String>, IDistributionClientResult> handleParsingError(Exception e) {
+		Either<List<String>, IDistributionClientResult> result;
+		log.error("failed to parse response from ASDC. error: " + e.getMessage());
+		IDistributionClientResult response = new DistributionClientResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "failed to parse response from ASDC");
+		result = Either.right(response);
+		return result;
+	}
+
+	Either<TopicRegistrationResponse, DistributionClientResultImpl> parseRegistrationResponse(HttpAsdcResponse registerResponse) {
+
+		String jsonMessage;
+		try {
+			jsonMessage = IOUtils.toString(registerResponse.getMessage().getContent());
+
+			Gson gson = new GsonBuilder().create();
+			TopicRegistrationResponse registrationResponse = gson.fromJson(jsonMessage, TopicRegistrationResponse.class);
+
+			if (registrationResponse.getDistrNotificationTopicName() == null) {
+				DistributionClientResultImpl response = new DistributionClientResultImpl(DistributionActionResultEnum.FAIL, "failed to receive notification topic from ASDC");
+				return Either.right(response);
+			}
+
+			if (registrationResponse.getDistrStatusTopicName() == null) {
+				DistributionClientResultImpl response = new DistributionClientResultImpl(DistributionActionResultEnum.FAIL, "failed to receive status topic from ASDC");
+				return Either.right(response);
+			}
+			return Either.left(registrationResponse);
+
+		} catch (UnsupportedOperationException | IOException e) {
+			log.error("failed to pars response from ASDC. error: " + e.getMessage());
+			DistributionClientResultImpl response = new DistributionClientResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "failed to parse response from ASDC");
+			return Either.right(response);
+		}
+	}
+
+	protected Map<String, String> addHeadersToHttpRequest(String requestId) {
+		Map<String, String> requestHeaders = new HashMap<>();
+		requestHeaders.put(DistributionClientConstants.HEADER_REQUEST_ID, requestId);
+		requestHeaders.put(DistributionClientConstants.HEADER_INSTANCE_ID, configuration.getConsumerID());
+		requestHeaders.put(HttpHeaders.CONTENT_TYPE, ContentType.APPLICATION_JSON.toString());
+
+		return requestHeaders;
+	}
+
+	private DistributionClientResultImpl handleAsdcError(HttpAsdcResponse registerResponse) {
+		int status = registerResponse.getStatus();
+		DistributionClientResultImpl errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "failed to send request to ASDC");
+		if (status == HttpStatus.SC_UNAUTHORIZED) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.ASDC_AUTHENTICATION_FAILED, "authentication to ASDC failed for user " + configuration.getUser());
+		} else if (status == HttpStatus.SC_FORBIDDEN) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.ASDC_AUTHORIZATION_FAILED, "authorization failure for user " + configuration.getUser());
+		} else if (status == HttpStatus.SC_BAD_REQUEST) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.BAD_REQUEST, "ASDC call failed due to missing information");
+		} else if (status == HttpStatus.SC_NOT_FOUND) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.ASDC_NOT_FOUND, "ASDC not found");
+		} else if (status == HttpStatus.SC_INTERNAL_SERVER_ERROR) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.ASDC_SERVER_PROBLEM, "ASDC server problem");
+		} else if (status == HttpStatus.SC_BAD_GATEWAY) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.ASDC_CONNECTION_FAILED, "ASDC server problem");
+		} else if (status == HttpStatus.SC_GATEWAY_TIMEOUT) {
+			errorResponse = new DistributionClientResultImpl(DistributionActionResultEnum.ASDC_SERVER_TIMEOUT, "ASDC server problem");
+		}
+		log.error("status from ASDC is " + registerResponse);
+		log.error(errorResponse.toString());
+		try {
+			String errorString = IOUtils.toString(registerResponse.getMessage().getContent());
+			log.debug("error from ASDC is: " + errorString);
+		} catch (UnsupportedOperationException | IOException e) {
+		}
+		return errorResponse;
+
+	}
+
+	private DistributionClientDownloadResultImpl handleAsdcDownloadArtifactError(HttpAsdcResponse registerResponse) {
+		int status = registerResponse.getStatus();
+		DistributionClientDownloadResultImpl errorResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "failed to send request to ASDC");
+		if (status == HttpStatus.SC_UNAUTHORIZED) {
+			errorResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.ASDC_AUTHENTICATION_FAILED, "authentication to ASDC failed for user " + configuration.getUser());
+		} else if (status == HttpStatus.SC_FORBIDDEN) {
+			errorResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.ASDC_AUTHORIZATION_FAILED, "authorization failure for user " + configuration.getUser());
+		} else if (status == HttpStatus.SC_BAD_REQUEST || status == HttpStatus.SC_NOT_FOUND) {
+			errorResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.ARTIFACT_NOT_FOUND, "Specified artifact is  not found");
+			// } else if (status == 404){
+			// errorResponse = new DistributionClientDownloadResultImpl(
+			// DistributionActionResultEnum.ASDC_NOT_FOUND,
+			// "ASDC not found");
+		} else if (status == HttpStatus.SC_INTERNAL_SERVER_ERROR) {
+			errorResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.ASDC_SERVER_PROBLEM, "ASDC server problem");
+		}
+		log.error("status from ASDC is " + registerResponse);
+		log.error(errorResponse.toString());
+		try {
+			String errorString = IOUtils.toString(registerResponse.getMessage().getContent());
+			log.debug("error from ASDC is: " + errorString);
+		} catch (UnsupportedOperationException | IOException e) {
+		}
+		return errorResponse;
+
+	}
+
+	private DistributionClientDownloadResultImpl parseDownloadArtifactResponse(IArtifactInfo artifactInfo, HttpAsdcResponse getServersResponse) {
+		HttpEntity entity = getServersResponse.getMessage();
+		InputStream is;
+		try {
+			is = entity.getContent();
+			String artifactName = "";
+			if (getServersResponse.getHeadersMap().containsKey(contentDispositionHeader))
+				artifactName = getServersResponse.getHeadersMap().get(contentDispositionHeader);
+
+			byte[] payload = IOUtils.toByteArray(is);
+			if (artifactInfo.getArtifactChecksum() == null || artifactInfo.getArtifactChecksum().isEmpty()) {
+				return new DistributionClientDownloadResultImpl(DistributionActionResultEnum.DATA_INTEGRITY_PROBLEM, "failed to get artifact from ASDC. Empty checksum");
+			}
+
+			DistributionClientDownloadResultImpl resResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.SUCCESS, "success", artifactName, payload);
+			return resResponse;
+			
+			//Validate checksum removed as fix to bug 293657 
+			/*if (validateChecksum(artifactInfo, payload)) {
+				DistributionClientDownloadResultImpl resResponse = new DistributionClientDownloadResultImpl(DistributionActionResultEnum.SUCCESS, "success", artifactName, payload);
+				return resResponse;
+
+			} else {
+
+				return new DistributionClientDownloadResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "Invalid checksum. ArtifactInfo checksum ");
+			}*/
+
+		} catch (UnsupportedOperationException | IOException e) {
+			log.error("failed to get artifact from response ");
+			return new DistributionClientDownloadResultImpl(DistributionActionResultEnum.GENERAL_ERROR, "UnsupportedOperationException ");
+		}
+
+	}
+
+	private boolean validateChecksum(IArtifactInfo artifactInfo, byte[] payload) {
+		boolean bRes = false;
+		String calculatedMD5 = GeneralUtils.calculateMD5(payload);
+		if (artifactInfo.getArtifactChecksum().equals(calculatedMD5)) {
+			bRes = true;
+		}
+
+		return bRes;
+	}
+
+}