package com.gkhy.labRiskManage.api.controller.common;

import com.gkhy.labRiskManage.config.authorization.TokenConfig;
import com.google.common.net.HttpHeaders;
import org.springframework.web.bind.annotation.ModelAttribute;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class BaseController {

    @Resource
    protected TokenConfig tokenConfig;

    protected HttpServletRequest request;
    protected HttpServletResponse response;

    @ModelAttribute
    public void setReqAndResp(HttpServletRequest request, HttpServletResponse response){
        this.request = request;
        this.response = response;
    }

    protected Long getCurrentUserId(){
        String userId = this.request.getHeader(tokenConfig.getLoginUserHeader());
        if(userId == null || userId.isEmpty())
            return -1L;
        try {
            Long uid = Long.parseLong(userId);
            return uid;
        } catch (NumberFormatException e) {
            return -1L;
        }
    }
}