DocstoreUtil

package org.kuali.ole.util;

import org.jfree.util.Log;
import org.kuali.ole.OLEConstants;
import org.kuali.ole.deliver.bo.OleDeliverRequestBo;
import org.kuali.ole.deliver.bo.OleItemSearch;
import org.kuali.ole.deliver.processor.LoanProcessor;
import org.kuali.ole.describe.keyvalue.LocationValuesBuilder;
import org.kuali.ole.docstore.common.client.DocstoreClientLocator;
import org.kuali.ole.docstore.common.document.*;
import org.kuali.ole.docstore.common.document.content.instance.Item;
import org.kuali.ole.docstore.common.document.content.instance.Location;
import org.kuali.ole.docstore.common.document.content.instance.OleHoldings;
import org.kuali.ole.docstore.common.document.content.instance.xstream.HoldingOlemlRecordProcessor;
import org.kuali.ole.docstore.common.document.content.instance.xstream.ItemOlemlRecordProcessor;
import org.kuali.ole.docstore.common.search.SearchParams;
import org.kuali.ole.docstore.common.search.SearchResponse;
import org.kuali.ole.docstore.common.search.SearchResult;
import org.kuali.ole.docstore.common.search.SearchResultField;
import org.kuali.ole.sys.context.SpringContext;
import org.kuali.rice.krad.util.GlobalVariables;
import org.kuali.rice.krad.util.KRADConstants;

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

/**
 * Created with IntelliJ IDEA.
 * User: arunag
 * Date: 1/9/14
 * Time: 2:48 PM
 * To change this template use File | Settings | File Templates.
 */
public class DocstoreUtil {
    private static final org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(DocstoreUtil.class);
    private DocstoreClientLocator docstoreClientLocator;

    public DocstoreClientLocator getDocstoreClientLocator() {
        if (docstoreClientLocator == null) {
            docstoreClientLocator = SpringContext.getBean(DocstoreClientLocator.class);
        }
        return docstoreClientLocator;
    }

