package com.github.dkharrat.nexusdata.modelgen.metamodel;

import java.util.ArrayList;
import java.util.List;

public class Entity {
    String baseClass = "ManagedObject";
    String name;
    List<Attribute> attributes;
    List<Relationship> relationships;
    List<EnumProperty> enums;

    public Entity() {
        attributes = new ArrayList<>();
        relationships = new ArrayList<>();
        enums = new ArrayList<>();
    }

    public String getBaseClass() {
        return baseClass;
    }

    public String getName() {
        return name;
    }

    public List<Attribute> getAttributes() {
        return attributes;
    }

    public List<Relationship> getRelationships() {
        return relationships;
    }

    public List<EnumProperty> getEnums() {
        return enums;
    }
}

@@ -52,16 +52,68 @@
 model;%0A%0A
+import com.google.gson.annotations.SerializedName;%0A%0A
 import j
@@ -177,24 +177,51 @@
 Entity %7B%0A   
+ @SerializedName(%22extends%22)
  String base

support 'extends'