1 files changed, 134 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/impl/heattotosca/HeatToToscaTranslatorImpl.java b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/impl/heattotosca/HeatToToscaTranslatorImpl.java
new file mode 100644
index 0000000000..0a6986816c
--- /dev/null
+++ b/openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/main/java/org/openecomp/sdc/translator/impl/heattotosca/HeatToToscaTranslatorImpl.java
@@ -0,0 +1,134 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * SDC
+ * ================================================================================
+ * 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.sdc.translator.impl.heattotosca;
+
+import org.apache.commons.collections4.MapUtils;
+import org.openecomp.core.translator.api.HeatToToscaTranslator;
+import org.openecomp.core.translator.datatypes.TranslatorOutput;
+import org.openecomp.core.utilities.file.FileUtils;
+import org.openecomp.core.utilities.json.JsonUtil;
+import org.openecomp.core.validation.api.ValidationManager;
+import org.openecomp.core.validation.errors.Messages;
+import org.openecomp.core.validation.factory.ValidationManagerFactory;
+import org.openecomp.core.validation.types.MessageContainerUtil;
+import org.openecomp.sdc.common.utils.AsdcCommon;
+import org.openecomp.sdc.datatypes.error.ErrorLevel;
+import org.openecomp.sdc.datatypes.error.ErrorMessage;
+import org.openecomp.sdc.heat.datatypes.manifest.FileData;
+import org.openecomp.sdc.heat.datatypes.manifest.ManifestContent;
+import org.openecomp.sdc.heat.datatypes.manifest.ManifestFile;
+import org.openecomp.sdc.translator.services.heattotosca.TranslationContext;
+import org.openecomp.sdc.translator.services.heattotosca.TranslationService;
+
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class HeatToToscaTranslatorImpl implements HeatToToscaTranslator {
+
+  private TranslationContext translationContext = new TranslationContext();
+  private ValidationManager validationManager =
+      ValidationManagerFactory.getInstance().createInterface();
+  private boolean isValid = false;
+
+
+  @Override
+  public void addManifest(String name, byte[] content) {
+    ManifestContent manifestData = JsonUtil.json2Object(new String(content), ManifestContent.class);
+    ManifestFile manifest = new ManifestFile();
+    manifest.setName(name);
+    manifest.setContent(manifestData);
+    translationContext.setManifest(manifest);
+    translationContext.addFile(name, content);
+    validationManager.addFile(AsdcCommon.MANIFEST_NAME, content);
+    addFilesFromManifestToTranslationContextManifestFilesMap(manifestData.getData());
+    isValid = false;
+  }
+
+  @Override
+  public void addFile(String name, byte[] content) {
+    translationContext.addFile(name, content);
+    validationManager.addFile(name, content);
+    isValid = false;
+  }
+
+  @Override
+  public void addFile(String name, InputStream content) {
+    addFile(name, FileUtils.toByteArray(content));
+  }
+
+
+  @Override
+  public Map<String, List<ErrorMessage>> validate() {
+
+    Map<String, List<ErrorMessage>> errors = new HashMap<>();
+    if (translationContext.getManifest() == null) {
+      ErrorMessage.ErrorMessageUtil.addMessage(AsdcCommon.MANIFEST_NAME, errors)
+          .add(new ErrorMessage(ErrorLevel.ERROR, Messages.MANIFEST_NOT_EXIST.getErrorMessage()));
+      return errors;
+    }
+
+    if (MapUtils.isEmpty(errors)) {
+      errors = validationManager.validate();
+    }
+    if (MapUtils.isEmpty(errors)) {
+      isValid = true;
+    }
+    return errors;
+  }
+
+  @Override
+  public TranslatorOutput translate() {
+    TranslationService translationService = new TranslationService();
+    TranslatorOutput translatorOutput = new TranslatorOutput();
+    if (!isValid) {
+      Map<String, List<ErrorMessage>> errors = validate();
+
+      if (MapUtils.isNotEmpty(MessageContainerUtil.getMessageByLevel(ErrorLevel.ERROR, errors))) {
+        translatorOutput.setErrorMessages(errors);
+        return translatorOutput;
+      }
+    }
+
+    translatorOutput = translationService.translateHeatFiles(translationContext);
+    return translatorOutput;
+  }
+
+  @Override
+  public void addExternalArtifacts(String name, byte[] content) {
+    translationContext.addExternalArtifacts(name, content);
+  }
+
+  @Override
+  public void addExternalArtifacts(String name, InputStream content) {
+    addExternalArtifacts(name, FileUtils.toByteArray(content));
+  }
+
+  private void addFilesFromManifestToTranslationContextManifestFilesMap(
+      List<FileData> fileDataListFromManifest) {
+    for (FileData fileFromManfiest : fileDataListFromManifest) {
+      translationContext.addManifestFile(fileFromManfiest.getFile(), fileFromManfiest.getType());
+    }
+  }
+
+
+}