org.cgiar.ccafs.marlo.action.powb.FlagshipPlansAction.java Source code

Java tutorial

Introduction

Here is the source code for org.cgiar.ccafs.marlo.action.powb.FlagshipPlansAction.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.powb;

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.CrpProgramManager;
import org.cgiar.ccafs.marlo.data.manager.FileDBManager;
import org.cgiar.ccafs.marlo.data.manager.GlobalUnitManager;
import org.cgiar.ccafs.marlo.data.manager.LiaisonInstitutionManager;
import org.cgiar.ccafs.marlo.data.manager.PowbFlagshipPlansManager;
import org.cgiar.ccafs.marlo.data.manager.PowbSynthesisManager;
import org.cgiar.ccafs.marlo.data.manager.UserManager;
import org.cgiar.ccafs.marlo.data.model.CrpProgram;
import org.cgiar.ccafs.marlo.data.model.GlobalUnit;
import org.cgiar.ccafs.marlo.data.model.LiaisonInstitution;
import org.cgiar.ccafs.marlo.data.model.LiaisonUser;
import org.cgiar.ccafs.marlo.data.model.PowbFlagshipPlans;
import org.cgiar.ccafs.marlo.data.model.PowbSynthesis;
import org.cgiar.ccafs.marlo.data.model.ProgramType;
import org.cgiar.ccafs.marlo.data.model.User;
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.powb.FlagshipPlansValidator;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
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 Andres Valencia - CIAT/CCAFS
 */
public class FlagshipPlansAction extends BaseAction {

    private static final long serialVersionUID = -1417643537265271428L;

    // Managers
    private GlobalUnitManager crpManager;
    private LiaisonInstitutionManager liaisonInstitutionManager;
    private PowbSynthesisManager powbSynthesisManager;
    private CrpProgramManager crpProgramManager;
    private AuditLogManager auditLogManager;
    private UserManager userManager;
    private PowbFlagshipPlansManager powbFlagshipPlansManager;
    private FileDBManager fileDBManager;
    // Model for the front-end
    private PowbSynthesis powbSynthesis;
    private Long powbSynthesisID;
    private List<LiaisonInstitution> liaisonInstitutions;
    private String transaction;
    private LiaisonInstitution liaisonInstitution;
    private Long liaisonInstitutionID;
    private GlobalUnit loggedCrp;
    private FlagshipPlansValidator validator;

    @Inject
    public FlagshipPlansAction(APConfig config, GlobalUnitManager crpManager,
            LiaisonInstitutionManager liaisonInstitutionManager, AuditLogManager auditLogManager,
            CrpProgramManager crpProgramManager, UserManager userManager, PowbSynthesisManager powbSynthesisManager,
            FlagshipPlansValidator validator, PowbFlagshipPlansManager powbFlagshipPlansManager,
            FileDBManager fileDBManager) {
        super(config);
        this.crpManager = crpManager;
        this.liaisonInstitutionManager = liaisonInstitutionManager;
        this.crpProgramManager = crpProgramManager;
        this.auditLogManager = auditLogManager;
        this.userManager = userManager;
        this.powbSynthesisManager = powbSynthesisManager;
        this.validator = validator;
        this.powbFlagshipPlansManager = powbFlagshipPlansManager;
        this.fileDBManager = fileDBManager;
    }

    @Override
    public String cancel() {
        return SUCCESS;
    }

    private void createEmptyFlagshipPlan() {
        if (powbSynthesis.getPowbFlagshipPlans() == null && this.isFlagship()) {
            PowbFlagshipPlans newPowbFlagshipPlans = new PowbFlagshipPlans();
            newPowbFlagshipPlans.setPlanSummary("");
            newPowbFlagshipPlans.setPowbSynthesis(powbSynthesis);
            powbSynthesis.setPowbFlagshipPlans(newPowbFlagshipPlans);
            powbSynthesis = powbSynthesisManager.savePowbSynthesis(powbSynthesis);
        }

    }

    public Long firstFlagship() {
        List<LiaisonInstitution> liaisonInstitutions = new ArrayList<>(loggedCrp.getLiaisonInstitutions().stream()
                .filter(c -> c.getCrpProgram() != null && c.isActive()
                        && c.getCrpProgram().getProgramType() == ProgramType.FLAGSHIP_PROGRAM_TYPE.getValue())
                .collect(Collectors.toList()));
        liaisonInstitutions.sort(Comparator.comparing(LiaisonInstitution::getAcronym));
        long liaisonInstitutionId = liaisonInstitutions.get(0).getId();
        return liaisonInstitutionId;
    }

