package com.fintech.pub.config.datasource;

import com.alibaba.druid.pool.DruidDataSource;
import com.xiaomi.mifi.common.utils.cryptor.KeyCenterCryptor;
import com.xiaomi.mifi.common.utils.db.DatabaseConnectionConfigurer;
import com.xiaomi.mifi.common.utils.db.keyproxy.KeyProxyClient;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.annotation.Conditional;
import org.springframework.stereotype.Component;
import java.util.Properties;

/**
 * @author HealerJean
 * @ClassName DataSourcePostProcessor
 * @date 2020/6/9  17:13.
 * @Description
 */
@Conditional(DataSourceEncryptCondition.class)
@Component
@Slf4j
public class DataSourcePostProcessor implements BeanPostProcessor {

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        if (bean instanceof DruidDataSource) {
            DruidDataSource druidDataSource = (DruidDataSource) bean;
            encryptDataSource(druidDataSource);
        }
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    private void encryptDataSource(DruidDataSource druidDataSource) {
        druidDataSource.setUrl(properties.getProperty("fintech.database.url"));
        druidDataSource.setUsername(properties.getProperty("fintech.database.username"));
        druidDataSource.setPassword(properties.getProperty("fintech.database.password"));
    }



}

@Configuration
@Slf4j
public class DataSourceEncryptCondition implements Condition {

    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata metadata) {
        Environment environment = conditionContext.getEnvironment();
        String encrypt = environment.getProperty("datasource.encrypt");
        if (StringUtils.isNotBlank(encrypt) && "true".equals(encrypt)) {
            return true;
        }
        return false;
    }
}