@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface Login {}

// import
import org.springframework.web.method.support.HandlerMethodArgumentResolver;

// main class
public class LoginMemberArgumentResolver implements HandlerMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter parameter) {

        // @Login 애노테이션이 파라미터에 있는지?
        boolean hasLoginAnnotation = parameter.hasParameterAnnotation(Login.class);

        /** 
        * parameter.getParameterType() : @Login 애노테이션 뒤에 붙은 클래스
        * isAssignableFrom : 할당할 수 있는지.
        */
        boolean hasModelType = Member.class.isAssignableFrom(parameter.getParameterType()); 

        return hasLoginAnnotation && hasModelType;
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        
        HttpServletRequest request = (HttpServletRequest) webRequest.getNativeRequest();

        HttpSession session = request.getSession();
        if (session == null) {
            return null;
        }
        return session.getAttribute("LOGIN_MEMBER");
    }
}

@Configuration
public class WebConfig implements WebMvcConfigurer {
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
          resolvers.add(new LoginMemberArgumentResolver());
      }
}

@RequiredArgsConstructor
public class LoginMemberArgumentResolver implements HandlerMethodArgumentResolver {

    private final UserMapper userMapper;

    @Override
    public boolean supportsParameter(MethodParameter parameter) {

        
        boolean hasLoginAnnotation = parameter.hasParameterAnnotation(Login.class);
        boolean hasModelType = User.class.isAssignableFrom(parameter.getParameterType()); 

        return hasLoginAnnotation && hasModelType;
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        
        // Token 정보 확인 
        HttpServletRequest request = (HttpServletRequest) webRequest.getNativeRequest();
        String header = request.getHeader("Authorization");
        String token = JwtUtils.getTokenFromHeader(header);


        if (JwtUtils.isValideToken(token)) {

            // Email 정보 가져오기
            String email = JwtUtils.getUserEmailFromToken(token);

            // 가져온 email정보로 해당 User객체 반환
            User checkUser = findUserByEmail(String email);
            return checkUser;

        } else if (!JwtUtils.isValideToken(token)) {

            /** Access Token이 유효하지 않을 경우
            * 만료된 토큰에서 Email 가져온다.
            * DB에서 Email로 해당 객체의 Refresh Token을 가져온다.
            * 다시 새로운 AccessToken을 만들어 준다.
            */
            String email = JwtUtils.getUserEmailFromToken(token);
            User checkUser = findUserByEmail(String email);
            String reToken = checkUser.getRetoken();

            if (JwtUtils.isValideReToken(reToken)) {

                // Refresh Token을 기반으로 새로운 Access Token을 생성
                User user = new User();
                user.setAuthCode(JwtUtils.getClaimsCodeFromToken(reToken));
                user.setEmail(JwtUtils.getUserEmailFromToken(reToken));
                user.setUserSeq(Long.parseLong(JwtUtils.getClaimsSeqFromToken(reToken)));

                response.setHeader(AuthConstants.AUTH_HEADER, AuthConstants.TOKEN_TYPE + " " + JwtUtils.generateJwtToken(user));
                response.setHeader(AuthConstants.RE_AUTH_HEADER, AuthConstants.TOKEN_TYPE + " " + reToken);
                
                HttpSession session = request.getSession();
                session.setAttribute(user.getEmail()+"_token", reToken);
        
                return checkUser;
                
            } else {
                response.addHeader("Content-Type", "application/json");
                response.getWriter().print("Fail");
                return false;
            }
        }
    }
}