org.cgiar.ccafs.marlo.action.projects.ProjectBudgetByFlagshipAction.java Source code

Java tutorial

Introduction

Here is the source code for org.cgiar.ccafs.marlo.action.projects.ProjectBudgetByFlagshipAction.java

Source

/*****************************************************************
 * This file is part of Managing Agricultural Research for Learning &
 * Outcomes Platform (MARLO).
 * MARLO 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.
 * MARLO 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 MARLO. If not, see <http://www.gnu.org/licenses/>.
 *****************************************************************/

package org.cgiar.ccafs.marlo.action.projects;

import org.cgiar.ccafs.marlo.action.BaseAction;
import org.cgiar.ccafs.marlo.config.APConstants;
import org.cgiar.ccafs.marlo.data.manager.AuditLogManager;
import org.cgiar.ccafs.marlo.data.manager.BudgetTypeManager;
import org.cgiar.ccafs.marlo.data.manager.CrpProgramManager;
import org.cgiar.ccafs.marlo.data.manager.GlobalUnitManager;
import org.cgiar.ccafs.marlo.data.manager.PhaseManager;
import org.cgiar.ccafs.marlo.data.manager.ProjectBudgetsFlagshipManager;
import org.cgiar.ccafs.marlo.data.manager.ProjectFocusManager;
import org.cgiar.ccafs.marlo.data.manager.ProjectManager;
import org.cgiar.ccafs.marlo.data.model.BudgetType;
import org.cgiar.ccafs.marlo.data.model.CrpProgram;
import org.cgiar.ccafs.marlo.data.model.GlobalUnit;
import org.cgiar.ccafs.marlo.data.model.Phase;
import org.cgiar.ccafs.marlo.data.model.ProgramType;
import org.cgiar.ccafs.marlo.data.model.Project;
import org.cgiar.ccafs.marlo.data.model.ProjectBudget;
import org.cgiar.ccafs.marlo.data.model.ProjectBudgetsFlagship;
import org.cgiar.ccafs.marlo.data.model.ProjectFocus;
import org.cgiar.ccafs.marlo.security.APCustomRealm;
import org.cgiar.ccafs.marlo.security.Permission;
import org.cgiar.ccafs.marlo.utils.APConfig;
import org.cgiar.ccafs.marlo.utils.AutoSaveReader;
import org.cgiar.ccafs.marlo.validation.projects.ProjectBudgetsFlagshipValidator;

import java.io.BufferedReader;
import java.io.FileReader;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import javax.inject.Inject;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonObject;
import org.apache.commons.lang3.StringUtils;

/**
 * @author Hermes Jimnez - CIAT/CCAFS
 */
public class ProjectBudgetByFlagshipAction extends BaseAction {

    private static final long serialVersionUID = 8935913606376386782L;

    private CrpProgramManager crpProgramManager;

    private BudgetTypeManager budgetTypeManager;

    private ProjectManager projectManager;

    private ProjectBudgetsFlagshipManager projectBudgetsFlagshipManager;

    private AuditLogManager auditLogManager;

    private GlobalUnitManager crpManager;

    private long projectID;

    private GlobalUnit loggedCrp;

    private Project project;

    private String transaction;

    private List<BudgetType> budgetTypesList;

    private ProjectBudgetsFlagshipValidator validator;
    private ProjectFocusManager projectFocusManager;
    private PhaseManager phaseManager;

    @Inject
    public ProjectBudgetByFlagshipAction(APConfig config, CrpProgramManager crpProgramManager,
            BudgetTypeManager budgetTypeManager, ProjectManager projectManager,
            ProjectBudgetsFlagshipManager projectBudgetsFlagshipManager, AuditLogManager auditLogManager,
            GlobalUnitManager crpManager, ProjectBudgetsFlagshipValidator validator,
            ProjectFocusManager projectFocusManager, PhaseManager phaseManager) {
        super(config);
        this.crpProgramManager = crpProgramManager;
        this.budgetTypeManager = budgetTypeManager;
        this.projectManager = projectManager;
        this.projectBudgetsFlagshipManager = projectBudgetsFlagshipManager;
        this.auditLogManager = auditLogManager;
        this.crpManager = crpManager;
        this.validator = validator;
        this.phaseManager = phaseManager;
        this.projectFocusManager = projectFocusManager;

    }

    @Override
    public String cancel() {
        Path path = this.getAutoSaveFilePath();

        if (path.toFile().exists()) {
            boolean fileDeleted = path.toFile().delete();
        }

        this.setDraft(false);
        Collection<String> messages = this.getActionMessages();
        if (!messages.isEmpty()) {
            String validationMessage = messages.iterator().next();
            this.setActionMessages(null);
            this.addActionMessage("draft:" + this.getText("cancel.autoSave"));
        } else {
            this.addActionMessage("draft:" + this.getText("cancel.autoSave"));
        }
        messages = this.getActionMessages();

        return SUCCESS;
    }

