package com.ycl.config;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ycl.entity.NewsAdmin;
import com.ycl.entity.NewsIp;
import com.ycl.entity.auth.AuthInfo;
import com.ycl.exception.ApiException;
import com.ycl.service.NewsAdminService;
import com.ycl.service.NewsIpService;
import com.ycl.utils.JwtTokenUtil;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;

public class Intercept implements HandlerInterceptor {

    @Resource
    NewsIpService newsIpService;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws IOException {
        List<String> ips = newsIpService.list().stream().map(item -> item.getIpAddess()).collect(Collectors.toList());
        if (!ips.contains(request.getRemoteAddr())) {
            return false;

        }
//        String token = request.getHeader("token");
//        JwtTokenUtil jwtTokenUtil = new JwtTokenUtil();
//        if (token==null||token.isEmpty()){
//            response.sendRedirect("");
//            return true;
//            throw new ApiException("用户未登录");
//        }
//        AuthInfo authInfo = jwtTokenUtil.parseToken(token);
//        if (authInfo == null) {
//            throw new ApiException("未认证用户");
//        }
//        if (newsAdminService.getOne(new LambdaQueryWrapper<NewsAdmin>().eq(NewsAdmin::getUsername, authInfo.getUsername())) == null) {
//            throw new ApiException("不是系统用户");
//        } else {
            return true;
//        }

    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        HandlerInterceptor.super.afterCompletion(request, response, handler, ex);
    }
}