package com.xwbing.handler;

import com.alibaba.fastjson.JSON;
import com.xwbing.configuration.DispatcherServletConfig;
import com.xwbing.constant.CommonConstant;
import com.xwbing.util.CommonDataUtil;
import com.xwbing.util.RestMessage;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.util.HashSet;
import java.util.Set;

/**
 * 说明:  登录拦截器
 * 项目名称: boot-module-demo
 * 创建时间: 2017/5/10 16:36
 * 作者:  xiangwb
 */
public class LoginInterceptor extends HandlerInterceptorAdapter {
    private final Logger logger = LoggerFactory.getLogger(DispatcherServletConfig.class);
    private static final Set<String> set = new HashSet<>();//拦截器白名单

    static {
        set.add("/v2/api-docs");
        set.add("/servlet/captchaCode");
        set.add("/swagger-ui.html");
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String servletPath = request.getServletPath();
        if (!set.contains(servletPath) && !servletPath.contains("login")) {
//            HttpSession session = request.getSession();
//            if (session.getAttribute(CommonConstant.CURRENT_USER)!=null) {
//                return true;
//            }else {
//                return false;
//            }
            if (CommonDataUtil.getToken(CommonConstant.CURRENT_USER) != null) {
                return true;
            } else {
                logger.info("用户未登录");
                OutputStream outputStream = response.getOutputStream();
                RestMessage restMessage = new RestMessage();
                restMessage.setMessage("请登录");
                outputStream.write(JSON.toJSONString(restMessage).getBytes("utf-8"));
                outputStream.close();
                return false;
            }
        }
        return true;
    }
}