package org.kuali.ole.service;
   
   import org.apache.commons.lang.StringUtils;
   import org.kuali.ole.docstore.discovery.service.QueryServiceImpl;
   import org.kuali.ole.docstore.model.bo.OleDocument;
   import org.kuali.ole.docstore.model.bo.WorkBibDocument;
   import org.kuali.ole.docstore.model.bo.WorkInstanceDocument;
   import org.kuali.ole.select.bo.OleAgreementSearch;
   
  import java.util.ArrayList;
  import java.util.Iterator;
  import java.util.List;
  import java.util.Map;
  
  /**
   *OleAgreementSearchService performs search operation and return list of agreement related information.
   */
  public class OleAgreementSearchService {
      private final String queryString = "(DocType:license AND DocFormat:onixpl)";
  
  
      /**
       *  This method returns the responses from Solr.
       * @param queryField
       * @param value
       * @return  List
       */
      public List getResponseFromSOLR(String queryField, String value) {
          String queryString = queryField + ":" + value;
          return QueryServiceImpl.getInstance().retriveResults(queryString);
      }
  
      /**
       *  This method returns list of agreement information based on search criteria.
       * @param searchCriteria
       * @return   List<OleAgreementSearch>
       */
      public List<OleAgreementSearch> getAgreementInformation(Map searchCriteria) {
          List<OleAgreementSearch> agreementSearchResults = new ArrayList<OleAgreementSearch>();
          OleAgreementSearch agreemetSearch;
          List solrResponse = getSolrResponse(searchCriteria);
          if(solrResponse.size() > 0) {
              Iterator listIterator = solrResponse.iterator();
              while(listIterator.hasNext()) {
                  Map results = (Map)listIterator.next();
                  agreemetSearch = new OleAgreementSearch();
                  ArrayList titleValue = (ArrayList<String>)results.get("Title_search");
                  if(titleValue != null && titleValue.size() > 0) {
                      String title = titleValue.toString();
                      agreemetSearch.setAgreementTitle(StringUtils.substring(title,1, title.length() - 1));
                  }
                  ArrayList contractNumValue = (ArrayList<String>)results.get("ContractNumber_search");
                  if(contractNumValue != null && contractNumValue.size() > 0) {
                      String contractNum = contractNumValue.toString();
                      agreemetSearch.setContractNumber(StringUtils.substring(contractNum,1, contractNum.length() - 1));
                  }
                  /*ArrayList licenseeValue = (ArrayList<String>)results.get("Licensee_search");
                  if(licenseeValue.size() > 0) {
                      String licensee = licenseeValue.toString();
                      agreemetSearch.setLicensee(licensee);
                  }
                  ArrayList licensorValue = (ArrayList<String>)results.get("Licensor_search");
                  if(licensorValue.size() > 0) {
                      String licensor = licensorValue.toString();
                      agreemetSearch.setLicensor(licensor);
                  }*/
                  ArrayList methodValue = (ArrayList<String>)results.get("Method_search");
                  if(methodValue != null && methodValue.size() > 0) {
                      String agrMethod = methodValue.toString();
                      agreemetSearch.setMethodName(StringUtils.substring(agrMethod,1, agrMethod.length() - 1));
                  }
                  ArrayList typeValue = (ArrayList<String>)results.get("Type_search");
                  if(typeValue!= null && typeValue.size() > 0) {
                      String agrType = typeValue.toString();
                      agreemetSearch.setType(StringUtils.substring(agrType,1, agrType.length() - 1));
                  }
                  ArrayList statusValue = (ArrayList<String>)results.get("Status_search");
                  if(statusValue != null && statusValue.size() > 0) {
                      String agrStatus = statusValue.toString();
                      agreemetSearch.setStatus(StringUtils.substring(agrStatus,1, agrStatus.length() - 1));
                  }
                  agreemetSearch.setUniqueId((String)results.get("uniqueId"));
                  agreementSearchResults.add(agreemetSearch);
              }
          }
          return agreementSearchResults;
      }
  
      /**
       *  This method returns the SolrResponse as List based on searchCriteria.
       * @param searchCriteria
       * @return  List
       */
      private List getSolrResponse(Map searchCriteria) {
          String query = queryString;
          String key = null;
          String value = null;
          if(!searchCriteria.isEmpty()) {
              OleDocument oleDocument = new WorkBibDocument();
             WorkInstanceDocument workInstance = new WorkInstanceDocument();
             if ((searchCriteria.containsKey("agreementTitle")) &&
                     searchCriteria.get("agreementTitle") != null & !searchCriteria.get("agreementTitle").equals("")) {
                 query = query + " AND Title_search:" +searchCriteria.get("agreementTitle");
             }
             if ((searchCriteria.containsKey("contractNumber")) &&
                         searchCriteria.get("contractNumber") != null & !searchCriteria.get("contractNumber").equals("")) {
                 query = query + " AND ContractNumber_search:" +searchCriteria.get("contractNumber");
             }
             /*if ((searchCriteria.containsKey("licensee")) &&
                     searchCriteria.get("licensee") != null & !searchCriteria.get("licensee").equals("")){
                 query = query + " AND Licensee_search:" +searchCriteria.get("licensee");
             }
             if ((searchCriteria.containsKey("licensor")) &&
                     searchCriteria.get("licensor") != null & !searchCriteria.get("licensor").equals("")) {
                 query = query + " AND Licensor_search:" +searchCriteria.get("licensor");
             }*/
             //query = query+"&fl=id,Title_search,ContractNumber_search,Licensee_search,Licensor_search";
             if(searchCriteria.containsKey("uuid")) {
                 query = query + "id:"+ searchCriteria.get("uuid");
             }
         }
         return QueryServiceImpl.getInstance().retriveResults(query);
     }
 
 
 }