package org.msh.tb.cases;

import org.jboss.seam.annotations.In;
import org.jboss.seam.annotations.Name;
import org.jboss.seam.annotations.RaiseEvent;
import org.jboss.seam.annotations.Transactional;
import org.jboss.seam.core.Events;
import org.jboss.seam.faces.FacesMessages;
import org.jboss.seam.framework.Controller;
import org.msh.tb.cases.treatment.TreatmentHome;
import org.msh.tb.entities.TbCase;
import org.msh.tb.entities.enums.CaseClassification;
import org.msh.tb.entities.enums.CaseState;
import org.msh.tb.entities.enums.DiagnosisType;
import org.msh.utils.date.Period;

import javax.persistence.EntityManager;
import java.util.Date;

@Name("caseCloseHome")
public class CaseCloseHome extends Controller{
    private static final long serialVersionUID = 805003311511687697L;

    @In(create=true)
    protected CaseHome caseHome;
    @In(create=true) FacesMessages facesMessages;
    @In(create=true) TreatmentHome treatmentHome;
    @In EntityManager entityManager;


    private Date date;
    private CaseState state;
    private String comment;

    protected static final CaseState[] outcomesMDR = {
            CaseState.CURED,
            CaseState.TREATMENT_COMPLETED,
            CaseState.FAILED,
            CaseState.DIED,
            CaseState.DEFAULTED,
            CaseState.NOT_EVALUATED,
            CaseState.OTHER
    };

    protected static final CaseState[] outcomesTB = {
            CaseState.CURED,
            CaseState.TREATMENT_COMPLETED,
            CaseState.FAILED,
            CaseState.DIED,
            CaseState.DEFAULTED,
            CaseState.NOT_EVALUATED,
            CaseState.OTHER
    };

    protected static final CaseState[] suspectOutcomes = {
            CaseState.NOT_TB,
            CaseState.DIED,
            CaseState.DEFAULTED,
            CaseState.OTHER
    };



    /**
     * Closes an opened case, setting its outcome
     * @return
     */
    @Transactional
    public String closeCase() {
        if (state == null) {
            return "errror";
        }

        TbCase tbcase = caseHome.getInstance();

        if (!validateClose())
            return "error";

        if ((tbcase.getTreatmentPeriod() != null) && (!tbcase.getTreatmentPeriod().isEmpty()))
            treatmentHome.cropTreatmentPeriod(new Period(tbcase.getTreatmentPeriod().getIniDate(), date));

        tbcase.setOutcomeDate(date);

        tbcase.setState(state);
        if (state.equals(CaseState.OTHER)) {
            tbcase.setOtherOutcome(comment);
        }
        else {
            tbcase.setOtherOutcome(null);
        }

        // save case changes
        caseHome.setTransactionLogActive(false);
        caseHome.persist();
        caseHome.updateCaseTags();

        Events.instance().raiseEvent("case.close");

        return "case-closed";
    }


    /**
     * Check if case can be closed
     * @return
     */
    public boolean validateClose() {
        TbCase tbcase = caseHome.getInstance();

        Date dt = tbcase.getDiagnosisDate();
        if ((dt != null) && (date.before(dt))) {
            facesMessages.addToControlFromResourceBundle("edtdate", "cases.close.msg1");
            return false;
        }

        if (tbcase.getTreatmentPeriod() != null && this.date.after(tbcase.getTreatmentPeriod().getEndDate())) {
            facesMessages.addToControlFromResourceBundle("edtdate", "cases.close.msg2");
            return false;
        }

        return true;
    }


    /**
     * Reopen a closed case
     * @return
     */
    @Transactional
    @RaiseEvent("case.reopen")
    public String reopenCase() {
        TbCase tbcase = caseHome.getInstance();

        if ((tbcase.getTreatmentPeriod() == null) || (tbcase.getTreatmentPeriod().isEmpty()))
            tbcase.setState(CaseState.WAITING_TREATMENT);
        else tbcase.setState(CaseState.ONTREATMENT);

        tbcase.setOtherOutcome(null);
        tbcase.setMovedSecondLineTreatment(false);

        caseHome.setTransactionLogActive(false);
        caseHome.persist();
        caseHome.updateCaseTags();

        return "case-reopened";
    }


    /**
     * Return the available outcomes of a case, according to its classification (TB or DR-TB)
     * @return Array of {@link CaseState} enumerations
     */
    public CaseState[] getOutcomes() {
        if (caseHome.getInstance().getDiagnosisType() == DiagnosisType.SUSPECT)
            return suspectOutcomes;

        if (caseHome.getInstance().getClassification() == CaseClassification.DRTB)
            return outcomesMDR;
        else return outcomesTB;
    }


    /**
     * get close date or default value - today
     * @return
     */
    public Date getDate() {
        return date;
    }

    public void setDate(Date date) {
        this.date = date;
    }

    public CaseState getState() {
        return state;
    }

    public void setState(CaseState state) {
        this.state = state;
    }

    public String getComment() {
        return comment;
    }

    public void setComment(String comment) {
        this.comment = comment;
    }


}