    public boolean isItemAvailableInDocStore(OleDeliverRequestBo oleDeliverRequestBo) {
        LOG.info("Inside isItemAvailableInDocStore");
        boolean available = false;
        Map<String, String> itemMap = new HashMap<String, String>();
        LocationValuesBuilder locationValuesBuilder = new LocationValuesBuilder();
        String holdingsId = "";
        try {

            try {
                org.kuali.ole.docstore.common.document.Item item = new ItemOleml();
                org.kuali.ole.docstore.common.search.SearchParams search_Params = new org.kuali.ole.docstore.common.search.SearchParams();
                SearchResponse searchResponse = null;
                search_Params.getSearchConditions().add(search_Params.buildSearchCondition("", search_Params.buildSearchField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.ITEM_BARCODE, oleDeliverRequestBo.getItemId()), ""));
                search_Params.getSearchResultFields().add(search_Params.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), "id"));
                search_Params.getSearchResultFields().add(search_Params.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.HOLDINGS.getCode(), "id"));
                searchResponse = getDocstoreClientLocator().getDocstoreClient().search(search_Params);
                for (SearchResult searchResult : searchResponse.getSearchResults()) {
                    for (SearchResultField searchResultField : searchResult.getSearchResultFields()) {
                        String fieldName = searchResultField.getFieldName();
                        String fieldValue = searchResultField.getFieldValue() != null ? searchResultField.getFieldValue() : "";
                        if (fieldName.equalsIgnoreCase("id") && !fieldValue.isEmpty() && searchResultField.getDocType().equalsIgnoreCase("holdings")) {
                            holdingsId = fieldValue;
                        } else {
                            oleDeliverRequestBo.setItemUuid(fieldValue);
                            String itemUUID = fieldValue;
                        }
                    }
                }
            } catch (Exception ex) {
                GlobalVariables.getMessageMap().putError(KRADConstants.GLOBAL_ERRORS, "Item Exists");
                LOG.error(OLEConstants.ITEM_EXIST + ex);
            }
            OleItemSearch itemSearchList = getOleItemSearchList(oleDeliverRequestBo.getItemUuid());
            if (itemSearchList != null) {
                oleDeliverRequestBo.setTitle(itemSearchList.getTitle());
                oleDeliverRequestBo.setAuthor(itemSearchList.getAuthor());
                oleDeliverRequestBo.setCallNumber(itemSearchList.getCallNumber());
                oleDeliverRequestBo.setItemType(itemSearchList.getItemType());
                oleDeliverRequestBo.setItemLocation(itemSearchList.getShelvingLocation());
            }
            LoanProcessor loanProcessor = new LoanProcessor();
            String itemXml = loanProcessor.getItemXML(oleDeliverRequestBo.getItemUuid());
            Item oleItem = loanProcessor.getItemPojo(itemXml);
            oleDeliverRequestBo.setOleItem(oleItem);
            oleDeliverRequestBo.setCopyNumber(oleItem.getCopyNumber());
            oleDeliverRequestBo.setItemStatus(oleItem.getItemStatus().getCodeValue());
            oleDeliverRequestBo.setClaimsReturnedFlag(oleItem.isClaimsReturnedFlag());
            locationValuesBuilder.getLocation(oleItem, oleDeliverRequestBo, holdingsId);
            available = true;
        } catch (Exception e) {
            LOG.error(OLEConstants.INVAL_LOC + e);
        }
        return available;
    }

    public OleItemSearch getOleItemSearchList(String itemUuid) {

        OleItemSearch oleItemSearch = new OleItemSearch();
        try {
            org.kuali.ole.docstore.common.document.Item item = getDocstoreClientLocator().getDocstoreClient().retrieveItem(itemUuid);
            ItemOlemlRecordProcessor itemOlemlRecordProcessor = new ItemOlemlRecordProcessor();
            Item itemContent = itemOlemlRecordProcessor.fromXML(item.getContent());
            oleItemSearch.setTitle(item.getHolding().getBib().getTitle());
            oleItemSearch.setAuthor(item.getHolding().getBib().getAuthor());
            oleItemSearch.setVolumeNumber(itemContent.getVolumeNumber());
            oleItemSearch.setBibUUID(item.getHolding().getBib().getId());
            StringBuffer locationLevel = new StringBuffer("");
            String location = "";
            if (itemContent.getLocation().getLocationLevel() != null) {
                location = getLocation(itemContent.getLocation(), locationLevel);
                oleItemSearch.setShelvingLocation(location);
            } else {
                HoldingOlemlRecordProcessor holdingOlemlRecordProcessor = new HoldingOlemlRecordProcessor();
                OleHoldings oleHoldings = holdingOlemlRecordProcessor.fromXML(item.getHolding().getContent());
                oleItemSearch.setShelvingLocation(getLocation(oleHoldings.getLocation(), locationLevel));
            }
            oleItemSearch.setPublisher(item.getHolding().getBib().getPublisher());
            oleItemSearch.setCallNumber(itemContent.getCallNumber().getNumber());
            oleItemSearch.setHoldingUUID(item.getHolding().getId());
            oleItemSearch.setInstanceUUID(item.getHolding().getId());
            oleItemSearch.setCopyNumber(itemContent.getCopyNumber());
            oleItemSearch.setItemBarCode(itemContent.getAccessInformation().getBarcode());
            oleItemSearch.setItemStatus(itemContent.getItemStatus().getFullValue());
            oleItemSearch.setItemType(itemContent.getItemType().getCodeValue());
            oleItemSearch.setItemUUID(item.getId());
        } catch (Exception ex) {
            ex.printStackTrace();
            Log.info("Item does not Exist");
        }
        return oleItemSearch;
    }


    public boolean isItemAvailableInDocStore(String itemBarcode) {
        LOG.info("Inside isItemAvailableInDocStore");
        boolean available = false;
        try {
            org.kuali.ole.docstore.common.document.Item item = new ItemOleml();
            org.kuali.ole.docstore.common.search.SearchParams search_Params = new org.kuali.ole.docstore.common.search.SearchParams();
            SearchResponse searchResponse = null;
            search_Params.getSearchConditions().add(search_Params.buildSearchCondition("", search_Params.buildSearchField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.ITEM_BARCODE, itemBarcode), ""));
            search_Params.getSearchResultFields().add(search_Params.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), "id"));
            searchResponse = getDocstoreClientLocator().getDocstoreClient().search(search_Params);
            if (searchResponse.getSearchResults() != null && searchResponse.getSearchResults().size() > 0) {
                available = true;
            }
        } catch (Exception ex) {

            LOG.error(OLEConstants.ITEM_EXIST + ex);
        }
        return available;
    }


    public List<OleItemSearch> getOlItemSearchList(SearchParams searchParams) {
        SearchResponse searchResponse = new SearchResponse();
        List<OleItemSearch> oleItemSearchList = new ArrayList<>();
        Bib bib = new BibMarc();
        org.kuali.ole.docstore.common.document.Item item = new ItemOleml();
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), "id"));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.ITEM_BARCODE));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.CALL_NUMBER));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.ITEM_TYPE));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.ITEM_STATUS));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.COPY_NUMBER));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.HOLDINGS.getCode(), Holdings.LOCATION_NAME));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.ITEM.getCode(), item.VOLUME_NUMBER));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.HOLDINGS.getCode(), "id"));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.BIB.getCode(), "id"));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.BIB.getCode(), bib.TITLE));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.BIB.getCode(), bib.AUTHOR));
        searchParams.getSearchResultFields().add(searchParams.buildSearchResultField(org.kuali.ole.docstore.common.document.content.enums.DocType.BIB.getCode(), bib.PUBLISHER));
        try {
            searchResponse = getDocstoreClientLocator().getDocstoreClient().search(searchParams);
            if (searchResponse.getSearchResults() != null && searchResponse.getSearchResults().size() > 0) {
                for (SearchResult searchResult : searchResponse.getSearchResults()) {
                    OleItemSearch oleItemSearch = new OleItemSearch();
                    if (searchResult.getSearchResultFields().get(0).getFieldName().equalsIgnoreCase("id") && searchResult.getSearchResultFields().get(0).getFieldValue() != null && !searchResult.getSearchResultFields().get(0).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(0).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setItemUUID(searchResult.getSearchResultFields().get(0).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(8).getFieldName().equalsIgnoreCase("id") && searchResult.getSearchResultFields().get(8).getFieldValue() != null && !searchResult.getSearchResultFields().get(8).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(8).getDocType().equalsIgnoreCase("holdings")) {
                        oleItemSearch.setHoldingUUID(searchResult.getSearchResultFields().get(8).getFieldValue());
                        oleItemSearch.setInstanceUUID(searchResult.getSearchResultFields().get(8).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(1).getFieldName().equalsIgnoreCase(item.ITEM_BARCODE) && searchResult.getSearchResultFields().get(1).getFieldValue() != null && !searchResult.getSearchResultFields().get(1).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(1).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setItemBarCode(searchResult.getSearchResultFields().get(1).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(2).getFieldName().equalsIgnoreCase(item.CALL_NUMBER) && searchResult.getSearchResultFields().get(2).getFieldValue() != null && !searchResult.getSearchResultFields().get(2).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(2).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setCallNumber(searchResult.getSearchResultFields().get(2).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(5).getFieldName().equalsIgnoreCase(item.COPY_NUMBER) && searchResult.getSearchResultFields().get(5).getFieldValue() != null && !searchResult.getSearchResultFields().get(5).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(5).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setCopyNumber(searchResult.getSearchResultFields().get(5).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(6).getFieldName().equalsIgnoreCase(Holdings.LOCATION_NAME) && searchResult.getSearchResultFields().get(6).getFieldValue() != null && !searchResult.getSearchResultFields().get(6).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(6).getDocType().equalsIgnoreCase("holdings")) {
                        oleItemSearch.setShelvingLocation(searchResult.getSearchResultFields().get(6).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(4).getFieldName().equalsIgnoreCase(item.ITEM_STATUS) && searchResult.getSearchResultFields().get(4).getFieldValue() != null && !searchResult.getSearchResultFields().get(4).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(4).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setItemStatus(searchResult.getSearchResultFields().get(4).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(3).getFieldName().equalsIgnoreCase(item.ITEM_TYPE) && searchResult.getSearchResultFields().get(3).getFieldValue() != null && !searchResult.getSearchResultFields().get(3).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(3).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setItemType(searchResult.getSearchResultFields().get(3).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(7).getFieldName().equalsIgnoreCase(item.VOLUME_NUMBER) && searchResult.getSearchResultFields().get(7).getFieldValue() != null && !searchResult.getSearchResultFields().get(7).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(7).getDocType().equalsIgnoreCase("item")) {
                        oleItemSearch.setVolumeNumber(searchResult.getSearchResultFields().get(7).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(10).getFieldName().equalsIgnoreCase(bib.TITLE) && searchResult.getSearchResultFields().get(10).getFieldValue() != null && !searchResult.getSearchResultFields().get(10).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(10).getDocType().equalsIgnoreCase("bibliographic")) {
                        oleItemSearch.setTitle(searchResult.getSearchResultFields().get(10).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(11).getFieldName().equalsIgnoreCase(bib.AUTHOR) && searchResult.getSearchResultFields().get(11).getFieldValue() != null && !searchResult.getSearchResultFields().get(11).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(11).getDocType().equalsIgnoreCase("bibliographic")) {
                        oleItemSearch.setAuthor(searchResult.getSearchResultFields().get(11).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(12).getFieldName().equalsIgnoreCase(bib.PUBLISHER) && searchResult.getSearchResultFields().get(12).getFieldValue() != null && !searchResult.getSearchResultFields().get(12).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(12).getDocType().equalsIgnoreCase("bibliographic")) {
                        oleItemSearch.setPublisher(searchResult.getSearchResultFields().get(12).getFieldValue());
                    }
                    if (searchResult.getSearchResultFields().get(9).getFieldName().equalsIgnoreCase("id") && searchResult.getSearchResultFields().get(9).getFieldValue() != null && !searchResult.getSearchResultFields().get(9).getFieldValue().isEmpty() && searchResult.getSearchResultFields().get(9).getDocType().equalsIgnoreCase("bibliographic")) {
                        oleItemSearch.setBibUUID(searchResult.getSearchResultFields().get(9).getFieldValue());
                    }
                    oleItemSearchList.add(oleItemSearch);
                }
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return oleItemSearchList;

    }

    public String getLocation(Location location, StringBuffer locationLevel) {
        StringBuffer locationName = new StringBuffer("");
        if (location != null && location.getLocationLevel() != null) {
            locationName = locationName.append(location.getLocationLevel().getName());
            locationLevel = locationLevel.append(location.getLocationLevel().getLevel());
            if (location.getLocationLevel().getLocationLevel() != null) {
                locationName = locationName.append("/").append(location.getLocationLevel().getLocationLevel().getName());
                locationLevel = locationLevel.append("/").append(location.getLocationLevel().getLocationLevel().getLevel());
                if (location.getLocationLevel().getLocationLevel().getLocationLevel() != null) {
                    locationName = locationName.append("/").append(location.getLocationLevel().getLocationLevel().getLocationLevel().getName());
                    locationLevel = locationLevel.append("/").append(location.getLocationLevel().getLocationLevel().getLocationLevel().getLevel());
                    if (location.getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel() != null) {
                        locationName = locationName.append("/").append(location.getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getName());
                        locationLevel = locationLevel.append("/").append(location.getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getLevel());
                        if (location.getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel() != null) {
                            locationName = locationName.append("/").append(location.getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getName());
                            locationLevel = locationLevel.append("/").append(location.getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getLocationLevel().getLevel());
                        }
                    }
                }
            }
        }
        return locationName.toString();
    }

    public OleHoldings getOleHoldings(String instanceUUID) throws Exception {
        LOG.info("--Inside getOleHoldings---");
        Holdings holdings = new Holdings();
        holdings = getDocstoreClientLocator().getDocstoreClient().retrieveHoldings(instanceUUID);
        HoldingOlemlRecordProcessor holdingOlemlRecordProcessor = new HoldingOlemlRecordProcessor();
        OleHoldings oleHoldings = holdingOlemlRecordProcessor.fromXML(holdings.getContent());
        return oleHoldings;
    }

    public List<OleItemSearch> getItemBibInfo(String bibId) {
        List<OleItemSearch> oleItemSearchList = new ArrayList<>();
        try {
            BibTree bibTree = getDocstoreClientLocator().getDocstoreClient().retrieveBibTree(bibId);
            for (HoldingsTree holdingsTree : bibTree.getHoldingsTrees()) {
                for (org.kuali.ole.docstore.common.document.Item item : holdingsTree.getItems()) {
                    OleItemSearch oleItemSearch = new OleItemSearch();
                    ItemOlemlRecordProcessor itemOlemlRecordProcessor = new ItemOlemlRecordProcessor();
                    Item itemContent = itemOlemlRecordProcessor.fromXML(item.getContent());
                    oleItemSearch.setBibUUID(bibTree.getBib().getId());
                    oleItemSearch.setTitle(bibTree.getBib().getTitle());
                    oleItemSearch.setAuthor(bibTree.getBib().getAuthor());
                    oleItemSearch.setHoldingUUID(holdingsTree.getHoldings().getId());
                    oleItemSearch.setInstanceUUID(holdingsTree.getHoldings().getId());
                    oleItemSearch.setItemBarCode(itemContent.getAccessInformation().getBarcode());
                    oleItemSearch.setItemUUID(item.getId());
                    oleItemSearch.setCallNumber(itemContent.getCallNumber().getNumber());
                    oleItemSearch.setCopyNumber(itemContent.getCopyNumber());
                    StringBuffer locationLevel = new StringBuffer("");
                    String location = "";
                    if (itemContent.getLocation().getLocationLevel() != null) {
                        location = getLocation(itemContent.getLocation(), locationLevel);
                        oleItemSearch.setShelvingLocation(location);
                    } else {
                        HoldingOlemlRecordProcessor holdingOlemlRecordProcessor = new HoldingOlemlRecordProcessor();
                        OleHoldings oleHoldings = holdingOlemlRecordProcessor.fromXML(item.getHolding().getContent());
                        oleItemSearch.setShelvingLocation(getLocation(oleHoldings.getLocation(), locationLevel));
                    }

                    oleItemSearch.setItemStatus(itemContent.getItemStatus().getCodeValue());
                    oleItemSearch.setItemType(itemContent.getItemType().getFullValue());
                    oleItemSearch.setVolumeNumber(itemContent.getVolumeNumber());
                    oleItemSearchList.add(oleItemSearch);
                }
            }
        } catch (Exception ex) {
            ex.printStackTrace();
        }

        return oleItemSearchList;
    }


}