/*

  * Copyright 2007-2009 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.opensource.org/licenses/ecl2.php

  *

  * 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.

  */

 package org.kuali.rice.krad.keyvalues;

 import org.apache.commons.lang.StringUtils;

 import org.apache.log4j.Logger;

 import org.kuali.rice.core.api.util.ClassLoaderUtils;

 import org.kuali.rice.core.api.util.ConcreteKeyValue;

 import org.kuali.rice.core.api.util.KeyValue;

 import org.kuali.rice.kim.api.services.KimApiServiceLocator;

 import org.kuali.rice.kim.api.type.KimType;

 import org.kuali.rice.kim.api.type.KimTypeAttribute;

 import org.kuali.rice.kim.api.type.KimTypeService;

 import org.kuali.rice.kim.service.KIMServiceLocatorWeb;

 import org.kuali.rice.krad.datadictionary.AttributeDefinition;

 import org.kuali.rice.krad.datadictionary.BusinessObjectEntry;

 import org.kuali.rice.krad.service.DataDictionaryService;

 import org.kuali.rice.krad.service.KRADServiceLocatorWeb;

 import java.util.ArrayList;

 import java.util.Collections;

 import java.util.List;

 /**

  * @author Kuali Rice Team (rice.collab@kuali.org)

  */

 public class KimAttributeValuesFinder extends KeyValuesBase {

         private static final Logger LOG = Logger.getLogger( KimAttributeValuesFinder.class );

         protected String kimTypeId;

         protected String kimAttributeName;

     private DataDictionaryService dataDictionaryService;

     protected DataDictionaryService getDataDictionaryService() {

                 if ( dataDictionaryService == null ) {

                         dataDictionaryService = KRADServiceLocatorWeb.getDataDictionaryService();

                 }

                 return this.dataDictionaryService;

         }

         /**

          * @see KeyValuesFinder#getKeyValues()

          */

         @Override

         public List<KeyValue> getKeyValues() {

         KimType kimType = KimApiServiceLocator.getKimTypeInfoService().getKimType(kimTypeId);

         if ( kimType != null ) {

                 KimTypeService service = KIMServiceLocatorWeb.getKimTypeService(kimType);

                 if ( service != null ) {

                                 return getAttributeValidValues(kimTypeId,kimAttributeName);

                 } 

                 LOG.error( "Unable to get type service " + kimType.getServiceName() );

         } else {

                 LOG.error( "Unable to obtain KIM type for kimTypeId=" + kimTypeId );

         }

         return Collections.emptyList();

         }

     private List<KeyValue> getAttributeValidValues(String kimTypeId, String attributeName) {

                 if ( LOG.isDebugEnabled() ) {

                         LOG.debug( "getAttributeValidValues(" + kimTypeId + "," + attributeName + ")");

                 }

                 KimTypeAttribute attrib = KimApiServiceLocator.getKimTypeInfoService().getKimType(kimTypeId).getAttributeDefinitionByName(attributeName);

                 if ( LOG.isDebugEnabled() ) {

                         LOG.debug( "Found Attribute definition: " + attrib );

                 }

                 List<KeyValue> pairs = null;

                 if ( StringUtils.isNotBlank(attrib.getKimAttribute().getComponentName()) ) {

                         try {

                                 Class.forName(attrib.getKimAttribute().getComponentName());

                                 try {

                                         pairs = getLocalDataDictionaryAttributeValues(attrib);

                                 } catch ( ClassNotFoundException ex ) {

                                         LOG.error( "Got a ClassNotFoundException resolving a values finder - since this should have been executing in the context of the host system - this should not happen.");

                                         return Collections.emptyList();

                                 }

                         } catch ( ClassNotFoundException ex ) {

                                 LOG.error( "Got a ClassNotFoundException resolving a component name (" + attrib.getKimAttribute().getComponentName() + ") - since this should have been executing in the context of the host system - this should not happen.");

                         }

                 } else {

                         pairs = getCustomValueFinderValues(attrib);

                 }

         return pairs;

         }

     protected List<KeyValue> getCustomValueFinderValues(KimTypeAttribute attrib) {

                 return Collections.emptyList();

         }

     protected List<KeyValue> getLocalDataDictionaryAttributeValues(KimTypeAttribute attr) throws ClassNotFoundException {

                 BusinessObjectEntry entry = getDataDictionaryService().getDataDictionary().getBusinessObjectEntry(attr.getKimAttribute().getComponentName());

                 if ( entry == null ) {

                         LOG.warn( "Unable to obtain BusinessObjectEntry for component name: " + attr.getKimAttribute().getComponentName() );

                         return Collections.emptyList();

                 }

                 AttributeDefinition definition = entry.getAttributeDefinition(attr.getKimAttribute().getAttributeName());

                 if ( definition == null ) {

                         LOG.warn( "No attribute named " + attr.getKimAttribute().getAttributeName() + " found on BusinessObjectEntry for: " + attr.getKimAttribute().getComponentName() );

                         return Collections.emptyList();

                 }

         List<KeyValue> pairs = new ArrayList<KeyValue>();

                 String keyValuesFinderName = definition.getControl().getValuesFinderClass();

                 if ( StringUtils.isNotBlank(keyValuesFinderName)) {

                         try {

                                 KeyValuesFinder finder = (KeyValuesFinder)Class.forName(keyValuesFinderName).newInstance();

                                 if (finder instanceof PersistableBusinessObjectValuesFinder) {

                         ((PersistableBusinessObjectValuesFinder) finder).setBusinessObjectClass(

                             ClassLoaderUtils.getClass(definition.getControl().getBusinessObjectClass()));

                         ((PersistableBusinessObjectValuesFinder) finder).setKeyAttributeName(definition.getControl().getKeyAttribute());

                         ((PersistableBusinessObjectValuesFinder) finder).setLabelAttributeName(definition.getControl().getLabelAttribute());

                         if (definition.getControl().getIncludeBlankRow() != null) {

                                 ((PersistableBusinessObjectValuesFinder) finder).setIncludeBlankRow(definition.getControl().getIncludeBlankRow());

                         }

                         ((PersistableBusinessObjectValuesFinder) finder).setIncludeKeyInDescription(definition.getControl().getIncludeKeyInLabel());

                                 }

                 for (KeyValue pair : finder.getKeyValues()) {

                     pairs.add(new ConcreteKeyValue(pair));

                 }

                         } catch ( ClassNotFoundException ex ) {

                                 LOG.info( "Unable to find class: " + keyValuesFinderName + " in the current context." );

                                 throw ex;

                         } catch (Exception e) {

                                 LOG.error("Unable to build a KeyValuesFinder for " + attr.getKimAttribute().getAttributeName(), e);

                         }

                 } else {

                         LOG.warn( "No values finder class defined on the control definition (" + definition.getControl() + ") on BO / attr = " + attr.getKimAttribute().getComponentName() + " / " + attr.getKimAttribute().getAttributeName() );

                 }

                 return pairs;

         }

         /**

          * @return the kimAttributeName

          */

         public String getKimAttributeName() {

                 return this.kimAttributeName;

         }

         /**

          * @param kimAttributeName the kimAttributeName to set

          */

         public void setKimAttributeName(String kimAttributeName) {

                 this.kimAttributeName = kimAttributeName;

         }

         /**

          * @return the kimTypeId

          */

         public String getKimTypeId() {

                 return this.kimTypeId;

         }

         /**

          * @param kimTypeId the kimTypeId to set

          */

         public void setKimTypeId(String kimTypeId) {

                 this.kimTypeId = kimTypeId;

         }

 }