path: root/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/structure/HeatStructureTree.java

/*-
 * ============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 java.util.ArrayList;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

import lombok.AccessLevel;
import lombok.Data;
import lombok.Getter;
import lombok.Setter;
import org.openecomp.sdc.datatypes.error.ErrorMessage;
import org.openecomp.sdc.heat.datatypes.manifest.FileData;

@Data
public class HeatStructureTree implements Comparable<HeatStructureTree> {

    private String fileName;
    private FileData.Type type;

    @Getter(AccessLevel.NONE)
    @Setter(AccessLevel.NONE)
    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;

    public HeatStructureTree() {
    }

    public HeatStructureTree(String fileName, boolean isBase) {
        this.isBase = isBase;
        this.fileName = fileName;
    }

    public Boolean getBase() {
        return isBase;
    }

    public void setBase(Boolean isBase) {
        this.isBase = isBase;
    }

    /**
     * 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;
    }

    /**
     * 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);
    }

    /**
     * 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 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;
    }

    @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;
        }

        return errors != null ? errors.equals(heatStructureTree.errors) : heatStructureTree.errors == null;
    }

    /**
     * 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);
        }
    }

    @Override
    public int compareTo(HeatStructureTree obj) {
        return obj.getFileName().compareTo(this.getFileName());
    }
}