/*-
 * ============LICENSE_START=======================================================
 * VID
 * ================================================================================
 * 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.vid.asdc.memory;

import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import java.util.Collection;
import java.util.LinkedList;
import java.util.Map;
import java.util.Map.Entry;
import java.util.UUID;
import java.util.zip.ZipFile;

import org.codehaus.jackson.JsonParseException;
import org.codehaus.jackson.map.JsonMappingException;
import org.codehaus.jackson.map.ObjectMapper;
import org.json.JSONArray;
import org.json.JSONObject;
import org.yaml.snakeyaml.Yaml;

import org.onap.vid.asdc.AsdcCatalogException;
import org.onap.vid.asdc.AsdcClient;
import org.onap.vid.asdc.beans.Artifact;
import org.onap.vid.asdc.beans.Resource;
import org.onap.vid.asdc.beans.Service;
import org.onap.vid.asdc.beans.tosca.ToscaCsar;
import org.onap.vid.asdc.beans.tosca.ToscaMeta;
import org.onap.vid.asdc.beans.tosca.ToscaModel;

/**
 * The Class InMemoryAsdcClient.
 */
public class InMemoryAsdcClient implements AsdcClient {

	/** The catalog. */
	private final JSONObject catalog;
	
	/** The mapper. */
	private final ObjectMapper mapper;

	/**
	 * The Class Builder.
	 */
	public static class Builder {
		
		/** The catalog. */
		private JSONObject catalog = new JSONObject()
										.put("resources", new JSONObject())
										.put("services", new JSONObject());
		
		/** The mapper. */
		private ObjectMapper mapper = new ObjectMapper();

		/**
		 * Instantiates a new builder.
		 */
		public Builder() {}
		
		/**
		 * Catalog.
		 *
		 * @param catalog the catalog
		 * @return the builder
		 */
		public Builder catalog(JSONObject catalog) {
			this.catalog = catalog;
			return this;
		}
		
		/**
		 * Mapper.
		 *
		 * @param mapper the mapper
		 * @return the builder
		 */
		public Builder mapper(ObjectMapper mapper) {
			this.mapper = mapper;
			return this;
		}
		
		/**
		 * Builds the.
		 *
		 * @return the in memory sdc client
		 */
		public InMemoryAsdcClient build() {
			return new InMemoryAsdcClient(this);
		}
	}
	
	/**
	 * Instantiates a new in memory sdc client.
	 *
	 * @param builder the builder
	 */
	private InMemoryAsdcClient(Builder builder) {
		catalog = builder.catalog;
		mapper = builder.mapper;
	}
	
	/**
	 * Gets the catalog.
	 *
	 * @return the catalog
	 */
	private JSONObject getCatalog() {
		return catalog;
	}
	
	/**
	 * Gets the mapper.
	 *
	 * @return the mapper
	 */
	private ObjectMapper getMapper() {
		return mapper;
	}
	
	/**
	 * Convert.
	 *
	 * @param <T> the generic type
	 * @param json the json
	 * @param clazz the clazz
	 * @return the t
	 * @throws AsdcCatalogException the sdc catalog exception
	 */
	private <T> T convert(JSONObject json, Class<T> clazz) throws AsdcCatalogException {
		try {
			return getMapper().readValue(json.toString(), clazz);
		} catch (JsonParseException e) {
			throw new AsdcCatalogException("Failed to parse SDC response (bad data)", e);
		} catch (JsonMappingException e) {
			throw new AsdcCatalogException("Failed to map SDC response to internal VID data structure(s)", e);
		} catch (IOException e) {
			throw new AsdcCatalogException("Failed to get a response from SDC", e);
		}
	}
	
	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getResource(java.util.UUID)
	 */
	public Resource getResource(UUID uuid) throws AsdcCatalogException {
		final JSONObject resource = getCatalog().getJSONObject("resources")
										.getJSONObject(uuid.toString());
		return convert(resource, Resource.class);
	}

