package com.ruoyi.common.mybatis;

import com.ruoyi.common.utils.SqlConverter;
import com.ruoyi.common.utils.SpringContextUtils;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.plugin.*;
import org.springframework.stereotype.Component;

import java.sql.Connection;
import java.util.Properties;

/**
 * SQL转换拦截器
 */
@Component
@Intercepts({
    @Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})
})
public class SqlConverterInterceptor implements Interceptor {

    private SqlConverter sqlConverter;

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        if (sqlConverter == null) {
            if (SpringContextUtils.getContext() == null) {
                // Spring容器还没初始化，直接跳过转换
                return invocation.proceed();
            }
            sqlConverter = SpringContextUtils.getBean(SqlConverter.class);
        }
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        String sql = statementHandler.getBoundSql().getSql();
        
        // 转换SQL
        String convertedSql = sqlConverter.convertSql(sql);
        
        // 使用反射设置转换后的SQL
        java.lang.reflect.Field field = statementHandler.getBoundSql().getClass().getDeclaredField("sql");
        field.setAccessible(true);
        field.set(statementHandler.getBoundSql(), convertedSql);
        
        return invocation.proceed();
    }

    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }

    @Override
    public void setProperties(Properties properties) {
        // 不需要设置属性
    }
}