/*
    * 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.ksb.messaging.quartz;
  
  import org.kuali.rice.core.api.config.ConfigurationException;
  import org.kuali.rice.core.api.config.property.ConfigContext;
  import org.kuali.rice.ksb.service.KSBServiceLocator;
  import org.kuali.rice.ksb.util.KSBConstants;
  import org.quartz.Scheduler;
  import org.quartz.SchedulerException;
  import org.quartz.SchedulerFactory;
  import org.springframework.scheduling.quartz.SchedulerFactoryBean;
  import org.springframework.transaction.PlatformTransactionManager;
  
  import javax.sql.DataSource;
  import javax.transaction.TransactionManager;
  
  /**
   * An implementation of the Quartz SchedulerFactoryBean which uses a database-backed quartz if the useQuartzDatabase property
   * is set.
   * 
   * @author Kuali Rice Team (rice.collab@kuali.org)
   */
  public class KSBSchedulerFactoryBean extends SchedulerFactoryBean {
  
      private PlatformTransactionManager jtaTransactionManager;
      private TransactionManager transactionManager;
      private DataSource dataSource;
      private DataSource nonTransactionalDataSource;
      private boolean transactionManagerSet = false;
      private boolean nonTransactionalDataSourceSet = false;
      private boolean nonTransactionalDataSourceNull = true;
      private boolean dataSourceSet = false;
      private boolean schedulerInjected = false;
      
      @Override
      protected Scheduler createScheduler(SchedulerFactory schedulerFactory, String schedulerName) throws SchedulerException {
          if (ConfigContext.getCurrentContextConfig().getObject(KSBConstants.Config.INJECTED_EXCEPTION_MESSAGE_SCHEDULER_KEY) != null) {
              try {
  	            schedulerInjected = true;
                  Scheduler scheduler = (Scheduler) ConfigContext.getCurrentContextConfig().getObject(KSBConstants.Config.INJECTED_EXCEPTION_MESSAGE_SCHEDULER_KEY);
                  scheduler.addJobListener(new MessageServiceExecutorJobListener());
                  return scheduler;
              } catch (Exception e) {
                  throw new ConfigurationException(e);
              }
          }
          return super.createScheduler(schedulerFactory, schedulerName);
      }
  
      @Override
      public void afterPropertiesSet() throws Exception {
  
          boolean useQuartzDatabase = new Boolean(ConfigContext.getCurrentContextConfig().getProperty(KSBConstants.Config.USE_QUARTZ_DATABASE));
          if (useQuartzDatabase && !schedulerInjected) {
              // require a transaction manager
              if (jtaTransactionManager == null) {
                  throw new ConfigurationException("No jta transaction manager was configured for the KSB Quartz Scheduler");
              }
              setTransactionManager(jtaTransactionManager);
              if (!nonTransactionalDataSourceSet) {
              	// since transaction manager is required... require a non transactional datasource
              	nonTransactionalDataSource = KSBServiceLocator.getMessageNonTransactionalDataSource();
              	if (nonTransactionalDataSource == null) {
              		throw new ConfigurationException("No non-transactional data source was found but is required for the KSB Quartz Scheduler");
              	}
              	super.setNonTransactionalDataSource(nonTransactionalDataSource);
              }
              if (!dataSourceSet) {
              	dataSource = KSBServiceLocator.getMessageDataSource();
              }
              super.setDataSource(dataSource);
              if (transactionManagerSet && nonTransactionalDataSourceNull) {
                  throw new ConfigurationException("A valid transaction manager was set but no non-transactional data source was found");
              }
          }
          super.afterPropertiesSet();
      }
      
      /**
       * This overridden method is used simply to keep track of whether the transactionManager property has been set
       * 
       * @see org.springframework.scheduling.quartz.SchedulerFactoryBean#setTransactionManager(org.springframework.transaction.PlatformTransactionManager)
       */
      @Override
     public void setTransactionManager(PlatformTransactionManager jtaTransactionManager) {
         transactionManagerSet = jtaTransactionManager != null;
         this.jtaTransactionManager = jtaTransactionManager;
     }
     
     /**
      * This overridden method is used to keep track of whether the non transactional data source is null
      * 
      * @see org.springframework.scheduling.quartz.SchedulerFactoryBean#setNonTransactionalDataSource(javax.sql.DataSource)
      */
     @Override
     public void setNonTransactionalDataSource(DataSource nonTransactionalDataSource) {
         nonTransactionalDataSourceSet = true;
         nonTransactionalDataSourceNull = (nonTransactionalDataSource == null);
         this.nonTransactionalDataSource = nonTransactionalDataSource;
     }
     
     @Override
     public void setDataSource(DataSource dataSource) {
     	dataSourceSet = true;
     	this.dataSource = dataSource;
     }
 }