OlePatronInquirableImpl

package org.kuali.ole.deliver.inquiry;

import org.apache.commons.collections.CollectionUtils;
import org.kuali.ole.OLEConstants;
import org.kuali.ole.OLEPropertyConstants;
import org.kuali.ole.deliver.bo.*;
import org.kuali.rice.core.api.config.property.ConfigContext;
import org.kuali.rice.kim.impl.identity.address.EntityAddressBo;
import org.kuali.rice.kim.impl.identity.affiliation.EntityAffiliationBo;
import org.kuali.rice.kim.impl.identity.email.EntityEmailBo;
import org.kuali.rice.kim.impl.identity.employment.EntityEmploymentBo;
import org.kuali.rice.kim.impl.identity.entity.EntityBo;
import org.kuali.rice.kim.impl.identity.phone.EntityPhoneBo;
import org.kuali.rice.krad.bo.BusinessObject;
import org.kuali.rice.krad.bo.ExternalizableBusinessObject;
import org.kuali.rice.krad.datadictionary.exception.UnknownBusinessClassAttributeException;
import org.kuali.rice.krad.inquiry.InquirableImpl;
import org.kuali.rice.krad.service.KRADServiceLocatorWeb;
import org.kuali.rice.krad.service.ModuleService;

import java.security.GeneralSecurityException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * OlePatronInquirableImpl supports to get the data object and patron document  .
 */
public class OlePatronInquirableImpl extends InquirableImpl {
    private static final org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(OlePatronInquirableImpl.class);

    /**
     * This method will retrieve the patron document based on the olePatronId
     *
     * @param parameters
     * @return patronDocument(Object)
     */
    String baseUrl = ConfigContext.getCurrentContextConfig().getProperty(OLEPropertyConstants.OLE_URL_BASE);

