Department.java

import java.util.HashMap;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Calendar;
import java.util.Date;
import java.io.Serializable;

public class Department implements Serializable {
	String code;
	DepartmentType type;
	HashMap<String, Employee> employees;
	HashMap<String, ArrayList<Ticket>> tickets;

	public Department(String code, DepartmentType type, HashMap<String, Employee> employees, HashMap<String, ArrayList<Ticket>> tickets) {
		this.code = code;
		this.type = type;
		this.employees = employees;
		this.tickets = tickets;
	}

	public Department(String code, DepartmentType type) {
		this(code, type, new HashMap<String, Employee>(), new HashMap<String, ArrayList<Ticket>>());
	}

	public Employee getEmployee(String id) {
		return employees.get(id);
	}

	public Employee addEmployee(String id) {
		Employee employee = new Employee(id);
		employees.put(employee.id, employee);
		return employee;
	}


	public Ticket punch(String employeeId, HourType hourType) {
		return punch(employeeId, hourType, new Date());
	}

	public void punch(String id, HourType hourType, int day, int hours, int minutes) {
		Calendar calStart = Calendar.getInstance();
		calStart.set(Calendar.DAY_OF_WEEK, day);
		calStart.set(Calendar.HOUR_OF_DAY, 1);
		calStart.set(Calendar.MINUTE, 0);
		calStart.set(Calendar.SECOND, 0);
		calStart.set(Calendar.MILLISECOND, 0);

		Calendar calEnd = Calendar.getInstance();
		calEnd.set(Calendar.DAY_OF_WEEK, day);
		calEnd.set(Calendar.HOUR_OF_DAY, hours+1);
		calEnd.set(Calendar.MINUTE, minutes);
		calEnd.set(Calendar.SECOND, 0);
		calEnd.set(Calendar.MILLISECOND, 0);

		Date start = calStart.getTime();
		Date end = calEnd.getTime();

		punch(id, hourType, start);
		punch(id, hourType, end);
	}

	public Timesheet getTimesheet(String employeeId, Date date) {
		ArrayList<Ticket> employeeTickets = getEmployeeTickets(employeeId, date);
		Timesheet sheet = new Timesheet(employeeTickets);
		return sheet;
	}

	public Timesheet getTimesheet(String id, Date[] dates) {
		ArrayList<Ticket> tickets = new ArrayList<Ticket>();
		for(Date date : dates) {
			ArrayList<Ticket> ticketsForDate = getEmployeeTickets(id, date);
			tickets.addAll(ticketsForDate);
		}
		return new Timesheet(tickets);
	}
	
	public ArrayList<Ticket> getEmployeeTickets(String id, Date date) {
		ArrayList<Ticket> employeeTickets = new ArrayList<Ticket>();
		ArrayList<Ticket> ticketsForDate = tickets.get(Ticket.keyFromDate(date));
		if(ticketsForDate == null) {
			return employeeTickets;
		}
		for(Ticket ticket : ticketsForDate) {
			if(ticket.employeeId.equals(id)) {
				employeeTickets.add(ticket);
			}
		}		
		return employeeTickets;
	}

	public Ticket punch(String employeeId, HourType hourType, Date date) {
		Employee employee = getEmployee(employeeId);
		if(employee == null) {
			return null;
		}
		if(tickets.isEmpty()) {
			return clockIn(employee.id, hourType, date);
		}
		ArrayList<Ticket> employeeTickets = getEmployeeTickets(employeeId, date);
		if(employeeTickets.isEmpty()) {
			return clockIn(employee.id, hourType, date);	
		}
		int lastIndex = employeeTickets.size() - 1;
		Ticket latestTicket = employeeTickets.get(lastIndex);
		if(latestTicket.ticketType == TicketType.ClockIn) {
			return clockOut(employee.id, hourType, date);
		}
		return clockIn(employee.id, hourType, date);
	}

	private Ticket clockIn(String employeeId, HourType hourType, Date date) {
		Ticket ticket = new Ticket(date, hourType, TicketType.ClockIn, employeeId);
		ArrayList<Ticket>ticketsForDay = tickets.get(ticket.dateKey());
		if(ticketsForDay == null) {
			ticketsForDay = new ArrayList<Ticket>();
			tickets.put(ticket.dateKey(), ticketsForDay);
		}
		ticketsForDay.add(ticket);
		return ticket;
	}

	private Ticket clockOut(String employeeId, HourType hourType, Date date) {
		Ticket ticket = new Ticket(date, hourType, TicketType.ClockOut, employeeId);
		ArrayList<Ticket>ticketsForDay = tickets.get(ticket.dateKey());
		if(ticketsForDay == null) {
			return null;
		}
		ticketsForDay.add(ticket);
		return ticket;
	}
}