All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.gravitee.policy.rest2soap.RestToSoapTransformerPolicy Maven / Gradle / Ivy

/**
 * Copyright (C) 2015 The Gravitee team (http://gravitee.io)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.gravitee.policy.rest2soap;

import io.gravitee.common.http.HttpMethod;
import io.gravitee.common.http.HttpStatusCode;
import io.gravitee.common.http.MediaType;
import io.gravitee.common.util.LinkedMultiValueMap;
import io.gravitee.el.exceptions.ELNullEvaluationException;
import io.gravitee.gateway.api.ExecutionContext;
import io.gravitee.gateway.api.Request;
import io.gravitee.gateway.api.Response;
import io.gravitee.gateway.api.buffer.Buffer;
import io.gravitee.gateway.api.http.stream.TransformableRequestStreamBuilder;
import io.gravitee.gateway.api.stream.ReadWriteStream;
import io.gravitee.policy.api.PolicyChain;
import io.gravitee.policy.api.annotations.OnRequest;
import io.gravitee.policy.api.annotations.OnRequestContent;
import io.gravitee.policy.rest2soap.configuration.SoapTransformerPolicyConfiguration;
import io.gravitee.policy.rest2soap.el.EvaluableRequest;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author David BRASSELY (david.brassely at graviteesource.com)
 * @author GraviteeSource Team
 */
public class RestToSoapTransformerPolicy {

    /**
     * Logger.
     */
    private final static Logger LOGGER = LoggerFactory.getLogger(RestToSoapTransformerPolicy.class);

    private final static String SOAP_ACTION_HEADER = "SOAPAction";

    public final static String QUERY_PARAMS_ATTRIBUTE = ExecutionContext.ATTR_PREFIX + "rest-to-soap.query-parameters";

    /**
     * SOAP transformer configuration
     */
    private final SoapTransformerPolicyConfiguration soapTransformerPolicyConfiguration;

    public RestToSoapTransformerPolicy(SoapTransformerPolicyConfiguration soapTransformerPolicyConfiguration) {
        this.soapTransformerPolicyConfiguration = soapTransformerPolicyConfiguration;
    }

    @OnRequest
    public void onRequest(Request request, Response response, ExecutionContext executionContext, PolicyChain policyChain) {
        // Force method to POST for SOAP requests
        LOGGER.debug("Override HTTP methods for SOAP invocation");
        executionContext.setAttribute(ExecutionContext.ATTR_REQUEST_METHOD, HttpMethod.POST);

        if (soapTransformerPolicyConfiguration.isStripPath()) {
            executionContext.setAttribute(ExecutionContext.ATTR_REQUEST_ENDPOINT, "");
        }

        if (soapTransformerPolicyConfiguration.getSoapAction() != null &&
                !soapTransformerPolicyConfiguration.getSoapAction().trim().isEmpty()) {
            LOGGER.debug("Add a SOAPAction header to invoke SOAP WS: {}", soapTransformerPolicyConfiguration.getSoapAction());
            request.headers().set(SOAP_ACTION_HEADER, soapTransformerPolicyConfiguration.getSoapAction());
        }

        if (!soapTransformerPolicyConfiguration.isPreserveQueryParams()) {
            // We are removing parameters from request, but those one can be used from SOAP envelope template
            // So let's create a copy of them for then being reused later from template engine
            executionContext.setAttribute(QUERY_PARAMS_ATTRIBUTE, new LinkedMultiValueMap<>(request.parameters()));

            request.parameters().clear();
        }

        policyChain.doNext(request, response);
    }

    @OnRequestContent
    public ReadWriteStream onRequestContent(Request request, ExecutionContext executionContext, PolicyChain policyChain) {
        String charset =  soapTransformerPolicyConfiguration.getCharset() ;
        String contentType = (charset == null || charset.isEmpty() ? MediaType.TEXT_XML : MediaType.TEXT_XML + "; charset="+charset);
        return TransformableRequestStreamBuilder
                .on(request)
		.contentType(contentType)
                .transform(
                        buffer -> {
                            executionContext.getTemplateEngine().getTemplateContext().setVariable("request",
                                    new EvaluableRequest(executionContext, buffer.toString()));

                            String soapEnvelope = executionContext.getTemplateEngine().convert(
                                    soapTransformerPolicyConfiguration.getEnvelope());

                            if (soapEnvelope == null) {

                                policyChain.streamFailWith(io.gravitee.policy.api.PolicyResult.failure(
                                        HttpStatusCode.INTERNAL_SERVER_ERROR_500,  new ELNullEvaluationException(soapTransformerPolicyConfiguration.getEnvelope()).getMessage()));
                                return null;
                            }

                            return Buffer.buffer(soapEnvelope);
                        }
                ).build();
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy