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.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);
  
          DocstoreHelperService docstoreHelperService = new DocstoreHelperService();
          try {
              String responseFromDocstore = docstoreHelperService.persistNewToDocstoreForIngest(bibliographicRecord, profileAttributes);
              ResponseHandler responseHander = new ResponseHandler();
              Response response = responseHander.toObject(responseFromDocstore);
              String instanceUUID = getUUID(response, OLEConstants.INSTANCE_DOC_TYPE);
  
              List bibInfo = getDiscoveryHelperService().getBibInformationFromInsatnceId(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);
          }
      }
  
      /**
       *   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;
     }
 }