/*
 * 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.
 *
 * Copyright (C) 2011 WirelessGalicia, 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.planner.daos;

import java.util.ArrayList;
import java.util.List;

import org.hibernate.criterion.Restrictions;
import org.libreplan.business.common.daos.GenericDAOHibernate;
import org.libreplan.business.common.exceptions.InstanceNotFoundException;
import org.libreplan.business.orders.entities.Order;
import org.libreplan.business.orders.entities.OrderElement;
import org.libreplan.business.planner.entities.SubcontractedTaskData;
import org.libreplan.business.planner.entities.Task;
import org.libreplan.business.planner.entities.TaskElement;
import org.libreplan.business.scenarios.bootstrap.PredefinedScenarios;
import org.libreplan.business.scenarios.entities.Scenario;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;

/**
 * DAO for the {@link SubcontractedTaskDataDAO} entity.
 *
 * @author Manuel Rego Casasnovas <mrego@igalia.com>
 */
@Repository
@Scope(BeanDefinition.SCOPE_SINGLETON)
public class SubcontractedTaskDataDAO
        extends GenericDAOHibernate<SubcontractedTaskData, Long>
        implements ISubcontractedTaskDataDAO {

    @Override
    @Transactional(readOnly = true, propagation = Propagation.REQUIRES_NEW)
    public boolean existsInAnohterTransaction(Long id) {
        if (id == null) {
            return false;
        }

        try {
            SubcontractedTaskData found = find(id);
            return (found != null) && (found.getId().equals(id));
        } catch (InstanceNotFoundException e) {
            return false;
        }
    }

    @Override
    public List<SubcontractedTaskData> getAll() {
        return list(SubcontractedTaskData.class);
    }

    @Override
    public void removeOrphanedSubcontractedTaskData() {
        for (SubcontractedTaskData subcontractedTaskData : getAll()) {
            if (subcontractedTaskData.getTask() == null) {
                getSession().delete(subcontractedTaskData);
            }
        }
    }

    @Override
    public List<SubcontractedTaskData> getAllForMasterScenario() {
        Scenario masterScenario = PredefinedScenarios.MASTER.getScenario();

        List<SubcontractedTaskData> result = new ArrayList<>();
        for (Order order : masterScenario.getOrders().keySet()) {
            order.useSchedulingDataFor(masterScenario);
            for (OrderElement orderElement : order.getAllOrderElements()) {
                for (TaskElement taskElement : orderElement.getTaskElements()) {
                    if (taskElement.isSubcontracted()) {
                        result.add(((Task) taskElement).getSubcontractedTaskData());
                    }
                }
            }
        }

        return result;
    }

    @Override
    @Transactional(readOnly = true)
    public SubcontractedTaskData getSubcontractedTaskDataByOrderElement(OrderElement orderElement)
            throws InstanceNotFoundException {

        TaskElement taskElement = (TaskElement) getSession()
                .createCriteria(TaskElement.class)
                .createCriteria("taskSource", "ts")
                .createCriteria("schedulingData", "data")
                .add(Restrictions.eq("data.orderElement", orderElement))
                .uniqueResult();

        return (taskElement != null && taskElement.isSubcontracted())
                ? ((Task) taskElement).getSubcontractedTaskData()
                : null;
    }

}