/*

  * Copyright 2006-2011 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.krms.impl.repository;

 import java.util.ArrayList;

 import java.util.Collection;

 import java.util.Collections;

 import java.util.HashMap;

 import java.util.List;

 import java.util.Map;

 import javax.swing.text.html.HTMLDocument.Iterator;

 import org.apache.commons.collections.CollectionUtils;

 import org.apache.commons.lang.StringUtils;

 import org.kuali.rice.kns.service.BusinessObjectService;

 import org.kuali.rice.kns.service.KNSServiceLocator;

 import org.kuali.rice.krms.api.repository.type.KrmsAttributeDefinition;

 public final class KrmsAttributeDefinitionServiceImpl implements KrmsAttributeDefinitionService {

     private BusinessObjectService businessObjectService;

     private final Map<String,KrmsAttributeDefinitionBo> krmsAttributeDefinitionIdCache = Collections.synchronizedMap( new HashMap<String,KrmsAttributeDefinitionBo>() );

         @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#convertAttributeKeys()

         */

         public Map<String,String> convertAttributeKeys(Map<String,String> attributesByName, String namespace) {

                 Map<String,String> attributesById = new HashMap<String,String>();

                 if(attributesByName != null) {

                         for(Map.Entry<String, String> attr : attributesByName.entrySet()) {

                                 String newKey = getKrmsAttributeId(attr.getKey(), namespace);

                                 if(StringUtils.isNotEmpty(newKey)) {

                                         attributesById.put(newKey, attr.getValue());

                                 }

                         }

                 }

                 return attributesById;

         }

         @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#getKrmsAttributeId()

         */

         public String getKrmsAttributeId( String attributeName, String namespace) {

                 String returnId = null;

                 KrmsAttributeDefinitionBo bo = getKrmsAttributeBo(attributeName, namespace);

                 if (bo != null){

                         returnId = bo.getId();

                 }

                 return returnId;

         }

         @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#getKrmsAttributeBo()

         */

         public KrmsAttributeDefinitionBo getKrmsAttributeBo( String attributeName, String namespace) {

                 String key = createKey(namespace, attributeName);

                 if (krmsAttributeDefinitionIdCache.containsKey(key))

                         return krmsAttributeDefinitionIdCache.get(key);                

                 KrmsAttributeDefinitionBo result = null;

                 Map<String,Object> criteria = new HashMap<String,Object>( 3 );

                 criteria.put( "name", attributeName );

                 criteria.put( "namespace", namespace );

         criteria.put( "active", Boolean.TRUE );

                 Collection<KrmsAttributeDefinitionBo> defs = getBusinessObjectService().findMatching( KrmsAttributeDefinitionBo.class, criteria );

                 if(CollectionUtils.isNotEmpty(defs)) {

                         if (defs.size() > 1){

                                 throw new IllegalStateException("Multiple KrmsAttributeDefinitions found with same name and namespace");

                         }

                         result = defs.iterator().next();

                         krmsAttributeDefinitionIdCache.put(key, result);

                 }

                 return result;

         }

         @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#createAttributeDefinition()

         */

         public KrmsAttributeDefinition createAttributeDefinition(KrmsAttributeDefinition attributeDefinition) {

                 if (attributeDefinition == null){

                 throw new IllegalArgumentException("attributeDefinition is null");

                 }

                 final String nameKey = attributeDefinition.getName();

                 final String namespaceKey = attributeDefinition.getNamespace();

                 final KrmsAttributeDefinition existing = getAttributeDefinitionByNameAndNamespace(nameKey, namespaceKey);

                 if (existing != null && existing.getName().equals(nameKey) && existing.getNamespace().equals(namespaceKey)){

             throw new IllegalStateException("the krms attribute definition to create already exists: " + attributeDefinition);                        

                 }

                 KrmsAttributeDefinitionBo bo = KrmsAttributeDefinitionBo.from(attributeDefinition);

                 getBusinessObjectService().save(bo);

                 updateCache(bo);

                 return KrmsAttributeDefinitionBo.to(bo);

         }

         @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#updateAttributeDefinition()

         */

         public void updateAttributeDefinition(KrmsAttributeDefinition attributeDefinition) {

                 if (attributeDefinition == null){

                 throw new IllegalArgumentException("attributeDefinition is null");

                 }

                 final String idKey = attributeDefinition.getId();

                 final KrmsAttributeDefinitionBo existing = getBusinessObjectService().findBySinglePrimaryKey(KrmsAttributeDefinitionBo.class, idKey);

                 if (existing == null){

             throw new IllegalStateException("the krms attribute definition does not exist: " + attributeDefinition);                        

                 }

                 final KrmsAttributeDefinition toUpdate;

                 if (!existing.getId().equals(attributeDefinition.getId())){

                         final KrmsAttributeDefinition.Builder builder = KrmsAttributeDefinition.Builder.create(attributeDefinition);

                         builder.setId(existing.getId());

                         toUpdate = builder.build();

                 } else {

                         toUpdate = attributeDefinition;

                 }

                 KrmsAttributeDefinitionBo bo = KrmsAttributeDefinitionBo.from(toUpdate);

                 getBusinessObjectService().save(bo);

                 updateCache(bo);

         }

         @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#getAttributeDefinitionById()

         */

     public KrmsAttributeDefinition getAttributeDefinitionById(final String id) {

         if (StringUtils.isBlank(id)) {

             throw new IllegalArgumentException("id is blank");

         }

                 if (krmsAttributeDefinitionIdCache.containsKey(id)){

                         return KrmsAttributeDefinitionBo.to(krmsAttributeDefinitionIdCache.get(id));        

                 } 

         KrmsAttributeDefinitionBo bo = getBusinessObjectService().findBySinglePrimaryKey(KrmsAttributeDefinitionBo.class, id);

         return KrmsAttributeDefinitionBo.to(bo);

     }

     @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#getAttributeDefinitionByNameAndNamespace()

         */

     public KrmsAttributeDefinition getAttributeDefinitionByNameAndNamespace(final String name, final String namespace) {

         if (StringUtils.isBlank(name)) {

             throw new IllegalArgumentException("name is blank");

         }

         if (StringUtils.isBlank(namespace)) {

             throw new IllegalArgumentException("namespace is blank");

         }

         KrmsAttributeDefinitionBo bo = getKrmsAttributeBo(name, namespace);

         if (bo == null) 

                 return null;

                return KrmsAttributeDefinitionBo.to(bo);

     }

     @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#findAttributeDefinitionsByNamespace()

         */

     public List<KrmsAttributeDefinition> findAttributeDefinitionsByNamespace(final String namespace) {

         final Map<String, Object> map = new HashMap<String, Object>();

         map.put("namespace", namespace);

         map.put("active", Boolean.TRUE);

         Collection<KrmsAttributeDefinitionBo> krmsAttributeDefinitionBos = getBusinessObjectService().findMatching(KrmsAttributeDefinitionBo.class, Collections.unmodifiableMap(map));

         return convertListOfBosToImmutables(krmsAttributeDefinitionBos);

     }

     @Override

         /**

         * @see org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService#findAllAttributeDefinitions()

         */

     public List<KrmsAttributeDefinition> findAllAttributeDefinitions() {

         final Map<String, Object> map = new HashMap<String, Object>();

         map.put("active", Boolean.TRUE);

         Collection<KrmsAttributeDefinitionBo> krmsAttributeDefinitionBos = getBusinessObjectService().findMatching(KrmsAttributeDefinitionBo.class, Collections.unmodifiableMap(map));

         return convertListOfBosToImmutables(krmsAttributeDefinitionBos);

     }

     /**

      * Sets the businessObjectService attribute value.

      *

      * @param businessObjectService The businessObjectService to set.

      */

     public void setBusinessObjectService(final BusinessObjectService businessObjectService) {

         this.businessObjectService = businessObjectService;

     }

     /**

      * This method returns a reference to the businessObjectService.

      * If the businessObjectService is not set, get it from the KNSServiceLocator.

      * @return businessObjectService

      */

     protected BusinessObjectService getBusinessObjectService() {

                 if ( businessObjectService == null ) {

                         businessObjectService = KNSServiceLocator.getBusinessObjectService();

                 }

                 return businessObjectService;

         }

     /**

      * Converts a List<KrmsAttributeDefinitionBo> to an Unmodifiable List<KrmsAttributeDefinition>

      *

      * @param krmsAttributeDefinitionBos a mutable List<KrmsAttributeDefinitionBo> to made completely immutable.

      * @return An unmodifiable List<KrmsAttributeDefinition>

      */

     public List<KrmsAttributeDefinition> convertListOfBosToImmutables(final Collection<KrmsAttributeDefinitionBo> krmsAttributeDefinitionBos) {

         ArrayList<KrmsAttributeDefinition> krmsAttributeDefinitions = new ArrayList<KrmsAttributeDefinition>();

         for (KrmsAttributeDefinitionBo bo : krmsAttributeDefinitionBos) {

             KrmsAttributeDefinition krmsAttributeDefinition = KrmsAttributeDefinitionBo.to(bo);

             krmsAttributeDefinitions.add(krmsAttributeDefinition);

         }

         return Collections.unmodifiableList(krmsAttributeDefinitions);

     }

     // Cache related methods

     public void clearCache(){

             krmsAttributeDefinitionIdCache.clear();

     }

     private String createKey(String namespace, String name){

             String key = namespace + ":" + name;

             return key;

     }

     private void updateCache(KrmsAttributeDefinitionBo bo){

                 String key = createKey(bo.getNamespace(), bo.getName());

                 if (bo.isActive()){

                         krmsAttributeDefinitionIdCache.put(key, bo);

                 } else {

                         krmsAttributeDefinitionIdCache.remove(key);

                 }

     }

 }