package com.poyee.utils;

import com.alibaba.fastjson.JSONObject;
import com.poyee.res.UserInfo;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

/**
 * Servlet 工具类
 */
public class ServletUtils {

    /**
     * 获取当前请求
     */
    public static HttpServletRequest getRequest() {
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (attributes == null) {
            return null;
        }
        return attributes.getRequest();
    }

    public static HttpServletResponse getHttpResponse() {
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        if (attributes == null) {
            return null;
        }
        return attributes.getResponse();
    }

    /**
     * 获取当前 Session
     */
    public static HttpSession getSession() {
        HttpServletRequest request = getRequest();
        if (request == null) {
            return null;
        }
        return request.getSession();
    }

    /**
     * 获取当前登录用户
     */
    public static JSONObject getCurrentUser() {
        HttpServletRequest request = getRequest();
        if (request == null) {
            return null;
        }
        return (JSONObject) request.getAttribute("currentUser");
    }

    /**
     * 设置当前登录用户
     */
    public static void setCurrentUser(UserInfo userInfo) {
        HttpServletRequest request = getRequest();
        if (request != null) {
            request.setAttribute("currentUser", userInfo);
        }
    }
}