/**
    * Copyright 2004-2015 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.batch;
  
  import org.apache.log4j.Logger;
  import org.joda.time.DateTime;
  import org.joda.time.LocalDate;
  import org.kuali.kpme.core.api.assignment.Assignment;
  import org.kuali.kpme.core.api.assignment.service.AssignmentService;
  import org.kuali.kpme.core.api.leaveplan.LeavePlanContract;
  import org.kuali.kpme.core.api.leaveplan.LeavePlanService;
  import org.kuali.kpme.core.api.principal.PrincipalHRAttributes;
  import org.kuali.kpme.core.api.principal.service.PrincipalHRAttributesService;
  import org.kuali.kpme.core.batch.BatchJob;
  import org.kuali.kpme.tklm.api.leave.accrual.AccrualService;
  import org.kuali.kpme.tklm.leave.calendar.service.LeaveCalendarServiceImpl;
  import org.quartz.JobExecutionContext;
  import org.quartz.JobExecutionException;
  
  import java.util.List;
  
  public class AccrualJob extends BatchJob {
  
  	private static final Logger LOG = Logger.getLogger(LeaveCalendarServiceImpl.class);
  
  	private static AccrualService ACCRUAL_SERVICE;
  	private static AssignmentService ASSIGNMENT_SERVICE;
  	private static LeavePlanService LEAVE_PLAN_SERVICE;
  	private static PrincipalHRAttributesService PRINCIPAL_HR_ATTRIBUTES_SERVICE;
  	
  	@Override
  	public void execute(JobExecutionContext context) throws JobExecutionException {
          String batchUserPrincipalId = getBatchUserPrincipalId();
          
          if (batchUserPrincipalId != null) {
      		LocalDate asOfDate = LocalDate.now();
  			List<Assignment> assignments = getAssignmentService().getActiveAssignments(asOfDate);
  			
  			for (Assignment assignment : assignments) {
  				if (assignment.getJob().isEligibleForLeave()) {
  					String principalId = assignment.getPrincipalId();
  					
  					PrincipalHRAttributes principalHRAttributes = getPrincipalHRAttributesService().getPrincipalCalendar(principalId, asOfDate);
  					if (principalHRAttributes != null) {
  						LeavePlanContract leavePlan = getLeavePlanService().getLeavePlan(principalHRAttributes.getLeavePlan(), principalHRAttributes.getEffectiveLocalDate());
  						if (leavePlan != null) {
  							DateTime endDate = asOfDate.toDateTimeAtStartOfDay().plusMonths(Integer.parseInt(leavePlan.getPlanningMonths()));
  							getAccrualService().runAccrual(principalId, asOfDate.toDateTimeAtStartOfDay(), endDate, true, batchUserPrincipalId);
  						}
  					}
  				}
  			}
          } else {
          	String principalName = getBatchUserPrincipalName();
          	LOG.error("Could not run batch jobs due to missing batch user " + principalName);
          }
  	}
  
  	public static AccrualService getAccrualService() {
  		return ACCRUAL_SERVICE;
  	}
  
  	public static void setAccrualService(AccrualService accrualService) {
  		ACCRUAL_SERVICE = accrualService;
  	}
  
  	public static AssignmentService getAssignmentService() {
  		return ASSIGNMENT_SERVICE;
  	}
  
  	public static void setAssignmentService(AssignmentService assignmentService) {
  		ASSIGNMENT_SERVICE = assignmentService;
  	}
  
  	public static LeavePlanService getLeavePlanService() {
  		return LEAVE_PLAN_SERVICE;
  	}
  
  	public static void setLeavePlanService(LeavePlanService leavePlanService) {
  		LEAVE_PLAN_SERVICE = leavePlanService;
  	}
  
  	public static PrincipalHRAttributesService getPrincipalHRAttributesService() {
  		return PRINCIPAL_HR_ATTRIBUTES_SERVICE;
  	}
  
 	public static void setPrincipalHRAttributesService(PrincipalHRAttributesService principalHRAttributesService) {
 		PRINCIPAL_HR_ATTRIBUTES_SERVICE = principalHRAttributesService;
 	}
 
 }