X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Faai%2Fbabel%2Fxml%2Fgenerator%2Fmodel%2FResource.java;h=d655ca62152eca6594cf93334b3ee423d469c381;hb=be30876e3a3872a7274c944995544836ff31913c;hp=1b64907a1f30103751ebcb8b62b185113a6e0bd7;hpb=047862bba53addd381fc7c715ac9e3dff76b740d;p=aai%2Fbabel.git

diff --git a/src/main/java/org/onap/aai/babel/xml/generator/model/Resource.java b/src/main/java/org/onap/aai/babel/xml/generator/model/Resource.java
index 1b64907..d655ca6 100644
--- a/src/main/java/org/onap/aai/babel/xml/generator/model/Resource.java
+++ b/src/main/java/org/onap/aai/babel/xml/generator/model/Resource.java
@@ -2,8 +2,8 @@
  * ============LICENSE_START=======================================================
  * org.onap.aai
  * ================================================================================
- * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
- * Copyright © 2017-2018 European Software Marketing Ltd.
+ * Copyright © 2017-2019 AT&T Intellectual Property. All rights reserved.
+ * Copyright © 2017-2019 European Software Marketing Ltd.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -18,10 +18,31 @@
  * limitations under the License.
  * ============LICENSE_END=========================================================
  */
+
 package org.onap.aai.babel.xml.generator.model;
 
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import org.onap.aai.babel.xml.generator.model.Widget.Type;
+
 public class Resource extends Model {
 
+    private Type type;
+    private boolean deleteFlag;
+    private boolean isResource = true;
+    private Map<String, Object> properties = Collections.emptyMap();
+
+    Widget vserver = null;
+    boolean addlintf = false;
+    boolean addvolume = false;
+    List<String> members;
+
+    public Resource(Type type, boolean deleteFlag) {
+        this.type = type;
+        this.deleteFlag = deleteFlag;
+    }
+
     @Override
     public int hashCode() {
         final String uuid = getModelNameVersionId();
@@ -36,20 +57,116 @@ public class Resource extends Model {
         return false;
     }
 
+    @Override
+    public boolean getDeleteFlag() {
+        return deleteFlag;
+    }
+
+    @Override
+    public String getWidgetInvariantId() {
+        return Widget.getWidget(getWidgetType()).getWidgetId();
+    }
+
+    @Override
+    public String getWidgetId() {
+        return Widget.getWidget(getWidgetType()).getId();
+    }
+
+    public void setProperties(Map<String, Object> properties) {
+        this.properties = properties;
+    }
+
+    @Override
+    public Map<String, Object> getProperties() {
+        return properties;
+    }
+
+    public void setIsResource(boolean isResource) {
+        this.isResource = isResource;
+    }
+
+    @Override
+    public boolean isResource() {
+        return isResource;
+    }
+
+    public void setMembers(List<String> members) {
+        this.members = members;
+    }
+
     @Override
     public boolean addResource(Resource resource) {
         return resources.add(resource);
     }
 
+    /**
+     * Adds a Widget.
+     *
+     * @param widget
+     *     the widget
+     * @return the boolean
+     */
     @Override
     public boolean addWidget(Widget widget) {
-        return widgets.add(widget);
+        if (type == Type.VFMODULE) {
+            if (widget.memberOf(members)) {
+                if (vserver == null && widget instanceof VServerWidget) {
+                    addVserverWidget(widget);
+                } else if (widget instanceof LIntfWidget) {
+                    return addLIntfWidget(widget);
+                } else if (widget instanceof VolumeWidget) {
+                    addVolumeWidget(widget);
+                    return true;
+                }
+                if (!(widget instanceof OamNetwork)) {
+                    return widgets.add(widget);
+                }
+            }
+            return false;
+        } else {
+            return widgets.add(widget);
+        }
+    }
+
+    public Type getWidgetType() {
+        return type;
     }
 
     @Override
-    public Widget.Type getWidgetType() {
-        org.onap.aai.babel.xml.generator.types.Model model =
-                this.getClass().getAnnotation(org.onap.aai.babel.xml.generator.types.Model.class);
-        return model.widget();
+    public String toString() {
+        return "Widget type " + getWidgetType() + ", isResource=" + isResource() + ", deleteFlag=" + deleteFlag;
+    }
+
+    private void addVolumeWidget(Widget widget) {
+        if (vserver != null) {
+            vserver.addWidget(widget);
+        } else {
+            addvolume = true;
+        }
+    }
+
+    /**
+     * @param widget
+     * @return
+     */
+    private boolean addLIntfWidget(Widget widget) {
+        if (vserver != null) {
+            vserver.addWidget(widget);
+            return true;
+        } else {
+            addlintf = true;
+            return false;
+        }
     }
+
+    private void addVserverWidget(Widget widget) {
+        vserver = widget;
+        if (addlintf) {
+            vserver.addWidget(new LIntfWidget());
+        }
+        if (addvolume) {
+            vserver.addWidget(new VolumeWidget());
+        }
+    }
+
 }