package com.qianwen.smartman.common.config;

import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.autoconfigure.ConfigurationCustomizer;
import com.baomidou.mybatisplus.extension.MybatisMapWrapperFactory;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.DynamicTableNameInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.InnerInterceptor;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Properties;
import org.apache.ibatis.mapping.DatabaseIdProvider;
import org.apache.ibatis.mapping.VendorDatabaseIdProvider;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.qianwen.smartman.common.constant.CommonConstant;
import com.qianwen.smartman.common.enums.LongTableNameEnum;
import com.qianwen.core.mp.plugins.BladePaginationInterceptor;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
/* loaded from: blade-api.jar:BOOT-INF/classes/org/springblade/common/config/BladeMybatisConfiguration.class */
public class BladeMybatisConfiguration {
    private static final Logger log = LoggerFactory.getLogger(BladeMybatisConfiguration.class);

    @Bean
    public ConfigurationCustomizer configurationCustomizer() {
        return configuration -> {
            configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory());
        };
    }

    @Bean
    public DatabaseIdProvider getDatabaseIdProvider() {
        VendorDatabaseIdProvider vendorDatabaseIdProvider = new VendorDatabaseIdProvider();
        Properties properties = new Properties();
        properties.setProperty("Oracle", CommonConstant.ORACLE);
        properties.setProperty("MySQL", "mysql");
        properties.setProperty("DB2", "db2");
        properties.setProperty("SQL Server", "sqlserver");
        properties.setProperty("DM DBMS", "dm");
        vendorDatabaseIdProvider.setProperties(properties);
        return vendorDatabaseIdProvider;
    }

    @ConditionalOnProperty(prefix = "datasource", name = {"type"}, havingValue = CommonConstant.ORACLE)
    @Bean
    public DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor(MybatisPlusInterceptor mybatisPlusInterceptor) {
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
            LongTableNameEnum realTableName = LongTableNameEnum.getByRealTableName(tableName);
            return !realTableName.equals(LongTableNameEnum.DEFAULT) ? realTableName.getShortTableName() : tableName;
        });
        addInterceptor(mybatisPlusInterceptor, dynamicTableNameInnerInterceptor);
        log.info("==========注册动态表名替换拦截器成功==============");
        return dynamicTableNameInnerInterceptor;
    }

    private void addInterceptor(MybatisPlusInterceptor mybatisPlusInterceptor, DynamicTableNameInnerInterceptor tableNameInnerInterceptor) {
        List<InnerInterceptor> interceptors = new ArrayList<>(mybatisPlusInterceptor.getInterceptors());
        if (CollectionUtil.isEmpty(interceptors)) {
            return;
        }
        InnerInterceptor oneInterceptor = interceptors.get(CommonConstant.ZERO.intValue());
        if (!(oneInterceptor instanceof DynamicTableNameInnerInterceptor)) {
            interceptors.set(CommonConstant.ZERO.intValue(), tableNameInnerInterceptor);
            interceptors.add(oneInterceptor);
        }
        InnerInterceptor lastInterceptor = interceptors.get(interceptors.size() - CommonConstant.ONE.intValue());
        if (!(lastInterceptor instanceof BladePaginationInterceptor)) {
            InnerInterceptor pageInterceptor = interceptors.stream().filter(innerInterceptor -> {
                return innerInterceptor instanceof BladePaginationInterceptor;
            }).findFirst().orElse(null);
            if (!Objects.isNull(pageInterceptor)) {
                interceptors.remove(pageInterceptor);
                interceptors.add(pageInterceptor);
            }
        }
        mybatisPlusInterceptor.setInterceptors(interceptors);
    }
}