org.kuali.kpme.tklm.time.timesheet.TimesheetDocument.java Source code

Introduction

Here is the source code for org.kuali.kpme.tklm.time.timesheet.TimesheetDocument.java

Source

/**
 * 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.kpme.tklm.time.timesheet;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.joda.time.LocalDate;
import org.kuali.kpme.core.assignment.Assignment;
import org.kuali.kpme.core.calendar.entry.CalendarEntry;
import org.kuali.kpme.core.document.calendar.CalendarDocument;
import org.kuali.kpme.core.job.Job;
import org.kuali.kpme.core.service.HrServiceLocator;
import org.kuali.kpme.core.util.TKUtils;
import org.kuali.kpme.tklm.time.rules.timecollection.TimeCollectionRule;
import org.kuali.kpme.tklm.time.service.TkServiceLocator;
import org.kuali.kpme.tklm.time.timeblock.TimeBlock;
import org.kuali.kpme.tklm.time.timesummary.TimeSummary;
import org.kuali.kpme.tklm.time.workflow.TimesheetDocumentHeader;
import org.kuali.rice.krad.util.GlobalVariables;

public class TimesheetDocument extends CalendarDocument {

    public static final String TIMESHEET_DOCUMENT_TYPE = "TimesheetDocument";

    private List<Job> jobs = new LinkedList<Job>();
    private List<TimeBlock> timeBlocks = new LinkedList<TimeBlock>();
    private TimeSummary timeSummary = new TimeSummary();
    private Map<Long, Job> jobNumberToJobMap = new HashMap<Long, Job>();

    public TimesheetDocument(TimesheetDocumentHeader documentHeader) {
        this.documentHeader = documentHeader;
        setCalendarType(TIMESHEET_DOCUMENT_TYPE);
    }

    public TimesheetDocumentHeader getDocumentHeader() {
        return (TimesheetDocumentHeader) super.getDocumentHeader();
    }

    public void setDocumentHeader(TimesheetDocumentHeader documentHeader) {
        this.documentHeader = documentHeader;
    }

    @Override
    public List<Assignment> getAssignments() {
        return assignments;
    }

    public void setAssignments(List<Assignment> assignments) {
        this.assignments = assignments;
    }

    public List<Job> getJobs() {
        return jobs;
    }

    public void setJobs(List<Job> jobs) {
        this.jobs = jobs;
        jobNumberToJobMap.clear();
        if (jobs != null) {
            for (Job job : jobs) {
                jobNumberToJobMap.put(job.getJobNumber(), job);
            }
        }
    }

    public List<TimeBlock> getTimeBlocks() {
        return timeBlocks;
    }

    public void setTimeBlocks(List<TimeBlock> timeBlocks) {
        this.timeBlocks = timeBlocks;
    }

    @Override
    public CalendarEntry getCalendarEntry() {
        return calendarEntry;
    }

    public void setCalendarEntry(CalendarEntry calendarEntry) {
        this.calendarEntry = calendarEntry;
    }

    public void setTimeSummary(TimeSummary timeSummary) {
        this.timeSummary = timeSummary;
    }

    public TimeSummary getTimeSummary() {
        return timeSummary;
    }

    public String getPrincipalId() {
        return getDocumentHeader().getPrincipalId();
    }

    public Job getJob(Long jobNumber) {
        return jobNumberToJobMap.get(jobNumber);
    }

    @Override
    public LocalDate getAsOfDate() {
        return getCalendarEntry().getBeginPeriodFullDateTime().toLocalDate();
    }

    public LocalDate getDocEndDate() {
        return getCalendarEntry().getEndPeriodFullDateTime().toLocalDate();
    }

    public String getDocumentId() {
        return this.getDocumentHeader().getDocumentId();
    }

    public Map<String, List<LocalDate>> getEarnCodeMap() {
        Map<String, List<LocalDate>> earnCodeMap = new HashMap<String, List<LocalDate>>();
        for (TimeBlock tb : getTimeBlocks()) {
            if (!earnCodeMap.containsKey(tb.getEarnCode())) {
                List<LocalDate> lst = new ArrayList<LocalDate>();
                lst.add(tb.getBeginDateTime().toLocalDate());
                earnCodeMap.put(tb.getEarnCode(), lst);
            } else {
                earnCodeMap.get(tb.getEarnCode()).add(tb.getBeginDateTime().toLocalDate());
            }
        }
        return earnCodeMap;
    }

    public Map<String, String> getAssignmentDescriptions(boolean clockOnlyAssignments) {
        Map<String, String> assignmentDescriptions = new LinkedHashMap<String, String>();

        for (Assignment assignment : assignments) {
            String principalId = GlobalVariables.getUserSession().getPrincipalId();

            if (HrServiceLocator.getHRPermissionService().canViewCalendarDocumentAssignment(principalId, this,
                    assignment)) {
                TimeCollectionRule tcr = TkServiceLocator.getTimeCollectionRuleService().getTimeCollectionRule(
                        assignment.getJob().getDept(), assignment.getWorkArea(), LocalDate.now());
                boolean isSynchronous = tcr == null || tcr.isClockUserFl();
                if (!clockOnlyAssignments || isSynchronous) {
                    assignmentDescriptions.putAll(TKUtils.formatAssignmentDescription(assignment));
                }
            }
        }

        return assignmentDescriptions;
    }
}