/**
    * Copyright 2004-2014 The Kuali Foundation
    *
    * Licensed under the Educational Community License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    * http://www.opensource.org/licenses/ecl2.php
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.kuali.kpme.tklm.leave.block;
  
  import java.math.BigDecimal;
  import java.sql.Timestamp;
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.Date;
  import java.util.List;
  
  import javax.persistence.Transient;
  
  import org.apache.commons.collections.CollectionUtils;
  import org.apache.commons.lang.StringUtils;
  import org.apache.commons.lang.builder.EqualsBuilder;
  import org.apache.commons.lang.builder.HashCodeBuilder;
  import org.joda.time.DateTime;
  import org.joda.time.LocalDate;
  import org.joda.time.LocalTime;
  import org.kuali.kpme.core.accrualcategory.AccrualCategory;
  import org.kuali.kpme.core.accrualcategory.rule.AccrualCategoryRule;
  import org.kuali.kpme.core.api.assignment.Assignable;
  import org.kuali.kpme.core.assignment.Assignment;
  import org.kuali.kpme.core.assignment.AssignmentDescriptionKey;
  import org.kuali.kpme.core.block.CalendarBlock;
  import org.kuali.kpme.core.calendar.Calendar;
  import org.kuali.kpme.core.calendar.entry.CalendarEntry;
  import org.kuali.kpme.core.earncode.EarnCode;
  import org.kuali.kpme.core.principal.PrincipalHRAttributes;
  import org.kuali.kpme.core.service.HrServiceLocator;
  import org.kuali.kpme.core.task.Task;
  import org.kuali.kpme.core.util.HrConstants;
  import org.kuali.kpme.core.util.HrContext;
  import org.kuali.kpme.core.util.TKUtils;
  import org.kuali.kpme.core.workarea.WorkArea;
  import org.kuali.kpme.tklm.api.leave.block.LeaveBlockContract;
  import org.kuali.kpme.tklm.common.TkConstants;
  import org.kuali.kpme.tklm.leave.service.LmServiceLocator;
  import org.kuali.kpme.tklm.leave.workflow.LeaveCalendarDocumentHeader;
  import org.kuali.kpme.tklm.leave.workflow.LeaveRequestDocument;
  import org.kuali.kpme.tklm.time.service.TkServiceLocator;
  import org.kuali.kpme.tklm.time.workflow.TimesheetDocumentHeader;
  import org.kuali.rice.kew.api.KewApiServiceLocator;
  import org.kuali.rice.kew.api.document.DocumentStatus;
  import org.kuali.rice.krad.util.ObjectUtils;
  import org.springframework.expression.spel.ast.Assign;
  
  public class LeaveBlock extends CalendarBlock implements Assignable, LeaveBlockContract {
  	
  	private static final long serialVersionUID = -8240826812581295376L;
  	public static final String CACHE_NAME = TkConstants.CacheNamespace.NAMESPACE_PREFIX + "LeaveBlock";
  	
  	protected String earnCode;
  	protected Long workArea;
  	protected Long jobNumber;
  	protected Long task;
  	protected BigDecimal leaveAmount = new BigDecimal("0.0");
  	
  	private Date leaveDate;
  	private String description;
  	private String scheduleTimeOffId;
  	private String accrualCategory;
  	private Boolean accrualGenerated;
  	private Long blockId;
  	private String requestStatus;
  	private String leaveBlockType;
  	private String documentStatus;
  	private String principalIdModified;
  	
  	private List<LeaveBlockHistory> leaveBlockHistories = new ArrayList<LeaveBlockHistory>();
      private String leaveRequestDocumentId;
          
  	protected String lmLeaveBlockId;
  	private String userPrincipalId;
      
  	@Transient
  	private Date beginDate;
      @Transient
  	private boolean submit;
  	@Transient
  	private String reason;
  
  	private String assignmentKey;
  	
  	@Transient
 	private String assignmentTitle;
 	@Transient
 	private String calendarId;
 	@Transient
 	private String planningDescription;
 
     @Transient
     private AccrualCategoryRule accrualCategoryRule;
 
     @Transient
     private AccrualCategory accrualCategoryObj;
 
     @Transient
     private PrincipalHRAttributes principalHRAttributes;
   
     @Transient
     private String affectPay;
 
 	private String transactionalDocId;
 	private LeaveCalendarDocumentHeader leaveCalendarDocumentHeader;
 
 	public String getAccrualCategoryRuleId() {
         AccrualCategoryRule aRule = getAccrualCategoryRule();
 		return ObjectUtils.isNull(aRule) ? null : aRule.getLmAccrualCategoryRuleId();
 	}
 	
 	public static class Builder {
 
 		// required parameters for the constructor
 		private final Date leaveDate;
 		private final String principalId;
 		private final String documentId;
 		private final String earnCode;
 		private final BigDecimal leaveAmount;
 
 		private String description = null;
 		private String principalIdModified = null;
 		private Timestamp timestamp = null;
 		private Boolean accrualGenerated = Boolean.FALSE;
 		private Long blockId = 0L;
 		private String scheduleTimeOffId;
 		private String accrualCategory;
 		private String requestStatus;
 		private Long workArea;
 		private Long jobNumber;
 		private Long task;
 		private String leaveBlockType;
 		private String userPrincipalId;
 		private LeaveCalendarDocumentHeader leaveCalendarDocumentHeader;
 		
 		public Builder(LocalDate leaveDate, String documentId,
 				String principalId, String earnCode, BigDecimal leaveAmount) {
 			this.leaveDate = leaveDate.toDate();
 			this.documentId = documentId;
 			this.principalId = principalId;
 			this.earnCode = earnCode;
 			this.leaveAmount = leaveAmount;
 		}
 
 		// validations could be done in the builder methods below
 
 		public Builder description(String val) {
 			this.description = val;
 			return this;
 		}
 
 
 		public Builder principalIdModified(String val) {
 			this.principalIdModified = val;
 			return this;
 		}
 
 		public Builder timestamp(Timestamp val) {
 			this.timestamp = val;
 			return this;
 		}
 
 		public Builder accrualGenerated(Boolean val) {
 			this.accrualGenerated = val;
 			return this;
 		}
 
 		public Builder blockId(Long val) {
 			this.blockId = val;
 			return this;
 		}
 
 		// TODO: need to hook up the objcets to get the real ids
 		public Builder scheduleTimeOffId(String val) {
 			this.scheduleTimeOffId = val;
 			return this;
 		}
 
 		public Builder accrualCategory(String val) {
 			this.accrualCategory = val;
 			return this;
 		}
 
 		public Builder workArea(Long val) {
 			this.workArea = val;
 			return this;
 		}
 
 		public Builder jobNumber(Long val) {
 			this.jobNumber = val;
 			return this;
 		}
 
 		public Builder task(Long val) {
 			this.task = val;
 			return this;
 		}
 
 		public Builder requestStatus(String val) {
 			this.requestStatus = val;
 			return this;
 		}
 		
 		public Builder leaveBlockType(String leaveBlockType) {
 			this.leaveBlockType = leaveBlockType;
 			return this;
 		}
 		
 		public Builder userPrincipalId(String userPrincipalId) {
 			this.userPrincipalId = userPrincipalId;
 			return this;
 		}
 
 		public Builder LeaveCalendarDocumentHeader(LeaveCalendarDocumentHeader leaveCalendarDocumentHeader) {
 			this.leaveCalendarDocumentHeader = leaveCalendarDocumentHeader;
 			return this;
 		}
 		
 		public LeaveBlock build() {
 			return new LeaveBlock(this);
 		}
 		
 	}
 
 	private LeaveBlock(Builder builder) {
 		leaveDate = builder.leaveDate;
 		description = builder.description;
 		principalId = builder.principalId;
 		earnCode = builder.earnCode;
 		leaveAmount = builder.leaveAmount;
 		documentId = builder.documentId;
 		principalIdModified = builder.principalIdModified;
 		timestamp = builder.timestamp;
 		accrualGenerated = builder.accrualGenerated;
 		blockId = builder.blockId;
 		scheduleTimeOffId = builder.scheduleTimeOffId;
 		accrualCategory = builder.accrualCategory;
 		requestStatus = builder.requestStatus;
 		workArea = builder.workArea;
 		jobNumber = builder.jobNumber;
 		task = builder.task;
 		leaveBlockType = builder.leaveBlockType;
 		userPrincipalId = builder.userPrincipalId;
 		leaveCalendarDocumentHeader = builder.leaveCalendarDocumentHeader;
 		// TODO: need to hook up leaveCodeObj, systemScheduledTimeOffObj,
 		// accrualCategoryObj, and ids for individual obj
 	}
 
 	public LeaveBlock() {
 
 	}
 
 	
 	public String getAccrualCategory() {
 		return accrualCategory;
 	}
 
 	public void setAccrualCategory(String accrualCategory) {
 		this.accrualCategory = accrualCategory;
 	}
 
 	public Boolean getAccrualGenerated() {
 		return accrualGenerated;
 	}
 
 	public void setAccrualGenerated(Boolean accrualGenerated) {
 		this.accrualGenerated = accrualGenerated;
 	}
 
 	public boolean isSubmit() {
 		return submit;
 	}
 
 	public void setSubmit(boolean submit) {
 		this.submit = submit;
 	}
 
 	public Long getBlockId() {
 		return blockId;
 	}
 
 	public void setBlockId(Long blockId) {
 		this.blockId = blockId;
 	}
 
 	public String getDescription() {
 		return description;
 	}
 
 	public void setDescription(String description) {
 		this.description = description;
 	}
 
 	public Date getLeaveDate() {
 		return leaveDate;
 	}
 
 	public void setLeaveDate(Date leaveDate) {
 		this.leaveDate = leaveDate;
 	}
 	
 	public LocalDate getLeaveLocalDate() {
 		return leaveDate != null ? LocalDate.fromDateFields(leaveDate) : null;
 	}
 	
 	public void setLeaveLocalDate(LocalDate leaveLocalDate) {
 		this.leaveDate = leaveLocalDate != null ? leaveLocalDate.toDate() : null;
 	}
 
 	public String getScheduleTimeOffId() {
 		return scheduleTimeOffId;
 	}
 
 	public void setScheduleTimeOffId(String scheduleTimeOffId) {
 		this.scheduleTimeOffId = scheduleTimeOffId;
 	}
 
 	public String getRequestStatus() {
 		return requestStatus;
 	}
 
 	public void setRequestStatus(String requestStatus) {
 		this.requestStatus = requestStatus;
 	}
 
     public String getRequestStatusString() {
         String status = HrConstants.REQUEST_STATUS_STRINGS.get(getRequestStatus());
         return status == null ? "usage" : status;
     }
 
 	public List<LeaveBlockHistory> getLeaveBlockHistories() {
 		return leaveBlockHistories;
 	}
 
 	public void setLeaveBlockHistories(
 			List<LeaveBlockHistory> leaveBlockHistories) {
 		this.leaveBlockHistories = leaveBlockHistories;
 	}
 
 	public String getReason() {
 		return reason;
 	}
 
 	public void setReason(String reason) {
 		this.reason = reason;
 	}
 
     public List<Assignment> getAssignments() {
         AssignmentDescriptionKey key = AssignmentDescriptionKey.get(getAssignmentKey());
         return Collections.singletonList(
                 HrServiceLocator.getAssignmentService().getAssignment(getPrincipalId(), key, getLeaveLocalDate()));
     }
 	public String getAssignmentTitle() {
 		StringBuilder b = new StringBuilder();
 
 		if (this.workArea != null) {
 			WorkArea wa = HrServiceLocator.getWorkAreaService().getWorkAreaWithoutRoles(
 					this.workArea, LocalDate.now());
 			if (wa != null) {
 				b.append(wa.getDescription());
 			}
 			Task task = HrServiceLocator.getTaskService().getTask(
 					this.getTask(), this.getLeaveLocalDate());
 			if (task != null) {
 				// do not display task description if the task is the default
 				// one
 				// default task is created in getTask() of TaskService
 				if (!task.getDescription()
 						.equals(HrConstants.TASK_DEFAULT_DESP)) {
 					b.append("-" + task.getDescription());
 				}
 			}
 		}
 		return b.toString();
 	}
 
 	public void setAssignmentTitle(String assignmentTitle) {
 		this.assignmentTitle = assignmentTitle;
 	}
 
 	public String getCalendarId() {
         if (StringUtils.isEmpty(calendarId)) {
             PrincipalHRAttributes principalHRAttributes = getPrincipalHRAttributes();
             Calendar pcal= null;
             if(principalHRAttributes != null) {
                 //pcal = principalHRAttributes.getCalendar() != null ? principalHRAttributes.getCalendar() : principalHRAttributes.getLeaveCalObj() ;
                 pcal = principalHRAttributes.getLeaveCalObj() != null ? principalHRAttributes.getLeaveCalObj() : principalHRAttributes.getCalendar();
                 if(pcal!= null) {
                     CalendarEntry calEntries = HrServiceLocator.getCalendarEntryService().getCurrentCalendarEntryByCalendarId(pcal.getHrCalendarId(), getLeaveLocalDate().toDateTimeAtStartOfDay());
                     if(calEntries != null) {
                         this.calendarId = calEntries.getHrCalendarEntryId();
                     }
                 }
             }
         }
 		return calendarId;
 	}
 
 	public void setCalendarId(String calendarId) {
 		this.calendarId = calendarId;
 	}
 	
 	public String getEarnCodeDescription() {
 		String earnCodeDescription = "";
 		
 		EarnCode earnCodeObj = HrServiceLocator.getEarnCodeService().getEarnCode(earnCode, getLeaveLocalDate());
 		if (earnCodeObj != null) {
 			earnCodeDescription = earnCodeObj.getDescription();
 		}
 		
 		return earnCodeDescription;
 	}
 
 	public String getLeaveBlockType() {
 		return leaveBlockType;
 	}
 
 	public void setLeaveBlockType(String leaveBlockType) {
 		this.leaveBlockType = leaveBlockType;
 	}
 
 	public LeaveCalendarDocumentHeader getLeaveCalendarDocumentHeader() {
         if (leaveCalendarDocumentHeader == null && this.getDocumentId() != null) {
             setLeaveCalendarDocumentHeader(LmServiceLocator.getLeaveCalendarDocumentHeaderService().getDocumentHeader(this.getDocumentId()));
         }
         return leaveCalendarDocumentHeader;
     }
 
     public void setLeaveCalendarDocumentHeader(
     		LeaveCalendarDocumentHeader leaveCalendarDocumentHeader) {
         this.leaveCalendarDocumentHeader = leaveCalendarDocumentHeader;
     }
 
 
 	
     public boolean isEditable() {
         return LmServiceLocator.getLMPermissionService().canEditLeaveBlock(HrContext.getPrincipalId(), this);
     }
 
     public boolean isDeletable() {
         return LmServiceLocator.getLMPermissionService().canDeleteLeaveBlock(HrContext.getPrincipalId(), this);
     }
     
     public String getAssignmentKey() {
         if (assignmentKey == null) {
             this.setAssignmentKey(TKUtils.formatAssignmentKey(jobNumber, workArea, task));
         }
         return assignmentKey;
     }
     public void setAssignmentKey(String assignmentDescription) {
         this.assignmentKey = assignmentDescription;
     }
 
 	public String getDocumentStatus() {
 		return documentStatus;
 	}
 
 	public void setDocumentStatus(String documentStatus) {
 		this.documentStatus = documentStatus;
 	}
 
     public String getLeaveRequestDocumentId() {
         return leaveRequestDocumentId;
     }
 
     public void setLeaveRequestDocumentId(String leaveRequestDocumentId) {
         this.leaveRequestDocumentId = leaveRequestDocumentId;
     }
 
     @Override
     public int hashCode() {
         return new HashCodeBuilder()
                 .append(this.principalId)
                 .append(this.jobNumber)
                 .append(this.workArea)
                 .append(this.task)
                 .append(this.earnCode)
                 .append(this.leaveDate)
                 .append(this.leaveAmount)
                 .append(this.accrualCategory)
                 .append(this.earnCode)
                 .append(this.description)
                 .append(this.leaveBlockType)
                 .toHashCode();
     }
 
     @Override
     public boolean equals(Object obj) {
         if (obj == null) {
             return false;
         }
         if (obj == this) {
             return true;
         }
         if (obj.getClass() != getClass()) {
             return false;
         }
         LeaveBlock leaveBlock = (LeaveBlock) obj;
         return new EqualsBuilder()
                 .append(principalId, leaveBlock.principalId)
                 .append(jobNumber, leaveBlock.jobNumber)
                 .append(workArea, leaveBlock.workArea)
                 .append(task, leaveBlock.task)
                 .append(earnCode, leaveBlock.earnCode)
                 .append(leaveDate, leaveBlock.leaveDate)
                 .append(leaveAmount, leaveBlock.leaveAmount)
                 .append(accrualCategory, leaveBlock.accrualCategory)
                 .append(earnCode, leaveBlock.earnCode)
                 .append(description, leaveBlock.description)
                 .append(leaveBlockType, leaveBlock.leaveBlockType)
                 .isEquals();
     }
     
     public String getPlanningDescription() {
     	if(this.getRequestStatus().equals(HrConstants.REQUEST_STATUS.DEFERRED)) {
     		List<LeaveRequestDocument> lrdList = LmServiceLocator.getLeaveRequestDocumentService().getLeaveRequestDocumentsByLeaveBlockId(this.getLmLeaveBlockId());
     		if(CollectionUtils.isNotEmpty(lrdList)) {
     			for(LeaveRequestDocument lrd : lrdList) {    				
     				DocumentStatus status = KewApiServiceLocator.getWorkflowDocumentService().getDocumentStatus(lrd.getDocumentNumber());
     				if(status != null && DocumentStatus.CANCELED.getCode().equals(status.getCode())) {
 	    				String requestDescription = "";
 						if(StringUtils.isNotEmpty(this.getDescription())) {
 							requestDescription = this.getDescription() + " <br/>";
 						}
 						String actionDateString = TKUtils.formatDate(lrd.getDocumentHeader().getWorkflowDocument().getDateFinalized().toLocalDate());
 						requestDescription += "Approval deferred on " + actionDateString + ". Reason: " + lrd.getDescription();
 			    		this.setPlanningDescription(requestDescription);
 						return planningDescription;
     				}
     			}
     		}
     	}
     	this.setPlanningDescription(this.getDescription());
     	return planningDescription;
     }
 
 	public void setPlanningDescription(String planningDescription) {
 		this.planningDescription = planningDescription;
 	}
 
 	public void setTransactionDocId(String documentHeaderId) {
 		transactionalDocId = documentHeaderId;		
 	}
 	
 	public String getTransactionalDocId() {
 		return transactionalDocId;
 	}
 
 	public DateTime getBeginDateTime() {
 		return beginTimestamp != null ? new DateTime(beginTimestamp) : null;
 	}
 	
 	public void setBeginDateTime(DateTime beginDateTime) {
 		beginTimestamp = beginDateTime != null ? new Timestamp(beginDateTime.getMillis()) : null;
 	}
 	
 	
 	public Date getBeginDate() {
 	    	Date beginDate = null;	    	
 	    	if (beginTimestamp != null) {
 	    		beginDate = new Date(beginTimestamp.getTime());
 	    	}
 	        return beginDate;
 	}
 
 	public Timestamp getBeginTimestampVal() {
 	        return new Timestamp(beginTimestamp.getTime());
 	}
 
 	public Timestamp getEndTimestampVal() {
 	        return new Timestamp(endTimestamp.getTime());
 	}
 
 	public void setBeginDate(Date beginDate) {
 	    	DateTime dateTime = new DateTime(beginTimestamp);
 	    	LocalDate localDate = new LocalDate(beginDate);
 	    	LocalTime localTime = new LocalTime(beginTimestamp);
 	    	beginTimestamp = new Timestamp(localDate.toDateTime(localTime, dateTime.getZone()).getMillis());
 	}
 	
 	public DateTime getEndDateTime() {
 		return endTimestamp != null ? new DateTime(endTimestamp) : null;
 	}
 	
 	public void setEndDateTime(DateTime endDateTime) {
 		endTimestamp = endDateTime != null ? new Timestamp(endDateTime.getMillis()) : null;
 	}
 
 	public String getLmLeaveBlockId() {
 		return lmLeaveBlockId;
 	}
 
 	public void setLmLeaveBlockId(String lmLeaveBlockId) {
 		this.lmLeaveBlockId = lmLeaveBlockId;
 	}
 
 	public String getPrincipalIdModified() {
 		return principalIdModified;
 	}
 
 	public void setPrincipalIdModified(String principalIdModified) {
 		this.principalIdModified = principalIdModified;
 	}
 
 	public BigDecimal getLeaveAmount() {
 		return leaveAmount;
 	}
 
 	public void setLeaveAmount(BigDecimal leaveAmount) {
 		this.leaveAmount = leaveAmount;
 	}
 
 	public Long getWorkArea() {
 		return workArea;
 	}
 
 	public void setWorkArea(Long workArea) {
 		this.workArea = workArea;
 	}
 
 	public Long getJobNumber() {
 		return jobNumber;
 	}
 
 	public void setJobNumber(Long jobNumber) {
 		this.jobNumber = jobNumber;
 	}
 
 	public Long getTask() {
 		return task;
 	}
 
 	public void setTask(Long task) {
 		this.task = task;
 	}
 
 	public String getEarnCode() {
 		return earnCode;
 	}
 
 	public void setEarnCode(String earnCode) {
 		this.earnCode = earnCode;
 	}
 
     public AccrualCategoryRule getAccrualCategoryRule() {
         if (accrualCategoryRule == null) {
             AccrualCategory category = getAccrualCategoryObj();
             PrincipalHRAttributes pha = getPrincipalHRAttributes();
             accrualCategoryRule = HrServiceLocator.getAccrualCategoryRuleService().getAccrualCategoryRuleForDate(category, getLeaveLocalDate(), pha.getServiceLocalDate());
         }
         return accrualCategoryRule;
     }
 
     public AccrualCategory getAccrualCategoryObj() {
         if (accrualCategoryObj == null) {
             accrualCategoryObj = HrServiceLocator.getAccrualCategoryService().getAccrualCategory(accrualCategory, getLeaveLocalDate());
         }
         return accrualCategoryObj;
     }
 
     public PrincipalHRAttributes getPrincipalHRAttributes() {
         if (principalHRAttributes == null) {
             principalHRAttributes = HrServiceLocator.getPrincipalHRAttributeService().getPrincipalCalendar(principalId, getLeaveLocalDate());
         }
         return principalHRAttributes;
     }
 
     public void setPrincipalHRAttributes(PrincipalHRAttributes principalHRAttributes) {
         this.principalHRAttributes = principalHRAttributes;
     }
 
     public String getUserPrincipalId() {
 	    return userPrincipalId;
 	}
 
 	public void setUserPrincipalId(String userPrincipalId) {
 	   	this.userPrincipalId = userPrincipalId;
 	}
 
 	public String getAffectPay() {
 		return affectPay;
 	}
 
 	public void setAffectPay(String affectPay) {
 		this.affectPay = affectPay;
 	}
 	
 }