package com.yami.trading.security.common.util;

import cn.hutool.core.util.StrUtil;
import com.yami.trading.bean.model.RiskClient;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.AntPathMatcher;

import java.time.Instant;
import java.time.ZoneId;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

@Slf4j
public class RiskClientUtil {

    private static Map<String, RiskClient> riskClientMap = new ConcurrentHashMap<>();

    private static AntPathMatcher ipMatcher = new AntPathMatcher();

    public static void initCache(Map<String, RiskClient> newRiskClientMap) {
        riskClientMap = newRiskClientMap;
    }

    public static Map<String, RiskClient> getRiskClientMap() {
        return Collections.unmodifiableMap(riskClientMap);
    }

    public static void saveRiskConfig(RiskClient config) {
        if (config == null) {
            return;
        }

        String cacheKey = config.getType() + ":" + config.getClientType() + ":" + config.getClientKey();
        riskClientMap.put(cacheKey, config);
    }

    public static void disableRiskConfig(String type, String clientType, String clientKey) {
        String cacheKey = type + ":" + clientType + ":" + clientKey;
        riskClientMap.remove(cacheKey);
    }

    /**
     * 基于指定用户ID值，提取特定风控类型的配置信息；
     * userId 参数必填，如果未设置 type 参数，则提取相关所有配置信息
     *
     * @param userCode
     * @param type
     * @return
     */
    public static List<RiskClient> getRiskInfoByUserCode(String userCode, String type) {
        if (StrUtil.isBlank(userCode)) {
            throw new RuntimeException("userCode参数必填");
        }

        List<RiskClient> riskList = new ArrayList<>();
        if (riskClientMap.isEmpty()) {
            return riskList;
        }

        if (StrUtil.isBlank(type)) {
            String fieldKey1 = "white:userCode:" + userCode.trim();
            String fieldKey2 = "black:userCode:" + userCode.trim();
            String fieldKey3 = "badnetwork:userCode:" + userCode.trim();

            RiskClient config1 = riskClientMap.get(fieldKey1);
            if (config1 != null) {
                riskList.add(config1);
            }

            RiskClient config2 = riskClientMap.get(fieldKey2);
            if (config2 != null) {
                riskList.add(config2);
            }

            RiskClient config3 = riskClientMap.get(fieldKey3);
            if (config3 != null) {
                riskList.add(config3);
            }
        } else {
            String fieldKey1 = type + ":userCode:" + userCode.trim();
            RiskClient config1 = riskClientMap.get(fieldKey1);
            if (config1 != null) {
                riskList.add(config1);
            }
        }

        long currentTimeMillis = Instant.now().atZone(ZoneId.of("Asia/Shanghai")).toInstant().toEpochMilli();
        List<RiskClient> validRiskList = new ArrayList<>();
        for (RiskClient oneRisk : riskList) {
            if (oneRisk.getStatus() != 1) {
                continue;
            }
            if (oneRisk.getBeginTimeTs().longValue() == 0 && oneRisk.getEndTimeTs().longValue() == 0) {
                // 时间不做限制
                validRiskList.add(oneRisk);
                continue;
            }
            if (oneRisk.getBeginTimeTs() > currentTimeMillis) {
                // 还没开始
                continue;
            }
            if (oneRisk.getEndTimeTs() > 0 && oneRisk.getEndTimeTs() < currentTimeMillis) {
                // 已经结束
                continue;
            }

            // 当前时间在起止时间中间
            validRiskList.add(oneRisk);
        }

        return validRiskList;
    }

    /**
     * 基于指定 IP 值，提取特定风控类型的配置信息；
     * ip 参数必填，如果未设置 type 参数，则提取相关所有配置信息
     *
     * @param clientIp
     * @param type
     * @return
     */
    public static List<RiskClient> getRiskInfoByIp(String clientIp, String type) {
        if (StrUtil.isBlank(clientIp)) {
            return new ArrayList<>();
            //throw new RuntimeException("clientIp参数必填");
        }

        List<RiskClient> riskList = new ArrayList<>();
        if (riskClientMap.isEmpty()) {
            return riskList;
        }

        List<String> keyList = new ArrayList<>();
        if (clientIp.contains("*")) {
            Set<String> cacheKeys = new HashSet<>(riskClientMap.keySet());
            Set<String> optionIps = new HashSet<>();
            for (String oneCacheKey : cacheKeys) {
                if (!oneCacheKey.contains(":ip:")) {
                    continue;
                }

                int idx = oneCacheKey.indexOf(":ip:");
                String ipPatternValue = oneCacheKey.substring(idx + ":ip:".length());
                if (ipMatcher.match(ipPatternValue, clientIp)) {
                    optionIps.add(ipPatternValue);
                    RiskClient curRiskClient = riskClientMap.get(oneCacheKey);
                    if (curRiskClient != null) {
                        riskList.add(curRiskClient);
                    }
                }
            }
        } else {
            if (StrUtil.isBlank(type)) {
                String fieldKey1 = "white:ip:" + clientIp.trim();
                String fieldKey2 = "black:ip:" + clientIp.trim();
                String fieldKey3 = "badnetwork:ip:" + clientIp.trim();

                RiskClient config1 = riskClientMap.get(fieldKey1);
                if (config1 != null) {
                    riskList.add(config1);
                }

                RiskClient config2 = riskClientMap.get(fieldKey2);
                if (config2 != null) {
                    riskList.add(config2);
                }

                RiskClient config3 = riskClientMap.get(fieldKey3);
                if (config3 != null) {
                    riskList.add(config3);
                }
            } else {
                String fieldKey1 = type + ":ip:" + clientIp.trim();
                RiskClient config1 = riskClientMap.get(fieldKey1);
                if (config1 != null) {
                    riskList.add(config1);
                }
            }
        }

        Date now = new Date(); // 注意时区问题
        List<RiskClient> validRiskList = new ArrayList<>();
        for (RiskClient oneRisk : riskList) {
            if (oneRisk.getStatus() != 1) {
                continue;
            }
            if (oneRisk.getBeginTimeTs().longValue() == 0 && oneRisk.getEndTimeTs().longValue() == 0) {
                // 时间不做限制
                validRiskList.add(oneRisk);
                continue;
            }
            if (oneRisk.getBeginTimeTs() > now.getTime()) {
                // 还没开始
                continue;
            }
            if (oneRisk.getEndTimeTs() > 0 && oneRisk.getEndTimeTs() < now.getTime()) {
                // 已经结束
                continue;
            }

            // 当前时间在起止时间中间
            validRiskList.add(oneRisk);
        }

        return validRiskList;
    }

}