package com.terra.proxy.intercepter;

import cn.hutool.core.collection.CollectionUtil;
import com.terra.proxy.bean.BlackRule;
import com.terra.proxy.bean.VistorBean;
import com.terra.proxy.mapper.LogMapper;
import com.terra.proxy.mapper.ServiceMapper;
import com.terra.proxy.service.LogService;
import com.terra.proxy.service.ServerRegisterService;
import com.terra.proxy.util.JedisUtils;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpException;
import org.apache.http.HttpRequest;
import org.apache.http.HttpRequestInterceptor;
import org.apache.http.protocol.HttpContext;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
import redis.clients.jedis.Jedis;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;

public class ServletInterceptor  implements HandlerInterceptor {

    @Autowired
    private LogMapper logMapper;
    @Autowired
    private ServiceMapper serviceMapper;
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if (request != null) {
            //获取访问IP
            String remoteAddr = request.getRemoteAddr();
            HashMap<String, Object> param = new HashMap<>();
            param.put("status","1");
            param.put("limit",5);
            param.put("page", 0);
            param.put("ip",remoteAddr);
            List<VistorBean> vistorBeans = logMapper.queryBlackLists(param);
            if(CollectionUtil.isNotEmpty(vistorBeans)) return false;
            //先判断redis中是否存在 不存在则加入reids中计数，存在则判断是否超出限制，若无则计数+1，若超出则加入黑名单
            Jedis jedis = JedisUtils.getJedis();
            String visitCountStr = jedis.get(remoteAddr);
            BlackRule blackRule = logMapper.selectBlackRule();
            if(blackRule==null) return false;
            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)zyid = splitpath[1];
                    }

                    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 = serviceMapper.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");
                    logMapper.saveBlackLog(vb);
                    return false;
                } else {
                    jedis.setrange(remoteAddr, 0, String.valueOf(visitCount + 1));
                }

            }

        }

        return true;
    }

    public static boolean isdigit(String str) {

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



}