From 4371989e7a50d015b6eefa074d4988033223551b Mon Sep 17 00:00:00 2001
From: "Desai, Dhrumin (dd303q)" <dd303q@att.com>
Date: Fri, 27 Aug 2021 17:53:06 -0400
Subject: integrate helm-chart-generator to runtime api

Issue-ID: DCAEGEN2-2694
Issue-ID: DCAEGEN2-2805
Change-Id: I1b7d1bd7d2254280e0faa561e49223357c615090
Signed-off-by: Dhrumin Desai <dd303q@att.com>
---
 .../core/helm/HelmChartGeneratorClientImpl.java    | 112 +++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 mod/runtimeapi/runtime-core/src/main/java/org/onap/dcae/runtime/core/helm/HelmChartGeneratorClientImpl.java

(limited to 'mod/runtimeapi/runtime-core/src/main/java/org/onap/dcae/runtime/core/helm/HelmChartGeneratorClientImpl.java')

diff --git a/mod/runtimeapi/runtime-core/src/main/java/org/onap/dcae/runtime/core/helm/HelmChartGeneratorClientImpl.java b/mod/runtimeapi/runtime-core/src/main/java/org/onap/dcae/runtime/core/helm/HelmChartGeneratorClientImpl.java
new file mode 100644
index 0000000..49ca3de
--- /dev/null
+++ b/mod/runtimeapi/runtime-core/src/main/java/org/onap/dcae/runtime/core/helm/HelmChartGeneratorClientImpl.java
@@ -0,0 +1,112 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * Copyright (C) 2021 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.dcae.runtime.core.helm;
+
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.io.FileUtils;
+import org.onap.dcaegen2.platform.helmchartgenerator.chartbuilder.ChartBuilder;
+import org.onap.dcaegen2.platform.helmchartgenerator.distribution.ChartDistributor;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.ComponentScan;
+import org.springframework.stereotype.Component;
+
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.nio.file.Files;
+
+/**
+ * Implementation class for Helm Chart Generator Client
+ */
+@Slf4j
+@Component
+@ComponentScan(basePackages = "org.onap.dcaegen2.platform.helmchartgenerator")
+public class HelmChartGeneratorClientImpl implements HelmChartGeneratorClient {
+
+    @Autowired
+    private final ChartBuilder chartBuilder;
+
+    @Autowired
+    private final ChartDistributor distributor;
+
+    @Value("${helm.base.chart.template.location}")
+    private String templateLocation;
+
+    public HelmChartGeneratorClientImpl(ChartBuilder chartBuilder, ChartDistributor distributor) {
+        this.chartBuilder = chartBuilder;
+        this.distributor = distributor;
+    }
+
+    /**
+     * Generate Helm Chart for a component spec
+     * @param componentSpec component spec as String
+     * @return packaged helm chart
+     */
+    @Override
+    public File generateHelmChart(String componentSpec){
+        try {
+            return chartBuilder.build(createTempSpecFile(componentSpec),templateLocation,
+                    createTempChartOutputLocation(), "");
+        } catch (Exception e) {
+            log.error(e.getMessage(), e);
+            throw new RuntimeException("Error while generating the helm chart.");
+        }
+    }
+
+    private String createTempChartOutputLocation() {
+        try {
+            return Files.createTempDirectory("chart").toAbsolutePath().toString();
+        } catch (IOException e) {
+            log.error(e.getMessage(), e);
+            throw new RuntimeException("Error creating a temporary chart dir.");
+        }
+    }
+
+    private String createTempSpecFile(String componentSpec) throws IOException{
+            File tmpFile = File.createTempFile("spec",".json");
+        try (FileWriter writer = new FileWriter(tmpFile)) {
+            writer.write(componentSpec);
+        }
+        return tmpFile.getAbsolutePath();
+    }
+
+    /**
+     * Distributes helm chart to Chart Museum
+     * @param helmChart packaged chart location
+     */
+    @Override
+    public void distribute(File helmChart) {
+        try {
+            distributor.distribute(helmChart);
+        }catch (Exception e){
+            throw e;
+        } finally {
+            removeChartLocally(helmChart);
+        }
+    }
+
+    private void removeChartLocally(File helmChart) {
+        try {
+            FileUtils.forceDelete(helmChart);
+        } catch (IOException e) {
+            log.warn("Could not delete a temporary helm chart " + helmChart.getName());
+        }
+    }
+}
-- 
cgit 1.2.3-korg