package org.msh.tb.cases.dispensing;

import org.msh.tb.entities.TreatmentMonitoring;
import org.msh.utils.date.DateUtils;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.List;


/**
 * Keep temporary information during editing about a case and its dispensing days 
 * @author Ricardo Memoria
 *
 */
public class CaseDispensingInfo {

	private TreatmentMonitoring treatmentMonitoring;
	private List<WeekInfo> weeks = new ArrayList<WeekInfo>();
	private List<DayInfo> days = new ArrayList<DayInfo>();
	private WeekInfo selectedWeek;


	public CaseDispensingInfo(TreatmentMonitoring treatmentMonitoring) {
		super();
		this.treatmentMonitoring = treatmentMonitoring;
		mountDispensingDays();
	}
	
	/**
	 * Generate the list of treatment days with information about the dispensing days
	 */
	protected void mountDispensingDays() {
		int month = treatmentMonitoring.getMonth() - 1;
		int year = treatmentMonitoring.getYear();
		
		Calendar c = Calendar.getInstance();
		c.clear();
		c.set(Calendar.MONTH, month);
		c.set(Calendar.YEAR, year);
		c.set(Calendar.DAY_OF_MONTH, 1);
		int weekday = c.get(Calendar.DAY_OF_WEEK);
		int numdays = DateUtils.daysInAMonth(year, month);

		Date iniDate = treatmentMonitoring.getTbcase().getTreatmentPeriod().getIniDate();
		Date endDate = treatmentMonitoring.getTbcase().getTreatmentPeriod().getEndDate();
		
		int day = 0;
		int count = 1;
		int weeknumber = 1;
		weeks = new ArrayList<WeekInfo>();
		while (true) {
			if (day + 1 > numdays)
				break;
			
			WeekInfo weekInfo = new WeekInfo();
			weekInfo.setWeek(weeknumber);
			weeks.add(weekInfo);

			for (int i = 1; i <= 7; i++) {
				if (count >= weekday) {
					day++;
					if (weekInfo.getIniDay() == 0)
						weekInfo.setIniDay(day);
				}
				
				if (day > numdays)
					break;
				DayInfo dd = new DayInfo();
				dd.setWeekDay(i);

				if ((day > 0) && (day <= numdays)) {
					dd.setValue(treatmentMonitoring.getDay(day));
					dd.setDay(day);					
					weekInfo.setEndDay(day);

					c.set(Calendar.DAY_OF_MONTH, day);
					dd.setAvailable((!c.getTime().before(iniDate)) && (!c.getTime().after(endDate)));
				}
				else dd.setAvailable(false);

				weekInfo.getDays().add(dd);
				days.add(dd);
				count++;
			}
			weeknumber++;
		}
	}



	/**
	 * Update the dispensing days to the object dispensingDays of class {@link CaseDispensingDays}, 
	 * making it ready to be persisted in the database
	 */
	public void updateDispensingDays() {
		if (weeks == null)
			return;
		
		for (WeekInfo weekInfo: weeks) {
			for (DayInfo day: weekInfo.getDays()) {
				if (day.isAvailable())
					treatmentMonitoring.setDay(day.getDay(), day.getValue());
			}
		}
//		treatmentMo.setDispensingDays(dispensingDays);
//		caseDispensing.updateTotalDays();
	}


	
	/**
	 * @return the caseDispensing
	 */
	public TreatmentMonitoring getTreatmentMonitoring() {
		return treatmentMonitoring;
	}
	/**
	 * @param caseDispensing the caseDispensing to set
	 */
	public void setTreatmentMonitoring(TreatmentMonitoring treatmentMonitoring) {
		this.treatmentMonitoring = treatmentMonitoring;
	}


	/**
	 * @return the weeks
	 */
	public List<WeekInfo> getWeeks() {
		return weeks;
	}

	/**
	 * @return the selectedWeek
	 */
	public WeekInfo getSelectedWeek() {
		return selectedWeek;
	}

	/**
	 * @param selectedWeek the selectedWeek to set
	 */
	public void setSelectedWeek(WeekInfo selectedWeek) {
		this.selectedWeek = selectedWeek;
	}

	/**
	 * @return the days
	 */
	public List<DayInfo> getDays() {
		return days;
	}

}