SubcontractModel.java example

Explorer
libreplan-master
/*
 * 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.web.planner.order;

import java.util.Date;
import java.util.List;
import java.util.SortedSet;
import java.util.TreeSet;

import org.joda.time.LocalDate;
import org.libreplan.business.externalcompanies.daos.IExternalCompanyDAO;
import org.libreplan.business.externalcompanies.entities.EndDateCommunication;
import org.libreplan.business.externalcompanies.entities.ExternalCompany;
import org.libreplan.business.planner.daos.ISubcontractedTaskDataDAO;
import org.libreplan.business.planner.entities.SubcontractState;
import org.libreplan.business.planner.entities.SubcontractedTaskData;
import org.libreplan.business.planner.entities.SubcontractorDeliverDate;
import org.libreplan.business.planner.entities.Task;
import org.libreplan.business.workingday.IntraDayDate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

/**
 * Model for UI operations related with subcontract process and {@link SubcontractedTaskData} entity.
 *
 * @author Manuel Rego Casasnovas <mrego@igalia.com>
 * @author Susana Montes Pedreira <smontes@wirelessgalicia.com>
 */
@Service
@Scope(BeanDefinition.SCOPE_PROTOTYPE)
public class SubcontractModel implements ISubcontractModel {

    /**
     * Conversation state
     */
    private Task task;

    private org.zkoss.ganttz.data.Task ganttTask;

    private IntraDayDate startDate;

    private IntraDayDate endDate;

    private SubcontractedTaskData subcontractedTaskData;

    private SubcontractedTaskData currentSubcontractedTaskData;

    @Autowired
    private IExternalCompanyDAO externalCompanyDAO;

    @Autowired
    private ISubcontractedTaskDataDAO subcontractedTaskDataDAO;

    @Override
    @Transactional(readOnly = true)
    public void init(Task task, org.zkoss.ganttz.data.Task ganttTask) {
        this.task = task;
        this.startDate = task.getIntraDayStartDate();
        this.endDate = task.getIntraDayEndDate();

        this.ganttTask = ganttTask;

        SubcontractedTaskData subcontractedTaskData = task.getSubcontractedTaskData();

        this.currentSubcontractedTaskData = subcontractedTaskData;

        if (subcontractedTaskData == null) {
            this.subcontractedTaskData = SubcontractedTaskData.create(task);

            if (task.getDeadline() != null) {
                this.addDeliverDate(task.getDeadline().toDateTimeAtStartOfDay().toDate());
            } else {
                this.addDeliverDate(task.getEndDate());
            }
        } else {
            subcontractedTaskDataDAO.reattach(subcontractedTaskData);
            loadRequiredDeliveringDates(subcontractedTaskData);
            loadAskedEndDatesFromSubcontractor(subcontractedTaskData);
            this.subcontractedTaskData = SubcontractedTaskData.createFrom(subcontractedTaskData);
        }
    }

    private void loadAskedEndDatesFromSubcontractor(SubcontractedTaskData subcontractedTaskData) {
        if (subcontractedTaskData != null) {
            for (EndDateCommunication askedEndDate : subcontractedTaskData.getEndDatesCommunicatedFromSubcontractor()) {
                askedEndDate.getEndDate();
            }
        }
    }

    private void loadRequiredDeliveringDates(SubcontractedTaskData subcontractedTaskData){
        if(subcontractedTaskData != null){
            for (SubcontractorDeliverDate subDeliverDate : subcontractedTaskData.getRequiredDeliveringDates()) {
                subDeliverDate.getSaveDate();
            }
        }
    }

    @Override
    public SubcontractedTaskData getSubcontractedTaskData() {
        return subcontractedTaskData;
    }

    @Override
    @Transactional(readOnly = true)
    public void confirm() {
        if (task != null) {
            if (subcontractedTaskData == null) {
                task.setSubcontractedTaskData(null);
            } else {
                subcontractedTaskDataDAO.save(subcontractedTaskData);

                if (currentSubcontractedTaskData == null) {
                    task.setSubcontractedTaskData(subcontractedTaskData);
                    currentSubcontractedTaskData = subcontractedTaskData;
                } else {
                    currentSubcontractedTaskData.applyChanges(subcontractedTaskData);
                }

                task.removeAllSatisfiedResourceAllocations();
                task.setDeadline(new LocalDate(currentSubcontractedTaskData.getLastRequiredDeliverDate()));
                Task.convertOnStartInFixedDate(task);
            }

            recalculateTaskLength();
        }
    }

