1 files changed, 467 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java
new file mode 100644
index 0000000000..c86425fd43
--- /dev/null
+++ b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java
@@ -0,0 +1,467 @@
+/*-
+ * ============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.heat.datatypes.structure;
+
+
+import org.codehaus.jackson.annotate.JsonProperty;
+import org.openecomp.sdc.datatypes.error.ErrorMessage;
+import org.openecomp.sdc.heat.datatypes.manifest.FileData;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
+import java.util.TreeSet;
+
+/**
+ * The type Heat structure tree.
+ */
+public class HeatStructureTree implements Comparable<HeatStructureTree> {
+
+  private String fileName;
+  private FileData.Type type;
+  private Boolean isBase;
+  private HeatStructureTree env;
+  private List<ErrorMessage> errors;
+  private Set<HeatStructureTree> HEAT;
+  private Set<HeatStructureTree> volume;
+  private Set<HeatStructureTree> network;
+  private Set<HeatStructureTree> nested;
+  private Set<HeatStructureTree> other;
+  private Set<Artifact> artifacts;
+
+  /**
+   * Instantiates a new Heat structure tree.
+   */
+  public HeatStructureTree() {
+  }
+
+  ;
+
+  /**
+   * Instantiates a new Heat structure tree.
+   *
+   * @param fileName the file name
+   * @param isBase   the is base
+   */
+  public HeatStructureTree(String fileName, boolean isBase) {
+    setBase(isBase);
+    setFileName(fileName);
+  }
+
+  /**
+   * Gets heat structure tree by name.
+   *
+   * @param filesSet the files set
+   * @param filename the filename
+   * @return the heat structure tree by name
+   */
+  public static HeatStructureTree getHeatStructureTreeByName(Set<HeatStructureTree> filesSet,
+                                                             String filename) {
+    for (HeatStructureTree heatStructureTree : filesSet) {
+      if (heatStructureTree.getFileName().equals(filename)) {
+        return heatStructureTree;
+      }
+    }
+
+    return null;
+  }
+
+  /**
+   * Sets type.
+   *
+   * @param type the type
+   */
+  public void setType(FileData.Type type) {
+    this.type = type;
+  }
+
+  /**
+   * Gets base.
+   *
+   * @return the base
+   */
+  public Boolean getBase() {
+    return isBase;
+  }
+
+  /**
+   * Sets base.
+   *
+   * @param base the base
+   */
+  public void setBase(Boolean base) {
+    isBase = base;
+  }
+
+  /**
+   * Gets file name.
+   *
+   * @return the file name
+   */
+  public String getFileName() {
+    return fileName;
+  }
+
+  /**
+   * Sets file name.
+   *
+   * @param file the file
+   */
+  public void setFileName(String file) {
+    this.fileName = file;
+  }
+
+  /**
+   * Gets heat.
+   *
+   * @return the heat
+   */
+  @JsonProperty(value = "HEAT")
+  public Set<HeatStructureTree> getHEAT() {
+    return HEAT;
+  }
+
+  /**
+   * Sets heat.
+   *
+   * @param heat the heat
+   */
+  public void setHEAT(Set<HeatStructureTree> heat) {
+    this.HEAT = heat;
+  }
+
+  /**
+   * Gets nested.
+   *
+   * @return the nested
+   */
+  public Set<HeatStructureTree> getNested() {
+    return nested;
+  }
+
+  /**
+   * Sets nested.
+   *
+   * @param nested the nested
+   */
+  public void setNested(Set<HeatStructureTree> nested) {
+    this.nested = nested;
+  }
+
+  /**
+   * Gets artifacts.
+   *
+   * @return the artifacts
+   */
+  public Set<Artifact> getArtifacts() {
+    return artifacts;
+  }
+
+  /**
+   * Sets artifacts.
+   *
+   * @param artifacts the artifacts
+   */
+  public void setArtifacts(Set<Artifact> artifacts) {
+    this.artifacts = artifacts;
+  }
+
+  /**
+   * Add heat structure tree to nested heat list.
+   *
+   * @param heatStructureTree the heat structure tree
+   */
+  public void addHeatStructureTreeToNestedHeatList(HeatStructureTree heatStructureTree) {
+    if (this.nested == null) {
+      this.nested = new TreeSet<>();
+    }
+    if (!findItemInSetByName(this.nested, heatStructureTree)) {
+      this.nested.add(heatStructureTree);
+    }
+  }
+
+  /**
+   * Add artifact to artifact list.
+   *
+   * @param artifact the artifact
+   */
+  public void addArtifactToArtifactList(Artifact artifact) {
+    if (this.artifacts == null || this.artifacts.isEmpty()) {
+      this.artifacts = new TreeSet<>();
+    }
+    this.artifacts.add(artifact);
+  }
+
+  /**
+   * Gets env.
+   *
+   * @return the env
+   */
+  public HeatStructureTree getEnv() {
+    return env;
+  }
+
+  /**
+   * Sets env.
+   *
+   * @param env the env
+   */
+  public void setEnv(HeatStructureTree env) {
+    this.env = env;
+  }
+
+  /**
+   * Gets volume.
+   *
+   * @return the volume
+   */
+  public Set<HeatStructureTree> getVolume() {
+    return volume;
+  }
+
+  /**
+   * Sets volume.
+   *
+   * @param volume the volume
+   */
+  public void setVolume(Set<HeatStructureTree> volume) {
+    this.volume = volume;
+  }
+
+  /**
+   * Gets network.
+   *
+   * @return the network
+   */
+  public Set<HeatStructureTree> getNetwork() {
+    return network;
+  }
+
+  /**
+   * Sets network.
+   *
+   * @param network the network
+   */
+  public void setNetwork(Set<HeatStructureTree> network) {
+    this.network = network;
+  }
+
+  /**
+   * Add network to network list.
+   *
+   * @param heatStructureTree the heat structure tree
+   */
+  public void addNetworkToNetworkList(HeatStructureTree heatStructureTree) {
+    if (this.network == null) {
+      this.network = new TreeSet<>();
+    }
+    if (!findItemInSetByName(this.network, heatStructureTree)) {
+      this.network.add(heatStructureTree);
+    }
+  }
+
+  /**
+   * Add volume file to volume list.
+   *
+   * @param heatStructureTree the heat structure tree
+   */
+  public void addVolumeFileToVolumeList(HeatStructureTree heatStructureTree) {
+    if (this.volume == null) {
+      this.volume = new TreeSet<>();
+    }
+    if (!findItemInSetByName(this.volume, heatStructureTree)) {
+      this.volume.add(heatStructureTree);
+    }
+  }
+
+  /**
+   * Add heat to heat list.
+   *
+   * @param heat the heat
+   */
+  public void addHeatToHEATList(HeatStructureTree heat) {
+    if (this.HEAT == null) {
+      this.HEAT = new TreeSet<>();
+    }
+
+    this.HEAT.add(heat);
+  }
+
+  /**
+   * Add other to other list.
+   *
+   * @param other the other
+   */
+  public void addOtherToOtherList(HeatStructureTree other) {
+    if (this.other == null) {
+      this.other = new TreeSet<>();
+    }
+
+    this.other.add(other);
+  }
+
+  /**
+   * Find item in set by name boolean.
+   *
+   * @param searchSet the search set
+   * @param toFind    the to find
+   * @return the boolean
+   */
+  public boolean findItemInSetByName(Set<HeatStructureTree> searchSet, HeatStructureTree toFind) {
+    for (HeatStructureTree heatStructureTree : searchSet) {
+      if (heatStructureTree.getFileName().equals(toFind.getFileName())) {
+        return true;
+      }
+
+    }
+
+    return false;
+  }
+
+  /**
+   * Remove from volume or network.
+   *
+   * @param fileNameToRemove the file name to remove
+   * @param type             the type
+   */
+  public void removeFromVolumeOrNetwork(String fileNameToRemove, FileData.Type type) {
+    Set<HeatStructureTree> volumeOrNetworkSet =
+        type.equals(FileData.Type.HEAT_VOL) ? this.volume : this.network;
+    HeatStructureTree toRemove = getHeatStructureTreeByName(volumeOrNetworkSet, fileNameToRemove);
+
+    volumeOrNetworkSet.remove(toRemove);
+  }
+
+  @Override
+  public boolean equals(Object other) {
+    if (this == other) {
+      return true;
+    }
+    if (other == null || getClass() != other.getClass()) {
+      return false;
+    }
+
+    HeatStructureTree heatStructureTree = (HeatStructureTree) other;
+
+    if (fileName != null ? !fileName.equals(heatStructureTree.fileName)
+        : heatStructureTree.fileName != null) {
+      return false;
+    }
+    if (env != null ? !env.equals(heatStructureTree.env) : heatStructureTree.env != null) {
+      return false;
+    }
+    if (HEAT != null ? !HEAT.equals(heatStructureTree.HEAT) : heatStructureTree.HEAT != null) {
+      return false;
+    }
+    if (volume != null ? !volume.equals(heatStructureTree.volume)
+        : heatStructureTree.volume != null) {
+      return false;
+    }
+    if (network != null ? !network.equals(heatStructureTree.network)
+        : heatStructureTree.network != null) {
+      return false;
+    }
+    if (artifacts != null ? !artifacts.equals(heatStructureTree.artifacts)
+        : heatStructureTree.artifacts != null) {
+      return false;
+    }
+    if (nested != null ? !nested.equals(heatStructureTree.nested)
+        : heatStructureTree.nested != null) {
+      return false;
+    }
+    if (errors != null ? !errors.equals(heatStructureTree.errors)
+        : heatStructureTree.errors != null) {
+      return false;
+    }
+
+    return true;
+  }
+
+  @Override
+  public int hashCode() {
+    int result1 = fileName != null ? fileName.hashCode() : 0;
+    result1 = 31 * result1 + (env != null ? env.hashCode() : 0);
+    result1 = 31 * result1 + (HEAT != null ? HEAT.hashCode() : 0);
+    result1 = 31 * result1 + (volume != null ? volume.hashCode() : 0);
+    result1 = 31 * result1 + (network != null ? network.hashCode() : 0);
+    result1 = 31 * result1 + (artifacts != null ? artifacts.hashCode() : 0);
+    result1 = 31 * result1 + (nested != null ? nested.hashCode() : 0);
+    result1 = 31 * result1 + (errors != null ? errors.hashCode() : 0);
+
+
+    return result1;
+  }
+
+  /**
+   * Gets errors.
+   *
+   * @return the errors
+   */
+  public List<ErrorMessage> getErrors() {
+    return errors;
+  }
+
+  /**
+   * Sets errors.
+   *
+   * @param errors the errors
+   */
+  public void setErrors(List<ErrorMessage> errors) {
+    this.errors = errors;
+  }
+
+  /**
+   * Add error to errors list.
+   *
+   * @param error the error
+   */
+  public void addErrorToErrorsList(ErrorMessage error) {
+    if (this.errors == null || this.errors.isEmpty()) {
+      this.errors = new ArrayList<>();
+    }
+    if (!this.errors.contains(error)) {
+      this.errors.add(error);
+    }
+  }
+
+  /**
+   * Gets other.
+   *
+   * @return the other
+   */
+  public Set<HeatStructureTree> getOther() {
+    return other;
+  }
+
+  /**
+   * Sets other.
+   *
+   * @param other the other
+   */
+  public void setOther(Set<HeatStructureTree> other) {
+    this.other = other;
+  }
+
+  @Override
+  public int compareTo(HeatStructureTree heatStructureTree) {
+    return heatStructureTree.getFileName().compareTo(this.getFileName());
+  }
+}