    public Object retrieveDataObject(Map<String, String> parameters) {
        if (dataObjectClass == null) {
            LOG.error("Data object class must be set in inquirable before retrieving the object");
            throw new RuntimeException("Data object class must be set in inquirable before retrieving the object");
        }

        // build list of key values from the map parameters
        List<String> pkPropertyNames = getDataObjectMetaDataService().listPrimaryKeyFieldNames(dataObjectClass);

        // some classes might have alternate keys defined for retrieving
        List<List<String>> alternateKeyNames = this.getAlternateKeysForClass(dataObjectClass);

        // add pk set as beginning so it will be checked first for match
        alternateKeyNames.add(0, pkPropertyNames);

        List<String> dataObjectKeySet = retrieveKeySetFromMap(alternateKeyNames, parameters);
        if ((dataObjectKeySet == null) || dataObjectKeySet.isEmpty()) {
            LOG.warn("Matching key set not found in request for class: " + getDataObjectClass());

            return null;
        }
        String patronHome = parameters.get("flow");

        // found key set, now build map of key values pairs we can use to retrieve the object
        Map<String, Object> keyPropertyValues = new HashMap<String, Object>();
        for (String keyPropertyName : dataObjectKeySet) {
            String keyPropertyValue = parameters.get(keyPropertyName);

            // uppercase value if needed
            Boolean forceUppercase = Boolean.FALSE;
            try {
                forceUppercase = getDataDictionaryService().getAttributeForceUppercase(dataObjectClass,
                        keyPropertyName);
            } catch (UnknownBusinessClassAttributeException ex) {
                // swallowing exception because this check for ForceUppercase would
                // require a DD entry for the attribute, and we will just set force uppercase to false
                LOG.warn("Data object class "
                        + dataObjectClass
                        + " property "
                        + keyPropertyName
                        + " should probably have a DD definition.", ex);
            }

            if (forceUppercase.booleanValue() && (keyPropertyValue != null)) {
                keyPropertyValue = keyPropertyValue.toUpperCase();
            }

            // check security on key field
            if (getDataObjectAuthorizationService().attributeValueNeedsToBeEncryptedOnFormsAndLinks(dataObjectClass,
                    keyPropertyName)) {
                try {
                    keyPropertyValue = getEncryptionService().decrypt(keyPropertyValue);
                } catch (GeneralSecurityException e) {
                    LOG.error("Data object class "
                            + dataObjectClass
                            + " property "
                            + keyPropertyName
                            + " should have been encrypted, but there was a problem decrypting it.", e);
                    throw new RuntimeException("Data object class "
                            + dataObjectClass
                            + " property "
                            + keyPropertyName
                            + " should have been encrypted, but there was a problem decrypting it.", e);
                }
            }

            keyPropertyValues.put(keyPropertyName, keyPropertyValue);
        }

        // now retrieve the object based on the key set
        Object dataObject = null;

        ModuleService moduleService = KRADServiceLocatorWeb.getKualiModuleService().getResponsibleModuleService(
                getDataObjectClass());
        if (moduleService != null && moduleService.isExternalizable(getDataObjectClass())) {
            dataObject = moduleService.getExternalizableBusinessObject(getDataObjectClass().asSubclass(
                    ExternalizableBusinessObject.class), keyPropertyValues);
        } else if (BusinessObject.class.isAssignableFrom(getDataObjectClass())) {
            dataObject = getBusinessObjectService().findByPrimaryKey(getDataObjectClass().asSubclass(
                    BusinessObject.class), keyPropertyValues);
        }
        //OleEntityAddressBo entityAddressBo = new OleEntityAddressBo();
        EntityAddressBo entityAddress = new EntityAddressBo();
        List<OleEntityAddressBo> oleEntityAddressList = new ArrayList<OleEntityAddressBo>();
        List<OleEntityPhoneBo> oleEntityPhoneBoList = new ArrayList<>();
        List<OleEntityEmailBo> oleEntityEmailBoList = new ArrayList<>();
        OlePatronDocument patronDocument = (OlePatronDocument) dataObject;
        if (patronDocument != null) {
            EntityBo kimEnity = patronDocument.getEntity();
            if (kimEnity.getNames().size() > 0) {
                patronDocument.setName(kimEnity.getNames().get(0));
            }
            if(kimEnity.getEntityTypeContactInfos().size()>0){
                List<EntityAddressBo> entityAddressList = kimEnity.getEntityTypeContactInfos().get(0).getAddresses();
                for (EntityAddressBo entityAdd : entityAddressList) {
                    OleEntityAddressBo entityAddressBo = new OleEntityAddressBo();
                    entityAddressBo.setEntityAddressBo(entityAdd);
                    Map addMap = new HashMap();
                    //addMap.put(OLEConstants.OlePatron.OLE_ADDRESS_ID, entityAdd.getId());
                    addMap.put(OLEConstants.OlePatron.ENTITY_BO_ID, entityAdd.getId());
                    addMap.put(OLEConstants.OlePatron.PATRON_ID, patronDocument.getId());
                    List<OleAddressBo> oleAddressBos = (List<OleAddressBo>)getBusinessObjectService().findMatching(OleAddressBo.class, addMap);
                    if(CollectionUtils.isNotEmpty(oleAddressBos)) {
                        entityAddressBo.setOleAddressBo(oleAddressBos.get(0));
                    }
                    oleEntityAddressList.add(entityAddressBo);
                }
                patronDocument.setOleEntityAddressBo(oleEntityAddressList);
                List<EntityPhoneBo> entityPhoneBoList = kimEnity.getEntityTypeContactInfos().get(0).getPhoneNumbers();
                for(EntityPhoneBo entityPhoneBo : entityPhoneBoList) {
                    OleEntityPhoneBo oleEntityPhoneBo = new OleEntityPhoneBo();
                    oleEntityPhoneBo.setEntityPhoneBo(entityPhoneBo);
                    Map map = new HashMap();
                    map.put(OLEConstants.OlePatron.ENTITY_BO_ID, entityPhoneBo.getId());
                    map.put(OLEConstants.OlePatron.PATRON_ID, patronDocument.getId());
                    List<OlePhoneBo> olePhoneBos = (List<OlePhoneBo>)getBusinessObjectService().findMatching(OlePhoneBo.class, map);
                    if(CollectionUtils.isNotEmpty(olePhoneBos)) {
                        oleEntityPhoneBo.setOlePhoneBo(olePhoneBos.get(0));
                    }
                    oleEntityPhoneBoList.add(oleEntityPhoneBo);
                }
                patronDocument.setOleEntityPhoneBo(oleEntityPhoneBoList);
                List<EntityEmailBo> entityEmailBoList = kimEnity.getEntityTypeContactInfos().get(0).getEmailAddresses();
                for(EntityEmailBo entityEmailBo : entityEmailBoList) {
                    OleEntityEmailBo oleEntityEmailBo = new OleEntityEmailBo();
                    oleEntityEmailBo.setEntityEmailBo(entityEmailBo);
                    Map map = new HashMap();
                    map.put(OLEConstants.OlePatron.ENTITY_BO_ID, entityEmailBo.getId());
                    map.put(OLEConstants.OlePatron.PATRON_ID, patronDocument.getId());
                    List<OleEmailBo> oleEmailBos = (List<OleEmailBo>)getBusinessObjectService().findMatching(OleEmailBo.class, map);
                    if(CollectionUtils.isNotEmpty(oleEmailBos)) {
                        oleEntityEmailBo.setOleEmailBo(oleEmailBos.get(0));
                    }
                    oleEntityEmailBoList.add(oleEntityEmailBo);
                }
                patronDocument.setOleEntityEmailBo(oleEntityEmailBoList);
                patronDocument.setPatronAffiliations(getPatronAffiliationFromEntity(kimEnity.getAffiliations(), kimEnity.getEmploymentInformation()));
                if (patronHome != null) {
                    if (patronHome.equalsIgnoreCase("start"))
                        patronDocument.setPatronHomePage(false);
                    else
                        patronDocument.setPatronHomePage(true);
                }
            }
            if (null != patronDocument.getName()) {
                String createBillUrl = OLEConstants.OlePatron.PATRON_CREATE_BILL_URL + patronDocument.getOlePatronId() + "&firstName=" + patronDocument.getName().getFirstName() + "&lastName=" + patronDocument.getName().getLastName();
                patronDocument.setCreateBillUrl(createBillUrl);
            }
            if (CollectionUtils.isNotEmpty(patronDocument.getPatronBillPayments())) {
                patronDocument.setPatronBillFileName(OLEConstants.OlePatron.PATRON_VIEW_BILLS);
                patronDocument.setViewBillUrl(OLEConstants.OlePatron.PATRON_VIEW_BILL_URL + patronDocument.getOlePatronId());
            }
            patronDocument.setShowBillUrlsFlag(true);


        }

        if (CollectionUtils.isNotEmpty(patronDocument.getOleProxyPatronDocuments())) {
            OlePatronDocument olePatronDocument;
            for (OleProxyPatronDocument oleProxyPatronDocument : patronDocument.getOleProxyPatronDocuments()) {
                olePatronDocument = oleProxyPatronDocument.getOlePatronDocument();
                oleProxyPatronDocument.setProxyForPatronFirstName(olePatronDocument.getEntity().getNames().get(0).getFirstName());
                oleProxyPatronDocument.setProxyForPatronLastName(olePatronDocument.getEntity().getNames().get(0).getLastName());
            }
        }
        if (CollectionUtils.isNotEmpty(patronDocument.getOleProxyPatronDocumentList())) {
            OlePatronDocument olePatronDocument;
            for (OleProxyPatronDocument oleProxyPatronDocument : patronDocument.getOleProxyPatronDocumentList()) {
                olePatronDocument = oleProxyPatronDocument.getOlePatronDocument();
                oleProxyPatronDocument.setProxyForPatronFirstName(olePatronDocument.getEntity().getNames().get(0).getFirstName());
                oleProxyPatronDocument.setProxyForPatronLastName(olePatronDocument.getEntity().getNames().get(0).getLastName());
            }
        }

        if(patronDocument.getOleLoanDocuments()!=null){
            patronDocument.setLoanCount(patronDocument.getOleLoanDocuments().size());
        }
        if(patronDocument.getOleTemporaryCirculationHistoryRecords()!=null){
            patronDocument.setTempCirculationHistoryCount(patronDocument.getOleTemporaryCirculationHistoryRecords().size());
        }
        if(patronDocument.getOleDeliverRequestBos()!=null){
            patronDocument.setRequestedItemRecordsCount(patronDocument.getOleDeliverRequestBos().size());
        }


        return patronDocument;
    }