    private void recalculateTaskLength() {
        task.setIntraDayStartDate(startDate);
        task.setIntraDayEndDate(endDate);

        ganttTask.enforceDependenciesDueToPositionPotentiallyModified();
    }

    @Override
    public void cancel() {
        task = null;
        startDate = null;
        endDate = null;
        currentSubcontractedTaskData = null;
        subcontractedTaskData = null;
    }

    @Override
    @Transactional(readOnly = true)
    public List<ExternalCompany> getSubcontractorExternalCompanies() {
        return externalCompanyDAO.findSubcontractor();
    }

    @Override
    public void setExternalCompany(ExternalCompany externalCompany) {
        if (subcontractedTaskData != null) {
            subcontractedTaskData.setExternalCompany(externalCompany);
        }
    }

    @Override
    public boolean hasResourceAllocations() {
        return task != null && !task.getSatisfiedResourceAllocations().isEmpty();
    }

    @Override
    public void setEndDate(Date endDate) {
        this.endDate = IntraDayDate.startOfDay(LocalDate.fromDateFields(endDate));
    }

    @Override
    public void removeSubcontractedTaskData() {
        subcontractedTaskData = null;
    }

    @Override
    public SortedSet<SubcontractorDeliverDate> getDeliverDates(){
        if(subcontractedTaskData != null){
            return subcontractedTaskData.getRequiredDeliveringDates();
        }

        return new TreeSet<>();
    }

    @Override
    public void addDeliverDate(Date subDeliverDate){
        if(subcontractedTaskData != null){
            SubcontractorDeliverDate subcontractorDeliverDate = SubcontractorDeliverDate
                    .create(new Date(), subDeliverDate, null);
            subcontractedTaskData.addRequiredDeliveringDates(subcontractorDeliverDate);

            // Update the state of the subcontracted task data
            updateStateToPendingUpdateDeliveringDate();
        }
    }

    private void updateStateToPendingUpdateDeliveringDate(){
        if ((subcontractedTaskData.getState() != null) &&
                (subcontractedTaskData.getState().equals(SubcontractState.SUCCESS_SENT))) {
            subcontractedTaskData.setState(SubcontractState.PENDING_UPDATE_DELIVERING_DATE);
        }
    }

    @Override
    public boolean alreadyExistsRepeatedDeliverDate(Date newDeliverDate) {
        if (this.getSubcontractedTaskData().getRequiredDeliveringDates().isEmpty()) {
            return false;
        }

        SubcontractorDeliverDate currentSubDeliverDate = this
                .getSubcontractedTaskData().getRequiredDeliveringDates()
                .first();

        Date deliverDate = new LocalDate(newDeliverDate)
                .toDateTimeAtStartOfDay().toDate();
        Date currentDeliverDate = new LocalDate(
                currentSubDeliverDate.getSubcontractorDeliverDate())
                .toDateTimeAtStartOfDay().toDate();

        return currentDeliverDate.compareTo(deliverDate) == 0;
    }

    @Override
    public void removeRequiredDeliverDate(SubcontractorDeliverDate subcontractorDeliverDate) {
        if(subcontractedTaskData != null){
            subcontractedTaskData.removeRequiredDeliveringDates(subcontractorDeliverDate);
            updateStateFromPendingDeliveringDateToSuccessSent();
        }
    }

    private void updateStateFromPendingDeliveringDateToSuccessSent(){
        if (subcontractedTaskData.getState() != null) {
            switch (subcontractedTaskData.getState()) {

                case PENDING_UPDATE_DELIVERING_DATE:
                case FAILED_UPDATE:
                    subcontractedTaskData.setState(SubcontractState.SUCCESS_SENT);
                    break;

                default:
                    break;
            }
        }
    }

    @Override
    public SortedSet<EndDateCommunication> getAskedEndDates() {
        if (subcontractedTaskData != null) {
            return subcontractedTaskData.getEndDatesCommunicatedFromSubcontractor();
        }

        return new TreeSet<>();
    }

}