path: root/catalog-be/src/main/java/org/openecomp/sdc/be/components/distribution/engine/ServiceDistributionArtifactsBuilder.java

/*-
 * ============LICENSE_START=======================================================
 * SDC
 * ================================================================================
 * 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.openecomp.sdc.be.components.distribution.engine;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

import javax.annotation.PostConstruct;

import org.openecomp.sdc.be.config.ConfigurationManager;
import org.openecomp.sdc.be.model.ArtifactDefinition;
import org.openecomp.sdc.be.model.ComponentInstance;
import org.openecomp.sdc.be.model.ComponentParametersView;
import org.openecomp.sdc.be.model.Resource;
import org.openecomp.sdc.be.model.Service;
import org.openecomp.sdc.be.model.category.CategoryDefinition;
import org.openecomp.sdc.be.model.category.SubCategoryDefinition;
import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
import org.openecomp.sdc.be.model.operations.api.IArtifactOperation;
import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
import org.openecomp.sdc.be.model.operations.impl.InterfaceLifecycleOperation;
import org.openecomp.sdc.common.api.ArtifactTypeEnum;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import fj.data.Either;

@Component("serviceDistributionArtifactsBuilder")
public class ServiceDistributionArtifactsBuilder {

	private int defaultArtifactInstallTimeout = 60;

	private static Logger logger = LoggerFactory.getLogger(ServiceDistributionArtifactsBuilder.class.getName());

	final static String BASE_ARTIFACT_URL = "/sdc/v1/catalog/services/%s/%s/";
	final static String RESOURCE_ARTIFACT_URL = BASE_ARTIFACT_URL + "resources/%s/%s/artifacts/%s";
	final static String SERVICE_ARTIFACT_URL = BASE_ARTIFACT_URL + "artifacts/%s";

	final static String RESOURCE_INSTANCE_ARTIFACT_URL = BASE_ARTIFACT_URL + "resourceInstances/%s/artifacts/%s";

	@javax.annotation.Resource
	InterfaceLifecycleOperation interfaceLifecycleOperation;

	@javax.annotation.Resource
	IArtifactOperation artifactOperation;
	
	@Autowired
	ToscaOperationFacade toscaOperationFacade;

	/*
	 * @javax.annotation.Resource private
	 * InformationDeployedArtifactsBusinessLogic
	 * informationDeployedArtifactsBusinessLogic;
	 */

	@PostConstruct
	private void init() {
		defaultArtifactInstallTimeout = ConfigurationManager.getConfigurationManager().getConfiguration()
				.getDefaultHeatArtifactTimeoutMinutes();
	}

	public InterfaceLifecycleOperation getInterfaceLifecycleOperation() {
		return interfaceLifecycleOperation;
	}

	public void setInterfaceLifecycleOperation(InterfaceLifecycleOperation interfaceLifecycleOperation) {
		this.interfaceLifecycleOperation = interfaceLifecycleOperation;
	}

	public INotificationData buildResourceInstanceForDistribution(Service service, String distributionId) {
		INotificationData notificationData = new NotificationDataImpl();

		notificationData.setResources(convertRIToJsonContanier(service));
		notificationData.setServiceName(service.getName());
		notificationData.setServiceVersion(service.getVersion());
		notificationData.setDistributionID(distributionId);
		notificationData.setServiceUUID(service.getUUID());
		notificationData.setServiceDescription(service.getDescription());
		notificationData.setServiceInvariantUUID(service.getInvariantUUID());
		String workloadContext= ConfigurationManager.getConfigurationManager().getConfiguration().getWorkloadContext();
		if(workloadContext!=null){
			notificationData.setWorkloadContext(workloadContext);
		}
		logger.debug("Before returning notification data object {}", notificationData);

		return notificationData;

	}

	public INotificationData buildServiceForDistribution(INotificationData notificationData, Service service) {

		notificationData.setServiceArtifacts(convertServiceArtifactsToArtifactInfo(service));

		logger.debug("Before returning notification data object {}", notificationData);

		return notificationData;

	}

	private List<ArtifactInfoImpl> convertServiceArtifactsToArtifactInfo(Service service) {
		
		Map<String, ArtifactDefinition> serviceArtifactsMap = service.getDeploymentArtifacts();
		List<ArtifactDefinition> extractedServiceArtifacts = serviceArtifactsMap.values().stream()
				//filters all artifacts with existing EsId
				.filter(artifactDef -> artifactDef.checkEsIdExist())
				//collects all filtered artifacts with existing EsId to List
				.collect(Collectors.toList());
		
		Optional<ArtifactDefinition> toscaTemplateArtifactOptl = exrtactToscaTemplateArtifact(service);
		if(toscaTemplateArtifactOptl.isPresent()){
			extractedServiceArtifacts.add(toscaTemplateArtifactOptl.get());
		}
		
		Optional<ArtifactDefinition> toscaCsarArtifactOptl = exrtactToscaCsarArtifact(service);
		if(toscaCsarArtifactOptl.isPresent()){
			extractedServiceArtifacts.add(toscaCsarArtifactOptl.get());
		}
		
		List<ArtifactInfoImpl> artifacts = ArtifactInfoImpl.convertServiceArtifactToArtifactInfoImpl(service, extractedServiceArtifacts);
		return artifacts;
	}

	private Optional<ArtifactDefinition> exrtactToscaTemplateArtifact(Service service) {
		return service.getToscaArtifacts().values().stream()
				//filters TOSCA_TEMPLATE artifact
				.filter(e -> e.getArtifactType().equals(ArtifactTypeEnum.TOSCA_TEMPLATE.getType())).findAny();
	}
	
	private Optional<ArtifactDefinition> exrtactToscaCsarArtifact(Service service) {
		return service.getToscaArtifacts().values().stream()
				//filters TOSCA_CSAR artifact
				.filter(e -> e.getArtifactType().equals(ArtifactTypeEnum.TOSCA_CSAR.getType())).findAny();
	}

	private List<JsonContainerResourceInstance> convertRIToJsonContanier(Service service) {
		List<JsonContainerResourceInstance> ret = new ArrayList<JsonContainerResourceInstance>();
		if (service.getComponentInstances() != null) {
			for (ComponentInstance resourceInstance : service.getComponentInstances()) {
				String resoucreType = resourceInstance.getOriginType().getValue();
				List<ArtifactDefinition> artifactsDefList = getArtifactsWithPayload(resourceInstance);
				List<ArtifactInfoImpl> artifacts = ArtifactInfoImpl.convertToArtifactInfoImpl(service, resourceInstance,
						artifactsDefList);

				String resourceInvariantUUID = null;
				String resourceCategory = null;
				String resourceSubcategory = null;

				ComponentParametersView componentParametersView = new ComponentParametersView();
				componentParametersView.disableAll();
				componentParametersView.setIgnoreCategories(false);
				Either<Resource, StorageOperationStatus> componentResponse = toscaOperationFacade
						.getToscaElement(resourceInstance.getComponentUid(), componentParametersView);

				if (componentResponse.isRight()) {
					logger.debug("Resource {} Invariant UUID & Categories retrieving failed", resourceInstance.getComponentUid());
				} else {
					Resource resource = componentResponse.left().value();
					resourceInvariantUUID = resource.getInvariantUUID();

					List<CategoryDefinition> categories = resource.getCategories();

					if (categories != null) {
						CategoryDefinition categoryDefinition = categories.get(0);

						if (categoryDefinition != null) {
							resourceCategory = categoryDefinition.getName();
							List<SubCategoryDefinition> subcategories = categoryDefinition.getSubcategories();
							if (null != subcategories) {
								SubCategoryDefinition subCategoryDefinition = subcategories.get(0);

								if (subCategoryDefinition != null) {
									resourceSubcategory = subCategoryDefinition.getName();
								}
							}
						}
					}
				}

				JsonContainerResourceInstance jsonContainer = new JsonContainerResourceInstance(resourceInstance, resoucreType,
						rebuildArtifactswith120TimeoutInsteadOf60(artifacts)/*TODO used to send artifacts, the function is a fix to the short timeout bug in distribution*/);
				jsonContainer.setResourceInvariantUUID(resourceInvariantUUID);
				jsonContainer.setCategory(resourceCategory);
				jsonContainer.setSubcategory(resourceSubcategory);
				ret.add(jsonContainer);
			}
		}
		return ret;
	}

	private List<ArtifactInfoImpl> rebuildArtifactswith120TimeoutInsteadOf60(List<ArtifactInfoImpl> artifacts) {
		for(ArtifactInfoImpl artifact : artifacts){
			if(artifact.getArtifactTimeout().equals(60)){
				artifact.setArtifactTimeout(120);
			}
		}
		return artifacts;
	}

	private List<ArtifactDefinition> getArtifactsWithPayload(ComponentInstance resourceInstance) {
		List<ArtifactDefinition> ret = new ArrayList<ArtifactDefinition>();

		// List<ArtifactDefinition> informationDeployedArtifacts =
		// informationDeployedArtifactsBusinessLogic.getInformationalDeployedArtifactsForResourceInstance(resourceInstance);
		List<ArtifactDefinition> deployableArtifacts = new ArrayList<ArtifactDefinition>();
		// deployableArtifacts.addAll(informationDeployedArtifacts);
		if (resourceInstance.getDeploymentArtifacts() != null) {
			deployableArtifacts.addAll(resourceInstance.getDeploymentArtifacts().values());
		}

		for (ArtifactDefinition artifactDef : deployableArtifacts) {
			if (artifactDef.checkEsIdExist()) {
				ret.add(artifactDef);
			}
		}

		return ret;
	}

	/**
	 * build the url for resource intance artifact
	 * 
	 * @param service
	 * @param resourceData
	 * @param artifactName
	 * @return
	 */
	public static String buildResourceInstanceArtifactUrl(Service service, ComponentInstance resourceInstance,
			String artifactName) {

		String url = String.format(RESOURCE_INSTANCE_ARTIFACT_URL, service.getSystemName(), service.getVersion(),
				resourceInstance.getNormalizedName(), artifactName);

		logger.debug("After building artifact url {}", url);

		return url;
	}

	/**
	 * build the url for resource intance artifact
	 * 
	 * @param service
	 * @param resourceData
	 * @param artifactName
	 * @return
	 */
	public static String buildServiceArtifactUrl(Service service, String artifactName) {

		String url = String.format(SERVICE_ARTIFACT_URL, service.getSystemName(), service.getVersion(), artifactName);

		logger.debug("After building artifact url {}", url);

		return url;

	}

	/**
	 * Retrieve all deployment artifacts of all resources under a given service
	 * 
	 * @param resourceArtifactsResult
	 * @param service
	 * @param deConfiguration
	 * @return
	 */
	public Either<Boolean, StorageOperationStatus> isServiceContainsDeploymentArtifacts(Service service) {

		Either<Boolean, StorageOperationStatus> result = Either.left(false);
		Map<String, ArtifactDefinition> serviseArtifactsMap = service.getDeploymentArtifacts();
		if (serviseArtifactsMap != null && !serviseArtifactsMap.isEmpty()) {
			result = Either.left(true);
			return result;
		}

		List<ComponentInstance> resourceInstances = service.getComponentInstances();

		if (resourceInstances != null) {
			for (ComponentInstance resourceInstance : resourceInstances) {

				Map<String, ArtifactDefinition> deploymentArtifactsMapper = resourceInstance.getDeploymentArtifacts();
				// List<ArtifactDefinition> informationDeployedArtifacts =
				// informationDeployedArtifactsBusinessLogic.getInformationalDeployedArtifactsForResourceInstance(resourceInstance);

				boolean isDeployableArtifactFound = isContainsPayload(deploymentArtifactsMapper.values());// ||
																											// isContainsPayload(informationDeployedArtifacts);
				if (isDeployableArtifactFound) {
					result = Either.left(true);
					break;
				}

			}

		}

		return result;
	}

	private boolean isContainsPayload(Collection<ArtifactDefinition> collection) {
		boolean payLoadFound = collection != null && collection.stream().anyMatch(p -> p.checkEsIdExist());
		return payLoadFound;
	}

}