/*
    *    Copyright 2009-2023 the original author or authors.
    *
    *    Licensed under the Apache 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
    *
    *       https://www.apache.org/licenses/LICENSE-2.0
    *
   *    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.mybatis.guice;
  
  import static com.google.inject.matcher.Matchers.annotatedWith;
  import static com.google.inject.matcher.Matchers.any;
  import static com.google.inject.matcher.Matchers.not;
  import static org.mybatis.guice.Preconditions.checkArgument;
  
  import jakarta.inject.Provider;
  import jakarta.transaction.TransactionManager;
  
  import javax.transaction.xa.XAResource;
  
  import org.apache.ibatis.logging.Log;
  import org.apache.ibatis.logging.LogFactory;
  import org.apache.ibatis.transaction.TransactionFactory;
  import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
  import org.apache.ibatis.transaction.managed.ManagedTransactionFactory;
  import org.mybatis.guice.transactional.Transactional;
  import org.mybatis.guice.transactional.TransactionalMethodInterceptor;
  import org.mybatis.guice.transactional.TxTransactionalMethodInterceptor;
  import org.mybatis.guice.transactional.XASqlSessionManagerProvider;
  
  public abstract class MyBatisJtaModule extends MyBatisModule {
    private final Log log = LogFactory.getLog(getClass());
  
    private TransactionManager transactionManager;
    private Class<? extends Provider<? extends XAResource>> xaResourceProvider = XASqlSessionManagerProvider.class;
  
    public MyBatisJtaModule() {
    }
  
    public MyBatisJtaModule(TransactionManager transactionManager) {
      this.transactionManager = transactionManager;
    }
  
    @Override
    protected void bindTransactionInterceptors() {
      TransactionManager manager = getTransactionManager();
  
      if (manager == null) {
        log.debug("bind default transaction interceptors");
        super.bindTransactionInterceptors();
      } else {
        log.debug("bind XA transaction interceptors");
  
        // transactional interceptor
        TransactionalMethodInterceptor interceptor = new TransactionalMethodInterceptor();
        requestInjection(interceptor);
  
        // jta transactional interceptor
        TxTransactionalMethodInterceptor interceptorTx = new TxTransactionalMethodInterceptor();
        requestInjection(interceptorTx);
        bind(XAResource.class).toProvider(xaResourceProvider);
  
        bind(TransactionManager.class).toInstance(manager);
  
        bindInterceptor(any(), not(DECLARED_BY_OBJECT).and(annotatedWith(Transactional.class)), interceptorTx,
            interceptor);
        // Intercept classes annotated with Transactional, but avoid "double"
        // interception when a mathod is also annotated inside an annotated
        // class.
        bindInterceptor(annotatedWith(Transactional.class),
            not(DECLARED_BY_OBJECT).and(not(annotatedWith(Transactional.class))), interceptorTx, interceptor);
      }
    }
  
    protected TransactionManager getTransactionManager() {
      return transactionManager;
    }
  
    protected void setTransactionManager(TransactionManager transactionManager) {
      this.transactionManager = transactionManager;
    }
  
    protected void bindDefaultTransactionProvider() {
      Class<? extends TransactionFactory> factoryType = getTransactionManager() == null ? JdbcTransactionFactory.class
          : ManagedTransactionFactory.class;
  
      bindTransactionFactoryType(factoryType);
    }
  
    protected void bindXAResourceProvider(Class<? extends Provider<? extends XAResource>> xaResourceProvider) {
      checkArgument(xaResourceProvider != null, "Parameter 'xaResourceProvider' must be not null");
      this.xaResourceProvider = xaResourceProvider;
   }
 
   protected static class ProviderImpl<T> implements Provider<T> {
     private T wrapper;
 
     public ProviderImpl(T wrapper) {
       this.wrapper = wrapper;
     }
 
     @Override
     public T get() {
       return wrapper;
     }
 
   }
 }