    private Path getAutoSaveFilePath() {
        String composedClassName = powbSynthesis.getClass().getSimpleName();
        String actionFile = this.getActionName().replace("/", "_");
        String autoSaveFile = powbSynthesis.getId() + "_" + composedClassName + "_"
                + this.getActualPhase().getDescription() + "_" + this.getActualPhase().getYear() + "_" + actionFile
                + ".json";
        return Paths.get(config.getAutoSaveFolder() + autoSaveFile);
    }

    public PowbFlagshipPlans getFlagshipPlansByliaisonInstitutionID(Long liaisonInstitutionID) {
        LiaisonInstitution liaisonInstitution = liaisonInstitutionManager
                .getLiaisonInstitutionById(liaisonInstitutionID);
        if (liaisonInstitution != null) {
            List<PowbSynthesis> powbSynthesisList = liaisonInstitution.getPowbSynthesis().stream().filter(
                    ps -> ps.isActive() && ps.getPhase() != null && ps.getPhase().equals(this.getActualPhase()))
                    .collect(Collectors.toList());
            if (powbSynthesisList != null && !powbSynthesisList.isEmpty()
                    && powbSynthesisList.get(0).getPowbFlagshipPlans() != null) {
                return powbSynthesisList.get(0).getPowbFlagshipPlans();
            }
        }
        return null;
    }

    public List<LiaisonInstitution> getFlagships() {
        List<LiaisonInstitution> flagshipsList = loggedCrp.getLiaisonInstitutions().stream()
                .filter(c -> c.getCrpProgram() != null
                        && c.getCrpProgram().getProgramType() == ProgramType.FLAGSHIP_PROGRAM_TYPE.getValue()
                        && c.isActive())
                .collect(Collectors.toList());
        if (flagshipsList != null) {
            flagshipsList.sort(Comparator.comparing(LiaisonInstitution::getAcronym));
            return flagshipsList;
        } else {
            return new ArrayList<>();
        }
    }

    public LiaisonInstitution getLiaisonInstitution() {
        return liaisonInstitution;
    }

    public Long getLiaisonInstitutionID() {
        return liaisonInstitutionID;
    }

    public List<LiaisonInstitution> getLiaisonInstitutions() {
        return liaisonInstitutions;
    }

    public GlobalUnit getLoggedCrp() {
        return loggedCrp;
    }

    // Method to download link file
    public String getPath(Long liaisonInstitutionID) {
        return config.getDownloadURL() + "/" + this.getPowbSourceFolder(liaisonInstitutionID).replace('\\', '/');
    }

    // Method to get the download folder
    private String getPowbSourceFolder(Long liaisonInstitutionID) {
        LiaisonInstitution liaisonInstitution = liaisonInstitutionManager
                .getLiaisonInstitutionById(liaisonInstitutionID);
        return APConstants.POWB_FOLDER.concat(File.separator).concat(this.getCrpSession()).concat(File.separator)
                .concat(liaisonInstitution.getAcronym()).concat(File.separator)
                .concat(this.getActionName().replace("/", "_")).concat(File.separator);
    }

    public PowbSynthesis getPowbSynthesis() {
        return powbSynthesis;
    }

    public Long getPowbSynthesisID() {
        return powbSynthesisID;
    }

    public String getTransaction() {
        return transaction;
    }

    public boolean isFlagship() {
        boolean isFP = false;
        if (liaisonInstitution.getCrpProgram() != null) {
            CrpProgram crpProgram = crpProgramManager
                    .getCrpProgramById(liaisonInstitution.getCrpProgram().getId().longValue());
            if (crpProgram.getProgramType() == ProgramType.FLAGSHIP_PROGRAM_TYPE.getValue()) {
                isFP = true;
            }
        }
        return isFP;
    }

    @Override
    public boolean isPMU() {
        boolean isFP = false;
        if (liaisonInstitution.getCrpProgram() == null) {
            isFP = true;
        }
        return isFP;
    }

    @Override
    public String next() {
        String result = this.save();
        if (result.equals(BaseAction.SUCCESS)) {
            return BaseAction.NEXT;
        } else {
            return result;
        }
    }

