/*
* This file is part of LibrePlan
*
* Copyright (C) 2009-2010 Fundación para o Fomento da Calidade Industrial e
* Desenvolvemento Tecnolóxico de Galicia
* Copyright (C) 2010-2011 Igalia, S.L.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.libreplan.business.workreports.daos;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.hibernate.Criteria;
import org.hibernate.NonUniqueResultException;
import org.hibernate.Query;
import org.hibernate.criterion.Restrictions;
import org.joda.time.LocalDate;
import org.libreplan.business.common.IAdHocTransactionService;
import org.libreplan.business.common.IOnTransaction;
import org.libreplan.business.common.daos.IntegrationEntityDAO;
import org.libreplan.business.common.entities.PersonalTimesheetsPeriodicityEnum;
import org.libreplan.business.common.exceptions.InstanceNotFoundException;
import org.libreplan.business.orders.daos.IOrderDAO;
import org.libreplan.business.orders.entities.OrderElement;
import org.libreplan.business.resources.entities.Resource;
import org.libreplan.business.workreports.entities.PredefinedWorkReportTypes;
import org.libreplan.business.workreports.entities.WorkReport;
import org.libreplan.business.workreports.entities.WorkReportLine;
import org.libreplan.business.workreports.entities.WorkReportType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Repository;
/**
* Dao for {@link WorkReportDAO}
*
* @author Diego Pino García <dpino@igalia.com>
*/
@Repository
@Scope(BeanDefinition.SCOPE_SINGLETON)
public class WorkReportDAO extends IntegrationEntityDAO<WorkReport>
implements IWorkReportDAO {
@Autowired
private IAdHocTransactionService adHocTransactionService;
@Autowired
private IOrderDAO orderDAO;
@Autowired
private IWorkReportTypeDAO workReportTypeDAO;
@SuppressWarnings("unchecked")
@Override
public List<WorkReport> getAllByWorkReportType(WorkReportType workReportType) {
final Criteria criteria = getSession().createCriteria(WorkReport.class);
return criteria.add(Restrictions.eq("workReportType", workReportType)).list();
}
@Override
public List<WorkReport> getAll() {
return list(WorkReport.class);
}
@Override
public List<WorkReport> allWorkReportsWithAssociatedOrdersUnproxied() {
forceOrdersUnproxied();
return list(WorkReport.class);
}
private void forceOrdersUnproxied() {
List<OrderElement> elements = adHocTransactionService
.runOnAnotherReadOnlyTransaction(new IOnTransaction<List<OrderElement>>() {
@Override
public List<OrderElement> execute() {
return getOrderElementsAssociatedWithWorkReports();
}
});
orderDAO.loadOrdersAvoidingProxyFor(elements);
}
private List<OrderElement> getOrderElementsAssociatedWithWorkReports() {
Set<OrderElement> result = new HashSet<OrderElement>();
result.addAll(elementsFrom(getSession().createQuery(
"select w.orderElement from WorkReport w")));
result
.addAll(elementsFrom(getSession()
.createQuery(
"select line.orderElement from WorkReport w JOIN w.workReportLines line")));
return new ArrayList<OrderElement>(result);
}
@SuppressWarnings("unchecked")
private List<OrderElement> elementsFrom(Query orderElementsQuery) {
return orderElementsQuery.list();
}
@Override
public int getFirstReportYear() {
final String query = "select min(w.date) from WorkReportLine w";
Date minDate = (Date) getSession().createQuery(query).uniqueResult();
if (minDate != null) {
Calendar calendar = Calendar.getInstance();
calendar.setTime(minDate);
return calendar.get(Calendar.YEAR);
}
return 0;
}
@Override
public int getLastReportYear() {
final String query = "select max(w.date) from WorkReportLine w";
Date maxDate = (Date) getSession().createQuery(query).uniqueResult();
if (maxDate != null) {
Calendar calendar = Calendar.getInstance();
calendar.setTime(maxDate);
return calendar.get(Calendar.YEAR);
}
return 0;
}
@Override
@SuppressWarnings("unchecked")
public WorkReport getPersonalTimesheetWorkReport(Resource resource,
LocalDate date, PersonalTimesheetsPeriodicityEnum periodicity) {
Criteria criteria = getSession().createCriteria(WorkReport.class);
criteria.add(Restrictions.eq("workReportType",
getPersonalTimesheetsWorkReportType()));
List<WorkReport> personalTimesheets = criteria.add(
Restrictions.eq("resource", resource)).list();
LocalDate start = periodicity.getStart(date);
LocalDate end = periodicity.getEnd(date);
for (WorkReport workReport : personalTimesheets) {
Set<WorkReportLine> workReportLines = workReport
.getWorkReportLines();
if (workReportLines.size() > 0) {
LocalDate workReportDate = LocalDate
.fromDateFields(workReportLines.iterator().next()
.getDate());
if (workReportDate.compareTo(start) >= 0
&& workReportDate.compareTo(end) <= 0) {
return workReport;
}
}
}
return null;
}
private WorkReportType getPersonalTimesheetsWorkReportType() {
WorkReportType workReportType;
try {
workReportType = workReportTypeDAO
.findUniqueByName(PredefinedWorkReportTypes.PERSONAL_TIMESHEETS
.getName());
} catch (NonUniqueResultException e) {
throw new RuntimeException(e);
} catch (InstanceNotFoundException e) {
throw new RuntimeException(e);
}
return workReportType;
}
@Override
public boolean isAnyPersonalTimesheetAlreadySaved() {
WorkReportType workReportType = getPersonalTimesheetsWorkReportType();
Criteria criteria = getSession().createCriteria(WorkReport.class);
criteria.add(Restrictions.eq("workReportType", workReportType));
return criteria.list().isEmpty();
}
@Override
@SuppressWarnings("unchecked")
public List<WorkReport> findPersonalTimesheetsByResourceAndOrderElement(
Resource resource) {
Criteria criteria = getSession().createCriteria(WorkReport.class);
criteria.add(Restrictions.eq("workReportType",
getPersonalTimesheetsWorkReportType()));
criteria.add(Restrictions.eq("resource", resource));
return criteria.list();
}
}