Updating Nokia driver

Change-Id: I950afe6acbdb359cd67a448024f006a45e8fc293
Signed-off-by: Denes Nemeth <denes.nemeth@nokia.com>
Issue-ID: VFC-728

1 files changed, 85 insertions, 0 deletions

diff --git a/nokiav2/driver/src/main/java/org/onap/vfc/nfvo/driver/vnfm/svnfm/nokia/packagetransformer/OnapVnfPackageBuilder.java b/nokiav2/driver/src/main/java/org/onap/vfc/nfvo/driver/vnfm/svnfm/nokia/packagetransformer/OnapVnfPackageBuilder.java
new file mode 100644
index 00000000..f769becb
--- /dev/null
+++ b/nokiav2/driver/src/main/java/org/onap/vfc/nfvo/driver/vnfm/svnfm/nokia/packagetransformer/OnapVnfPackageBuilder.java
@@ -0,0 +1,85 @@
+/*
+ * Copyright 2016-2017, Nokia Corporation
+ *
+ * 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.
+ */
+
+package org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.packagetransformer;
+
+import com.google.common.io.ByteStreams;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.InputStream;
+import java.nio.charset.StandardCharsets;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.util.SystemFunctions.systemFunctions;
+import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.CatalogManager.getFileInZip;
+import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.CatalogManager.getVnfdLocation;
+
+/**
+ * Transforms a CBAM package into an ONAP package
+ */
+
+public class OnapVnfPackageBuilder {
+
+    /**
+     * Entry point for the command line package transformer
+     *
+     * @param args not used (required due to signature)
+     */
+    public static void main(String[] args) throws Exception {
+        byte[] covert = new OnapVnfPackageBuilder().covert(systemFunctions().in());
+        systemFunctions().out().write(covert);
+    }
+
+    /**
+     * @param zip the original CBAM package
+     * @return the converted ONAP package
+     */
+    public byte[] covert(InputStream zip) throws Exception {
+        byte[] cbamVnfPackage = ByteStreams.toByteArray(zip);
+        String vnfdLocation = getVnfdLocation(new ByteArrayInputStream(cbamVnfPackage));
+        ByteArrayOutputStream vnfdContent = getFileInZip(new ByteArrayInputStream(cbamVnfPackage), vnfdLocation);
+        byte[] cbamVnfdContent = vnfdContent.toByteArray();
+        String onapVnfd = new OnapVnfdBuilder().toOnapVnfd(new String(cbamVnfdContent, StandardCharsets.UTF_8));
+        byte[] modifiedCbamPackage = new CbamVnfPackageBuilder().toModifiedCbamVnfPackage(cbamVnfPackage, vnfdLocation, new CbamVnfdBuilder().build(new String(cbamVnfdContent)));
+        return buildNewOnapPackage(modifiedCbamPackage, onapVnfd);
+    }
+
+    private byte[] buildNewOnapPackage(byte[] modifiedCbamPackage, String onapVnfd) throws Exception {
+        ByteArrayOutputStream result = new ByteArrayOutputStream();
+        ZipOutputStream out = new ZipOutputStream(result);
+        out.putNextEntry(new ZipEntry("Artifacts/Deployment/OTHER/cbam.package.zip"));
+        out.write(modifiedCbamPackage);
+        out.closeEntry();
+        out.putNextEntry(new ZipEntry("TOSCA-Metadata/TOSCA.meta"));
+        out.write(systemFunctions().loadFile("TOSCA.meta"));
+        out.closeEntry();
+        out.putNextEntry(new ZipEntry("MainServiceTemplate.yaml"));
+        out.write(onapVnfd.getBytes());
+        out.closeEntry();
+        out.putNextEntry(new ZipEntry("Definitions/MainServiceTemplate.yaml"));
+        out.write(onapVnfd.getBytes());
+        out.closeEntry();
+        out.putNextEntry(new ZipEntry("MainServiceTemplate.meta"));
+        out.write(systemFunctions().loadFile("MainServiceTemplate.meta"));
+        out.closeEntry();
+        out.close();
+        return result.toByteArray();
+    }
+
+
+}