/*-
 * ============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.impl;

import java.util.ArrayList;
import java.util.List;

import org.onap.sdc.api.notification.IArtifactInfo;
import org.onap.sdc.api.notification.IResourceInstance;
import org.onap.sdc.api.results.IDistributionClientResult;
import org.onap.sdc.utils.DistributionActionResultEnum;
import org.onap.sdc.api.consumer.INotificationCallback;
import org.onap.sdc.api.notification.INotificationData;
import org.onap.sdc.utils.ArtifactTypeEnum;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.att.nsa.cambria.client.CambriaConsumer;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

class NotificationConsumer implements Runnable {

	private static Logger log = LoggerFactory.getLogger(NotificationConsumer.class.getName());

	private CambriaConsumer cambriaConsumer;
	private INotificationCallback clientCallback;
	private List<String> artifactsTypes;
	private DistributionClientImpl distributionClient;

	public NotificationConsumer(CambriaConsumer cambriaConsumer, INotificationCallback clientCallback, List<String> artifactsTypes, DistributionClientImpl distributionClient) {
		this.cambriaConsumer = cambriaConsumer;
		this.clientCallback = clientCallback;
		this.artifactsTypes = artifactsTypes;
		this.distributionClient = distributionClient;
	}

	@Override
	public void run() {

		try {
			Gson gson = new GsonBuilder().setPrettyPrinting().create();
			long currentTimeMillis = System.currentTimeMillis();
			for (String notificationMsg : cambriaConsumer.fetch()) {
				log.debug("received message from topic");
				log.debug("recieved notification from broker: {}", notificationMsg);

				final NotificationDataImpl notificationFromUEB = gson.fromJson(notificationMsg, NotificationDataImpl.class);
				NotificationDataImpl notificationForCallback = buildCallbackNotificationLogic(currentTimeMillis, notificationFromUEB);
				if (isActivateCallback(notificationForCallback)) {
					String stringNotificationForCallback = gson.toJson(notificationForCallback);
					log.debug("sending notification to client: {}", stringNotificationForCallback);
					clientCallback.activateCallback(notificationForCallback);
				}
			}

		} catch (Exception e) {
			log.error("Error exception occured when fetching with Cambria Client:{}", e.getMessage());
			log.debug("Error exception occured when fetching with Cambria Client:{}", e.getMessage(), e);
		}
	}

	private boolean isActivateCallback(NotificationDataImpl notificationForCallback) {
		boolean hasRelevantArtifactsInResourceInstance = notificationForCallback.getResources() != null && !notificationForCallback.getResources().isEmpty();
		boolean hasRelevantArtifactsInService = notificationForCallback.getServiceArtifacts() != null && !notificationForCallback.getServiceArtifacts().isEmpty();

		return hasRelevantArtifactsInResourceInstance || hasRelevantArtifactsInService;
	}

	protected NotificationDataImpl buildCallbackNotificationLogic(long currentTimeMillis, final NotificationDataImpl notificationFromUEB) {
		List<IResourceInstance> relevantResourceInstances = buildResourceInstancesLogic(notificationFromUEB, currentTimeMillis);
		List<ArtifactInfoImpl> relevantServiceArtifacts = handleRelevantArtifacts(notificationFromUEB, currentTimeMillis, notificationFromUEB.getServiceArtifactsImpl());
		notificationFromUEB.setResources(relevantResourceInstances);
		notificationFromUEB.setServiceArtifacts(relevantServiceArtifacts);
		return notificationFromUEB;
	}

	private List<IResourceInstance> buildResourceInstancesLogic(NotificationDataImpl notificationFromUEB, long currentTimeMillis) {

		List<IResourceInstance> relevantResourceInstances = new ArrayList<>();

		for (JsonContainerResourceInstance resourceInstance : notificationFromUEB.getResourcesImpl()) {
			final List<ArtifactInfoImpl> artifactsImplList = resourceInstance.getArtifactsImpl();
			List<ArtifactInfoImpl> foundRelevantArtifacts = handleRelevantArtifacts(notificationFromUEB, currentTimeMillis, artifactsImplList);
			if (!foundRelevantArtifacts.isEmpty() || distributionClient.getConfiguration().isFilterInEmptyResources()) {
				resourceInstance.setArtifacts(foundRelevantArtifacts);
				relevantResourceInstances.add(resourceInstance);
			}
		}
		return relevantResourceInstances;

	}

	private List<ArtifactInfoImpl> handleRelevantArtifacts(NotificationDataImpl notificationFromUEB, long currentTimeMillis, final List<ArtifactInfoImpl> artifactsImplList) {
		List<ArtifactInfoImpl> relevantArtifacts = new ArrayList<>();
		if (artifactsImplList != null) {
			for (ArtifactInfoImpl artifactInfo : artifactsImplList) {
				handleRelevantArtifact(notificationFromUEB, currentTimeMillis, artifactsImplList, relevantArtifacts, artifactInfo);
			}
		}
		return relevantArtifacts;
	}

	private void handleRelevantArtifact(NotificationDataImpl notificationFromUEB, long currentTimeMillis, final List<ArtifactInfoImpl> artifactsImplList, List<ArtifactInfoImpl> relevantArtifacts, ArtifactInfoImpl artifactInfo) {
		boolean isArtifactRelevant = artifactsTypes.contains(artifactInfo.getArtifactType());
		String artifactType = artifactInfo.getArtifactType();
		if (artifactInfo.getGeneratedFromUUID() != null && !artifactInfo.getGeneratedFromUUID().isEmpty()) {
			IArtifactInfo generatedFromArtInfo = findGeneratedFromArtifact(artifactInfo.getGeneratedFromUUID(), artifactsImplList);
			if (generatedFromArtInfo != null)
				isArtifactRelevant = isArtifactRelevant && artifactsTypes.contains(generatedFromArtInfo.getArtifactType());
			else
				isArtifactRelevant = false;
		}
		if (isArtifactRelevant) {
			setRelatedArtifacts(artifactInfo, notificationFromUEB);
			if (artifactType.equals(ArtifactTypeEnum.HEAT.name()) || artifactType.equals(ArtifactTypeEnum.HEAT_VOL.name()) || artifactType.equals(ArtifactTypeEnum.HEAT_NET.name())) {
				setGeneratedArtifact(artifactsImplList, artifactInfo);
			}
			relevantArtifacts.add(artifactInfo);

		}
		IDistributionClientResult notificationStatus = distributionClient.sendNotificationStatus(currentTimeMillis, notificationFromUEB.getDistributionID(), artifactInfo, isArtifactRelevant);
		if (notificationStatus.getDistributionActionResult() != DistributionActionResultEnum.SUCCESS) {
			log.error("Error failed to send notification status to UEB failed status:{}, error message:{}", notificationStatus.getDistributionActionResult().name(), notificationStatus.getDistributionMessageResult());
		}
	}

	private void setRelatedArtifacts(ArtifactInfoImpl artifact, INotificationData notificationData) {
		if (artifact.getRelatedArtifactsUUID() != null) {
			List<IArtifactInfo> relatedArtifacts = new ArrayList<>();
			for (String relatedArtifactUUID : artifact.getRelatedArtifactsUUID()) {
				relatedArtifacts.add(notificationData.getArtifactMetadataByUUID(relatedArtifactUUID));
			}
			artifact.setRelatedArtifactsInfo(relatedArtifacts);
		}

	}

	private void setGeneratedArtifact(final List<ArtifactInfoImpl> artifactsImplList, ArtifactInfoImpl artifactInfo) {
		IArtifactInfo found = null;
		String artifactUUID = artifactInfo.getArtifactUUID();
		for (ArtifactInfoImpl generatedArtifactInfo : artifactsImplList) {
			if (generatedArtifactInfo.getArtifactType().equals(ArtifactTypeEnum.HEAT_ENV.name()) && artifactUUID.equals(generatedArtifactInfo.getGeneratedFromUUID())) {
				found = generatedArtifactInfo;
				break;
			}
		}

		artifactInfo.setGeneratedArtifact(found);
	}

	private IArtifactInfo findGeneratedFromArtifact(String getGeneratedFromUUID, List<ArtifactInfoImpl> list) {
		IArtifactInfo found = null;
		for (ArtifactInfoImpl artifactInfo : list) {
			if (getGeneratedFromUUID.equals(artifactInfo.getArtifactUUID())) {
				found = artifactInfo;
				break;
			}
		}
		return found;
	}

}