package quant.rich.emoney.config;

import java.io.File;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import javax.sql.DataSource;

import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionTemplate;
import org.mybatis.spring.annotation.MapperScan;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.ClassPathResource;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
import com.zaxxer.hikari.HikariDataSource;

import lombok.extern.slf4j.Slf4j;
import quant.rich.emoney.EmoneyAutoApplication;

@Slf4j
@Configuration
@MapperScan(basePackages = "quant.rich.emoney.mapper.sqlite", sqlSessionTemplateRef = "sqliteSqlSessionTemplate")
public class SqliteMybatisConfig {
	
	private static final String RESOURCE_PATH = "database.db";
	
    public static final String SQLITE_TRANSACTION_MANAGER = "sqliteTransactionManager";
    
    /**
     * 配置数据库连接
     * <ul>
     * <li>如果非打包状态，则直接选取当前项目内数据库位置</li>
     * <li>如果打包状态，以 JDBC 链接位置为主，如果位置不存在则覆盖</li>
     * </ul>
     * @param dataSource
     */
    public void initSQLiteLocation(DataSource dataSource) {
    	// 指定 sqlite 路径
    	if (dataSource instanceof HikariDataSource hikariDataSource) {
    		
    		String filePath = hikariDataSource.getJdbcUrl();
    		if (filePath == null || !filePath.startsWith("jdbc:sqlite:")) {
    			log.warn(
			        "无法在 SQLite HikariDataSource 中找到合法 SQLite JDBC url, "
			        + "数据库可能会加载失败。合法的 url 需在 application.yml(properties) "
			        + "中配置，以 jdbc:sqlite: 开头。当前获取到的 jdbc-url: {}", filePath);
    			return;
    		}
    		filePath = filePath.substring("jdbc:sqlite:".length()).trim();
    		
    		ClassPathResource original = new ClassPathResource(RESOURCE_PATH);
    		if (!original.exists()) {
                log.warn("未找到 SQLite 资源: {}", RESOURCE_PATH);
                return;
    		}
    		String protocol = EmoneyAutoApplication.class.getProtectionDomain().getCodeSource().getLocation().getProtocol();
    	    boolean isJar = "jar".equals(protocol);
    	    
    	    if (isJar) {
    	    	// 复制到外部 yml 指定路径，已存在则不复制
    	    	File dest = new File(filePath), parentDir = dest.getParentFile();
    	    	String destAbsolutePath = dest.getAbsolutePath();
        		if (!parentDir.exists() && !parentDir.mkdirs()) {
        			log.warn("无法创建放置 SQLite 文件的目录: {}", parentDir.getAbsolutePath());
        			return;
        		}
        		
        		if (dest.exists()) {
        			// 已存在
        			log.warn("目标资源 {} 已存在，忽略", destAbsolutePath);
        			return;
        		}
        		
    			try (InputStream in = getClass().getClassLoader().getResourceAsStream(RESOURCE_PATH)) {
    	            if (in == null) {
    	                log.warn("无法读取 SQLite 资源: {}", RESOURCE_PATH);
    	                return;
    	            }
    	            Files.copy(in, dest.toPath(), java.nio.file.StandardCopyOption.REPLACE_EXISTING);
    	            log.info("SQLite 数据库文件已复制至：{}", destAbsolutePath);
    	        } catch (Exception e) {
	                log.warn("复制 SQLite 数据库文件失败", e);
    	        }
    	    }
    	    else {
    	    	// 使用当前绝对路径
    	    	Path path = Path.of("src/main/resources", RESOURCE_PATH);
    	    	hikariDataSource.setJdbcUrl("jdbc:sqlite:" + path.toAbsolutePath().toString());
    	    }
    	}
    }

    @Bean("sqliteSqlSessionFactory")
    public SqlSessionFactory sqliteSqlSessionFactory(
        @Qualifier("sqliteDataSource") DataSource dataSource) throws Exception {
    	
    	initSQLiteLocation(dataSource);
    	
        MybatisSqlSessionFactoryBean factory = new MybatisSqlSessionFactoryBean();
        factory.setDataSource(dataSource);

        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.SQLITE));
        interceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor());
        factory.setPlugins(interceptor);
        
        return factory.getObject();
        
    }
    
    @Bean("sqliteSqlSessionTemplate")
    public SqlSessionTemplate sqliteSqlSessionTemplate(@Qualifier("sqliteSqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
    
    @Bean(SQLITE_TRANSACTION_MANAGER)
    public DataSourceTransactionManager postgreTransacetionManager(@Qualifier("sqliteDataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }
}