63067e3ede5f62a4ab43895f06966a9ce2061e11
[sdc.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.vendorsoftwareproduct.types.candidateheat;
22
23 import java.util.Collection;
24 import java.util.HashSet;
25 import java.util.Set;
26 import lombok.Getter;
27 import lombok.NoArgsConstructor;
28 import org.apache.commons.collections4.CollectionUtils;
29
30 @Getter
31 @NoArgsConstructor
32 public class AnalyzedZipHeatFiles {
33
34     private final Set<String> nestedFiles = new HashSet<>();
35     private final Set<String> otherNonModuleFiles = new HashSet<>();
36     private final Set<String> moduleFiles = new HashSet<>();
37
38     public void addNestedFile(String fileName) {
39         nestedFiles.add(fileName);
40         moduleFiles.remove(fileName);
41     }
42
43     public void addNestedFiles(Collection<String> fileNames) {
44         nestedFiles.addAll(fileNames);
45         moduleFiles.removeAll(fileNames);
46     }
47
48     public void addOtherNonModuleFile(String fileName) {
49         otherNonModuleFiles.add(fileName);
50         moduleFiles.remove(fileName);
51     }
52
53     public void addOtherNonModuleFiles(Collection<String> fileNames) {
54         otherNonModuleFiles.addAll(fileNames);
55         moduleFiles.removeAll(fileNames);
56     }
57
58     public void addModuleFile(String fileName) {
59         moduleFiles.add(fileName);
60     }
61
62     public Collection<String> getFilesNotEligbleForModules() {
63         return CollectionUtils.union(this.getNestedFiles(), this.getOtherNonModuleFiles());
64     }
65
66 }