org.onap migration

Change-Id: I52f0b2851f2c765752b6d21f49b32136d7d72a3d
Issue-ID: VID-86
Signed-off-by: Ofir Sonsino <os0695@att.com>

1 files changed, 394 insertions, 0 deletions

diff --git a/vid-app-common/src/main/java/org/onap/vid/asdc/local/LocalAsdcClient.java b/vid-app-common/src/main/java/org/onap/vid/asdc/local/LocalAsdcClient.java
new file mode 100644
index 000000000..3c04e5ce8
--- /dev/null
+++ b/vid-app-common/src/main/java/org/onap/vid/asdc/local/LocalAsdcClient.java
@@ -0,0 +1,394 @@
+package org.onap.vid.asdc.local;
+
+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.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;
+import org.yaml.snakeyaml.Yaml;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardCopyOption;
+import java.util.Collection;
+import java.util.LinkedList;
+import java.util.Map;
+import java.util.UUID;
+import java.util.zip.ZipFile;
+
+/**
+ * The Class LocalAsdcClient.
+ */
+public class LocalAsdcClient implements AsdcClient {
+
+
+    /**
+     * The catalog.
+     */
+    private final JSONObject catalog;
+
+    /**
+     * The mapper.
+     */
+    private final ObjectMapper mapper;
+
+    /**
+     * Instantiates a new in local sdc client.
+     *
+     * @param builder the builder
+     */
+    public LocalAsdcClient(org.onap.vid.asdc.local.LocalAsdcClient.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)
+     * @see org.openecomp.vid.asdc.AsdcClient#getResources()
+     */
+    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 (Map.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 {
+
+        JSONObject serviceJsonObject = null;
+        final JSONArray categoryJsonArray = getCatalog().getJSONArray("services");
+
+        for (int i = 0; i < categoryJsonArray.length(); i++) {
+            JSONObject jsonServiceObject = categoryJsonArray.getJSONObject(i);
+            if (jsonServiceObject.get("uuid").equals(uuid.toString())) {
+                serviceJsonObject = jsonServiceObject;
+                break;
+            }
+        }
+
+        if (serviceJsonObject != null)
+            return convert(serviceJsonObject, Service.class);
+        else return null;
+    }
+
+    /* (non-Javadoc)
+     * @see org.openecomp.vid.asdc.AsdcClient#getServices()
+     */
+    public Collection<Service> getServices() throws AsdcCatalogException {
+        final Collection<Service> services = new LinkedList<Service>();
+
+        JSONArray servicesArr = getCatalog().getJSONArray("services");
+
+        for (Object objService : servicesArr) {
+            JSONObject jsonServiceItem = (JSONObject) objService;
+            final Service service = convert(jsonServiceItem, 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>();
+
+        JSONArray catalogServices = catalog.getJSONArray("services");
+
+        for (int i = 0; i < catalogServices.length(); i++) {
+
+            JSONObject serviceJson = catalogServices.getJSONObject(i);
+
+            boolean filterMatch = true;
+
+            for (Map.Entry<String, String[]> entry : filter.entrySet()) {
+                for (int j = 0; j < entry.getValue().length; j++) {
+                    if (!serviceJson.getString(entry.getKey()).equals(entry.getValue()[j])) {
+                        filterMatch = false;
+                        break;
+                    }
+                }
+            }
+            if (filterMatch) services.add(convert(serviceJson, 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 {
+
+        String toscaModelURL = null;
+
+        final JSONArray categoryJsonArray = getCatalog().getJSONArray("services");
+
+        for (int i = 0; i < categoryJsonArray.length(); i++) {
+
+            JSONObject jsonServiceObject = categoryJsonArray.getJSONObject(i);
+            if (jsonServiceObject.get("uuid").equals(serviceUuid.toString())) {
+                toscaModelURL = jsonServiceObject.getString("toscaModelURL");
+                break;
+            }
+        }
+        if (toscaModelURL == null) {
+            return null;
+        }
+        final InputStream toscaModelStream = getClass().getClassLoader().getResourceAsStream(toscaModelURL);
+
+        ClassLoader classLoader = getClass().getClassLoader();
+        File file = new File(classLoader.getResource(toscaModelURL).getFile());
+        Path path = Paths.get(file.getPath());
+
+        if (toscaModelStream == null) return null;
+
+        return path;
+    }
+
+    /**
+     * 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 (Map.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);
+        }
+    }
+
+    /**
+     * 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 org.onap.vid.asdc.local.LocalAsdcClient.Builder catalog(JSONObject catalog) {
+            this.catalog = catalog;
+            return this;
+        }
+
+        /**
+         * Mapper.
+         *
+         * @param mapper the mapper
+         * @return the builder
+         */
+        public org.onap.vid.asdc.local.LocalAsdcClient.Builder mapper(ObjectMapper mapper) {
+            this.mapper = mapper;
+            return this;
+        }
+
+        /**
+         * Builds the.
+         *
+         * @return the in local sdc client
+         */
+        public org.onap.vid.asdc.local.LocalAsdcClient build() {
+            return new org.onap.vid.asdc.local.LocalAsdcClient(this);
+        }
+    }
+
+}