    @Override
    public void prepare() throws Exception {
        // Get current CRP
        loggedCrp = (GlobalUnit) this.getSession().get(APConstants.SESSION_CRP);
        loggedCrp = crpManager.getGlobalUnitById(loggedCrp.getId());
        // Check history version
        if (this.getRequest().getParameter(APConstants.TRANSACTION_ID) != null) {
            this.setPowbSynthesisIdHistory();
        } else {
            this.setPowbSynthesisParameters();
        }
        // Validate draft version
        if (powbSynthesis != null) {
            Path path = this.getAutoSaveFilePath();
            if (path.toFile().exists() && this.getCurrentUser().isAutoSave()) {
                this.readJsonAndLoadPowbSynthesis(path);
            } else {
                this.setDraft(false);
                this.createEmptyFlagshipPlan();
            }
        }

        // Get the list of liaison institutions Flagships and PMU.
        liaisonInstitutions = loggedCrp.getLiaisonInstitutions().stream()
                .filter(c -> c.getCrpProgram() != null
                        && c.getCrpProgram().getProgramType() == ProgramType.FLAGSHIP_PROGRAM_TYPE.getValue()
                        && c.isActive())
                .collect(Collectors.toList());
        liaisonInstitutions.addAll(loggedCrp.getLiaisonInstitutions().stream()
                .filter(c -> c.getCrpProgram() == null && c.getAcronym().equals("PMU") && c.isActive())
                .collect(Collectors.toList()));
        liaisonInstitutions.sort(Comparator.comparing(LiaisonInstitution::getAcronym));
        // Base Permission
        String params[] = { loggedCrp.getAcronym(), powbSynthesis.getId() + "" };
        this.setBasePermission(this.getText(Permission.POWB_SYNTHESIS_FLAGSHIPPLANS_BASE_PERMISSION, params));

        if (this.isHttpPost()) {
            if (powbSynthesis.getPowbFlagshipPlans() != null) {
                powbSynthesis.getPowbFlagshipPlans().setFlagshipProgramFile(null);
            }
        }
    }

    private void readJsonAndLoadPowbSynthesis(Path path) throws IOException {
        BufferedReader reader = null;
        reader = new BufferedReader(new FileReader(path.toFile()));
        Gson gson = new GsonBuilder().create();
        JsonObject jReader = gson.fromJson(reader, JsonObject.class);
        AutoSaveReader autoSaveReader = new AutoSaveReader();
        // We read the JSON serialized by the front-end and cast it to the object
        powbSynthesis = (PowbSynthesis) autoSaveReader.readFromJson(jReader);
        if (powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile() != null
                && powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId() != null) {
            powbSynthesis.getPowbFlagshipPlans().setFlagshipProgramFile(fileDBManager
                    .getFileDBById(powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId()));
        }
        powbSynthesisID = powbSynthesis.getId();
        this.setDraft(true);
        reader.close();
    }

