/**
    * Copyright 2005-2016 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.app.persistence.jpa;
  
  import org.springframework.orm.jpa.persistenceunit.MutablePersistenceUnitInfo;
  import org.springframework.orm.jpa.persistenceunit.PersistenceUnitPostProcessor;
  
  import javax.sql.DataSource;
  
  public class RicePersistenceUnitPostProcessor implements PersistenceUnitPostProcessor {
  	static org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(RicePersistenceUnitPostProcessor.class);
  	
  	public static final String KRAD_APPLICATION_PERSISTENCE_UNIT_NAME = "krad-application-unit";
  	public static final String KRAD_SERVER_PERSISTENCE_UNIT_NAME = "krad-server-unit";
  
  	private DataSource jtaDataSource;
      
      public void postProcessPersistenceUnitInfo(MutablePersistenceUnitInfo mutablePersistenceUnitInfo) {
          mutablePersistenceUnitInfo.setJtaDataSource(getJtaDataSource());
          addKRADManagedClassNames(mutablePersistenceUnitInfo);
          if (mutablePersistenceUnitInfo.getPersistenceUnitName().equals(KRAD_APPLICATION_PERSISTENCE_UNIT_NAME) || mutablePersistenceUnitInfo.getPersistenceUnitName().equals(
                  KRAD_SERVER_PERSISTENCE_UNIT_NAME)) {
          	addRiceManagedClassNamesToKRADPersistenceUnit(mutablePersistenceUnitInfo);
          }
      }
      
      /**
       * 
       * Adds all the KNS Managed entities to the persistence unit - which is important, becuase all
       * persistence units get the KNS entities to manage
       * 
       * @param mutablePersistenceUnitInfo
       */
      public void addKRADManagedClassNames(MutablePersistenceUnitInfo mutablePersistenceUnitInfo) {
      	addManagedClassNames(mutablePersistenceUnitInfo, new KRADPersistableBusinessObjectClassExposer());
      }
      
      /**
       * Adds the class names listed by exposed by the given exposer into the persistence unit
       * 
       * @param mutablePersistenceUnitInfo the persistence unit to add managed JPA entity class names to
       * @param exposer the exposer for class names to manage
       */
      public void addManagedClassNames(MutablePersistenceUnitInfo mutablePersistenceUnitInfo, PersistableBusinessObjectClassExposer exposer) {
      	for (String exposedClassName : exposer.exposePersistableBusinessObjectClassNames()) {
      		if (LOG.isDebugEnabled()) {
      			LOG.debug("JPA will now be managing class: "+exposedClassName);
      		}
      		mutablePersistenceUnitInfo.addManagedClassName(exposedClassName);
      	}
      }
      
      public void addRiceManagedClassNamesToKRADPersistenceUnit(MutablePersistenceUnitInfo mutablePersistenceUnitInfo) {
      	addManagedClassNames(mutablePersistenceUnitInfo, new RiceToNervousSystemBusinessObjectClassExposer());
      }
  
      public DataSource getJtaDataSource() {
          return jtaDataSource;
      }
  
      public void setJtaDataSource(DataSource jtaDataSource) {
          this.jtaDataSource = jtaDataSource;
      }
  	
  }