f5f7c500cfd359dc9386604b69b14dc40ddf201a
[aai/babel.git] / src / main / java / org / onap / aai / babel / xml / generator / model / Resource.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright (c) 2017-2019 AT&T Intellectual Property. All rights reserved.
6  * Copyright (c) 2017-2019 European Software Marketing Ltd.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *       http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.aai.babel.xml.generator.model;
23
24 import java.util.Collections;
25 import java.util.List;
26 import java.util.Map;
27 import org.onap.aai.babel.xml.generator.XmlArtifactGenerationException;
28 import org.onap.aai.babel.xml.generator.types.ModelType;
29
30 public class Resource extends Model {
31
32     private WidgetType type;
33     private boolean deleteFlag;
34     private ModelType modelType = ModelType.RESOURCE;
35     private Map<String, Object> properties = Collections.emptyMap();
36
37     Widget vserver = null;
38     boolean addlintf = false;
39     boolean addvolume = false;
40     List<String> members;
41
42     public Resource(WidgetType type, boolean deleteFlag) {
43         this.type = type;
44         this.deleteFlag = deleteFlag;
45     }
46
47     @Override
48     public int hashCode() {
49         final String uuid = getModelNameVersionId();
50         return uuid == null ? 0 : uuid.hashCode();
51     }
52
53     @Override
54     public boolean equals(Object obj) {
55         if (obj instanceof Resource) {
56             return getModelNameVersionId().equals(((Resource) obj).getModelNameVersionId());
57         }
58         return false;
59     }
60
61     @Override
62     public boolean getDeleteFlag() {
63         return deleteFlag;
64     }
65
66     public void setProperties(Map<String, Object> properties) {
67         this.properties = properties;
68     }
69
70     public Map<String, Object> getProperties() {
71         return properties;
72     }
73
74     public void setModelType(ModelType type) {
75         this.modelType = type;
76     }
77
78     public ModelType getModelType() {
79         return modelType;
80     }
81
82     public void setMembers(List<String> members) {
83         this.members = members;
84     }
85
86     /**
87      * Adds a Widget.
88      *
89      * @param widget
90      *            the widget
91      * @return the boolean
92      * @throws XmlArtifactGenerationException
93      */
94     @Override
95     public boolean addWidget(Widget widget) throws XmlArtifactGenerationException {
96         if (type == WidgetType.valueOf("VFMODULE")) {
97             if (widget.memberOf(members)) {
98                 if (vserver == null && widget.getWidgetType() == WidgetType.valueOf("VSERVER")) {
99                     addVserverWidget(widget);
100                 } else if (widget.getWidgetType() == WidgetType.valueOf("LINT")) {
101                     return addLIntfWidget(widget);
102                 } else if (widget.getWidgetType() == WidgetType.valueOf("VOLUME")) {
103                     addVolumeWidget(widget);
104                     return true;
105                 }
106                 if (widget.getWidgetType() != WidgetType.valueOf("OAM_NETWORK")) {
107                     return widgets.add(widget);
108                 }
109             }
110             return false;
111         } else {
112             return widgets.add(widget);
113         }
114     }
115
116     public WidgetType getWidgetType() {
117         return type;
118     }
119
120     public String getModelTypeName() {
121         return "resource";
122     }
123
124     @Override
125     public String toString() {
126         return "Resource [widget type=" + getWidgetType() + ", deleteFlag=" + deleteFlag + ", modelType=" + modelType
127                 + ", properties=" + properties + ", vserver=" + vserver + ", addlintf=" + addlintf + ", addvolume="
128                 + addvolume + ", members=" + members + "]";
129     }
130
131     private void addVolumeWidget(Widget widget) {
132         if (vserver != null) {
133             vserver.addWidget(widget);
134         } else {
135             addvolume = true;
136         }
137     }
138
139     /**
140      * @param widget
141      * @return
142      */
143     private boolean addLIntfWidget(Widget widget) {
144         if (vserver != null) {
145             vserver.addWidget(widget);
146             return true;
147         } else {
148             addlintf = true;
149             return false;
150         }
151     }
152
153     private void addVserverWidget(Widget widget) throws XmlArtifactGenerationException {
154         vserver = widget;
155         if (addlintf) {
156             vserver.addWidget(Widget.getWidget(WidgetType.valueOf("LINT")));
157         }
158         if (addvolume) {
159             vserver.addWidget(Widget.getWidget(WidgetType.valueOf("VOLUME")));
160         }
161     }
162
163 }