    private List<OlePatronAffiliation> getPatronAffiliationFromEntity(List<EntityAffiliationBo> affiliations,
                                                                      List<EntityEmploymentBo> employeeDetails) {
        List<OlePatronAffiliation> patronAffiliations = new ArrayList<OlePatronAffiliation>();
        for (EntityAffiliationBo entityAffiliationBo : affiliations) {
            OlePatronAffiliation patronAffiliation = new OlePatronAffiliation(entityAffiliationBo);
            List<EntityEmploymentBo> employmentBos = new ArrayList<EntityEmploymentBo>();
            for (EntityEmploymentBo entityEmploymentBo : employeeDetails) {
                if (patronAffiliation.getEntityAffiliationId().equalsIgnoreCase(entityEmploymentBo.getEntityAffiliationId())) {
                    employmentBos.add(entityEmploymentBo);
                }
                patronAffiliation.setEmployments(employmentBos);
            }
            patronAffiliations.add(patronAffiliation);
        }
        return patronAffiliations;
    }

    public String getTempCircRecords(String olePatronId) {
        String url = baseUrl + "/portal.do?channelTitle=Patron&channelUrl=" + baseUrl + "/ole-kr-krad/temporaryCirculationRecord?viewId=OleTemporaryCirculationHistoryRecordView&amp;methodToCall=viewTempCircRecords&amp;patronId=" + olePatronId;
        return url;
    }

