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.util.List;
import java.util.stream.Collectors;

public class Intercept implements HandlerInterceptor {

    @Resource
    NewsAdminService newsAdminService;
    @Resource
    NewsIpService newsIpService;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
        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");
        System.out.println(token);
        JwtTokenUtil jwtTokenUtil = new JwtTokenUtil();
        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);
    }
}