/**
 * 自定義數(shù)據(jù)源配置類
 * 支持從命令行參數(shù)或環(huán)境變量讀取解密密鑰，解密數(shù)據(jù)庫用戶名和密碼后構(gòu)造 DataSource
 */
@Configuration
public class DataSourceSimpleConfig {

    private static final Logger logger = LoggerFactory.getLogger(DataSourceSimpleConfig.class);

    private static final String ENC_PREFIX = "{enc}";

    // 密鑰獲取優(yōu)先級：命令行參數(shù) > 環(huán)境變量
    @Value("${db.encrypt.key:#{systemEnvironment['ENCRYPT_KEY']}}")
    private String encryptKey;

    private final Environment environment;

    public DataSourceSimpleConfig(Environment environment) {
        this.environment = environment;
    }

    /**
     * 自定義 DataSource Bean
     * 解密數(shù)據(jù)庫配置后構(gòu)造 HikariDataSource
     */
    @Bean
    public DataSource dataSource() throws Exception {
        logger.info("========== 開始配置數(shù)據(jù)源 ==========");

        // 獲取原始配置
        String url = environment.getProperty("spring.datasource.url");
        String encryptedUsername = environment.getProperty("spring.datasource.username");
        String encryptedPassword = environment.getProperty("spring.datasource.password");
        String driverClassName = environment.getProperty("spring.datasource.driver-class-name");

        // 檢查是否需要解密
        String username;
        String password;

        if (encryptedUsername.startsWith(ENC_PREFIX)) {
            // 需要解密用戶名
            logger.info("檢測到加密的用戶名，開始解密...");
            String encryptedUser = encryptedUsername.substring(ENC_PREFIX.length());
            username = decrypt(encryptedUser, encryptKey);
            logger.info("用戶名解密成功");
        } else {
            username = encryptedUsername;
            logger.info("用戶名未加密，直接使用");
        }

        if (encryptedPassword.startsWith(ENC_PREFIX)) {
            // 需要解密密碼
            logger.info("檢測到加密的密碼，開始解密...");
            String encryptedPass = encryptedPassword.substring(ENC_PREFIX.length());
            password = decrypt(encryptedPass, encryptKey);
            logger.info("密碼解密成功");
        } else {
            password = encryptedPassword;
            logger.info("密碼未加密，直接使用");
        }

        // 創(chuàng)建 HikariConfig
        HikariConfig hikariConfig = new HikariConfig();
        hikariConfig.setJdbcUrl(url);
        hikariConfig.setUsername(username);
        hikariConfig.setPassword(password);
        
        if (driverClassName != null && !driverClassName.isEmpty()) {
            hikariConfig.setDriverClassName(driverClassName);
        }

        // 創(chuàng)建數(shù)據(jù)源
        HikariDataSource dataSource = new HikariDataSource(hikariConfig);
        
        logger.info("? 數(shù)據(jù)源配置完成");
        logger.info("   JDBC URL: {}", url);
        logger.info("========== 數(shù)據(jù)源配置完成 ==========");

        return dataSource;
    }

    /**
     * 解密方法
     *
     * @param encryptedText 加密文本（Base64 編碼）
     * @param key           解密密鑰（Base64 編碼）
     * @return 解密后的明文
     */
    private String decrypt(String encryptedText, String key) throws Exception {
        try {
            return AesUtil.decryptGCM(encryptedText, key);
        } catch (Exception e) {
            logger.error("? 解密失??！", e);
            logger.error("   錯誤信息: {}", e.getMessage());
            throw new RuntimeException("解密失敗，請檢查密鑰是否正確", e);
        }
    }
}

url: jdbc:postgresql://
username: "{enc}111111"
password: "{enc}222222"

========== 開始配置數(shù)據(jù)源 ==========
檢測到加密的用戶名，開始解密...
已獲取解密密鑰（長度 4）
用戶名解密成功
檢測到加密的密碼，開始解密...
已獲取解密密鑰（長度 4）
密碼解密成功