/*
* 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.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;
private final String WORK_REPORT_TYPE_COLUMN = "workReportType";
@SuppressWarnings("unchecked")
@Override
public List<WorkReport> getAllByWorkReportType(WorkReportType workReportType) {
return getSession()
.createCriteria(WorkReport.class)
.add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, 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(() -> getOrderElementsAssociatedWithWorkReports());
orderDAO.loadOrdersAvoidingProxyFor(elements);
}
private List<OrderElement> getOrderElementsAssociatedWithWorkReports() {
Set<OrderElement> result = new HashSet<>();
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<>(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(WORK_REPORT_TYPE_COLUMN, 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.isEmpty() ) {
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 | InstanceNotFoundException e) {
throw new RuntimeException(e);
}
return workReportType;
}
@Override
public boolean isAnyPersonalTimesheetAlreadySaved() {
return getSession()
.createCriteria(WorkReport.class)
.add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, getPersonalTimesheetsWorkReportType()))
.list()
.isEmpty();
}
@Override
@SuppressWarnings("unchecked")
public List<WorkReport> findPersonalTimesheetsByResourceAndOrderElement(Resource resource) {
return getSession()
.createCriteria(WorkReport.class)
.add(Restrictions.eq(WORK_REPORT_TYPE_COLUMN, getPersonalTimesheetsWorkReportType()))
.add(Restrictions.eq("resource", resource))
.list();
}
}