package com.qianwen.core.tenant.config;

import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure;
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import com.baomidou.dynamic.datasource.aop.DynamicDataSourceAnnotationAdvisor;
import com.baomidou.dynamic.datasource.creator.DruidDataSourceCreator;
import com.baomidou.dynamic.datasource.processor.DsHeaderProcessor;
import com.baomidou.dynamic.datasource.processor.DsProcessor;
import com.baomidou.dynamic.datasource.processor.DsSessionProcessor;
import com.baomidou.dynamic.datasource.processor.DsSpelExpressionProcessor;
import com.baomidou.dynamic.datasource.provider.DynamicDataSourceProvider;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DataSourceProperty;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceAutoConfiguration;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceCreatorAutoConfiguration;
import com.baomidou.dynamic.datasource.spring.boot.autoconfigure.DynamicDataSourceProperties;
import javax.sql.DataSource;
import com.qianwen.core.tenant.constant.TenantBaseConstant;
import com.qianwen.core.tenant.dynamic.DsTenantIdProcessor;
import com.qianwen.core.tenant.dynamic.TenantDataSourceAnnotationInterceptor;
import com.qianwen.core.tenant.dynamic.TenantDataSourceGlobalAdvisor;
import com.qianwen.core.tenant.dynamic.TenantDataSourceGlobalInterceptor;
import com.qianwen.core.tenant.dynamic.TenantDataSourceHolder;
import com.qianwen.core.tenant.dynamic.TenantDataSourceJdbcProvider;
import org.springframework.beans.factory.SmartInitializingSingleton;
import org.springframework.boot.autoconfigure.AutoConfigureBefore;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.context.annotation.Primary;
import org.springframework.context.annotation.Role;
import org.springframework.core.annotation.Order;
import org.springframework.jdbc.core.JdbcTemplate;

@AutoConfigureBefore({DruidDataSourceAutoConfigure.class, DynamicDataSourceAutoConfiguration.class})
@EnableConfigurationProperties({DataSourceProperties.class, DynamicDataSourceProperties.class})
@Configuration(proxyBeanMethods = false)
@ConditionalOnProperty(value = {TenantBaseConstant.TENANT_DYNAMIC_DATASOURCE_PROP}, havingValue = "true")
@Import({DynamicDataSourceCreatorAutoConfiguration.class})

public class TenantDataSourceConfiguration {
    @Bean
    @Primary
    public DynamicDataSourceProvider dynamicDataSourceProvider(DataSourceProperties dataSourceProperties, DynamicDataSourceProperties dynamicDataSourceProperties) {
        String driverClassName = dataSourceProperties.getDriverClassName();
        String url = dataSourceProperties.getUrl();
        String username = dataSourceProperties.getUsername();
        String password = dataSourceProperties.getPassword();
        DataSourceProperty master = (DataSourceProperty) dynamicDataSourceProperties.getDatasource().get(dynamicDataSourceProperties.getPrimary());
        if (master != null) {
            driverClassName = master.getDriverClassName();
            url = master.getUrl();
            username = master.getUsername();
            password = master.getPassword();
        }
        return new TenantDataSourceJdbcProvider(dynamicDataSourceProperties, driverClassName, url, username, password);
    }

    @Bean
    @Primary
    public DataSource dataSource(DynamicDataSourceProvider dynamicDataSourceProvider, DynamicDataSourceProperties dynamicDataSourceProperties) {
        DynamicRoutingDataSource dataSource = new DynamicRoutingDataSource();
        dataSource.setPrimary(dynamicDataSourceProperties.getPrimary());
        dataSource.setStrict(dynamicDataSourceProperties.getStrict());
        dataSource.setStrategy(dynamicDataSourceProperties.getStrategy());
        dataSource.setProvider(dynamicDataSourceProvider);
        dataSource.setP6spy(dynamicDataSourceProperties.getP6spy());
        dataSource.setSeata(dynamicDataSourceProperties.getSeata());
        return dataSource;
    }

    @ConditionalOnMissingBean
    @Bean
    public TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor(DsProcessor dsProcessor, DynamicDataSourceProperties dynamicDataSourceProperties) {
        return new TenantDataSourceAnnotationInterceptor(Boolean.valueOf(dynamicDataSourceProperties.isAllowedPublicOnly()), dsProcessor);
    }

    @ConditionalOnMissingBean
    @Bean
    @Role(2)
    public DynamicDataSourceAnnotationAdvisor dynamicDatasourceAnnotationAdvisor(TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor, DynamicDataSourceProperties dynamicDataSourceProperties) {
        DynamicDataSourceAnnotationAdvisor advisor = new DynamicDataSourceAnnotationAdvisor(tenantDataSourceAnnotationInterceptor);
        advisor.setOrder(dynamicDataSourceProperties.getOrder().intValue());
        return advisor;
    }

