ReminderCalendarHandler.java

package org.sigmah.server.handler.calendar;

/*
 * #%L
 * Sigmah
 * %%
 * Copyright (C) 2010 - 2016 URD
 * %%
 * This program 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, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program 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/gpl-3.0.html>.
 * #L%
 */

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;

import javax.persistence.EntityManager;
import javax.persistence.NoResultException;
import javax.persistence.Query;

import org.sigmah.server.domain.reminder.Reminder;
import org.sigmah.server.domain.reminder.ReminderList;
import org.sigmah.shared.command.result.Calendar;
import org.sigmah.shared.dto.calendar.CalendarIdentifier;
import org.sigmah.shared.dto.calendar.Event;
import org.sigmah.shared.dto.calendar.ReminderCalendarIdentifier;

import com.google.inject.Inject;

/**
 * Retrieve monitored points as events.
 * 
 * @author tmi
 */
public class ReminderCalendarHandler implements CalendarHandler {

	
	@Inject
	public ReminderCalendarHandler() {
	
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	@SuppressWarnings("deprecation")
	public Calendar getCalendar(CalendarIdentifier identifier,EntityManager em) {

		if (!(identifier instanceof ReminderCalendarIdentifier)) {
			throw new IllegalArgumentException("Identifier must be an instance of ReminderCalendarIdentifier, received an instance of "
				+ identifier.getClass().getSimpleName());
		}

		final ReminderCalendarIdentifier calendarIdentifier = (ReminderCalendarIdentifier) identifier;

		final SimpleDateFormat format = new SimpleDateFormat(calendarIdentifier.getDateFormat());

		final Query query = em.createQuery("SELECT l FROM ReminderList l WHERE l.id = :listId");
		query.setParameter("listId", calendarIdentifier.getReminderListId());

		// Configuring the calendar
		final Calendar calendar = new Calendar();
		calendar.setIdentifier(identifier);
		calendar.setName(calendarIdentifier.getCalendarName());
		calendar.setEditable(false);

		final HashMap<Date, List<Event>> eventMap = new HashMap<Date, List<Event>>();
		calendar.setEvents(eventMap);

		try {

			final ReminderList list = (ReminderList) query.getSingleResult();

			if (list.getReminders() != null) {
				for (final Reminder reminder : list.getReminders()) {
                    if(reminder.getDeleted() == null || !reminder.getDeleted()) {
					final Event event = new Event();
					event.setParent(calendar);
					event.setIdentifier(reminder.getId());

					// A completed point is displayed at its completion date,
					// while a running point is displayed at its expected date.
					final Date date = reminder.isCompleted() ? reminder.getCompletionDate() : reminder.getExpectedDate();

					event.setDtstart(date);
					event.setDtend(new Date(date.getYear(), date.getMonth(), date.getDate() + 1));

					// Summary.
					StringBuilder sb = new StringBuilder();
					sb.append(reminder.getLabel());
					if (reminder.isCompleted()) {
						sb.append(" (");
						sb.append(calendarIdentifier.getCompletedEventString());
						sb.append(')');
					}
					event.setSummary(sb.toString());

					// Description.
					sb = new StringBuilder();
					sb.append(reminder.getLabel());
					if (reminder.isCompleted()) {
						sb.append(" (");
						sb.append(calendarIdentifier.getExpectedDateString());
						sb.append(": ");
						sb.append(format.format(reminder.getExpectedDate()));
						sb.append(')');
					}

					event.setDescription(sb.toString());

					// Adding the event to the event map
					final Date key = event.getKey();
					List<Event> events = eventMap.get(key);
					if (events == null) {
						events = new ArrayList<Event>();
						eventMap.put(key, events);
					}
					events.add(event);
				}
			}
			}

		} catch (NoResultException e) {
			// No monitored points in the current project.
		}

		return calendar;
	}
}