org.projectforge.web.humanresources.HRPlanningEventsProvider.java Source code

Introduction

Here is the source code for org.projectforge.web.humanresources.HRPlanningEventsProvider.java

Source

/////////////////////////////////////////////////////////////////////////////
//
// Project ProjectForge Community Edition
//         www.projectforge.org
//
// Copyright (C) 2001-2013 Kai Reinhard (k.reinhard@micromata.de)
//
// ProjectForge is dual-licensed.
//
// This community edition is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License as published
// by the Free Software Foundation; version 3 of the License.
//
// This community edition is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
// Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, see http://www.gnu.org/licenses/.
//
/////////////////////////////////////////////////////////////////////////////

package org.projectforge.web.humanresources;

import java.math.BigDecimal;
import java.util.List;

import net.ftlines.wicket.fullcalendar.Event;

import org.apache.commons.lang.StringUtils;
import org.apache.wicket.Component;
import org.joda.time.DateTime;
import org.projectforge.common.NumberHelper;
import org.projectforge.humanresources.HRPlanningDO;
import org.projectforge.humanresources.HRPlanningDao;
import org.projectforge.humanresources.HRPlanningEntryDO;
import org.projectforge.humanresources.HRPlanningFilter;
import org.projectforge.user.PFUserContext;
import org.projectforge.web.calendar.ICalendarFilter;
import org.projectforge.web.calendar.MyFullCalendarEventsProvider;

/**
 * Creates events corresponding to the hr planning entries.
 * @author Kai Reinhard (k.reinhard@micromata.de)
 * 
 */
public class HRPlanningEventsProvider extends MyFullCalendarEventsProvider {
    private static final long serialVersionUID = -8614136730204759894L;

    private static final org.apache.log4j.Logger log = org.apache.log4j.Logger
            .getLogger(HRPlanningEventsProvider.class);

    private final HRPlanningDao hrPlanningDao;

    private final ICalendarFilter calendarFilter;

    /**
     * the name of the event class.
     */
    public static final String EVENT_CLASS_NAME = "hrPlanning";

    /**
     * @param parent For i18n.
     * @param hrPlanningDao
     * @see Component#getString(String)
     */
    public HRPlanningEventsProvider(final Component parent, final ICalendarFilter calendarFilter,
            final HRPlanningDao hrPlanningDao) {
        super(parent);
        this.calendarFilter = calendarFilter;
        this.hrPlanningDao = hrPlanningDao;
    }

    /**
     * @see org.projectforge.web.calendar.MyFullCalendarEventsProvider#buildEvents(org.joda.time.DateTime, org.joda.time.DateTime)
     */
    @Override
    protected void buildEvents(final DateTime start, final DateTime end) {
        if (calendarFilter.isShowPlanning() == false) {
            // Don't show plannings.
            return;
        }
        final HRPlanningFilter filter = new HRPlanningFilter();
        Integer timesheetUserId = calendarFilter.getTimesheetUserId();
        if (timesheetUserId == null) {
            timesheetUserId = PFUserContext.getUserId();
        }
        filter.setUserId(timesheetUserId);
        filter.setStartTime(start.toDate());
        filter.setStopTime(end.toDate());
        final List<HRPlanningDO> list = hrPlanningDao.getList(filter);
        if (list == null) {
            return;
        }
        for (final HRPlanningDO planning : list) {
            if (planning.getEntries() == null) {
                continue;
            }
            final DateTime week = new DateTime(planning.getWeek(), PFUserContext.getDateTimeZone());
            for (final HRPlanningEntryDO entry : planning.getEntries()) {
                putEvent(entry, week, "week", 6, entry.getUnassignedHours());
                putEvent(entry, week, "mo", 0, entry.getMondayHours());
                putEvent(entry, week.plusDays(1), "tu", 0, entry.getTuesdayHours());
                putEvent(entry, week.plusDays(2), "we", 0, entry.getWednesdayHours());
                putEvent(entry, week.plusDays(3), "th", 0, entry.getThursdayHours());
                putEvent(entry, week.plusDays(4), "fr", 0, entry.getFridayHours());
                putEvent(entry, week.plusDays(5), "we", 1, entry.getWeekendHours());
            }
        }
    }

    private void putEvent(final HRPlanningEntryDO entry, final DateTime start, final String suffix,
            final int durationDays, final BigDecimal hours) {
        if (NumberHelper.isGreaterZero(hours) == false) {
            return;
        }
        if (log.isDebugEnabled() == true) {
            log.debug("Date: " + start + ", hours=" + hours + ", duration: " + durationDays);
        }
        final Event event = new Event().setAllDay(true);
        event.setClassName(EVENT_CLASS_NAME);
        final String id = "" + entry.getId() + "-" + suffix;
        event.setId(id);
        event.setStart(start);
        if (durationDays > 0) {
            event.setEnd(start.plusDays(durationDays));
        } else {
            event.setEnd(start);
        }
        final StringBuffer buf = new StringBuffer();
        buf.append(NumberHelper.formatFraction2(hours)).append(parent.getString("calendar.unit.hour")).append(" ")
                .append(entry.getProjektNameOrStatus());
        if (StringUtils.isNotBlank(entry.getDescription()) == true) {
            buf.append(": ");
            if (durationDays > 2) {
                buf.append(StringUtils.abbreviate(entry.getDescription(), 100));
            } else if (durationDays > 1) {
                buf.append(StringUtils.abbreviate(entry.getDescription(), 50));
            } else {
                buf.append(StringUtils.abbreviate(entry.getDescription(), 20));
            }
        }
        event.setTitle(buf.toString());
        events.put(id, event);
    }
}