OleItemRecordHandler

package org.kuali.ole;

import com.thoughtworks.xstream.XStream;
import org.kuali.ole.docstore.model.xmlpojo.ingest.AdditionalAttributes;
import org.kuali.ole.docstore.model.xmlpojo.work.bib.marc.DataField;
import org.kuali.ole.docstore.model.xmlpojo.work.bib.marc.SubField;
import org.kuali.ole.docstore.model.xmlpojo.work.instance.oleml.*;
import org.kuali.ole.pojo.ProfileAttribute;
import org.kuali.ole.pojo.bib.BibliographicRecord;

import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

//import org.kuali.ole.pojo.item.Extension;
//import org.kuali.ole.pojo.item.OleItem;

/**
 * Created by IntelliJ IDEA.
 * User: pvsubrah
 * Date: 4/9/12
 * Time: 3:22 PM
 * To change this template use File | Settings | File Templates.
 */
public class OleItemRecordHandler {

    public static final String COPY_NUMBER = "1";
    public static final String COPY_NUMBER_LABEL = "c";

    public String generateXML(BibliographicRecord bibliographicRecord, List<ProfileAttribute> profileAttributes) {
        return toXML(getOleItem(bibliographicRecord, profileAttributes));
    }

    public Item getOleItem(BibliographicRecord bibliographicRecord, List<ProfileAttribute> profileAttributes ){

        DataField dataFieldForTag = getDataFieldForTag(bibliographicRecord, "852");
        String dataField = dataFieldForTag.getTag();

        Item oleItem = new Item();
//        oleItem.setItemType(getSubFieldValueFor(bibliographicRecord, dataField, "$5"));
        ItemType itemType= new ItemType();
        itemType.setCodeValue(getSubFieldValueFor(bibliographicRecord, dataField, "$5"));
        itemType.setFullValue(getSubFieldValueFor(bibliographicRecord, dataField, "$5"));
        oleItem.setItemType(itemType);
        oleItem.setCopyNumber(COPY_NUMBER);
        oleItem.setCopyNumberLabel(COPY_NUMBER_LABEL);

        oleItem.setPurchaseOrderLineItemIdentifier("");

        String valueFor = getSubFieldValueFor(bibliographicRecord, dataField, "$4");
        if (null != valueFor) {
//            oleItem.setStatisticalSearchingCodes(valueFor);
            List<StatisticalSearchingCode> statisticalSearchingCodes = new ArrayList<StatisticalSearchingCode>();
            StatisticalSearchingCode searchingCodes = new StatisticalSearchingCode();
            searchingCodes.setCodeValue(valueFor);
            searchingCodes.setFullValue(valueFor);
            statisticalSearchingCodes.add(searchingCodes);
            oleItem.setStatisticalSearchingCode(statisticalSearchingCodes);
        }
        oleItem.setCopyNumber(COPY_NUMBER);
        oleItem.setCopyNumberLabel(COPY_NUMBER_LABEL);
//
//        PhysicalLocation location = new PhysicalLocation();
//        location.getLocationStatus().add(getAttributeValue(profileAttributes, "locationStatus"));

//        LocationLevel locationLevel = new LocationLevel();
//        CodeOrIdentifier levelName = new CodeOrIdentifier();

//        levelName.setValue(getAttributeValue(profileAttributes, "levelName"));
//
//        TypeOrSource typeOrSource1 = new TypeOrSource();
//        typeOrSource1.setPointer("Pointer 1");
//        typeOrSource1.setText("Text for Level Name 1");
//        levelName.setTypeOrSource(typeOrSource1);
//
//        locationLevel.setLevelName(levelName);
//        location.getLocationLevel().add(locationLevel);
//
//        location.getCallNumberPrefix().add(getSubFieldValueFor(bibliographicRecord, dataField, "$k"));
//        location.getCallNumberSuffix().add(getSubFieldValueFor(bibliographicRecord, dataField, "$,"));

//        CodeOrIdentifier classification = new CodeOrIdentifier();
//        TypeOrSource typeOrSource4 = new TypeOrSource();
//        typeOrSource4.setPointer("Classification Pointer -1");
//        typeOrSource4.setText("Classification Text -1");
//        classification.setTypeOrSource(typeOrSource4);
//        classification.setValue(getClassificationOrShelvingSchemeSource(bibliographicRecord));
//        location.setClassificationOrShelvingSchemeSource(classification);

//        location.setClassificationPart(getSubFieldValueFor(bibliographicRecord, dataField, "$h"));
//        location.getItemPart().add(getSubFieldValueFor(bibliographicRecord, dataField, "$i"));
//        location.setShelvingControlNumber(getSubFieldValueFor(bibliographicRecord, dataField, "$j"));
//
//        oleItem.getLocation().add(location);

        //Eill
//        oleItem.getVendorLineItemIdentifier().add("Vendoer Line Item Identifier-1");

        // Extension
        DateFormat df = new SimpleDateFormat("dd-MM-yyyy");
        Extension extension = new Extension();
        AdditionalAttributes additionalAttributes = new AdditionalAttributes();
        additionalAttributes.setDateEntered(String.valueOf(df.format(new Date())));
        additionalAttributes.setLastUpdated(String.valueOf(df.format(new Date())));
        additionalAttributes.setSupressFromPublic("false");
        additionalAttributes.setFastAddFlag("false");
        additionalAttributes.setHarvestable("true");
        additionalAttributes.setStatus("n"); // new Record
        additionalAttributes.setAttribute("dateEntered",additionalAttributes.getDateEntered());
        additionalAttributes.setAttribute("lastUpdated",additionalAttributes.getLastUpdated());
        additionalAttributes.setAttribute("fastAddFlag",additionalAttributes.getFastAddFlag());
        additionalAttributes.setAttribute("supressFromPublic",additionalAttributes.getSupressFromPublic());
        additionalAttributes.setAttribute("harvestable",additionalAttributes.getHarvestable());
        additionalAttributes.setAttribute("status",additionalAttributes.getStatus());
        extension.getContent().add(additionalAttributes);
        oleItem.setExtension(extension);

        return oleItem;
    }

