package org.kuali.student.r2.core.class1.atp.model;

 import org.kuali.student.common.entity.KSEntityConstants;

 import org.kuali.student.r2.common.dto.AttributeInfo;

 import org.kuali.student.r2.common.entity.MetaEntity;

 import org.kuali.student.r2.common.infc.Attribute;

 import org.kuali.student.r2.common.util.RichTextHelper;

 import org.kuali.student.r2.core.atp.dto.AtpInfo;

 import org.kuali.student.r2.core.atp.infc.Atp;

 import javax.persistence.CascadeType;

 import javax.persistence.Column;

 import javax.persistence.Entity;

 import javax.persistence.OneToMany;

 import javax.persistence.Table;

 import javax.persistence.Temporal;

 import javax.persistence.TemporalType;

 import java.util.ArrayList;

 import java.util.Date;

 import java.util.List;

 @Entity

 @Table(name = "KSEN_ATP")

 public class AtpEntity extends MetaEntity {

     @Column(name = "NAME")

     private String name;

     @Column(name = "ADMIN_ORG_ID")

     private String adminOrgId;

     @Column(name = "ATP_CD")

     private String atpCode;

     @Temporal(TemporalType.TIMESTAMP)

     @Column(name = "START_DT", nullable = false)

     private Date startDate;

     @Temporal(TemporalType.TIMESTAMP)

     @Column(name = "END_DT", nullable = false)

     private Date endDate;

     @Column(name = "DESCR_FORMATTED", length = KSEntityConstants.EXTRA_LONG_TEXT_LENGTH)

     private String formatted;

     @Column(name = "DESCR_PLAIN", length = KSEntityConstants.EXTRA_LONG_TEXT_LENGTH, nullable = false)

     private String plain;

     @Column(name = "ATP_TYPE", nullable = false)

     private String atpType;

     @Column(name = "ATP_STATE", nullable = false)

     private String atpState;

     @OneToMany(cascade = CascadeType.ALL, mappedBy = "owner")

     private List<AtpAttributeEntity> attributes = new ArrayList<AtpAttributeEntity>();

     public AtpEntity() {

     }

     public AtpEntity(Atp atp) {

         super(atp);

         this.setId(atp.getId());

         this.setAtpType(atp.getTypeKey());

         this.fromDTO(atp);

     }

     public void fromDTO(Atp atp) {

         this.setAtpCode(atp.getCode());

         this.setName(atp.getName());

         if (atp.getDescr() != null) {

             this.setDescrFormatted(atp.getDescr().getFormatted());

             this.setDescrPlain(atp.getDescr().getPlain());

         } else {

             this.setDescrFormatted(null);

             this.setDescrPlain(null);

         }

         this.setAdminOrgId(atp.getAdminOrgId());

         this.setAtpState(atp.getStateKey());

         this.setStartDate(atp.getStartDate());

         this.setEndDate(atp.getEndDate());

         this.setAttributes(new ArrayList<AtpAttributeEntity>());

         for (Attribute att : atp.getAttributes()) {

             this.getAttributes().add(new AtpAttributeEntity(att, this));

         }

     }

     public String getName() {

         return name;

     }

     public void setName(String name) {

         this.name = name;

     }

     public Date getStartDate() {

         return startDate;

     }

     public void setStartDate(Date startDate) {

         this.startDate = startDate;

     }

     public Date getEndDate() {

         return endDate;

     }

     public void setEndDate(Date endDate) {

         this.endDate = endDate;

     }

     public String getAtpType() {

         return atpType;

     }

     public void setAtpType(String atpType) {

         this.atpType = atpType;

     }

     public String getAtpState() {

         return atpState;

     }

     public void setAtpState(String atpState) {

         this.atpState = atpState;

     }

     public void setAttributes(List<AtpAttributeEntity> attributes) {

         this.attributes = attributes;

     }

     public List<AtpAttributeEntity> getAttributes() {

         return attributes;

     }

     public String getAdminOrgId() {

         return adminOrgId;

     }

     public void setAdminOrgId(String adminOrgId) {

         this.adminOrgId = adminOrgId;

     }

     public AtpInfo toDto() {

         AtpInfo atp = new AtpInfo();

         atp.setId(getId());

         atp.setCode(atpCode);

         atp.setName(name);

         atp.setStartDate(startDate);

         atp.setEndDate(endDate);

         atp.setAdminOrgId(getAdminOrgId());

         atp.setTypeKey(atpType);

         atp.setStateKey(atpState);

         atp.setMeta(super.toDTO());

         atp.setDescr(new RichTextHelper().toRichTextInfo(getDescrPlain(), getDescrFormatted()));

         if (getAttributes() != null) {

             for (AtpAttributeEntity att : getAttributes()) {

                 AttributeInfo attInfo = att.toDto();

                 atp.getAttributes().add(attInfo);

             }

         }

         return atp;

     }

     public String getDescrFormatted() {

         return formatted;

     }

     public void setDescrFormatted(String formatted) {

         this.formatted = formatted;

     }

     public String getDescrPlain() {

         return plain;

     }

     public void setDescrPlain(String plain) {

         this.plain = plain;

     }

     public String getAtpCode() {

         return atpCode;

     }

     public void setAtpCode(String atpCode) {

         this.atpCode = atpCode;

     }

 }