src/main/java/org/onap/sdc/toscaparser/api/parameters/Annotation.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

package org.onap.sdc.toscaparser.api.parameters;

import java.util.ArrayList;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;

import org.onap.sdc.toscaparser.api.Property;
import org.onap.sdc.toscaparser.api.elements.enums.ToscaElementNames;

public class Annotation{
	
	private final static String HEAT = "HEAT";
	
	private String name;
	private String type;
	private ArrayList<Property> properties;
	
	public Annotation(){}
	@SuppressWarnings("unchecked")
	public Annotation(Map.Entry<String,Object> annotationEntry){
		if(annotationEntry != null){
			name = annotationEntry.getKey();
			Map<String, Object> annValue = (Map<String, Object>) annotationEntry.getValue();
			type = (String) annValue.get(ToscaElementNames.TYPE.getName());
			properties = fetchProperties((Map<String, Object>) annValue.get(ToscaElementNames.PROPERTIES.getName()));
		}
	}

	public String getName() {
		return name;
	}
	
	public void setName(String name) {
		this.name = name;
	}
	
	public String getType() {
		return type;
	}
	
	public void setType(String type) {
		this.type = type;
	}
	
	public ArrayList<Property> getProperties() {
		return properties;
	}
	
	public void setProperties(ArrayList<Property> properties) {
		this.properties = properties;
	}
	
	private ArrayList<Property> fetchProperties(Map<String, Object> properties) {
		if(properties != null){
			return (ArrayList<Property>) properties.entrySet().stream()
					.map(Property::new)
					.collect(Collectors.toList());
		}
		return null;
	}
	
	public boolean isHeatSourceType(){
		if(properties == null){
			return false;
		}
		Optional<Property> sourceType = properties.stream()
				.filter(p -> p.getName().equals(ToscaElementNames.SOURCE_TYPE.getName()))
				.findFirst();
		if(!sourceType.isPresent()){
			return false;
		}
		return sourceType.get().getValue() != null && ((String)sourceType.get().getValue()).equals(HEAT);
	}
	
}