package com.terra.proxy.util;

import cn.hutool.core.collection.CollectionUtil;
import com.terra.proxy.bean.BlackRule;
import com.terra.proxy.bean.VistorBean;
import com.terra.proxy.service.Impl.LogServiceImpl;
import com.terra.proxy.service.Impl.ServerRegisterServiceImpl;
import org.apache.commons.lang3.StringUtils;
import redis.clients.jedis.Jedis;

import javax.servlet.http.HttpServletRequest;
import java.sql.DriverManager;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class BlackUtils {

    /**
     *
     * @param request
     * @param logService
     * @param service
     * @return true 可以访问   flase 过滤
     */
    public static Boolean blackInterceptor(HttpServletRequest request, LogServiceImpl logService, ServerRegisterServiceImpl service){
        if (request != null) {
            //获取访问IP
            String requestURI = request.getRequestURI();
            //系统内部接口不拦截
            if(
                    requestURI.startsWith("/log")||
                    requestURI.startsWith("/token")||
                    requestURI.startsWith("/serv")||
                    requestURI.startsWith("/serverRegister")||
                    requestURI.startsWith("/test")
            ){
                return true;
            }

            String remoteAddr = request.getRemoteAddr();
            HashMap<String, Object> param = new HashMap<>();
            param.put("status","3"); //白名单生效
            param.put("ip",remoteAddr);
            //白名单用户不拦截
            if(CollectionUtil.isNotEmpty(logService.selectWhiteList(param))) return true;
            param.put("status","1"); //黑名单生效
            List<VistorBean> vistorBeans = logService.queryBlackLists(param);
            if(CollectionUtil.isNotEmpty(vistorBeans)) return false;
            //先判断redis中是否存在 不存在则加入reids中计数，存在则判断是否超出限制，若无则计数+1，若超出则加入黑名单
            Jedis jedis = JedisUtils.getJedis();
            String visitCountStr = jedis.get(remoteAddr);
            BlackRule blackRule = logService.selectBlackRule();
            if(blackRule==null) return true;
            if (StringUtils.isEmpty(visitCountStr)) {
                //不存在则加入redis中
                if(blackRule.getSeltime().equals("m")){
                    blackRule.setTime(blackRule.getTime()*60);
                }else if(blackRule.getSeltime().equals("h")){
                    blackRule.setTime(blackRule.getTime()*3600);
                }
                jedis.setex(remoteAddr, blackRule.getTime(), "1");
            } else {
                //已经存在 判断是否大于1000次 是则加入黑名单 否计数+1
                Integer visitCount = Integer.valueOf(visitCountStr);
                if (visitCount >= blackRule.getCount()) {
                    //加入黑名单 表visitlog中status状态为1：黑名单状态
                    VistorBean vb = new VistorBean();
                    Map<String, Object> params = new HashMap();
                    String zyid="0000";
                    String [] splitpath=null;
                    List<Map> serverlist = new ArrayList<>();
                    vb.setRequestip(request.getRemoteAddr());
                    String path = request.getPathInfo();
                    if(StringUtils.isNotEmpty(path)){
                        splitpath = path.split("/");
                        if(null!=splitpath && splitpath.length>1){
                            String numZid = splitpath[1];
                            if(isdigit(numZid)) zyid=numZid;
                        }
                    }
                    String suffix = null;
                    vb.setResourceid(Integer.valueOf(zyid));
                    vb.setVisitfromsys("HYJGPT");
                    if (null!=splitpath&&splitpath.length > 3 && splitpath[3] != null) {
                        suffix = isdigit(splitpath[2]) ? splitpath[2] : null;
                    }
                    params.put("resourceid", Integer.valueOf(zyid));
                    params.put("suffix", suffix != null ? Integer.valueOf(suffix) : null);
                    params.put("enable", 1);
                    params.put("fromsys", "HYJGPT");
                    params.put("SERVERURL", "test");
                    serverlist = service.queryServer(params);
                    if(CollectionUtil.isNotEmpty(serverlist)){
                        params.put("SERVERURL", serverlist.get(0).get("SERVERURL").toString());
                        serverlist.add(param);
                        vb.setServerurl(serverlist.get(0).get("SERVERURL").toString());
                    }
                    vb.setResult("false");
                    logService.saveBlackLog(vb);
                    //JedisUtils.close(jedis);
                    return false;
                } else {
                    jedis.setrange(remoteAddr, 0, String.valueOf(visitCount + 1));
                }

            }
            JedisUtils.close(jedis);
        }

        return true;
    }

    public static boolean isdigit(String str){

        return str.matches("[0-9]+");
    }

    public static void main(String[] args) {

    }
}