com.github.masonm.JwtStubMappingTransformer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wiremock-jwt-extension Show documentation
Wiremock request matcher and stub mapping transformer for JSON Web Tokens (JWT)
There is a newer version: 1.0.0
Show newest version
package com.github.masonm;

import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.github.tomakehurst.wiremock.common.FileSource;
import com.github.tomakehurst.wiremock.extension.Parameters;
import com.github.tomakehurst.wiremock.extension.StubMappingTransformer;
import com.github.tomakehurst.wiremock.matching.MultiValuePattern;
import com.github.tomakehurst.wiremock.matching.RequestPattern;
import com.github.tomakehurst.wiremock.matching.RequestPatternBuilder;
import com.github.tomakehurst.wiremock.stubbing.StubMapping;

import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.Map;

public class JwtStubMappingTransformer extends StubMappingTransformer {
    public static final String PAYLOAD_FIELDS = "payloadFields";

    @Override
    public String getName() {
        return "jwt-stub-mapping-transformer";
    }

    @Override
    public boolean applyGlobally() {
        return false;
    }

    @Override
    public StubMapping transform(StubMapping stubMapping, FileSource files, Parameters parameters) {
        if (!parameters.containsKey(PAYLOAD_FIELDS)) {
            return stubMapping;
        }

        if (stubMapping.getRequest().getCustomMatcher() != null) {
            // already has a custom matcher. Don't overwrite
            return stubMapping;
        }

        Map requestHeaders = stubMapping.getRequest().getHeaders();
        if (requestHeaders == null || !requestHeaders.containsKey("Authorization")) {
            return stubMapping;
        }

        String authHeader = requestHeaders.get("Authorization").getExpected();
        Parameters requestMatcherParameters = getRequestMatcherParameter(
            Jwt.fromAuthHeader(authHeader),
            parameters.get(PAYLOAD_FIELDS)
        );

        if (requestMatcherParameters == null) {
            return stubMapping;
        }

        requestMatcherParameters.put("request", this.getInnerRequestPattern(stubMapping.getRequest()));

        RequestPattern newRequest = new RequestPatternBuilder(JwtMatcherExtension.NAME, requestMatcherParameters).build();
        stubMapping.setRequest(newRequest);
        return stubMapping;
    }

    private RequestPattern getInnerRequestPattern(RequestPattern outer) {
        Map newHeaders = null;
        if (outer.getHeaders() != null) {
            newHeaders = new LinkedHashMap<>(outer.getHeaders());
            newHeaders.remove("Authorization");
            if (newHeaders.isEmpty()) {
                newHeaders = null;
            }
        }

        return new RequestPattern(
            outer.getUrlMatcher(),
            outer.getMethod(),
            newHeaders,
            outer.getQueryParameters(),
            outer.getCookies(),
            outer.getBasicAuthCredentials(),
            outer.getBodyPatterns(),
            null,
            outer.getMultipartPatterns()
        );
    }

    private Parameters getRequestMatcherParameter(Jwt token, Object payloadParamValue) {
        if (token.getPayload().isMissingNode()) {
            return null;
        }

        Iterable payloadFields = new ObjectMapper().convertValue(
            payloadParamValue,
            new TypeReference>() {}
        );
        Parameters params = new Parameters();

        Map payload = new HashMap<>();
        for (String field: payloadFields) {
            payload.put(field, token.getPayload().path(field).asText());
        }
        params.put(JwtMatcherExtension.PARAM_NAME_PAYLOAD, payload);

        return params;
    }
}