    /**
     * This method clears the cache and re-load the user permissions in the next iteration.
     */
    @Override
    public void clearPermissionsCache() {
        ((APCustomRealm) securityContext.getRealm())
                .clearCachedAuthorizationInfo(securityContext.getSubject().getPrincipals());
    }

    public boolean existOnYear(Long focusID, int year) {

        Phase phase = phaseManager.findCycle(this.getActualPhase().getDescription(), year, this.getCrpID());
        if (phase == null) {
            phase = phaseManager.findCycle(APConstants.PLANNING, APConstants.FIRST_YEAR, this.getCrpID());
        }
        if (phase != null) {
            List<ProjectFocus> focus = phase.getProjectFocuses().stream()
                    .filter(c -> c.getProject().getId().longValue() == projectID
                            && focusID.equals(c.getCrpProgram().getId()) && c.isActive())
                    .collect(Collectors.toList());
            if (!focus.isEmpty()) {
                return true;
            }

        }
        return false;
    }

    private Path getAutoSaveFilePath() {
        String composedClassName = project.getClass().getSimpleName();
        // get the action name and replace / for _
        String actionFile = this.getActionName().replace("/", "_");
        // concatane name and add the .json extension
        String autoSaveFile = project.getId() + "_" + composedClassName + "_"
                + this.getActualPhase().getDescription() + "_" + this.getActualPhase().getYear() + "_" + actionFile
                + ".json";

        return Paths.get(config.getAutoSaveFolder() + autoSaveFile);
    }

    /**
     * @param programID
     * @param year
     * @param type
     * @return
     */
    public ProjectBudgetsFlagship getBudget(long programID, int year, long type) {
        if (project.getBudgetsFlagship() == null) {
            project.setBudgetsFlagship(new ArrayList<>());
        }
        return project.getBudgetsFlagship().get(this.getIndexBudget(programID, year, type));
    }

    public List<BudgetType> getBudgetTypesList() {
        return budgetTypesList;
    }

    /**
     * @param programID
     * @param year
     * @param type
     * @return
     */
    public int getIndexBudget(long programID, int year, long type) {

        if (project.getBudgetsFlagship() != null) {
            int i = 0;
            for (ProjectBudgetsFlagship projectBudget : project.getBudgetsFlagship()) {
                if (projectBudget.getCrpProgram() != null) {
                    if (projectBudget.getCrpProgram().getId() == programID && year == projectBudget.getYear()
                            && type == projectBudget.getBudgetType().getId().longValue()) {
                        return i;
                    }

                }
                i++;
            }

        } else {
            project.setBudgetsCluserActvities(new ArrayList<>());
        }

        ProjectBudgetsFlagship projectBudget = new ProjectBudgetsFlagship();
        projectBudget.setCrpProgram(crpProgramManager.getCrpProgramById(programID));
        projectBudget.setYear(year);
        projectBudget.setBudgetType(budgetTypeManager.getBudgetTypeById(type));
        project.getBudgetsFlagship().add(projectBudget);

        return this.getIndexBudget(programID, year, type);
    }

    public GlobalUnit getLoggedCrp() {
        return loggedCrp;
    }

    public Project getProject() {
        return project;
    }

    public long getProjectID() {
        return projectID;
    }

    /**
     * @param type
     * @param year
     * @return
     * @throws ParseException
     */
    public double getRemaining(Long type, int year) throws ParseException {
        double remaining = 100;

        if (project.getBudgetsFlagship() != null) {
            for (ProjectBudgetsFlagship budgetsFlagship : project.getBudgetsFlagship()) {
                if (budgetsFlagship.getYear() == year
                        && budgetsFlagship.getBudgetType().getId().longValue() == type.longValue()) {
                    if (budgetsFlagship.getAmount() != null) {

                        remaining = remaining - budgetsFlagship.getAmount().doubleValue();
                        remaining = this.round(remaining, 2);
                    }
                }
            }
        }

        return remaining;
    }

    /**
     * @param budget type
     * @param year
     * @return
     */
    public Long getTotalAmount(Long type, int year) {

        long totalAmount = 0;
        double porcentage;
        try {
            porcentage = Math.abs(this.getRemaining(type, year) - 100);
            totalAmount = (long) (this.getTotalYearPartners(year, type) * (porcentage / 100));
            return totalAmount;
        } catch (ParseException e) {
            return new Long(0);
        }

    }

