HLTech / pact-gen

import java.util.List;

import java.util.stream.Collectors;

public class JaxRsMethodRepresentationExtractor implements ClientMethodRepresentationExtractor {

    private final Collection<AnnotatedMethodHandler> annotatedMethodHandlers;

    public JaxRsMethodRepresentationExtractor(Collection<AnnotatedMethodHandler> annotatedMethodHandlers) {

        this.annotatedMethodHandlers = annotatedMethodHandlers;

    }

    @Override

    public ClientMethodRepresentation extractClientMethodRepresentation(Method clientMethod) {

        return ClientMethodRepresentation.builder()

                .requestRepresentation(extractRequestProperties(clientMethod))

                .responseRepresentationList(extractResponseProperties(clientMethod))

                .build();

    }

    private RequestRepresentation extractRequestProperties(Method clientMethod) {

        return this.annotatedMethodHandlers.stream()

                .filter(annotationHandler -> annotationHandler.isSupported(clientMethod))

                .findFirst().orElseThrow(() -> new IllegalArgumentException("Unknown HTTP method"))

                .handleRequest(clientMethod);

    }

    private List<ResponseRepresentation> extractResponseProperties(Method method) {

        String[] responseHeaders = this.annotatedMethodHandlers.stream()

            .filter(annotationHandler -> annotationHandler.isSupported(method))

            .findFirst().orElseThrow(() -> new IllegalArgumentException("Unknown HTTP method"))

            .getResponseMediaHeaders(method);

        List<ResponseRepresentation> results = Arrays

            .stream(method.getDeclaredAnnotationsByType(InteractionInfo.class))

            .map(annotation -> ResponseRepresentation.from(

                annotation.responseStatus(),

                RawHeadersParser.parseAll(annotation.responseHeaders()),

                method,

                annotation.description(),

                annotation.emptyBodyExpected()))

            .collect(Collectors.toList());

        return !results.isEmpty()

            ? results

            : Lists.newArrayList(ResponseRepresentation.getDefaultForMethod(method, responseHeaders));

    }

}

import java.util.List;

import java.util.stream.Collectors;

public class FeignMethodRepresentationExtractor implements ClientMethodRepresentationExtractor {

    private final Collection<AnnotatedMethodHandler> annotatedMethodHandlers;

    public FeignMethodRepresentationExtractor(Collection<AnnotatedMethodHandler> annotatedMethodHandlers) {

        this.annotatedMethodHandlers = annotatedMethodHandlers;

    }

    @Override

    public ClientMethodRepresentation extractClientMethodRepresentation(Method method) {

        return ClientMethodRepresentation.builder()

            .requestRepresentation(extractRequestProperties(method))

            .responseRepresentationList(extractResponseProperties(method))

            .build();

    }

    private RequestRepresentation extractRequestProperties(Method method) {

        return this.annotatedMethodHandlers.stream()

            .filter(annotationHandler -> annotationHandler.isSupported(method))

            .findFirst().orElseThrow(() -> new IllegalArgumentException("Unknown HTTP method"))

            .handleRequest(method);

    }

    private List<ResponseRepresentation> extractResponseProperties(Method method) {

        String[] responseHeaders = this.annotatedMethodHandlers.stream()

            .filter(annotationHandler -> annotationHandler.isSupported(method))

            .findFirst().orElseThrow(() -> new IllegalArgumentException("Unknown HTTP method"))

            .getResponseMediaHeaders(method);

        List<ResponseRepresentation> results = Arrays

            .stream(method.getDeclaredAnnotationsByType(InteractionInfo.class))

            .map(annotation -> ResponseRepresentation.from(

                annotation.responseStatus(),

                RawHeadersParser.parseAll(ArrayUtils.addAll(annotation.responseHeaders(), responseHeaders)),

                method,

                annotation.description(),

                annotation.emptyBodyExpected()))

            .collect(Collectors.toList());

        return !results.isEmpty()

            ? results

            : Lists.newArrayList(ResponseRepresentation.getDefaultForMethod(method, responseHeaders));

    }

}