HLTech / pact-gen

import com.hltech.pact.gen.domain.client.annotation.MappingMethodHandler;

import com.hltech.pact.gen.domain.client.model.Param;

import com.hltech.pact.gen.domain.client.model.RequestRepresentation;

import com.hltech.pact.gen.domain.client.util.PathParametersExtractor;

import com.hltech.pact.gen.domain.client.util.RawHeadersParser;

import com.hltech.pact.gen.domain.client.util.RequestBodyExtractor;

import com.hltech.pact.gen.domain.client.util.RequestHeaderParamsExtractor;

import com.hltech.pact.gen.domain.client.util.RequestParametersExtractor;

import org.apache.commons.lang3.ArrayUtils;

import org.springframework.http.HttpMethod;

import org.springframework.web.bind.annotation.PutMapping;

import java.lang.reflect.Method;

import java.util.Arrays;

import java.util.List;

import java.util.stream.Collectors;

import java.util.stream.Stream;

public class PutMappingMethodsHandler implements AnnotatedMethodHandler {

    @Override

    public boolean isSupported(Method method) {

        return method.isAnnotationPresent(PutMapping.class);

    }

    @Override

    public RequestRepresentation handleRequest(Method method) {

        return RequestRepresentation.builder()

            .httpMethod(HttpMethod.PUT)

            .path(getPathFromMethod(method))

            .headers(combineHeaders(

                ArrayUtils.addAll(method.getAnnotation(PutMapping.class).headers(), getRequestMediaHeaders(method)),

                RequestHeaderParamsExtractor.extractAll(method)))

            .body(RequestBodyExtractor.extract(method.getParameters()))

            .requestParameters(RequestParametersExtractor.extractAll(method))

            .pathParameters(PathParametersExtractor.extractAll(method))

            .build();

    }

    @Override

    public String[] getResponseMediaHeaders(Method method) {

        return Arrays.stream(method.getAnnotation(PutMapping.class).produces())

            .map(header -> "Content-Type=" + header)

            .toArray(String[]::new);

    }

    private String getPathFromMethod(Method method) {

        PutMapping annotation = method.getAnnotation(PutMapping.class);

        return annotation.path().length == 1 ? annotation.path()[0] : annotation.value()[0];

    }

    private static List<Param> combineHeaders(String[] rawHeaders, List<Param> headers) {

        return Stream

            .concat(RawHeadersParser.parseAll(rawHeaders).stream(), headers.stream())

            .collect(Collectors.toList());

    }

}