	/* (non-Javadoc)
	 */
	public Collection<Resource> getResources() throws AsdcCatalogException {
		final Collection<Resource> resources = new LinkedList<Resource> ();
		
		for (String key : getCatalog().getJSONObject("resources").keySet()) {
			final JSONObject json = getCatalog().getJSONObject("resources").getJSONObject(key);
			final Resource resource = convert(json, Resource.class);
			resources.add(resource);
		}
		
		return resources;
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getResources(java.util.Map)
	 */
	public Collection<Resource> getResources(Map<String, String[]> filter) throws AsdcCatalogException {
		final Collection<Resource> resources = new LinkedList<Resource> ();
		
		for (String key : getCatalog().getJSONObject("resources").keySet()) {
			final JSONObject json = getCatalog().getJSONObject("resources").getJSONObject(key);
			
			boolean filterMatch = true;
			
			for (Entry<String, String[]> entry : filter.entrySet()) {
				for (int i = 0; i < entry.getValue().length; i++) {
					if (!json.getString(entry.getKey()).equals(entry.getValue()[i])) {
						filterMatch = false;
						break;
					}
				}
			}
			
			if (filterMatch) resources.add(convert(json, Resource.class));
		}
		
		return resources;
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getService(java.util.UUID)
	 */
	public Service getService(UUID uuid) throws AsdcCatalogException {
		final JSONObject service = getCatalog().getJSONObject("services")
				.getJSONObject(uuid.toString());
		return convert(service, Service.class);
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getServices()
	 */
	public Collection<Service> getServices() throws AsdcCatalogException {
		final Collection<Service> services = new LinkedList<Service> ();
		
		for (String key : getCatalog().getJSONObject("services").keySet()) {
			final JSONObject json = getCatalog().getJSONObject("services").getJSONObject(key);
			final Service service = convert(json, Service.class);
			services.add(service);
		}
		
		return services;
	}

	/* (non-Javadoc)
	 * @see org.openecompt.vid.asdc.AsdcClient#getServices(java.util.Map)
	 */
	public Collection<Service> getServices(Map<String, String[]> filter) throws AsdcCatalogException {
		final Collection<Service> services = new LinkedList<Service> ();
		
		for (String key : getCatalog().getJSONObject("services").keySet()) {
			final JSONObject json = getCatalog().getJSONObject("services").getJSONObject(key);
			
			boolean filterMatch = true;
			
			for (Entry<String, String[]> entry : filter.entrySet()) {
				for (int i = 0; i < entry.getValue().length; i++) {
					if (!json.getString(entry.getKey()).equals(entry.getValue()[i])) {
						filterMatch = false;
						break;
					}
				}
			}
			
			if (filterMatch) services.add(convert(json, Service.class));
		}
		
		return services;
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getResourceArtifact(java.util.UUID, java.util.UUID)
	 */
	public Artifact getResourceArtifact(UUID resourceUuid, UUID artifactUuid) throws AsdcCatalogException {
		final JSONArray artifacts = getCatalog().getJSONObject("resources")
				.getJSONObject(resourceUuid.toString())
				.getJSONArray("artifacts");
		
		for (int i = 0; i < artifacts.length(); i++) {
			final JSONObject artifact = artifacts.getJSONObject(i);
			
			if (artifact.getString("artifactUUID").equals(artifactUuid.toString())) {
				return convert(artifact, Artifact.class);
			}
		}
		
		return null;
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getServiceArtifact(java.util.UUID, java.util.UUID)
	 */
	public Artifact getServiceArtifact(UUID serviceUuid, UUID artifactUuid) throws AsdcCatalogException {
		final JSONArray artifacts = getCatalog().getJSONObject("services")
				.getJSONObject(serviceUuid.toString())
				.getJSONArray("artifacts");
		
		for (int i = 0; i < artifacts.length(); i++) {
			final JSONObject artifact = artifacts.getJSONObject(i);
			
			if (artifact.getString("artifactUUID").equals(artifactUuid.toString())) {
				return convert(artifact, Artifact.class);
			}
		}
		
		return null;
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getResourceToscaModel(java.util.UUID)
	 */
	public Path getResourceToscaModel(UUID resourceUuid) throws AsdcCatalogException {
		final String toscaModelURL = getCatalog().getJSONObject("resources")
				.getJSONObject(resourceUuid.toString())
				.getString("toscaModelURL");


		final InputStream toscaModelStream = getClass().getClassLoader().getResourceAsStream(toscaModelURL);
		
		if (toscaModelStream == null) return null;
		
		return null;//getToscaModel(toscaModelStream);
	}

	/* (non-Javadoc)
	 * @see org.openecomp.vid.asdc.AsdcClient#getServiceToscaModel(java.util.UUID)
	 */
	public Path getServiceToscaModel(UUID serviceUuid) throws AsdcCatalogException {
		final String toscaModelURL = getCatalog().getJSONObject("services")
						.getJSONObject(serviceUuid.toString())
						.getString("toscaModelURL");
		
		final InputStream toscaModelStream = getClass().getClassLoader().getResourceAsStream(toscaModelURL);
		
		if (toscaModelStream == null) return null;
		
		return null;//getToscaModel(toscaModelStream);
	}

	/**
	 * Gets the tosca model.
	 *
	 * @param csarInputStream the csar input stream
	 * @return the tosca model
	 * @throws AsdcCatalogException the asdc catalog exception
	 */
	private ToscaCsar getToscaModel(InputStream csarInputStream) throws AsdcCatalogException {
		final Path csarFile;
		
		try {
			csarFile = Files.createTempFile("csar", ".zip");
			Files.copy(csarInputStream, csarFile, StandardCopyOption.REPLACE_EXISTING);
		} catch (IOException e) {
			throw new AsdcCatalogException("Caught IOException while creating CSAR", e);
		}
		
		try (final ZipFile csar = new ZipFile(csarFile.toFile())) {
			
			final InputStream toscaMetaStream = csar.getInputStream(csar.getEntry("TOSCA-Metadata/TOSCA.meta"));
			final ToscaMeta toscaMeta = new ToscaMeta.Builder(toscaMetaStream).build();
			final String entryDefinitions = toscaMeta.get("Entry-Definitions");
			final InputStream toscaParentEntryYamlStream = csar.getInputStream(csar.getEntry(entryDefinitions));
			
			final Yaml yaml = new Yaml();
			final ToscaModel parentModel = yaml.loadAs(toscaParentEntryYamlStream, ToscaModel.class);

			final ToscaCsar.Builder csarBuilder = new ToscaCsar.Builder(parentModel);
			
			for (Map<String, Map<String, String>> imports : parentModel.getImports()) {
				for (Entry<String, Map<String, String>> entry : imports.entrySet()) {
					final InputStream toscaChildEntryYamlStream = csar.getInputStream(csar.getEntry("Definitions/" + entry.getValue().get("file")));
					final ToscaModel childModel = yaml.loadAs(toscaChildEntryYamlStream, ToscaModel.class);
					csarBuilder.addVnf(childModel);
				}
			}
			
			return csarBuilder.build();
		} catch (IOException e) {
			throw new AsdcCatalogException("Caught IOException while processing CSAR", e);
		}
	}

}