com.thebund1st.daming.sdk.web.handler.SmsVerificationRequiredHandlerInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sms-verification-sdk Show documentation

Your description of the lib

The newest version!

package com.thebund1st.daming.sdk.web.handler;

import com.thebund1st.daming.sdk.jwt.SmsVerificationClaims;
import com.thebund1st.daming.sdk.jwt.SmsVerificationJwtVerifier;
import com.thebund1st.daming.sdk.web.annotation.SmsVerificationRequired;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

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

public class SmsVerificationRequiredHandlerInterceptor extends HandlerInterceptorAdapter {

    private final SmsVerificationJwtVerifier smsVerificationJwtVerifier;
    private final String jwtHeaderName;

    public SmsVerificationRequiredHandlerInterceptor(SmsVerificationJwtVerifier smsVerificationJwtVerifier,
                                                     String jwtHeaderName) {
        this.smsVerificationJwtVerifier = smsVerificationJwtVerifier;
        this.jwtHeaderName = jwtHeaderName;
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        HandlerMethod handlerMethod = (HandlerMethod) handler;
        if (handlerMethod.hasMethodAnnotation(SmsVerificationRequired.class)) {
            SmsVerificationRequired smsVerificationRequired = handlerMethod
                    .getMethodAnnotation(SmsVerificationRequired.class);
            final String jwt = request.getHeader(jwtHeaderName);
            SmsVerificationClaims claims = smsVerificationJwtVerifier.verify(jwt, smsVerificationRequired.scope());
            request.setAttribute("smsVerificationClaims", claims);
            return true;
        } else {
            return true;
        }
    }
}