package org.kuali.ole.ingest.action;
   
   import org.kuali.ole.DataCarrierService;
   import org.kuali.ole.OLEConstants;
   import org.kuali.ole.docstore.model.xmlpojo.ingest.Response;
   import org.kuali.ole.docstore.model.xmlpojo.ingest.ResponseDocument;
   import org.kuali.ole.docstore.model.xstream.ingest.ResponseHandler;
   import org.kuali.ole.describe.service.DiscoveryHelperService;
   import org.kuali.ole.describe.service.DocstoreHelperService;
  import org.kuali.ole.ingest.pojo.ProfileAttributeBo;
  import org.kuali.ole.pojo.OleBibRecord;
  import org.kuali.ole.pojo.OleOrderRecord;
  import org.kuali.ole.pojo.bib.BibliographicRecord;
  import org.kuali.ole.pojo.edi.EDIOrder;
  import org.kuali.ole.pojo.edi.LineItemOrder;
  import org.kuali.ole.service.OleOrderRecordService;
  import org.kuali.ole.service.OleOverlayActionService;
  import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader;
  import org.kuali.rice.krms.api.engine.ExecutionEnvironment;
  import org.kuali.rice.krms.framework.engine.Action;
  
  import java.util.HashMap;
  import java.util.Iterator;
  import java.util.List;
  import java.util.Map;
  
  /**
   * CreateBibAction is the action class for BatchIngest(Staff upload screen) which creates a bibliographic
   * record in Docstore.
   */
  public class CreateBibAction implements Action {
      private DocstoreHelperService docstoreHelperService;
      private DiscoveryHelperService discoveryHelperService;
      private OleOrderRecordService oleOrderRecordService;
      private  OleOverlayActionService oleOverlayActionService;
      /**
       *  This method takes the initial request when creating the BibAction.
       * @param executionEnvironment
       */
      @Override
      public void execute(ExecutionEnvironment executionEnvironment) {
          DataCarrierService dataCarrierService = GlobalResourceLoader.getService(OLEConstants.DATA_CARRIER_SERVICE);
          List<ProfileAttributeBo> profileAttributes = (List<ProfileAttributeBo>) dataCarrierService.getData(OLEConstants.PROFILE_ATTRIBUTE_LIST);
          BibliographicRecord bibliographicRecord = (BibliographicRecord) dataCarrierService.getData(OLEConstants.REQUEST_BIB_RECORD);
          LineItemOrder lineItemOrder = (LineItemOrder) dataCarrierService.getData(OLEConstants.REQUEST_LINE_ITEM_ORDER_RECORD);
          EDIOrder ediOrder = (EDIOrder) dataCarrierService.getData(OLEConstants.EDI_ORDER);
          String profileName = (String)dataCarrierService.getData(OLEConstants.PROFILE_NM);
          DocstoreHelperService docstoreHelperService = new DocstoreHelperService();
          try {
              String responseFromDocstore = docstoreHelperService.persistNewToDocstoreForIngest(lineItemOrder,bibliographicRecord, profileAttributes);
              HashMap<String,Object> objects = new HashMap<String, Object>();
              objects.put(OLEConstants.MRC,bibliographicRecord);
              objects.put(OLEConstants.EDI,lineItemOrder);
              ResponseHandler responseHander = new ResponseHandler();
              Response response = responseHander.toObject(responseFromDocstore);
              String instanceUUID = getUUID(response, OLEConstants.INSTANCE_DOC_TYPE);
              if(null == instanceUUID){
                  throw new Exception("instance id returned from docstore is null");
              }
              List bibInfo = getBibInfo(instanceUUID);
              if(bibInfo.isEmpty()){
                  Thread.sleep(2000);
                  bibInfo = getBibInfo(instanceUUID);
              }
              OleBibRecord oleBibRecord = new OleBibRecord();
              Map<String, List> bibFieldValues = (Map<String, List>) bibInfo.get(0);
              oleBibRecord.setBibAssociatedFieldsValueMap(bibFieldValues);
              oleBibRecord.setLinkedInstanceId(instanceUUID);
              oleBibRecord.setBibUUID(getUUID(response, OLEConstants.BIB_DOC_TYPE));
              executionEnvironment.getEngineResults().setAttribute(OLEConstants.BIB_CREATION_FLAG, true);
              OleOrderRecord oleOrderRecord =oleOrderRecordService.fetchOleOrderRecord(executionEnvironment.getEngineResults(), oleBibRecord, ediOrder, bibliographicRecord, profileAttributes);
              oleOverlayActionService.performOverlayLookupAction(profileName,objects,instanceUUID,oleOrderRecord);
              executionEnvironment.getEngineResults().setAttribute(OLEConstants.OLE_ORDER_RECORD, oleOrderRecord);
          } catch (Exception e) {
              e.printStackTrace();
              executionEnvironment.getEngineResults().setAttribute(OLEConstants.OLE_ORDER_RECORD, null);
          }
      }
  
      private List getBibInfo(String instanceUUID) {
          return getDiscoveryHelperService().getBibInformationFromInsatnceId(instanceUUID);
      }
  
      /**
       *   This method simulate the executionEnvironment.
       * @param executionEnvironment
       */
  
      @Override
      public void executeSimulation(ExecutionEnvironment executionEnvironment) {
          execute(executionEnvironment);
      }
  
      /**
       *      This method gets the uuid based on docType
       * @param response
       * @param docType
       * @return  uuid
       */
     private String getUUID(Response response, String docType) {
         List<ResponseDocument> documents = response.getDocuments();
         return getUUID(documents, docType);
     }
 
     /**
    *  This method gets the uuid based on List of documents and docType.
      * @param documents
      * @param docType
      * @return   uuid
      */
     private String getUUID(List<ResponseDocument> documents, String docType) {
         for (Iterator<ResponseDocument> iterator = documents.iterator(); iterator.hasNext(); ) {
             ResponseDocument responseDocument = iterator.next();
             if (responseDocument.getType().equals(docType)) {
                 return responseDocument.getUuid();
             } else {
                 return getUUID(responseDocument.getLinkedDocuments(), docType);
             }
         }
         return null;
     }
 
     /**
      *     Gets the docstoreHelperService attribute.
      * @return  Returns the docstoreHelperService.
      */
     public DocstoreHelperService getDocstoreHelperService() {
         return docstoreHelperService;
     }
 
     /**
      * Sets the docstoreHelperService attribute value.
      * @param docstoreHelperService .The docstoreHelperService to set.
      */
     public void setDocstoreHelperService(DocstoreHelperService docstoreHelperService) {
         this.docstoreHelperService = docstoreHelperService;
     }
 
     /**
      *   Sets the discoveryHelperService attribute value.
      * @param discoveryHelperService .The discoveryHelperService to set.
      */
     public void setDiscoveryHelperService(DiscoveryHelperService discoveryHelperService) {
         this.discoveryHelperService = discoveryHelperService;
     }
 
     /**
      * Gets the discoveryHelperService attribute.
      * @return  Returns discoveryHelperService.
      */
     public DiscoveryHelperService getDiscoveryHelperService() {
         return discoveryHelperService;
     }
 
     public OleOrderRecordService getOleOrderRecordService() {
         return oleOrderRecordService;
     }
 
     public void setOleOrderRecordService(OleOrderRecordService oleOrderRecordService) {
         this.oleOrderRecordService = oleOrderRecordService;
     }
 
     public OleOverlayActionService getOleOverlayActionService() {
         return oleOverlayActionService;
     }
 
     public void setOleOverlayActionService(OleOverlayActionService oleOverlayActionService) {
         this.oleOverlayActionService = oleOverlayActionService;
     }
 }