001package org.kuali.ole.ingest;
002
003import org.kuali.ole.OLEConstants;
004import org.kuali.ole.OleOrderRecordHandler;
005import org.kuali.ole.OleOrderRecords;
006import org.kuali.ole.converter.MarcXMLConverter;
007import org.kuali.ole.converter.OLEEDIConverter;
008import org.kuali.ole.docstore.model.xmlpojo.ingest.Request;
009import org.kuali.ole.docstore.model.xmlpojo.ingest.RequestDocument;
010import org.kuali.ole.docstore.model.xstream.ingest.RequestHandler;
011import org.kuali.ole.describe.service.DocstoreHelperService;
012import org.kuali.ole.pojo.OleBibRecord;
013import org.kuali.ole.pojo.OleOrderRecord;
014import org.kuali.ole.select.service.impl.OleExposedWebServiceImpl;
015import org.kuali.ole.sys.context.SpringContext;
016import org.kuali.rice.core.api.config.property.ConfigContext;
017import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader;
018import org.kuali.rice.krms.api.engine.EngineResults;
019import org.xml.sax.SAXException;
020
021import java.io.IOException;
022import java.util.ArrayList;
023import java.util.Arrays;
024import java.util.Iterator;
025import java.util.List;
026
027/**
028 * IngestProcessor converts the marcFileContent in to marcXmlContent,ediFileContent in to ediXMLContent and also
029 * creates Requisition and Purchase Order based on oleOrderRecordXml
030 */
031public class IngestProcessor extends AbstractIngestProcessor {
032    /**
033     *   This method modify the marcFileContent in to marcXmlContent.
034     * @param marcFileContent
035     * @return  modifiedXMLContent.
036     */
037    @Override
038    public String preProcessMarc(String marcFileContent) {
039        String marcXMLContent = null;
040        MarcXMLConverter marcXMLConverter = new MarcXMLConverter();
041        marcXMLContent = marcXMLConverter.convert(marcFileContent);
042
043        //TODO: hack to get rid of the extra xmlns entry. Not sure why the second entry gets generated when calling marc4J in ole-docstore-utility.
044        //TODO: the duplicate entry does not get genereated if its run directly in the ole-docstore-utilty project.
045        String modifiedXMLContent =
046                marcXMLContent.
047                        replace("collection xmlns=\"http://www.loc.gov/MARC21/slim\" xmlns=\"http://www.loc.gov/MARC21/slim",
048                                "collection xmlns=\"http://www.loc.gov/MARC21/slim");
049        return modifiedXMLContent;
050    }
051
052    /**
053     *  This method converts the ediFileContent in to ediXMLContent.
054     * @param ediFileContent
055     * @return  ediXMLContent.
056     */
057    @Override
058    public String preProcessEDI(String ediFileContent) {
059        String ediXMLContent = null;
060        OLEEDIConverter oleEDIConverter = new OLEEDIConverter();
061        try {
062            ediXMLContent = oleEDIConverter.convertToXML(ediFileContent);
063        } catch (IOException e) {
064            System.out.println(e.getMessage());
065        } catch (SAXException e) {
066            System.out.println(e.getMessage());
067        }
068        return ediXMLContent;
069    }
070
071    /**
072     * This method creates Requisition and Purchase Order based on oleOrderRecordXml.
073     * The oleExposedWebService will call the createReqAndPO method in oleRice1 and creates the requisition and PO.
074     */
075    @Override
076    public void postProcess() {
077        OleOrderRecords oleOrderRecords = null;
078        try {
079            oleOrderRecords = new OleOrderRecords();
080            List<EngineResults> engineResults = getEngineResults();
081            List<OleOrderRecord> oleOrderRecordList = new ArrayList();
082            for (Iterator<EngineResults> iterator = engineResults.iterator(); iterator.hasNext(); ) {
083                EngineResults results = iterator.next();
084                OleOrderRecord oleOrderRecord = (OleOrderRecord) results.getAttribute(OLEConstants.OLE_ORDER_RECORD);
085                oleOrderRecordList.add(oleOrderRecord);
086            }
087            oleOrderRecords.setRecords(oleOrderRecordList);
088            OleOrderRecordHandler oleEditorResponseHandler = new OleOrderRecordHandler();
089            String oleOrderRecordXml = oleEditorResponseHandler.toXML(oleOrderRecords);
090           OleExposedWebServiceImpl oleExposedWebService = (OleExposedWebServiceImpl)SpringContext.getBean("oleExposedWebService");
091            oleExposedWebService.createReqAndPO(oleOrderRecordXml);
092        } catch (Exception e) {
093            System.out.println(e.getMessage());
094            String xmlForRollback = buildRequestForRollback(oleOrderRecords);
095            DocstoreHelperService docstoreHelperService = GlobalResourceLoader.getService(OLEConstants.DOCSTORE_HELPER_SERVICE);
096            try {
097                docstoreHelperService.rollbackData(xmlForRollback);
098            } catch (Exception e1) {
099                System.out.println(e1.getMessage());
100            }
101        }
102    }
103
104    /**
105     *  This method builds the request to Rollback the LinkedDocs for failure transactions.
106     * @param oleOrderRecords
107     * @return  xml
108     */
109    private String buildRequestForRollback(OleOrderRecords oleOrderRecords) {
110        String xml = null;
111        for (Iterator<OleOrderRecord> iterator = oleOrderRecords.getRecords().iterator(); iterator.hasNext(); ) {
112            OleOrderRecord oleOrderRecord = iterator.next();
113            OleBibRecord oleBibRecord = oleOrderRecord.getOleBibRecord();
114            String bibUUID = oleBibRecord.getBibUUID();
115
116            RequestHandler requestHandler = new RequestHandler();
117            Request request = new Request();
118            request.setOperation("deleteWithLinkedDocs");
119            RequestDocument requestDocument = new RequestDocument();
120            requestDocument.setId(bibUUID);
121            request.setRequestDocuments(Arrays.asList(requestDocument));
122            xml = requestHandler.toXML(request);
123        }
124
125        return xml;
126    }
127
128    /**
129     *  Gets the oleExposedWebService url from PropertyUtil.
130     * @return  url.
131     */
132    public String getURL() {
133        String url = ConfigContext.getCurrentContextConfig().getProperty("oleExposedWebService.url");
134        return url;
135    }
136}