OpenSRP/opensrp-client-anc

package org.smartregister.anc.library.interactor;

import androidx.annotation.VisibleForTesting;

import org.apache.commons.lang3.StringUtils;
import org.jeasy.rules.api.Facts;
import org.json.JSONObject;
import org.smartregister.anc.library.AncLibrary;
import org.smartregister.anc.library.contract.BaseContactContract;
import org.smartregister.anc.library.contract.ContactSummarySendContract;
import org.smartregister.anc.library.model.ContactSummaryModel;
import org.smartregister.anc.library.repository.PatientRepository;
import org.smartregister.anc.library.util.ANCJsonFormUtils;
import org.smartregister.anc.library.util.AppExecutors;
import org.smartregister.anc.library.util.ConstantsUtils;
import org.smartregister.anc.library.util.DBConstantsUtils;
import org.smartregister.anc.library.util.Utils;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import timber.log.Timber;

public class ContactSummaryInteractor extends BaseContactInteractor implements ContactSummarySendContract.Interactor {

    private ANCJsonFormUtils formUtils = new ANCJsonFormUtils();

    public ContactSummaryInteractor() {
        this(new AppExecutors());
    }

    @VisibleForTesting
    ContactSummaryInteractor(AppExecutors appExecutors) {
        super(appExecutors);
    }

    @Override
    public void fetchWomanDetails(String baseEntityId, BaseContactContract.InteractorCallback callBack) {
        super.fetchWomanDetails(baseEntityId, callBack);
    }


    @Override
    public void fetchUpcomingContacts(final String entityId, final String referralContactNo,
                                      final ContactSummarySendContract.InteractorCallback callback) {

        Runnable runnable = () -> {
            try {
                Map<String, String> details = PatientRepository.getWomanProfileDetails(entityId);

                Map<String, String> clientDetails =
                        AncLibrary.getInstance().getDetailsRepository().getAllDetailsForClient(entityId);
                JSONObject rawContactSchedule;
                if (clientDetails.containsKey(ConstantsUtils.DetailsKeyUtils.CONTACT_SCHEDULE)) {
                    rawContactSchedule = new JSONObject(
                            AncLibrary.getInstance().getDetailsRepository().getAllDetailsForClient(entityId)
                                    .get(ConstantsUtils.DetailsKeyUtils.CONTACT_SCHEDULE));
                } else {
                    rawContactSchedule = new JSONObject();
                }

                List<String> contactSchedule = new ArrayList<>();
                if (StringUtils.isBlank(referralContactNo)) {
                    if (rawContactSchedule.has(ConstantsUtils.DetailsKeyUtils.CONTACT_SCHEDULE)) {
                        contactSchedule =
                                Utils.getListFromString(
                                        rawContactSchedule.getString(ConstantsUtils.DetailsKeyUtils.CONTACT_SCHEDULE));
                    }
                } else {
                    int previousContact = getPreviousContactNo(referralContactNo);
                    if (previousContact > 0) {
                        Facts facts =
                                AncLibrary.getInstance().getPreviousContactRepository()
                                        .getImmediatePreviousSchedule(entityId, String.valueOf(previousContact));
                        if (facts != null && facts.asMap().containsKey(ConstantsUtils.CONTACT_SCHEDULE)) {
                            String schedule = (String) facts.asMap().get(ConstantsUtils.CONTACT_SCHEDULE);
                            contactSchedule = Utils.getListFromString(schedule);
                        }
                    }
                }
                final List<ContactSummaryModel> contactDates;

                final Integer lastContact = Integer.valueOf(details.get(DBConstantsUtils.KeyUtils.NEXT_CONTACT));
                Integer lastContactSequence = Integer.valueOf(details.get(DBConstantsUtils.KeyUtils.NEXT_CONTACT));

                String edd = details.get(DBConstantsUtils.KeyUtils.EDD);
                contactDates = formUtils.generateNextContactSchedule(edd, contactSchedule, lastContactSequence);

                getAppExecutors().mainThread().execute(() -> {
                    int contact = lastContact - 1;
                    if (!StringUtils.isBlank(referralContactNo)) {
                        contact = Integer.parseInt(referralContactNo);
                    }
                    callback.onUpcomingContactsFetched(contactDates, contact);
                });
            } catch (Exception e) {
                Timber.e(e, "%s --> fetchUpcomingContacts()", this.getClass().getCanonicalName());
            }
        };
        getAppExecutors().diskIO().execute(runnable);
    }

    public int getPreviousContactNo(String referralContact) {
        int contactNo = 0;
        String[] contactArray = referralContact.split("-");
        if (contactArray.length > 0) {
            int currentContact = Integer.parseInt(contactArray[1]);
            contactNo = currentContact - 1;
        }
        return contactNo;
    }

}