/**
 *
 */
package org.msh.tb.cases;

import org.jboss.seam.Component;
import org.jboss.seam.annotations.In;
import org.jboss.seam.annotations.Name;
import org.jboss.seam.faces.FacesMessages;
import org.msh.tb.cases.exams.MedicalExaminationHome;
import org.msh.tb.entities.MedicalExamination;
import org.msh.tb.entities.TbCase;
import org.msh.tb.entities.enums.CaseClassification;
import org.msh.tb.entities.enums.DiagnosisType;
import org.msh.tb.entities.enums.InfectionSite;
import org.msh.tb.entities.enums.RoleAction;
import org.msh.tb.ng.entities.MedicalExaminationNG;
import org.msh.tb.ng.entities.TbCaseNG;
import org.msh.tb.ng.entities.enums.HIVPosition;

/**
 * Controller class to handle follow-up of suspect cases. Basically there are two
 * main methods that handle suspect follow-up: <p/>
 *  {@link SuspectFollowupHome#registerConfirmedFollowup()} for not confirmed cases and
 *  {@link SuspectFollowupHome#registerNotConfirmedFollowup()} for confirmed cases
 * <p/>
 * Depending on the follow-up of the suspect, other information must be provided.
 *
 * @author Ricardo Memoria
 *
 */
@Name("suspectFollowupHome")
public class SuspectFollowupHome {

    @In CaseHome caseHome;

    private TbCase dataModel;

    /**
     * Initialize the data model with default values
     */
    public void createDataModel() {
        dataModel = new TbCaseNG();

        TbCase tbcase = caseHome.getInstance();
        dataModel.setClassification( tbcase.getClassification() );
    }

    /**
     * Register the follow-up of the suspect case when the outcome will be as not confirmed
     */
    public String registerNotConfirmedFollowup() {
        CaseCloseHome caseCloseHome = (CaseCloseHome)Component.getInstance("caseCloseHome");

        if ("case-closed".equals(caseCloseHome.closeCase())) {
            return "followup-registered";
        }
        else {
            return "error";
        }
    }


    /**
     * Register the follow-up of the suspect case when the case is confirmed
     */
    public String registerConfirmedFollowup() {
        if (dataModel == null)
            throw new IllegalArgumentException("dataModel cannot be null");

        if (dataModel.getInfectionSite() == InfectionSite.PULMONARY) {
            dataModel.setExtrapulmonaryType(null);
            dataModel.setExtrapulmonaryType2(null);
        }
        else {
            if ((dataModel.getExtrapulmonaryType() == null) && (dataModel.getPulmonaryType() == null)) {
                FacesMessages.instance().addToControlFromResourceBundle("cbextrapulmonary", "javax.faces.component.UIInput.REQUIRED");
                return "error";
            }
            dataModel.setPulmonaryType(null);
        }

        if ((dataModel.getClassification() == CaseClassification.DRTB) && (dataModel.getDrugResistanceType() == null)) {
            FacesMessages.instance().addToControlFromResourceBundle("cbresistance", "javax.faces.component.UIInput.REQUIRED");
            return "error";
        }

        if (caseHome.getInstance() instanceof TbCaseNG) {
            TbCaseNG tbCaseNG = (TbCaseNG) caseHome.getInstance();
            if (HIVPosition.POSITIVE.equals(tbCaseNG.getHivPosition()) && tbCaseNG.getHivPositionDetail() == null) {
                FacesMessages.instance().addToControlFromResourceBundle("hivdetail", "javax.faces.component.UIInput.REQUIRED");
                return "error";
            }
        }
        if ((dataModel.getClassification() == CaseClassification.TB) && (dataModel.getDrugResistanceType() != null)) {
            dataModel.setDrugResistanceType(null);
        }

        caseHome.initTransactionLog(RoleAction.EDIT);
        caseHome.setTransactionLogActive(true);

        TbCaseNG tbcase = (TbCaseNG)caseHome.getInstance();
        tbcase.setDiagnosisType(DiagnosisType.CONFIRMED);
        tbcase.setClassification(dataModel.getClassification());
        tbcase.setDiagnosisDate(dataModel.getDiagnosisDate());
        tbcase.setPatientType(dataModel.getPatientType());
        tbcase.setDrugResistanceType(dataModel.getDrugResistanceType());
        tbcase.setInfectionSite(dataModel.getInfectionSite());
        tbcase.setPulmonaryType(dataModel.getPulmonaryType());
        tbcase.setExtrapulmonaryType(dataModel.getExtrapulmonaryType());
        tbcase.setExtrapulmonaryType2(dataModel.getExtrapulmonaryType2());

        caseHome.setRoleName("SUSPECT_FOLLOWUP");

        updateWeight();

        if ("persisted".equals(caseHome.persist()))
            return "followup-registered";
        else return "error";
    }


    /**
     * Update the weight of the patient
     */
    private void updateWeight() {
        MedicalExamination medexam = (MedicalExamination) Component.getInstance("medicalExamination");
        if (medexam != null && medexam.getWeight() != null) {
            medexam.setTbcase(caseHome.getTbCase());
            if (medexam.getDate() == null) {
                medexam.setDate(caseHome.getTbCase().getDiagnosisDate());
            }
            caseHome.getEntityManager().persist(medexam);
        }
    }

    /**
     * Return the instance of {@link TbCase} that will contain the changes from the UI
     * @return the dataModel as instance of {@link TbCase}
     */
    public TbCase getDataModel() {
        if (dataModel == null) {
            createDataModel();
        }
        return dataModel;
    }
}