/*
    *    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.datasource.dbcp;
  
  import com.google.inject.AbstractModule;
  import com.google.inject.Scopes;
  import com.google.inject.TypeLiteral;
  
  import jakarta.inject.Provider;
  
  import java.lang.annotation.Annotation;
  import java.util.Map;
  
  public final class PerUserPoolDataSourceModule extends AbstractModule {
  
    private static final TypeLiteral<Map<String, Boolean>> STRING_BOOLEAN_MAP = new TypeLiteral<Map<String, Boolean>>() {
    };
  
    private static final TypeLiteral<Map<String, Integer>> STRING_INTEGER_MAP = new TypeLiteral<Map<String, Integer>>() {
    };
  
    private static final TypeLiteral<Map<String, Long>> STRING_LONG_MAP = new TypeLiteral<Map<String, Long>>() {
    };
  
    private final Class<? extends Provider<Map<String, Boolean>>> perUserDefaultAutoCommitProviderClass;
  
    private final Class<? extends Provider<Map<String, Boolean>>> perUserDefaultReadOnlyProviderClass;
  
    private final Class<? extends Provider<Map<String, Integer>>> perUserDefaultTransactionIsolationProviderClass;
  
    private final Class<? extends Provider<Map<String, Integer>>> perUserMaxActiveProviderClass;
  
    private final Class<? extends Provider<Map<String, Integer>>> perUserMaxIdleProviderClass;
  
    private final Class<? extends Provider<Map<String, Long>>> perUserMaxWaitMillisProviderClass;
  
    private PerUserPoolDataSourceModule(
        Class<? extends Provider<Map<String, Boolean>>> perUserDefaultAutoCommitProviderClass,
        Class<? extends Provider<Map<String, Boolean>>> perUserDefaultReadOnlyProviderClass,
        Class<? extends Provider<Map<String, Integer>>> perUserDefaultTransactionIsolationProviderClass,
        Class<? extends Provider<Map<String, Integer>>> perUserMaxActiveProviderClass,
        Class<? extends Provider<Map<String, Integer>>> perUserMaxIdleProviderClass,
        Class<? extends Provider<Map<String, Long>>> perUserMaxWaitMillisProviderClass) {
      this.perUserDefaultAutoCommitProviderClass = perUserDefaultAutoCommitProviderClass;
      this.perUserDefaultReadOnlyProviderClass = perUserDefaultReadOnlyProviderClass;
      this.perUserDefaultTransactionIsolationProviderClass = perUserDefaultTransactionIsolationProviderClass;
      this.perUserMaxActiveProviderClass = perUserMaxActiveProviderClass;
      this.perUserMaxIdleProviderClass = perUserMaxIdleProviderClass;
      this.perUserMaxWaitMillisProviderClass = perUserMaxWaitMillisProviderClass;
    }
  
    @Override
    protected void configure() {
      doBind(perUserDefaultAutoCommitProviderClass, STRING_BOOLEAN_MAP, PerUserDefaultAutoCommit.class);
      doBind(perUserDefaultReadOnlyProviderClass, STRING_BOOLEAN_MAP, PerUserDefaultReadOnly.class);
      doBind(perUserDefaultTransactionIsolationProviderClass, STRING_INTEGER_MAP,
          PerUserDefaultTransactionIsolation.class);
      doBind(perUserMaxActiveProviderClass, STRING_INTEGER_MAP, PerUserMaxTotal.class);
      doBind(perUserMaxIdleProviderClass, STRING_INTEGER_MAP, PerUserMaxIdle.class);
      doBind(perUserMaxWaitMillisProviderClass, STRING_LONG_MAP, PerUserMaxWaitMillis.class);
    }
  
    private <T> void doBind(Class<? extends Provider<T>> providerClass, TypeLiteral<T> typeLiteral,
        Class<? extends Annotation> annotation) {
      if (providerClass != null) {
        bind(typeLiteral).annotatedWith(annotation).toProvider(providerClass).in(Scopes.SINGLETON);
      }
    }
  
    public static final class Builder {
  
      private Class<? extends Provider<Map<String, Boolean>>> perUserDefaultAutoCommitProviderClass;
  
      private Class<? extends Provider<Map<String, Boolean>>> perUserDefaultReadOnlyProviderClass;
  
      private Class<? extends Provider<Map<String, Integer>>> perUserDefaultTransactionIsolationProviderClass;
  
      private Class<? extends Provider<Map<String, Integer>>> perUserMaxActiveProviderClass;
  
      private Class<? extends Provider<Map<String, Integer>>> perUserMaxIdleProviderClass;
  
      private Class<? extends Provider<Map<String, Long>>> perUserMaxWaitProviderClass;
  
      /**
       * Sets the per user default auto commit provider class.
       *
      * @param perUserDefaultAutoCommitProviderClass
      *          the perUserDefaultAutoCommitProviderClass to set
      *
      * @return the builder
      */
     public Builder setPerUserDefaultAutoCommitProviderClass(
         Class<? extends Provider<Map<String, Boolean>>> perUserDefaultAutoCommitProviderClass) {
       this.perUserDefaultAutoCommitProviderClass = perUserDefaultAutoCommitProviderClass;
       return this;
     }
 
     /**
      * Sets the per user default read only provider class.
      *
      * @param perUserDefaultReadOnlyProviderClass
      *          the perUserDefaultReadOnlyProviderClass to set
      *
      * @return the builder
      */
     public Builder setPerUserDefaultReadOnlyProviderClass(
         Class<? extends Provider<Map<String, Boolean>>> perUserDefaultReadOnlyProviderClass) {
       this.perUserDefaultReadOnlyProviderClass = perUserDefaultReadOnlyProviderClass;
       return this;
     }
 
     /**
      * Sets the per user default transaction isolation provider class.
      *
      * @param perUserDefaultTransactionIsolationProviderClass
      *          the perUserDefaultTransactionIsolationProviderClass to set
      *
      * @return the builder
      */
     public Builder setPerUserDefaultTransactionIsolationProviderClass(
         Class<? extends Provider<Map<String, Integer>>> perUserDefaultTransactionIsolationProviderClass) {
       this.perUserDefaultTransactionIsolationProviderClass = perUserDefaultTransactionIsolationProviderClass;
       return this;
     }
 
     /**
      * Sets the per user max active provider class.
      *
      * @param perUserMaxActiveProviderClass
      *          the perUserMaxActiveProviderClass to set
      *
      * @return the builder
      */
     public Builder setPerUserMaxTotalProviderClass(
         Class<? extends Provider<Map<String, Integer>>> perUserMaxActiveProviderClass) {
       this.perUserMaxActiveProviderClass = perUserMaxActiveProviderClass;
       return this;
     }
 
     /**
      * Sets the per user max idle provider class.
      *
      * @param perUserMaxIdleProviderClass
      *          the perUserMaxIdleProviderClass to set
      *
      * @return the builder
      */
     public Builder setPerUserMaxIdleProviderClass(
         Class<? extends Provider<Map<String, Integer>>> perUserMaxIdleProviderClass) {
       this.perUserMaxIdleProviderClass = perUserMaxIdleProviderClass;
       return this;
     }
 
     /**
      * Sets the per user max wait provider class.
      *
      * @param perUserMaxWaitProviderClass
      *          the perUserMaxWaitProviderClass to set
      *
      * @return the builder
      */
     public Builder setPerUserMaxWaitMillisProviderClass(
         Class<? extends Provider<Map<String, Long>>> perUserMaxWaitProviderClass) {
       this.perUserMaxWaitProviderClass = perUserMaxWaitProviderClass;
       return this;
     }
 
     /**
      * Creates the pool.
      *
      * @return the per user pool data source module
      */
     public PerUserPoolDataSourceModule create() {
       return new PerUserPoolDataSourceModule(perUserDefaultAutoCommitProviderClass, perUserDefaultReadOnlyProviderClass,
           perUserDefaultTransactionIsolationProviderClass, perUserMaxActiveProviderClass, perUserMaxIdleProviderClass,
           perUserMaxWaitProviderClass);
     }
 
   }
 
 }