    /**
     * @param year
     * @param type
     * @return
     */
    public double getTotalYearPartners(int year, long type) {
        double total = 0;
        Project projectBD = projectManager.getProjectById(projectID);

        for (ProjectBudget projectBudget : projectBD.getProjectBudgets()) {
            if (year == projectBudget.getYear() && type == projectBudget.getBudgetType().getId().longValue()
                    && projectBudget.getPhase().equals(this.getActualPhase()) && projectBudget.isActive()) {
                if (projectBudget.getAmount() != null) {
                    total = total + projectBudget.getAmount();
                }

            }

        }
        return total;
    }

    public String getTransaction() {
        return transaction;
    }

    /**
     * @param type
     * @param year
     * @return
     */
    public boolean hasBudgets(Long type, int year) {
        Project projectBD = projectManager.getProjectById(projectID);
        List<ProjectBudget> budgets = projectBD.getProjectBudgets().stream()
                .filter(c -> c.isActive() && c.getYear() == year && c.getPhase().equals(this.getActualPhase())
                        && c.getBudgetType().getId().longValue() == type.longValue()
                        && (c.getAmount() != null && c.getAmount() >= 0))
                .collect(Collectors.toList());

        return budgets.size() > 0;
    }

    @Override
    public void prepare() throws Exception {
        projectID = Long
                .parseLong(StringUtils.trim(this.getRequest().getParameter(APConstants.PROJECT_REQUEST_ID)));
        loggedCrp = (GlobalUnit) this.getSession().get(APConstants.SESSION_CRP);
        loggedCrp = crpManager.getGlobalUnitById(loggedCrp.getId());

        // Budget Types list
        budgetTypesList = budgetTypeManager.findAll();

        // Transaction Audilog
        if (this.getRequest().getParameter(APConstants.TRANSACTION_ID) != null) {

            transaction = StringUtils.trim(this.getRequest().getParameter(APConstants.TRANSACTION_ID));
            Project history = (Project) auditLogManager.getHistory(transaction);

            if (history != null) {
                project = history;
            } else {
                this.transaction = null;

                this.setTransaction("-1");
            }

        } else {
            project = projectManager.getProjectById(projectID);
        }
        if (project != null) {
            Path path = this.getAutoSaveFilePath();

            if (path.toFile().exists() && this.getCurrentUser().isAutoSave()) {

                BufferedReader reader = null;

                reader = new BufferedReader(new FileReader(path.toFile()));

                Gson gson = new GsonBuilder().create();

                JsonObject jReader = gson.fromJson(reader, JsonObject.class);
                reader.close();

                AutoSaveReader autoSaveReader = new AutoSaveReader();

                project = (Project) autoSaveReader.readFromJson(jReader);
                Project projectDb = projectManager.getProjectById(project.getId());
                project.setProjectInfo(projectDb.getProjecInfoPhase(this.getActualPhase()));
                project.getProjectInfo().setProjectEditLeader(
                        projectDb.getProjecInfoPhase(this.getActualPhase()).isProjectEditLeader());
                reader.close();
                this.setDraft(true);

            } else {

                this.setDraft(false);
                project.setProjectInfo(project.getProjecInfoPhase(this.getActualPhase()));
                // Setup the Project Budgets Flaship list
                project.setBudgetsFlagship(project.getProjectBudgetsFlagships().stream()
                        .filter(c -> c.isActive() && c.getPhase().equals(this.getActualPhase()))
                        .collect(Collectors.toList()));
            }

            project.getBudgetsFlagship().removeIf(Objects::isNull);

            Project projectBD = projectManager.getProjectById(projectID);
            project.setProjectInfo(projectBD.getProjecInfoPhase(this.getActualPhase()));

            // List the Project flagships that contribute it
            List<CrpProgram> flagships = new ArrayList<>();
            List<ProjectFocus> projectFocuses = new ArrayList<>(projectBD.getProjectFocuses().stream()
                    .filter(pf -> pf.isActive() && pf.getPhase().equals(this.getActualPhase())
                            && pf.getCrpProgram().getProgramType() == ProgramType.FLAGSHIP_PROGRAM_TYPE.getValue())
                    .collect(Collectors.toList()));
            for (ProjectFocus projectFocus : projectFocuses) {
                flagships.add(projectFocus.getCrpProgram());
            }
            project.setFlagships(flagships);

            // Check Null Crp program Values inside Project Budget
            if (project.getBudgetsFlagship() != null) {
                for (ProjectBudgetsFlagship projectBudgetsFlagship : project.getBudgetsFlagship()) {

                    // This additional null check is required for auto save file after a CoA has been deleted from the project
                    if (projectBudgetsFlagship.getCrpProgram() != null) {
                        projectBudgetsFlagship.setCrpProgram(crpProgramManager
                                .getCrpProgramById(projectBudgetsFlagship.getCrpProgram().getId()));
                    }

                }
            }

        }

        // Base Permission
        String params[] = { loggedCrp.getAcronym(), project.getId() + "" };
        this.setBasePermission(this.getText(Permission.PROJECT_BUDGET_FLAGSHIP_BASE_PERMISSION, params));

        if (this.isHttpPost()) {
            if (project.getBudgetsFlagship() != null) {
                project.getBudgetsFlagship().clear();
            }
        }

    }