    private String getAttributeValue(List<ProfileAttribute> profileAttributes, String attributeName) {
        for (Iterator<ProfileAttribute> iterator = profileAttributes.iterator(); iterator.hasNext(); ) {
            ProfileAttribute attribute = iterator.next();
            if (attribute.getAttributeName().equals(attributeName)) {
                return attribute.getAttributeValue();
            }
        }
        return null;
    }

    private String getSubFieldValueFor(BibliographicRecord bibliographicRecord, String dataField, String tag) {
        String subFieldValue = null;
        DataField dataFieldForTag = getDataFieldForTag(bibliographicRecord, dataField);
        if (null != dataFieldForTag) {
            List<SubField> subfields = dataFieldForTag.getSubFields();
            for (Iterator<SubField> iterator = subfields.iterator(); iterator.hasNext(); ) {
                SubField marcSubField = iterator.next();
                if (marcSubField.getCode().equals(tag)) {
                    return subFieldValue = marcSubField.getValue();
                }
            }
        }
        return subFieldValue;
    }


    public DataField getDataFieldForTag(BibliographicRecord bibliographicRecord, String tag) {
        for (Iterator<DataField> iterator = bibliographicRecord.getDatafields().iterator(); iterator.hasNext(); ) {
            DataField marcDataField = iterator.next();
            if (marcDataField.getTag().equalsIgnoreCase(tag)) {
                return marcDataField;
            }
        }
        return null;
    }


    public String toXML(Item oleItem) {
        XStream xs = new XStream();
        xs.autodetectAnnotations(true);
        xs.processAnnotations(Item.class);
        String xml = xs.toXML(oleItem);
        return xml;
    }

    public String getClassificationOrShelvingSchemeSource(BibliographicRecord bibliographicRecord) {
        DataField dataFieldForTag = getDataFieldForTag(bibliographicRecord, "852");
        if (dataFieldForTag.getInd1().equals("7")) {
            List<SubField> subfields = dataFieldForTag.getSubFields();
            for (Iterator<SubField> iterator = subfields.iterator(); iterator.hasNext(); ) {
                SubField subField = iterator.next();
                if (subField.getCode().equals("2")) {
                    return subField.getValue();
                }
            }
        }

        return "";
    }
}