/**
    * Copyright 2004-2013 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.hr.lm.util;
  
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.Comparator;
  import java.util.List;
  
  import org.joda.time.DateTime;
  import org.joda.time.Interval;
  import org.joda.time.LocalDate;
  import org.joda.time.LocalDateTime;
  import org.kuali.hr.lm.leaveblock.LeaveBlock;
  import org.kuali.hr.time.calendar.CalendarEntries;
  import org.kuali.hr.time.calendar.LeaveCalendar;
  import org.kuali.hr.time.util.TKUtils;
  
  public class LeaveBlockAggregate {
  	public List<List<LeaveBlock>> dayLeaveBlockList = new ArrayList<List<LeaveBlock>>();
  	private CalendarEntries leaveCalendarEntry;
  	private LeaveCalendar leaveCalendar;
  
      
      /**
       * Provides the option to refer to the time zone adjusted time for the current
       * user.
       * @param LeaveBlocks
       * @param leaveCalendarEntry
       * @param leaveCalendar
       */
      public LeaveBlockAggregate(List<LeaveBlock> leaveBlocks, CalendarEntries leaveCalendarEntry, LeaveCalendar leaveCalendar) {
  		this.leaveCalendarEntry = leaveCalendarEntry;
  		this.leaveCalendar = leaveCalendar;
  		List<Interval> dayIntervals = TKUtils.getDaySpanForCalendarEntry(leaveCalendarEntry);
  		for(Interval dayInt : dayIntervals){
  			List<LeaveBlock> dayLeaveBlocks = new ArrayList<LeaveBlock>();
  			for(LeaveBlock leaveBlock : leaveBlocks){
  				LocalDate localDate = new LocalDate(leaveBlock.getLeaveDate());
                  LocalDate dayIntBegin = new LocalDate(dayInt.getStart());
  				if(localDate.equals(dayIntBegin)){
  					dayLeaveBlocks.add(leaveBlock);
  				}
  			}
  			dayLeaveBlockList.add(dayLeaveBlocks);
  		} 
  	}
      
      public LeaveBlockAggregate(List<LeaveBlock> leaveBlocks, CalendarEntries leaveCalendarEntry) {
  		this.leaveCalendarEntry = leaveCalendarEntry;
  		List<Interval> dayIntervals = TKUtils.getDaySpanForCalendarEntry(leaveCalendarEntry);
  		for(Interval dayInt : dayIntervals){
  			List<LeaveBlock> dayLeaveBlocks = new ArrayList<LeaveBlock>();
  			for(LeaveBlock leaveBlock : leaveBlocks){
                  LocalDate localDate = new LocalDate(leaveBlock.getLeaveDate());
                  LocalDate dayIntBegin = new LocalDate(dayInt.getStart());
                  if(localDate.equals(dayIntBegin)){
                      dayLeaveBlocks.add(leaveBlock);
                  }
  			}
  			dayLeaveBlockList.add(dayLeaveBlocks);
  		} 
  	}
      
      /**
       *  build leaveBlockAggregate with given leaveBlocks, calendarEntry and dayIntervals
       *  dayIntervals with full week span is for Time Calendar
       * @param LeaveBlocks
       * @param leaveCalendarEntry
       * @param dayIntervals
       */ 
      public LeaveBlockAggregate(List<LeaveBlock> leaveBlocks, CalendarEntries leaveCalendarEntry, List<Interval> dayIntervals) {
      	this.leaveCalendarEntry = leaveCalendarEntry;
  		for(Interval dayInt : dayIntervals){
  			List<LeaveBlock> dayLeaveBlocks = new ArrayList<LeaveBlock>();
  			DateTime localTime = (new DateTime(dayInt.getStart())).toLocalDateTime().toDateTime();
  			String intervalStartDateString = TKUtils.getTimelessDate(localTime.toDate()).toString();
  			
  			for(LeaveBlock leaveBlock : leaveBlocks){
  				// if the interval end time is 0, ie the beginning of a day, use the date string of the interval start time
  				// to check if the leave block should go into this interval. Leave blocks only have leaveDate, there's no leave time
  				if(dayInt.getEnd().getHourOfDay() == 0) {
  					String lbDateString = TKUtils.getTimelessDate(leaveBlock.getLeaveDate()).toString();
  					if(intervalStartDateString.equals(lbDateString)) {
  						dayLeaveBlocks.add(leaveBlock);
  					}
 				} else {
                     LocalDate localDate = new LocalDate(leaveBlock.getLeaveDate());
                     LocalDate dayIntBegin = new LocalDate(dayInt.getStart());
                     if(localDate.equals(dayIntBegin)){
                         dayLeaveBlocks.add(leaveBlock);
                     }
 				}
 			}
 			dayLeaveBlockList.add(dayLeaveBlocks);
 		} 
     }   
     
 	public List<LeaveBlock> getFlattenedLeaveBlockList(){
 		List<LeaveBlock> lstLeaveBlocks = new ArrayList<LeaveBlock>();
 		for(List<LeaveBlock> leaveBlocks : dayLeaveBlockList){
 			lstLeaveBlocks.addAll(leaveBlocks);
 		}
 
 		Collections.sort(lstLeaveBlocks, new Comparator<LeaveBlock>() { // Sort the Leave Blocks
 			public int compare(LeaveBlock tb1, LeaveBlock tb2) {
 				if (tb1 != null && tb2 != null) {
                     if (tb1.getTimestamp() != null && tb2.getTimestamp() != null) {
 					    return tb1.getTimestamp().compareTo(tb2.getTimestamp());
                     }
                 }
 				return 0;
 			}
 		});
 
 		return lstLeaveBlocks;
 	}
 
 	/**
 	 * Provides a way to access all of the leave blocks for a given week.
 	 *
 	 * Outer list is 0 indexed list representing days in a week.
 	 * Inner List are all of the time blocks for that day.
 	 *
 	 * Ex.
 	 *
 	 * List<List<LeaveBlock>> week0 = getWeekLeaveBlocks(0);
 	 * List<LeaveBlock> day0 = week0.get(0);
 	 *
 	 * @param week
 	 * @return
 	 */
 	public List<List<LeaveBlock>> getWeekLeaveBlocks(int week){
 		int startIndex = week*7;
 		int endIndex = (week*7)+7;
 		endIndex = endIndex > dayLeaveBlockList.size() ? dayLeaveBlockList.size() : endIndex;
 
         // Need to sort each day by clock time.
         List<List<LeaveBlock>> wList = dayLeaveBlockList.subList(startIndex, endIndex);
         for (List<LeaveBlock> dList : wList) {
             Collections.sort(dList, new Comparator<LeaveBlock>() { // Sort the Leave Blocks
                 public int compare(LeaveBlock tb1, LeaveBlock tb2) {
                     if (tb1 != null && tb2 != null)
                         return tb1.getTimestamp().compareTo(tb2.getTimestamp());
                     return 0;
                 }
             });
         }
 
 		return wList;
 	}
 
 	/**
 	 * @return the total number of weeks that this object represents.
 	 */
 	public int numberOfAggregatedWeeks() {
 		int weeks = 0;
 
 		if (this.dayLeaveBlockList.size() > 0) {
 			weeks = this.dayLeaveBlockList.size() / 7;
 			if (this.dayLeaveBlockList.size() % 7 > 0)
 				weeks++;
 		}
 
 		return weeks;
 	}
 
 	public List<List<LeaveBlock>> getDayLeaveBlockList() {
 		return dayLeaveBlockList;
 	}
 
 	public CalendarEntries getleaveCalendarEntry() {
 		return leaveCalendarEntry;
 	}
 
 	public void setleaveCalendarEntry(CalendarEntries leaveCalendarEntry) {
 		this.leaveCalendarEntry = leaveCalendarEntry;
 	}
 
 	public LeaveCalendar getLeaveCalendar() {
 		return leaveCalendar;
 	}
 
 	public void setLeaveCalendar(LeaveCalendar leaveCalendar) {
 		this.leaveCalendar = leaveCalendar;
 	}
 
 }