/*

  * Copyright 2007-2008 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 java.util.HashMap;

 import javax.persistence.EntityManager;

 import org.apache.commons.lang.StringUtils;

 import org.apache.ojb.broker.core.proxy.ProxyHelper;

 import org.hibernate.proxy.HibernateProxy;

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

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

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

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

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

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

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

 import org.kuali.rice.kns.util.ObjectUtils;

 import org.springframework.transaction.annotation.Transactional;

 @Transactional

 public class PersistenceDaoProxy implements PersistenceDao {

         private PersistenceDao persistenceDaoJpa;

         private PersistenceDao persistenceDaoOjb;        

         private static KualiModuleService kualiModuleService;

         private static HashMap<String, PersistenceDao> persistenceDaoValues = new HashMap<String, PersistenceDao>();

     public void setPersistenceDaoJpa(PersistenceDao persistenceDaoJpa) {

                 this.persistenceDaoJpa = persistenceDaoJpa;

         }

         public void setPersistenceDaoOjb(PersistenceDao persistenceDaoOjb) {

                 this.persistenceDaoOjb = persistenceDaoOjb;

         }

     private PersistenceDao 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 (persistenceDaoValues.get(dataSourceName) != null) {

                     return persistenceDaoValues.get(dataSourceName);

                 } else {

                         if (OrmUtils.isJpaAnnotated(clazz) && (OrmUtils.isJpaEnabled() || OrmUtils.isJpaEnabled("rice.kns"))) {

                         //return the default JPA values since PersistenceDaoJPA delegates to the BusinessObjectService which should grab the correct datasource

                         persistenceDaoValues.put(dataSourceName, persistenceDaoJpa);

                         return persistenceDaoJpa;

                     } else {

                         //using OJB

                         PersistenceDaoOjb persistDaoOjb = new PersistenceDaoOjb();

                         persistDaoOjb.setJcdAlias(dataSourceName);

                         persistenceDaoValues.put(dataSourceName, persistDaoOjb);

                         return persistDaoOjb;

                     }

                 }

             }

         }

             return (OrmUtils.isJpaAnnotated(clazz) && (OrmUtils.isJpaEnabled() || OrmUtils.isJpaEnabled("rice.kns"))) ?

                                                 persistenceDaoJpa : persistenceDaoOjb; 

     }

         /**

      * @see org.kuali.rice.kns.dao.PersistenceDao#clearCache()

      */

     public void clearCache() {

             //if (OrmUtils.isJpaEnabled()) {

                     persistenceDaoJpa.clearCache();

             //} else {

                     persistenceDaoOjb.clearCache();

             //}

     }

     /**

      * @see org.kuali.rice.kns.dao.PersistenceDao#resolveProxy(java.lang.Object)

      */

     public Object resolveProxy(Object o) {

             return getDao(ObjectUtils.materializeClassForProxiedObject(o)).resolveProxy(o);

     }

     /**

      * @see org.kuali.rice.kns.dao.PersistenceDao#retrieveAllReferences(java.lang.Object)

      */

     public void retrieveAllReferences(Object o) {

             getDao(ObjectUtils.materializeClassForProxiedObject(o)).retrieveAllReferences(o);

     }

     /**

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

      */

     public void retrieveReference(Object o, String referenceName) {

             getDao(ObjectUtils.materializeClassForProxiedObject(o)).retrieveReference(o, referenceName);

     }

     /**

          * Asks proper DAO implementation if the object is proxied

          * 

          * @see org.kuali.rice.kns.dao.PersistenceDao#isProxied(java.lang.Object)

          */

         public boolean isProxied(Object object) {

                 if (object instanceof HibernateProxy) return true;

                 if (ProxyHelper.isProxy(object)) return true;

                 return false;

         }

         private static KualiModuleService getKualiModuleService() {

         if (kualiModuleService == null) {

             kualiModuleService = KNSServiceLocatorWeb.getKualiModuleService();

         }

         return kualiModuleService;

     }

 }