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.editor.service.DiscoveryHelperService;
   import org.kuali.ole.editor.service.DocstoreHelperService;
  import org.kuali.ole.ingest.pojo.ProfileAttributeBo;
  import org.kuali.ole.pojo.OleBibRecord;
  import org.kuali.ole.pojo.bib.BibliographicRecord;
  import org.kuali.ole.pojo.edi.LineItemOrder;
  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.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;
  
      /**
       *  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);
  
          DocstoreHelperService docstoreHelperService = new DocstoreHelperService();
          try {
              String responseFromDocstore = docstoreHelperService.persistNewToDocstoreForIngest(lineItemOrder,bibliographicRecord, profileAttributes);
              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.OLE_BIB_RECORD, oleBibRecord);
              executionEnvironment.getEngineResults().setAttribute(OLEConstants.BIB_CREATION_FLAG, true);
          } catch (Exception e) {
              e.printStackTrace();
              executionEnvironment.getEngineResults().setAttribute(OLEConstants.OLE_BIB_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;
     }
 }