package org.kuali.ole.docstore.discovery.service;
   
   import com.thoughtworks.xstream.XStream;
   import com.thoughtworks.xstream.io.json.JettisonMappedXmlDriver;
   import org.apache.solr.client.solrj.SolrQuery;
   import org.apache.solr.client.solrj.SolrServer;
   import org.apache.solr.client.solrj.response.QueryResponse;
   import org.apache.solr.common.SolrDocument;
   import org.kuali.ole.docstore.model.xmlpojo.work.instance.oleml.InstanceCollection;
  import org.kuali.ole.repository.CheckoutManager;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import java.util.List;
  
  /**
   * Created with IntelliJ IDEA.
   * User: ?
   * Date: 2/7/13
   * Time: 5:42 PM
   * To change this template use File | Settings | File Templates.
   */
  public class OleDocstoreDataRetrieveService {
      private static final Logger LOG = LoggerFactory.getLogger(OleDocstoreDataRetrieveService.class);
  
      /**
       * @param bibUUIDs
       * @param contentType
       * @return the instance xml for the list of bibuuids
       * @throws Exception
       */
      public String getInstanceDetails(List<String> bibUUIDs, String contentType) {
          StringBuffer output = new StringBuffer();
          try {
              for (int i = 0; i < bibUUIDs.size(); i++) {
                  SolrQuery solrQuery = new SolrQuery();
                  solrQuery.setQuery("id:" + bibUUIDs.get(i));
                  solrQuery.setRows(500);
                  SolrServer server = SolrServerManager.getInstance().getSolrServer();
                  QueryResponse response = server.query(solrQuery);
                  List<SolrDocument> solrDocumentList = response.getResults();
                  for (SolrDocument solrDocument : solrDocumentList) {
                      if (solrDocument.getFieldValue("instanceIdentifier") instanceof List) {
                          List<String> instanceIdentifierList = (List<String>) solrDocument.getFieldValue("instanceIdentifier");
                          for (String instanceIdentifier : instanceIdentifierList) {
                              LOG.info("Instant Identifier Value is :" + instanceIdentifier);
                              CheckoutManager checkoutManager = new CheckoutManager();
                              String instanceOutput = checkoutManager.checkOut(instanceIdentifier, null, "checkout");
                              if (contentType.equals("xml")) {
                                  output.append(generateXmlInstanceContent(instanceOutput));
                              } else if (contentType.equals("json")) {
                                  InstanceCollection instanceCollection = getInstanceCollection(instanceOutput);
                                  output.append(generateJsonInstanceContent(instanceCollection));
                              }
                          }
                      } else {
                          String instanceIdentifier = (String) solrDocument.getFieldValue("instanceIdentifier");
                          LOG.info("Instant Identifier Value is :" + instanceIdentifier);
                          CheckoutManager checkoutManager = new CheckoutManager();
                          String instanceOutput = checkoutManager.checkOut(instanceIdentifier, null, "checkout");
                          if (contentType.equals("xml")) {
                              output.append(generateXmlInstanceContent(instanceOutput));
                          } else if (contentType.equals("json")) {
                              InstanceCollection instanceCollection = getInstanceCollection(instanceOutput);
                              output.append(generateJsonInstanceContent(instanceCollection));
                          }
  
                      }
                  }
              }
              LOG.info("Instance Output :" + output.toString());
  
          } catch (Exception e) {
              e.printStackTrace();
          }
          return output.toString();
      }
  
      /**
       * @param instanceData
       * @return the instance Collection object from the given xml content
       */
      public InstanceCollection getInstanceCollection(String instanceData) {
          XStream xs = new XStream();
          xs.autodetectAnnotations(true);
          xs.processAnnotations(InstanceCollection.class);
          InstanceCollection instanceCollection = (InstanceCollection) xs.fromXML(instanceData);
          return instanceCollection;
      }
  
  
      /**
       * @param instanceCollection
       * @return the json format of  the instance collection object
       */
      public String generateJsonInstanceContent(InstanceCollection instanceCollection) {
          XStream stream = new XStream(new JettisonMappedXmlDriver());
          stream.autodetectAnnotations(true);
          stream.processAnnotations(InstanceCollection.class);
         stream.setMode(XStream.NO_REFERENCES);
         String jsonContent = stream.toXML(instanceCollection);
         return jsonContent;
     }
 
     /**
      *
      * @param instanceContent
      * @return   the required xml format content.
      */
     public String generateXmlInstanceContent(String instanceContent) {
         InstanceCollection instanceCollection = getInstanceCollection(instanceContent);
         OleInstanceXmlConverterService oleInstanceXmlConverterService = new OleInstanceXmlConverterService();
         String instanceXml = oleInstanceXmlConverterService.generateInstanceCollectionsXml(instanceCollection);
         return instanceXml;
     }
 
 }