package com.yanzhen.framework.mvc; // 定义包名

import com.yanzhen.entity.Student; // 导入Student类
import com.yanzhen.entity.User; // 导入User类
import com.yanzhen.framework.exception.MyException; // 导入自定义异常类MyException
import com.yanzhen.framework.jwt.JWTUtil; // 导入JWT工具类JWTUtil
import org.springframework.web.servlet.HandlerInterceptor; // 导入Spring MVC的HandlerInterceptor接口
import org.springframework.web.servlet.ModelAndView; // 导入Spring MVC的ModelAndView类

import javax.servlet.http.HttpServletRequest; // 导入HttpServletRequest类
import javax.servlet.http.HttpServletResponse; // 导入HttpServletResponse类

public class TokenInterceptor implements HandlerInterceptor { // 定义TokenInterceptor类并实现HandlerInterceptor接口

    @Override // 重写preHandle方法，在请求处理之前进行调用
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String token = request.getHeader(JWTUtil.token); // 从请求头中获取token

        String type = JWTUtil.getType(token); // 根据token获取类型
        if("USER".equals(type)){ // 如果类型是USER
            //根据token获取user对象
            User user = JWTUtil.getUser(token); // 通过JWT工具类获取User对象
            if(user == null){ // 如果User对象为空
                throw  new MyException("超时或不合法的token"); // 抛出自定义异常
            }
            String newToken = JWTUtil.sign(user); // 重新生成新的token
            response.setHeader(JWTUtil.token,newToken); // 将新token设置到响应头中
            response.setHeader("Access-Control-Expose-Headers", JWTUtil.token); // 设置允许暴露的响应头
            request.setAttribute("user",user); // 将User对象设置到请求属性中
        }else if("STUDENT".equals(type)){ // 如果类型是STUDENT
            //根据token获取user对象
            Student student = JWTUtil.getStudent(token); // 通过JWT工具类获取Student对象
            if(student == null){ // 如果Student对象为空
                throw  new MyException("超时或不合法的token"); // 抛出自定义异常
            }
            String newToken = JWTUtil.signForStudent(student); // 重新生成新的token
            response.setHeader(JWTUtil.token,newToken); // 将新token设置到响应头中
            response.setHeader("Access-Control-Expose-Headers", JWTUtil.token); // 设置允许暴露的响应头
            request.setAttribute("student",student); // 将Student对象设置到请求属性中
        }
        return true; // 返回true表示继续执行后续的拦截器和处理器
    }

    @Override // 重写postHandle方法，在请求处理之后但在视图渲染之前进行调用
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

    }

    @Override // 重写afterCompletion方法，在整个请求结束之后进行调用
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {

    }
}