    public String getLoanedRecords(String olePatronId) {

        String url = baseUrl + "/portal.do?channelTitle=Patron&channelUrl=" + baseUrl + "/ole-kr-krad/patronLoanedRecord?viewId=OlePatronLoanedRecordView&amp;methodToCall=viewLoanedRecords&amp;patronId=" + olePatronId;
        return url;
    }

    public String getRequestedRecords(String olePatronId) {
        String url = baseUrl + "/portal.do?channelTitle=Patron&channelUrl=" + baseUrl + "/ole-kr-krad/patronRequestedRecord?viewId=OlePatronRequestedRecordView&amp;methodToCall=viewRequestedRecords&amp;patronId=" + olePatronId;
        return url;
    }

    public String getCountOfPendingRequests(String itemId){
        Map itemMap = new HashMap();
        itemMap.put(OLEConstants.OleDeliverRequest.ITEM_ID, itemId);
        List<OleDeliverRequestBo> oleDeliverRequestBoList = (List<OleDeliverRequestBo>) getBusinessObjectService().findMatching(OleDeliverRequestBo.class, itemMap);
        if (oleDeliverRequestBoList!=null && oleDeliverRequestBoList.size()>0) {

            return "View all requests";
        }
        else
            return " ";
    }

    public String getHrefLink(String item,String instance,String bib){
       String inquiryUrl= ConfigContext.getCurrentContextConfig().getProperty(OLEPropertyConstants.OLE_URL_BASE)+
               "/ole-kr-krad/editorcontroller?viewId=EditorView&amp;methodToCall=load&amp;docCategory=work&amp;docType=item&amp;editable=false&amp;docFormat=oleml&amp;docId="+item+"&amp;instanceId="+instance+"&amp;bibId="+bib+"";
        return inquiryUrl;
    }

}