updated to use latest vdu plugin code

Change-Id: I46cd41d32e0902a141a87d0c1e01aea278ec17ea
Issue-ID: SO-561
Signed-off-by: DeWayne Filppi <dewayne@cloudify.co>

3 files changed, 534 insertions, 2 deletions

diff --git a/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/adapters/vdu/VduStatus.java b/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/adapters/vdu/VduStatus.java
index 174bed90a2..4d9702f726 100644
--- a/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/adapters/vdu/VduStatus.java
+++ b/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/adapters/vdu/VduStatus.java
@@ -25,7 +25,11 @@ public class VduStatus {
 	private VduStateType state;	

 	private String errorMessage;

 	private PluginAction lastAction;	

-	

+

+	public VduStatus(){}

+	public VduStatus( VduStateType state) {

+		this.state = state;

+	}

 	public VduStateType getState() {

 		return state;

 	}

@@ -51,4 +55,4 @@ public class VduStatus {
 		lastAction.setRawMessage(rawCloudMessage);

 	}

 	

-}
\ No newline at end of file
+}

diff --git a/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/aria/AriaVduPlugin.java b/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/aria/AriaVduPlugin.java
new file mode 100644
index 0000000000..2d32f85d2d
--- /dev/null
+++ b/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/aria/AriaVduPlugin.java
@@ -0,0 +1,340 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * 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.mso.aria;
+
+import com.gigaspaces.aria.rest.client.AriaClient;
+import com.gigaspaces.aria.rest.client.AriaClientFactory;
+import com.gigaspaces.aria.rest.client.ExecutionDetails;
+import com.gigaspaces.aria.rest.client.Input;
+import com.gigaspaces.aria.rest.client.InputImpl;
+import com.gigaspaces.aria.rest.client.Output;
+import com.gigaspaces.aria.rest.client.Service;
+import com.gigaspaces.aria.rest.client.ServiceTemplate;
+import com.gigaspaces.aria.rest.client.ServiceTemplateImpl;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import org.openecomp.mso.adapters.vdu.CloudInfo;
+import org.openecomp.mso.adapters.vdu.VduException;
+import org.openecomp.mso.adapters.vdu.VduInstance;
+import org.openecomp.mso.adapters.vdu.VduModelInfo;
+import org.openecomp.mso.adapters.vdu.VduPlugin;
+import org.openecomp.mso.adapters.vdu.VduStateType;
+import org.openecomp.mso.adapters.vdu.VduStatus;
+import org.openecomp.mso.logger.MessageEnum;
+import org.openecomp.mso.logger.MsoLogger;
+
+/**
+ * ARIA VDU Plugin. Pluggable interface for the ARIA REST API to support TOSCA orchestration.
+ *
+ * @author DeWayne
+ */
+public class AriaVduPlugin implements VduPlugin {
+    private static final String API_VERSION = "0.1";
+    private static final MsoLogger logger = MsoLogger.getMsoLogger(MsoLogger.Catalog.RA);
+    private AriaClient client = null;
+    private Map<String, Integer> templateIds = new HashMap<>();
+    private Map<String, Integer> serviceIds = new HashMap<>();
+    private Map<String, Map<String, Object>> inputsCache = new HashMap<>();
+
+    public AriaVduPlugin() {
+        super();
+    }
+
+    public AriaVduPlugin(String host, int port) {
+        try {
+            client = new AriaClientFactory().createRestClient("http", host, port, API_VERSION);
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    "aria",
+                    MsoLogger.ErrorCode.AvailabilityError,
+                    "Connection to ARIA REST API failed",
+                    e);
+            throw e;
+        }
+    }
+
+    /**
+     * Instantiate VDU in ARIA. <code>instanceName</code> is used for both service template name and
+     * service name.
+     */
+    @SuppressWarnings("unchecked")
+    @Override
+    public VduInstance instantiateVdu(
+            CloudInfo cloudInfo,
+            String instanceName,
+            Map<String, Object> inputs,
+            VduModelInfo vduModel,
+            boolean rollbackOnFailure)
+            throws VduException {
+
+        String cloudSiteId = cloudInfo.getCloudSiteId();
+        String tenantId = cloudInfo.getTenantId();
+
+        // Currently only support simple CSAR with single main template
+        byte[] csar = new CSAR(vduModel).create();
+
+        ServiceTemplate template = new ServiceTemplateImpl(instanceName, csar);
+        try {
+            client.install_service_template(template);
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceName,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    "instantiate vdu via csar failed",
+                    e);
+            throw new VduException(e.getMessage());
+        }
+
+        /** Create a service */
+        try {
+            int templateId = -1;
+            for (ServiceTemplate stemplate :
+                    (List<ServiceTemplate>) client.list_service_templates()) {
+                if (stemplate.getName().equals(instanceName)) {
+                    templateId = stemplate.getId();
+                }
+            }
+            List<Input> sinputs = new ArrayList<Input>();
+            for (Map.Entry<String, ? extends Object> entry : inputs.entrySet()) {
+                Input inp = new InputImpl(entry.getKey(), entry.getValue().toString(), "");
+                sinputs.add(inp);
+            }
+            client.create_service(templateId, instanceName, sinputs);
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceName,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    "aria service creation failed",
+                    e);
+            throw new VduException(e.getMessage());
+        }
+
+        // Get the service ID and cache it
+        int sid = getServiceId(instanceName);
+        serviceIds.put(instanceName, sid);
+
+        /** Run install */
+        try {
+            client.start_execution(sid, "install", new ExecutionDetails());
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceName,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    "aria install workflow failed",
+                    e);
+            throw new VduException(e.getMessage());
+        }
+
+        /** Get the outputs and return */
+        try {
+            Map<String, Object> voutputs = getOutputs(sid);
+
+            VduInstance vi = new VduInstance();
+            vi.setVduInstanceName(instanceName);
+            vi.setInputs((Map<String, Object>) inputs);
+            inputsCache.put(instanceName, vi.getInputs());
+            vi.setOutputs(voutputs);
+            vi.setStatus(new VduStatus(VduStateType.INSTANTIATED));
+            return vi;
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceName,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    "aria service output fetch failed",
+                    e);
+            throw new VduException(e.getMessage());
+        }
+    }
+
+    /**
+     * Queries ARIA for VDU status. instanceId used as template and service name in ARIA (by
+     * convention).
+     */
+    @Override
+    public VduInstance queryVdu(CloudInfo cloudInfo, String instanceId) throws VduException {
+        if (client == null) {
+            throw new VduException("Internal error: no ARIA connection found");
+        }
+
+        VduInstance vif = new VduInstance();
+        vif.setVduInstanceId(instanceId);
+        Integer sid = serviceIds.get(instanceId);
+        if (sid == null) {
+            // service doesn't exist
+            vif.setStatus(new VduStatus(VduStateType.NOTFOUND));
+            return vif;
+        }
+        Service service = client.get_service(sid);
+        if (service == null) {
+            throw new VduException(
+                    String.format("Internal error: cached service id %s not found in ARIA", sid));
+        }
+        Map<String, Object> voutputs = getOutputs(sid);
+        vif.setOutputs(voutputs);
+        vif.setInputs(inputsCache.get(instanceId));
+        vif.setStatus(new VduStatus(VduStateType.INSTANTIATED));
+        return vif;
+    }
+
+    @Override
+    public VduInstance deleteVdu(CloudInfo cloudInfo, String instanceId, int timeoutMinutes)
+            throws VduException {
+        VduInstance vif = new VduInstance();
+        vif.setVduInstanceId(instanceId);
+
+        if (client == null) {
+            throw new VduException("Internal error: no ARIA connection found");
+        }
+        Integer sid = serviceIds.get(instanceId);
+        if (sid == null) {
+            // service doesn't exist
+            vif.setStatus(new VduStatus(VduStateType.NOTFOUND));
+            return vif;
+        }
+
+        /** Run uninstall */
+        try {
+            client.start_execution(sid, "uninstall", new ExecutionDetails());
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceId,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    "aria uninstall workflow failed",
+                    e);
+            throw new VduException(e.getMessage());
+        }
+
+        /** Delete the service */
+        try {
+            client.delete_service(sid);
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceId,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    String.format("aria service delete failed. Service id: %d", sid),
+                    e);
+            throw new VduException(e.getMessage());
+        }
+
+        /** Delete the blueprint */
+        try {
+            client.delete_service_template(templateIds.get(instanceId));
+        } catch (Exception e) {
+            logger.error(
+                    MessageEnum.RA_CREATE_VNF_ERR,
+                    "",
+                    "",
+                    "",
+                    "",
+                    instanceId,
+                    MsoLogger.ErrorCode.BusinessProcesssError,
+                    "aria template delete failed",
+                    e);
+            throw new VduException(e.getMessage());
+        }
+
+        vif.setStatus(new VduStatus(VduStateType.DELETED));
+        return vif;
+    }
+
+    /** Deployment update not possible with ARIA */
+    @Override
+    public VduInstance updateVdu(
+            CloudInfo cloudInfo,
+            String instanceId,
+            Map<String, Object> inputs,
+            VduModelInfo vduModel,
+            boolean rollbackOnFailure)
+            throws VduException {
+        throw new VduException("NOT IMPLEMENTED");
+    }
+
+    /** Private */
+
+    /**
+     * p Gets and repacks service outputs for internal use
+     *
+     * @param sid the service id (ARIA service id)
+     * @return
+     */
+    private Map<String, Object> getOutputs(int sid) {
+        @SuppressWarnings("unchecked")
+        List<Output> outputs = (List<Output>) client.list_service_outputs(sid);
+        Map<String, Object> voutputs = new HashMap<>();
+        for (Output output : outputs) {
+            voutputs.put(output.getName(), output.getValue());
+        }
+        return voutputs;
+    }
+
+    @SuppressWarnings("unchecked")
+    private int getServiceId(String service_name) throws VduException {
+        int sid = -1;
+        List<Service> services = (List<Service>) client.list_services();
+        for (Service service : services) {
+            if (service.getName().equals(service_name)) {
+                sid = service.getId();
+            }
+        }
+        if (sid == -1) {
+            throw new VduException(
+                    String.format(
+                            "Internal error: just created service not found: %s", service_name));
+        }
+        return sid;
+    }
+}
diff --git a/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/aria/CSAR.java b/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/aria/CSAR.java
new file mode 100644
index 0000000000..bb1201fba3
--- /dev/null
+++ b/adapters/mso-adapter-utils/src/main/java/org/openecomp/mso/aria/CSAR.java
@@ -0,0 +1,188 @@
+/*
+ * ============LICENSE_START===================================================
+ * Copyright (c) 2017 Cloudify.co.  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.mso.aria;
+
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.OutputStream;
+import java.io.PrintStream;
+import java.util.Map;
+import java.util.List;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+import org.openecomp.mso.adapters.vdu.VduModelInfo;
+import org.openecomp.mso.adapters.vdu.VduArtifact;
+import org.openecomp.mso.adapters.vdu.VduArtifact.ArtifactType;
+
+import com.google.common.io.Files;
+
+/**
+ * The purpose of this class is to create a CSAR byte array from Vdu inputs for the purpose
+ * of forwarding to a TOSCA orchestrator.
+ * 
+ * @author DeWayne
+ *
+ */
+public class CSAR {
+	private static final String MANIFEST_FILENAME = "MANIFEST.MF";
+	private VduModelInfo vduModel;
+
+	public CSAR(VduModelInfo model){
+	   this.vduModel = model;
+	}
+
+	/**
+	 * Creates a byte array representation of a CSAR corresponding to the VduBlueprint arg in the
+	 * constructor.  
+	 * 
+	 * @return
+	 */
+	public byte[] create() {
+		File dir = Files.createTempDir();
+
+		/**
+		 * Create subdir
+		 */
+		File metadir = new File(dir.getAbsolutePath() + "/" + "TOSCA-Metadata");
+		if (!metadir.mkdir()) {
+			throw new RuntimeException("CSAR TOSCA-Metadata directory create failed");
+		}
+
+		/**
+		 * Organize model info for consumption
+		 */
+		VduArtifact mainTemplate = null;
+		List<VduArtifact> extraFiles = null;
+		for(VduArtifact artifact: vduModel.getArtifacts()){
+				  if( artifact.getType() == ArtifactType.MAIN_TEMPLATE ){
+					 mainTemplate = artifact;
+				  }
+				  else{
+				    extraFiles.add(artifact);
+				  }
+		}	  
+
+		/**
+		 * Write template files
+		 */
+		OutputStream ofs = null;
+		try {
+			ofs = new FileOutputStream(new File(dir, mainTemplate.getName()));
+			ofs.write(mainTemplate.getContent());
+			ofs.close();
+
+			/**
+			 * Write other files
+			 */
+			if (extraFiles != null) {
+				for (VduArtifact artifact: extraFiles){
+					ofs = new FileOutputStream(new File(dir, artifact.getName()));
+					ofs.write(artifact.getContent());
+					ofs.close();
+				}
+			}
+
+
+			/**
+			 * Create manifest
+			 */
+			PrintStream mfstream = new PrintStream(new File(metadir.getAbsolutePath() + "/" + MANIFEST_FILENAME));
+			mfstream.println("TOSCA-Meta-File-Version: 1.0");
+			mfstream.println("CSAR-Version: 1.1");
+			mfstream.println("Created-by: ONAP");
+			mfstream.println("Entry-Definitions: " + mainTemplate.getName());
+			mfstream.close();
+
+			/**
+			 * ZIP it up
+			 */
+			ByteArrayOutputStream zipbytes = new ByteArrayOutputStream();
+			ZipOutputStream zos = new ZipOutputStream(zipbytes);
+			compressTree(zos, "", dir, dir);
+			zos.close();
+			return zipbytes.toByteArray();
+
+		} catch (Exception e) {
+			throw new RuntimeException("Failed to create CSAR: " + e.getMessage());
+		} finally {
+
+			/**
+			 * Clean up tmpdir
+			 */
+			deleteDirectory(dir);
+		}
+	}
+
+	/**
+	 * Private methods
+	 */
+
+	/**
+	 * Compresses (ZIPs) a directory tree
+	 * 
+	 * @param dir
+	 * @throws IOException
+	 */
+	private void compressTree(ZipOutputStream zos, String path, File basedir, File dir) throws IOException {
+		if (!dir.isDirectory())
+			return;
+
+		for (File f : dir.listFiles()) {
+			if (f.isDirectory()) {
+				String newpath = path + f.getName() + "/";
+				ZipEntry entry = new ZipEntry(newpath);
+				zos.putNextEntry(entry);
+				zos.closeEntry();
+				compressTree(zos, newpath, basedir, f);
+			} else {
+				ZipEntry ze = new ZipEntry(
+						f.getAbsolutePath().substring(basedir.getAbsolutePath().length() + 1).replaceAll("\\\\", "/"));
+				zos.putNextEntry(ze);
+				// read the file and write to ZipOutputStream
+				FileInputStream fis = new FileInputStream(f);
+				byte[] buffer = new byte[1024];
+				int len;
+				while ((len = fis.read(buffer)) > 0) {
+					zos.write(buffer, 0, len);
+				}
+				zos.closeEntry();
+				fis.close();
+			}
+		}
+	}
+
+	private boolean deleteDirectory(File directory) {
+		if (directory.exists()) {
+			File[] files = directory.listFiles();
+			if (null != files) {
+				for (int i = 0; i < files.length; i++) {
+					if (files[i].isDirectory()) {
+						deleteDirectory(files[i]);
+					} else {
+						files[i].delete();
+					}
+				}
+			}
+		}
+		return (directory.delete());
+	}
+}