/**
    * Copyright 2012 The Kuali Foundation Licensed under the
    * Educational Community License, Version 2.0 (the "License"); you may
    * not use this file except in compliance with the License. You may
    * obtain a copy of the License at
    *
    * http://www.osedu.org/licenses/ECL-2.0
    *
    * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an "AS IS"
   * BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
   * or implied. See the License for the specific language governing
   * permissions and limitations under the License.
   *
   * Created by mharmath on 7/17/12
   */
  package org.kuali.student.enrollment.class2.population.service.impl;
  
  import org.kuali.rice.core.api.criteria.Predicate;
  import org.kuali.rice.core.api.criteria.PredicateFactory;
  import org.kuali.rice.core.api.criteria.QueryByCriteria;
  import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader;
  import org.kuali.rice.krad.lookup.LookupableImpl;
  import org.kuali.rice.krad.web.form.LookupForm;
  import org.kuali.student.enrollment.class2.population.dto.PopulationWrapper;
  import org.kuali.student.r2.common.dto.ContextInfo;
  import org.kuali.student.r2.common.util.ContextUtils;
  import org.kuali.student.r2.core.constants.PopulationServiceConstants;
  import org.kuali.student.r2.core.population.dto.PopulationInfo;
  import org.kuali.student.r2.core.population.dto.PopulationRuleInfo;
  import org.kuali.student.r2.core.population.service.PopulationService;
  
  import javax.xml.namespace.QName;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  /**
   * This class performs lookups on Populations.
   *
   * @author Kuali Student Team
   */
  public class PopulationWrapperLookupableImpl extends LookupableImpl {
  
      private transient PopulationService populationService;
  
      protected List<?> getSearchResults(LookupForm lookupForm, Map<String, String> fieldValues, boolean unbounded) {
          List<PopulationWrapper> populationWrappers = new ArrayList<PopulationWrapper>();
  
          ContextInfo context = ContextUtils.createDefaultContextInfo();
  
          try {
              //perform the lookup using the service
              QueryByCriteria qbc = buildQueryByCriteria(fieldValues);
              List<PopulationInfo> populationInfoList = getPopulationService().searchForPopulations(qbc, context);
  
              //Transform each PopulationInfo to the wrapper class
              for (PopulationInfo populationInfo: populationInfoList) {
                  PopulationRuleInfo populationRuleInfo = getPopulationService().getPopulationRuleForPopulation(populationInfo.getId(), context);
                  PopulationWrapper wrapper = new PopulationWrapper();
                  wrapper.setPopulationRuleInfo(populationRuleInfo);
                  wrapper.setPopulationInfo(populationInfo);
                  wrapper.setId(populationInfo.getId());
                  //set display names
                  wrapper.setPopulationRuleTypeKeyName(populationRuleInfo.getTypeKey().substring(populationRuleInfo.getTypeKey().lastIndexOf('.')+1));
                  wrapper.setPopulationStateKeyName(populationInfo.getStateKey().substring(populationInfo.getStateKey().lastIndexOf('.')+1));
  
                  populationWrappers.add(wrapper);
              }
          } catch (Exception e) {
              throw new RuntimeException("PopulationWrapperLookupableImpl exception. ", e);
          }
  
          return populationWrappers;
      }
  
      /**
       * Builds a QueryByCriteria based on the KRAD field values passed in.
       * Performs fuzzy searching on the keyword field against the name and description fields on PopulationEntity
       *
       * @param fieldValues
       * @return a criteria query
       */
      private QueryByCriteria buildQueryByCriteria(Map<String, String> fieldValues){
          String keyword = fieldValues.get("keyword");
          String stateKey = fieldValues.get("populationInfo.stateKey");
  
          keyword = keyword.isEmpty()?"*":keyword; //search for all if empty
  
          List<Predicate> predicates = new ArrayList<Predicate>();
          predicates.add(PredicateFactory.or(PredicateFactory.like("name", "%"+keyword+"%"),
                                             PredicateFactory.like("descrPlain", "%"+keyword+"%")));
          if (stateKey.equals(PopulationServiceConstants.POPULATION_ACTIVE_STATE_KEY) ||
                  stateKey.equals(PopulationServiceConstants.POPULATION_INACTIVE_STATE_KEY)) {
              predicates.add(PredicateFactory.and(PredicateFactory.equal("populationState", stateKey)));
          }
  
          QueryByCriteria.Builder qbcBuilder = QueryByCriteria.Builder.create();
          qbcBuilder.setPredicates(predicates.toArray(new Predicate[predicates.size()]));
 
         return qbcBuilder.build();
     }
 
     private PopulationService getPopulationService() {
         if(populationService == null) {
             populationService = (PopulationService) GlobalResourceLoader.getService(new QName(PopulationServiceConstants.NAMESPACE, "PopulationService"));
         }
         return this.populationService;
     }
 
 }