    @ConditionalOnMissingBean
    @ConditionalOnProperty(value = {TenantBaseConstant.TENANT_DYNAMIC_GLOBAL_PROP}, havingValue = "true")
    @Bean
    public TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor() {
        return new TenantDataSourceGlobalInterceptor();
    }

    @ConditionalOnMissingBean
    @ConditionalOnProperty(value = {TenantBaseConstant.TENANT_DYNAMIC_GLOBAL_PROP}, havingValue = "true")
    @Bean
    @Role(2)
    public TenantDataSourceGlobalAdvisor tenantDataSourceGlobalAdvisor(TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor, DynamicDataSourceProperties dynamicDataSourceProperties) {
        TenantDataSourceGlobalAdvisor advisor = new TenantDataSourceGlobalAdvisor(tenantDataSourceGlobalInterceptor);
        advisor.setOrder(dynamicDataSourceProperties.getOrder().intValue() + 1);
        return advisor;
    }

    @ConditionalOnMissingBean
    @Bean
    public DsProcessor dsProcessor() {
        DsHeaderProcessor headerProcessor = new DsHeaderProcessor();
        DsSessionProcessor sessionProcessor = new DsSessionProcessor();
        DsTenantIdProcessor tenantIdProcessor = new DsTenantIdProcessor();
        DsSpelExpressionProcessor spelExpressionProcessor = new DsSpelExpressionProcessor();
        headerProcessor.setNextProcessor(sessionProcessor);
        sessionProcessor.setNextProcessor(tenantIdProcessor);
        tenantIdProcessor.setNextProcessor(spelExpressionProcessor);
        return headerProcessor;
    }

    @Configuration(proxyBeanMethods = false)
    @ConditionalOnProperty(value = {TenantBaseConstant.TENANT_DYNAMIC_DATASOURCE_PROP}, havingValue = "true")
    @Order
    
    public static class TenantDataSourceAnnotationConfiguration implements SmartInitializingSingleton {
        private final TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor;
        private final DataSource dataSource;
        private final DruidDataSourceCreator dataSourceCreator;
        private final JdbcTemplate jdbcTemplate;

        public TenantDataSourceAnnotationConfiguration(final TenantDataSourceAnnotationInterceptor tenantDataSourceAnnotationInterceptor, final DataSource dataSource, final DruidDataSourceCreator dataSourceCreator, final JdbcTemplate jdbcTemplate) {
            this.tenantDataSourceAnnotationInterceptor = tenantDataSourceAnnotationInterceptor;
            this.dataSource = dataSource;
            this.dataSourceCreator = dataSourceCreator;
            this.jdbcTemplate = jdbcTemplate;
        }

        public void afterSingletonsInstantiated() {
            TenantDataSourceHolder tenantDataSourceHolder = new TenantDataSourceHolder(this.dataSource, this.dataSourceCreator, this.jdbcTemplate);
            this.tenantDataSourceAnnotationInterceptor.setHolder(tenantDataSourceHolder);
        }
    }

    @Configuration(proxyBeanMethods = false)
    @ConditionalOnProperty(value = {TenantBaseConstant.TENANT_DYNAMIC_GLOBAL_PROP}, havingValue = "true")
    @Order
    
    public static class TenantDataSourceGlobalConfiguration implements SmartInitializingSingleton {
        private final TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor;
        private final DataSource dataSource;
        private final DruidDataSourceCreator dataSourceCreator;
        private final JdbcTemplate jdbcTemplate;

        public TenantDataSourceGlobalConfiguration(final TenantDataSourceGlobalInterceptor tenantDataSourceGlobalInterceptor, final DataSource dataSource, final DruidDataSourceCreator dataSourceCreator, final JdbcTemplate jdbcTemplate) {
            this.tenantDataSourceGlobalInterceptor = tenantDataSourceGlobalInterceptor;
            this.dataSource = dataSource;
            this.dataSourceCreator = dataSourceCreator;
            this.jdbcTemplate = jdbcTemplate;
        }

        public void afterSingletonsInstantiated() {
            TenantDataSourceHolder tenantDataSourceHolder = new TenantDataSourceHolder(this.dataSource, this.dataSourceCreator, this.jdbcTemplate);
            this.tenantDataSourceGlobalInterceptor.setHolder(tenantDataSourceHolder);
        }
    }
}