package com.common.aspect;


import com.common.core.utils.JwtTokenUtil;
import com.common.security.utils.SecurityHolderUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import javax.servlet.http.HttpServletRequest;

/**
 * @author 廖振钦
 * @date 2022-01-17
 */
@Slf4j
@Aspect
@Component
@Order(0)   //TOKEN记录切面优先级最高
public class TokenAspect {

    @Autowired
    private JwtTokenUtil jwtTokenUtil;


    @Around("execution(* com..controller..*.*(..))" +
            "&& (@annotation(org.springframework.web.bind.annotation.RequestMapping)" +
            "|| @annotation(org.springframework.web.bind.annotation.GetMapping)" +
            "|| @annotation(org.springframework.web.bind.annotation.PostMapping)" +
            "|| @annotation(org.springframework.web.bind.annotation.DeleteMapping)" +
            "|| @annotation(org.springframework.web.bind.annotation.PatchMapping))" +
            "&& !@annotation(com.common.annotation.NoToken)")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        try {
            this.checkToken();
            return pjp.proceed();
        } catch (Throwable e) {
            log.error("LogAspect>>>>>>>>", e);
            throw e;
        }
    }

    private void checkToken(){
        HttpServletRequest request = ((ServletRequestAttributes) (RequestContextHolder.currentRequestAttributes())).getRequest();
        String token = request.getHeader("pi-token");
        Object useridobject = jwtTokenUtil.getClaim(token).get(JwtTokenUtil.CLAIM_KEY_USERID);
        if(useridobject != null){
            String userid = useridobject.toString();
            SecurityHolderUtils.setUserId(userid);
        }
    }
}