/*

  * 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.kns.dao.proxy;

 import org.apache.commons.lang.StringUtils;

 import org.kuali.rice.core.api.config.ConfigurationException;

 import org.kuali.rice.core.framework.persistence.jpa.OrmUtils;

 import org.kuali.rice.kns.bo.ModuleConfiguration;

 import org.kuali.rice.kns.dao.LookupDao;

 import org.kuali.rice.kns.dao.impl.LookupDaoJpa;

 import org.kuali.rice.kns.dao.impl.LookupDaoOjb;

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

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

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

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

 import org.springframework.transaction.annotation.Transactional;

 import javax.persistence.EntityManager;

 import java.util.Collection;

 import java.util.Collections;

 import java.util.HashMap;

 import java.util.Map;

 @Transactional

 public class LookupDaoProxy implements LookupDao {

         private LookupDao lookupDaoJpa;

         private LookupDao lookupDaoOjb;

     private static KualiModuleService kualiModuleService;

     private static Map<String, LookupDao> lookupDaoValues = Collections.synchronizedMap(new HashMap<String, LookupDao>());

     public void setLookupDaoJpa(LookupDao lookupDaoJpa) {

                 this.lookupDaoJpa = lookupDaoJpa;

         }

         public void setLookupDaoOjb(LookupDao lookupDaoOjb) {

                 this.lookupDaoOjb = lookupDaoOjb;

         }

     private LookupDao getDao(Class clazz) {

         ModuleService moduleService = getKualiModuleService().getResponsibleModuleService(clazz);

         if (moduleService != null) {

             ModuleConfiguration moduleConfig = moduleService.getModuleConfiguration();

             String dataSourceName = "";

             EntityManager entityManager = null;

             if (moduleConfig != null) {

                 dataSourceName = moduleConfig.getDataSourceName();

                 entityManager = moduleConfig.getEntityManager();

             }

             if (StringUtils.isNotEmpty(dataSourceName)) {

                 if (lookupDaoValues.get(dataSourceName) != null) {

                     return lookupDaoValues.get(dataSourceName);

                 } else {         

                     if (OrmUtils.isJpaAnnotated(clazz) && OrmUtils.isJpaEnabled()) {

                         //using JPA               

                             LookupDaoJpa classSpecificLookupDaoJpa = new LookupDaoJpa();

                                 if (entityManager != null) {

                                         classSpecificLookupDaoJpa.setEntityManager(entityManager);

                                         classSpecificLookupDaoJpa.setPersistenceStructureService(KNSServiceLocator.getPersistenceStructureService());

                                 classSpecificLookupDaoJpa.setDateTimeService(KNSServiceLocator.getDateTimeService());

                                         lookupDaoValues.put(dataSourceName, classSpecificLookupDaoJpa);

                                         return classSpecificLookupDaoJpa;

                                 } else {

                                         throw new ConfigurationException("EntityManager is null. EntityManager must be set in the Module Configuration bean in the appropriate spring beans xml. (see nested exception for details).");

                                 }

                                         } else {

                                                 LookupDaoOjb classSpecificLookupDaoOjb = new LookupDaoOjb();

                         classSpecificLookupDaoOjb.setJcdAlias(dataSourceName);

                         classSpecificLookupDaoOjb.setPersistenceStructureService(KNSServiceLocator.getPersistenceStructureService());

                         classSpecificLookupDaoOjb.setDateTimeService(KNSServiceLocator.getDateTimeService());

                         classSpecificLookupDaoOjb.setBusinessObjectDictionaryService(KNSServiceLocatorWeb.getBusinessObjectDictionaryService());

                         lookupDaoValues.put(dataSourceName, classSpecificLookupDaoOjb);

                         return classSpecificLookupDaoOjb;

                     }

                 }

             }

         }

         //return lookupDaoJpa;

         return (OrmUtils.isJpaAnnotated(clazz) && OrmUtils.isJpaEnabled()) ? lookupDaoJpa : lookupDaoOjb;

     }

         /**

          * @see org.kuali.rice.kns.dao.LookupDao#createCriteria(java.lang.Object, java.lang.String, java.lang.String, java.lang.Object)

          */

         public boolean createCriteria(Object example, String searchValue, String propertyName, Object criteria) {

                 return getDao(example.getClass()).createCriteria(example, searchValue, propertyName, criteria);

         }

         /**

          * @see org.kuali.rice.kns.dao.LookupDao#createCriteria(java.lang.Object, java.lang.String, java.lang.String, boolean, java.lang.Object)

          */

         public boolean createCriteria(Object example, String searchValue, String propertyName, boolean caseInsensitive, boolean treatWildcardsAndOperatorsAsLiteral, Object criteria) {

                 return getDao(example.getClass()).createCriteria(example, searchValue, propertyName, caseInsensitive, treatWildcardsAndOperatorsAsLiteral, criteria);

         }

         /**

          * @see org.kuali.rice.kns.dao.LookupDao#findCollectionBySearchHelper(java.lang.Class, java.util.Map, boolean, boolean)

          */

         public Collection findCollectionBySearchHelper(Class example, Map formProps, boolean unbounded, boolean usePrimaryKeyValuesOnly) {

                 return getDao(example).findCollectionBySearchHelper(example, formProps, unbounded, usePrimaryKeyValuesOnly);

         }

         /**

          * @see org.kuali.rice.kns.dao.LookupDao#findCollectionBySearchHelper(java.lang.Class, java.util.Map, boolean, boolean, java.lang.Object)

          */

         public Collection findCollectionBySearchHelper(Class example, Map formProps, boolean unbounded, boolean usePrimaryKeyValuesOnly, Object additionalCriteria) {

                 return getDao(example).findCollectionBySearchHelper(example, formProps, unbounded, usePrimaryKeyValuesOnly, additionalCriteria);

         }

         /**

          * @see org.kuali.rice.kns.dao.LookupDao#findCountByMap(java.lang.Object, java.util.Map)

          */

         public Long findCountByMap(Object example, Map formProps) {

                 return getDao(example.getClass()).findCountByMap(example, formProps);

         }

         /**

          * @see org.kuali.rice.kns.dao.LookupDao#findObjectByMap(java.lang.Object, java.util.Map)

          */

         public Object findObjectByMap(Object example, Map formProps) {

                 return getDao(example.getClass()).findObjectByMap(example, formProps);

         }

         private static KualiModuleService getKualiModuleService() {

         if (kualiModuleService == null) {

             kualiModuleService = KNSServiceLocatorWeb.getKualiModuleService();

         }

         return kualiModuleService;

     }

 }