malinink / easy-tests

import com.atlassian.oai.validator.SwaggerRequestResponseValidator;

import com.atlassian.oai.validator.model.Request;

import com.atlassian.oai.validator.model.Response;

import com.atlassian.oai.validator.model.SimpleRequest;

import com.atlassian.oai.validator.model.SimpleResponse;

import com.atlassian.oai.validator.report.ValidationReport;

import java.io.IOException;

import java.io.Reader;

import java.util.*;

import javax.annotation.Nonnull;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.springframework.core.io.support.EncodedResource;

import org.springframework.web.util.ContentCachingResponseWrapper;

import org.springframework.web.util.UriComponents;

import org.springframework.web.util.UriComponentsBuilder;

/**

 * @author atlassian

 */

class SwaggerRequestValidationService {

    private static final String MESSAGE_REQUEST_PATH_MISSING = "validation.request.path.missing";

    private final SwaggerRequestResponseValidator requestValidator;

    SwaggerRequestValidationService(final EncodedResource restInterface) throws IOException {

        this(SwaggerRequestResponseValidator

                .createFor(readReader(restInterface.getReader()))

                .build());

    }

    SwaggerRequestValidationService(final SwaggerRequestResponseValidator requestValidator) {

        Objects.requireNonNull(requestValidator, "A request validator is required.");

        this.requestValidator = requestValidator;

    }

    /**

     * @param servletRequest the {@link HttpServletRequest}

     * @return the build {@link Request} created out of given {@link HttpServletRequest}

     * @throws IOException if the request body can't be read

     */

    /**

     * @param servletResponse the {@link ContentCachingResponseWrapper}

     * @return the build {@link Response} created out of given {@link ContentCachingResponseWrapper}

     * @throws IOException if the response body can't be write

     */

    Response buildResponse(final ContentCachingResponseWrapper servletResponse) throws IOException {

        final SimpleResponse.Builder builder = new SimpleResponse.Builder(servletResponse.getStatus());

        builder.withBody(new String(servletResponse.getContentAsByteArray()));

        for (final String headerName : servletResponse.getHeaderNames()) {

            builder.withHeader(headerName, new ArrayList(servletResponse.getHeaders(headerName)));

        }

        return builder.build();

    }

    /**

     * @param request the {@link Request} to validate against the Swagger schema

     * @return the {@link ValidationReport} for the validated {@link Request}

     */

    ValidationReport validateRequest(final Request request) {

        return requestValidator.validateRequest(request);

    }

    ValidationReport validateResponse(@Nonnull String path, @Nonnull Request.Method method, final Response response) {

        return requestValidator.validateResponse(path, method, response);

    }

    /**

     * @param validationReport the {@link ValidationReport}

     * @return {@code true} if the validated request is defined in the Swagger schema, otherwise {@code false}

     */

    boolean isDefinedSwaggerRequest(final ValidationReport validationReport) {

        for (final ValidationReport.Message message : validationReport.getMessages()) {

            if (MESSAGE_REQUEST_PATH_MISSING.equals(message.getKey())) {

                return false;

            }

        }

        return true;

    }

    private static String readReader(final Reader reader) throws IOException {

        return IOUtils.toString(reader);

    }

    private static String getCompleteRequestUri(final HttpServletRequest servletRequest) {

        if (StringUtils.isBlank(servletRequest.getQueryString())) {

            return servletRequest.getRequestURI();

        }

        return servletRequest.getRequestURI() + "?" + servletRequest.getQueryString();

    }

}