package com.hxzk.util;

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

import com.hxzk.pojo.TbUser;
import org.springframework.stereotype.Component;

@Component
public class SessionManager {

    /**
     * 获取当前用户信息
     * @param request HTTP请求对象
     * @return 当前用户信息对象
     */
    public TbUser getCurrentUser(HttpServletRequest request) {
        HttpSession session = request.getSession(false);
        if (session != null) {
            Object userObj = session.getAttribute("currentUser");
            if (userObj instanceof TbUser) {
                return (TbUser) userObj;
            }
        }
        return null;
    }

    /**
     * 设置当前用户信息
     * @param request HTTP请求对象
     * @param user 当前用户信息对象
     */
    public void setCurrentUser(HttpServletRequest request, TbUser user) {
        HttpSession session = request.getSession();
        session.setAttribute("currentUser", user);
    }

    /**
     * 清除当前用户信息（注销用户）
     * @param request HTTP请求对象
     */
    public void clearCurrentUser(HttpServletRequest request) {
        HttpSession session = request.getSession(false);
        if (session != null) {
            session.removeAttribute("currentUser");
            session.invalidate(); // 可选：使Session失效
        }
    }
}