org.connid.bundles.soap.cxf.ForceSoapActionOutInterceptor.java Source code

Introduction

Here is the source code for org.connid.bundles.soap.cxf.ForceSoapActionOutInterceptor.java

Source

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE
 * file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file
 * to you 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 org.connid.bundles.soap.cxf;

import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
import org.apache.commons.lang.StringUtils;

import org.apache.cxf.binding.soap.Soap11;
import org.apache.cxf.binding.soap.Soap12;
import org.apache.cxf.binding.soap.SoapBindingConstants;
import org.apache.cxf.binding.soap.SoapMessage;
import org.apache.cxf.binding.soap.interceptor.AbstractSoapInterceptor;
import org.apache.cxf.binding.soap.model.SoapOperationInfo;
import org.apache.cxf.helpers.CastUtils;
import org.apache.cxf.interceptor.Fault;
import org.apache.cxf.message.Message;
import org.apache.cxf.phase.Phase;
import org.apache.cxf.service.model.BindingOperationInfo;

/**
 * This interceptor is responsible for setting up the SOAP version and header, so that this is available to any
 * pre-protocol interceptors that require these to be available.
 */
public class ForceSoapActionOutInterceptor extends AbstractSoapInterceptor {

    private final String SOAPActionUriPrefix;

    public ForceSoapActionOutInterceptor(final String SOAPActionUriPrefix) {
        super(Phase.POST_LOGICAL);

        this.SOAPActionUriPrefix = SOAPActionUriPrefix == null ? null
                : (SOAPActionUriPrefix.endsWith("/") ? SOAPActionUriPrefix : SOAPActionUriPrefix + "/");
    }

    /**
     * Mediate a message dispatch.
     *
     * @param message the current message
     * @throws Fault
     */
    public void handleMessage(SoapMessage message) throws Fault {
        setSoapAction(message);

    }

    private void setSoapAction(SoapMessage message) {
        BindingOperationInfo boi = message.getExchange().getBindingOperationInfo();

        // The soap action is set on the wrapped operation.
        if (boi != null && boi.isUnwrapped()) {
            boi = boi.getWrappedOperation();
        }

        String action = getSoapAction(message, boi);

        if (message.getVersion() instanceof Soap11) {
            Map<String, List<String>> reqHeaders = CastUtils
                    .cast((Map<?, ?>) message.get(Message.PROTOCOL_HEADERS));
            if (reqHeaders == null) {
                reqHeaders = new TreeMap<String, List<String>>(String.CASE_INSENSITIVE_ORDER);
            }

            if (reqHeaders.size() == 0) {
                message.put(Message.PROTOCOL_HEADERS, reqHeaders);
            }

            if (!reqHeaders.containsKey(SoapBindingConstants.SOAP_ACTION)) {
                reqHeaders.put(SoapBindingConstants.SOAP_ACTION, Collections.singletonList(action));
            }
        } else if (message.getVersion() instanceof Soap12 && !"\"\"".equals(action)) {
            String ct = (String) message.get(Message.CONTENT_TYPE);

            if (ct.indexOf("action=\"") == -1) {
                ct = new StringBuilder().append(ct).append("; action=").append(action).toString();
                message.put(Message.CONTENT_TYPE, ct);
            }
        }
    }

    private String getSoapAction(SoapMessage message, BindingOperationInfo boi) {
        // allow an interceptor to override the SOAPAction if need be
        String action = (String) message.get(SoapBindingConstants.SOAP_ACTION);

        // Fall back on the SOAPAction in the operation info
        if (action == null) {
            if (boi == null) {
                action = "\"\"";
            } else {
                BindingOperationInfo dboi = (BindingOperationInfo) boi.getProperty("dispatchToOperation");
                if (null != dboi) {
                    boi = dboi;
                }

                SoapOperationInfo soi = boi.getExtensor(SoapOperationInfo.class);

                action = soi == null || StringUtils.isBlank(soi.getAction())
                        || StringUtils.isBlank(SOAPActionUriPrefix) ? "\"\""
                                : (SOAPActionUriPrefix + soi.getAction());
            }
        }

        if (!action.startsWith("\"")) {
            action = new StringBuilder().append("\"").append(action).append("\"").toString();
        }

        return action;
    }
}