    @Override
    public String save() {
        if (this.hasPermission("canEdit")) {
            this.saveUpdatePowbSynthesis();
            Path path = this.getAutoSaveFilePath();
            if (path.toFile().exists()) {
                path.toFile().delete();
            }
            if (!this.getInvalidFields().isEmpty()) {
                this.setActionMessages(null);
                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;
        }
    }

    private void saveUpdatePowbSynthesis() {
        PowbFlagshipPlans powbFlagshipPlan = powbSynthesis.getPowbFlagshipPlans();
        if (powbFlagshipPlan.getId() == null) {
            powbFlagshipPlan.setId(powbSynthesisID);
        }
        if (powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile() != null) {
            if (powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId() == null) {
                powbFlagshipPlan.setFlagshipProgramFile(null);
            } else {
                powbFlagshipPlan
                        .setFlagshipProgramFile(powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile());
            }
        }
        powbFlagshipPlan.setPlanSummary(powbSynthesis.getPowbFlagshipPlans().getPlanSummary());
        powbFlagshipPlan = powbFlagshipPlansManager.savePowbFlagshipPlans(powbFlagshipPlan);
        List<String> relationsName = new ArrayList<>();

        powbSynthesis = powbSynthesisManager.getPowbSynthesisById(powbSynthesisID);
        /**
         * The following is required because we need to update something on the @PowbSynthesis if we want a row created in
         * the auditlog table.
         */
        this.setModificationJustification(powbSynthesis);
        powbSynthesisManager.save(powbSynthesis, this.getActionName(), relationsName, this.getActualPhase());
    }

    public void setLiaisonInstitution(LiaisonInstitution liaisonInstitution) {
        this.liaisonInstitution = liaisonInstitution;
    }

    public void setLiaisonInstitutionID(Long liaisonInstitutionID) {
        this.liaisonInstitutionID = liaisonInstitutionID;
    }

    /**
     * Get Liaison institution ID Parameter
     */
    private void setLiaisonInstitutionIdParameter() {
        try {
            liaisonInstitutionID = Long.parseLong(
                    StringUtils.trim(this.getRequest().getParameter(APConstants.LIAISON_INSTITUTION_REQUEST_ID)));
        } catch (NumberFormatException e) {
            User user = userManager.getUser(this.getCurrentUser().getId());
            if (user.getLiasonsUsers() != null || !user.getLiasonsUsers().isEmpty()) {
                List<LiaisonUser> liaisonUsers = new ArrayList<>(user.getLiasonsUsers().stream()
                        .filter(lu -> lu.isActive() && lu.getLiaisonInstitution().isActive()
                                && lu.getLiaisonInstitution().getCrp().getId() == loggedCrp.getId()
                                && lu.getLiaisonInstitution().getInstitution() == null)
                        .collect(Collectors.toList()));
                if (!liaisonUsers.isEmpty()) {
                    boolean isLeader = false;
                    for (LiaisonUser liaisonUser : liaisonUsers) {
                        LiaisonInstitution institution = liaisonUser.getLiaisonInstitution();
                        if (institution.isActive()) {
                            if (institution.getCrpProgram() != null) {
                                if (institution.getCrpProgram()
                                        .getProgramType() == ProgramType.FLAGSHIP_PROGRAM_TYPE.getValue()) {
                                    liaisonInstitutionID = institution.getId();
                                    isLeader = true;
                                    break;
                                }
                            } else {
                                if (institution.getAcronym().equals("PMU")) {
                                    liaisonInstitutionID = institution.getId();
                                    isLeader = true;
                                    break;
                                }
                            }
                        }
                    }
                    if (!isLeader) {
                        liaisonInstitutionID = this.firstFlagship();
                    }
                } else {
                    liaisonInstitutionID = this.firstFlagship();
                }
            } else {
                liaisonInstitutionID = this.firstFlagship();
            }
        }
    }

    public void setLiaisonInstitutions(List<LiaisonInstitution> liaisonInstitutions) {
        this.liaisonInstitutions = liaisonInstitutions;
    }

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

    public void setPowbSynthesis(PowbSynthesis powbSynthesis) {
        this.powbSynthesis = powbSynthesis;
    }

    public void setPowbSynthesisID(Long powbSynthesisID) {
        this.powbSynthesisID = powbSynthesisID;
    }

    private void setPowbSynthesisIdHistory() {
        transaction = StringUtils.trim(this.getRequest().getParameter(APConstants.TRANSACTION_ID));
        PowbSynthesis history = (PowbSynthesis) auditLogManager.getHistory(transaction);
        if (history != null) {
            powbSynthesis = history;
            if (powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile() != null
                    && powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId() != null) {
                powbSynthesis.getPowbFlagshipPlans().setFlagshipProgramFile(fileDBManager
                        .getFileDBById(powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId()));
            }
            powbSynthesisID = powbSynthesis.getId();
            liaisonInstitutionID = powbSynthesis.getLiaisonInstitution().getId();
            liaisonInstitution = liaisonInstitutionManager.getLiaisonInstitutionById(liaisonInstitutionID);
        } else {
            this.transaction = null;
            this.setTransaction("-1");
        }
    }

    private void setPowbSynthesisIdParameter() {
        try {
            powbSynthesisID = Long
                    .parseLong(StringUtils.trim(this.getRequest().getParameter(APConstants.POWB_SYNTHESIS_ID)));
            powbSynthesis = powbSynthesisManager.getPowbSynthesisById(powbSynthesisID);

            if (!powbSynthesis.getPhase().equals(this.getActualPhase())) {
                powbSynthesis = powbSynthesisManager.findSynthesis(this.getActualPhase().getId(),
                        liaisonInstitutionID);
                if (powbSynthesis == null) {
                    powbSynthesis = this.createPowbSynthesis(this.getActualPhase().getId(), liaisonInstitutionID);
                }
                powbSynthesisID = powbSynthesis.getId();
            }
        } catch (NumberFormatException e) {
            PowbSynthesis powbSynthesis = powbSynthesisManager.findSynthesis(this.getActualPhase().getId(),
                    liaisonInstitutionID);
            if (powbSynthesis != null) {
                powbSynthesisID = powbSynthesis.getId();
            }
        }
    }

    private void setPowbSynthesisParameters() {
        this.setLiaisonInstitutionIdParameter();
        this.setPowbSynthesisIdParameter();
        liaisonInstitution = liaisonInstitutionManager.getLiaisonInstitutionById(liaisonInstitutionID);
        powbSynthesis = powbSynthesisManager.getPowbSynthesisById(powbSynthesisID);
    }

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

    @Override
    public void validate() {
        if (save) {
            if (powbSynthesis.getPowbFlagshipPlans() != null
                    && powbSynthesis.getPowbFlagshipPlans().getId() != null) {
                if (powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile() != null
                        && powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId() == null
                        || powbSynthesis.getPowbFlagshipPlans().getFlagshipProgramFile().getId()
                                .longValue() == -1) {
                    powbSynthesis.getPowbFlagshipPlans().setFlagshipProgramFile(null);
                }
            }
            validator.validate(this, powbSynthesis, true);
        }
    }

}