    /**
     * Roun up the value that the user put in the value field
     * 
     * @param value
     * @param places
     * @return round number value
     */
    public double round(double value, int places) {
        if (places < 0) {
            throw new IllegalArgumentException();
        }

        BigDecimal bd = new BigDecimal(value);
        bd = bd.setScale(places, RoundingMode.HALF_UP);
        return bd.doubleValue();
    }

    @Override
    public String save() {
        if (this.hasPermission("canEdit")) {
            this.saveBasicBudgets();

            List<String> relationsName = new ArrayList<>();
            relationsName.add(APConstants.PROJECT_BUDGETS_FLAGSHIP_RELATION);
            relationsName.add(APConstants.PROJECT_INFO_RELATION);

            project = projectManager.getProjectById(projectID);
            /**
             * The following is required because we need to update something on the @Project if we want a row created in
             * the auditlog table.
             */
            this.setModificationJustification(project);
            projectManager.saveProject(project, this.getActionName(), relationsName, this.getActualPhase());
            Path path = this.getAutoSaveFilePath();

            if (path.toFile().exists()) {
                path.toFile().delete();
            }
            Collection<String> messages = this.getActionMessages();
            if (!this.getInvalidFields().isEmpty()) {
                this.setActionMessages(null);
                // this.addActionMessage(Map.toString(this.getInvalidFields().toArray()));
                List<String> keys = new ArrayList<String>(this.getInvalidFields().keySet());
                for (String key : keys) {
                    this.addActionMessage(key + ": " + this.getInvalidFields().get(key));
                }

            } else {
                this.addActionMessage("message:" + this.getText("saving.saved"));
            }
            return SUCCESS;
        } else {

            return NOT_AUTHORIZED;
        }
    }

    /**
     * 
     */
    public void saveBasicBudgets() {
        Project projectDB = projectManager.getProjectById(projectID);

        for (ProjectBudgetsFlagship projectBudget : projectDB.getProjectBudgetsFlagships().stream()
                .filter(c -> c.isActive() && c.getPhase().equals(this.getActualPhase()))
                .collect(Collectors.toList())) {

            if (project.getBudgetsFlagship() == null) {
                project.setBudgetsFlagship(new ArrayList<>());
            }
            if (projectBudget.getYear() == this.getCurrentCycleYear()) {
                if (!project.getBudgetsFlagship().contains(projectBudget)) {
                    projectBudgetsFlagshipManager.deleteProjectBudgetsFlagship(projectBudget.getId());

                }

            }
        }
        if (project.getBudgetsFlagship() != null) {
            for (ProjectBudgetsFlagship budgetsFlagshipUI : project.getBudgetsFlagship()) {
                if (budgetsFlagshipUI != null) {
                    if (budgetsFlagshipUI.getId() == null) {

                        budgetsFlagshipUI.setProject(project);
                        budgetsFlagshipUI.setPhase(this.getActualPhase());

                        if (budgetsFlagshipUI.getCrpProgram() != null) {
                            projectBudgetsFlagshipManager.saveProjectBudgetsFlagship(budgetsFlagshipUI);
                        }

                    } else {
                        ProjectBudgetsFlagship ProjectBudgetDB = projectBudgetsFlagshipManager
                                .getProjectBudgetsFlagshipById(budgetsFlagshipUI.getId());
                        ProjectBudgetDB.setAmount(budgetsFlagshipUI.getAmount());
                        budgetsFlagshipUI.setPhase(this.getActualPhase());
                        budgetsFlagshipUI.setProject(project);
                        budgetsFlagshipUI = projectBudgetsFlagshipManager
                                .saveProjectBudgetsFlagship(budgetsFlagshipUI);
                    }

                }

            }

        }
    }

    public void setBudgetTypesList(List<BudgetType> budgetTypesList) {
        this.budgetTypesList = budgetTypesList;
    }

    public void setLoggedCrp(GlobalUnit loggedCrp) {
        this.loggedCrp = loggedCrp;
    }

    public void setProject(Project project) {
        this.project = project;
    }

    public void setProjectID(long projectID) {
        this.projectID = projectID;
    }

    public void setTransaction(String transaction) {
        this.transaction = transaction;
    }

    @Override
    public void validate() {
        if (save) {
            validator.validate(this, project, true);
        }
    }

}