package com.zy.asrs.wms.common.interceptor;

import com.zy.asrs.framework.common.SpringUtils;
import com.zy.asrs.wms.common.constant.RedisConstants;
import com.zy.asrs.wms.utils.RedisUtil;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.plugin.*;
import java.sql.Connection;
import java.util.Properties;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@Intercepts({
        @Signature(type = StatementHandler.class, method = "prepare", args = {Connection.class, Integer.class})
})
public class DataChangeInterceptor implements Interceptor {

    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
        String sql = statementHandler.getBoundSql().getSql();
        if (sql != null) {
            String trimmedSql = sql.trim().toUpperCase();
//            System.out.println("SQL: " + sql);
            if (trimmedSql.startsWith("INSERT")) {
                clearRedisCache(trimmedSql);
            } else if (trimmedSql.startsWith("UPDATE")) {
                clearRedisCache(trimmedSql);
            } else if (trimmedSql.startsWith("DELETE")) {
                clearRedisCache(trimmedSql);
            }
        }

        // 根据需要进行SQL解析
        return invocation.proceed();
    }

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

    @Override
    public void setProperties(Properties properties) {}

    public static String extractTableName(String sql) {
        if (sql == null || sql.trim().isEmpty()) {
            return "无效的SQL语句";
        }

        String trimmedSql = sql.trim().toUpperCase();
        String tableName = "";

        // 定义正则表达式
        Pattern insertPattern = Pattern.compile("INSERT INTO (\\S+)", Pattern.CASE_INSENSITIVE);
        Pattern updatePattern = Pattern.compile("UPDATE (\\S+)", Pattern.CASE_INSENSITIVE);
        Pattern deletePattern = Pattern.compile("DELETE FROM (\\S+)", Pattern.CASE_INSENSITIVE);

        Matcher matcher;

        // 检查INSERT语句
        matcher = insertPattern.matcher(trimmedSql);
        if (matcher.find()) {
            tableName = matcher.group(1);
        } else {
            // 检查UPDATE语句
            matcher = updatePattern.matcher(trimmedSql);
            if (matcher.find()) {
                tableName = matcher.group(1);
            } else {
                // 检查DELETE语句
                matcher = deletePattern.matcher(trimmedSql);
                if (matcher.find()) {
                    tableName = matcher.group(1);
                }
            }
        }

        if (tableName.isEmpty()) {
            return null;
        }

        return tableName;
    }

    private void clearRedisCache(String sql) {
        RedisUtil redisUtil = SpringUtils.getBean(RedisUtil.class);
        if (redisUtil == null) {
            return;
        }

        String tableName = extractTableName(sql);
        if (tableName == null) {
            return;
        }

        tableName = tableName.toLowerCase();

        Set<String> keys = redisUtil.searchRedisKeys(tableName);
        for (String key : keys) {
            if (!key.startsWith(RedisConstants.CACHE_DATA)) {
                continue;
            }
            redisUtil.del(key);
        }
    }
}