BaseEndpoint.java

/*
 * Copyright 2019 Global Crop Diversity Trust
 *
 * 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 org.gringlobal.soap;

import javax.xml.transform.Result;
import javax.xml.transform.Source;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.dom.DOMSource;

import org.jdom2.Document;
import org.jdom2.Element;
import org.jdom2.input.DOMBuilder;
import org.jdom2.transform.JDOMResult;
import org.springframework.xml.transform.TransformerHelper;
import org.w3c.dom.Node;

public abstract class BaseEndpoint {

	private final TransformerHelper transformerHelper = new TransformerHelper();

	protected Element getSource(final Source requestPayload) throws TransformerException {
		if (requestPayload instanceof DOMSource) {
			final Node node = ((DOMSource) requestPayload).getNode();
			final DOMBuilder domBuilder = new DOMBuilder();
			if (node.getNodeType() == Node.ELEMENT_NODE) {
				return domBuilder.build((org.w3c.dom.Element) node);
			} else if (node.getNodeType() == Node.DOCUMENT_NODE) {
				final Document document = domBuilder.build((org.w3c.dom.Document) node);
				return document.getRootElement();
			}
		}
		// we have no other option than to transform
		final JDOMResult jdomResult = new JDOMResult();
		transform(requestPayload, jdomResult);
		return jdomResult.getDocument().getRootElement();
	}

	protected Element getElement(final Node node) {
		final DOMBuilder domBuilder = new DOMBuilder();
		if (node.getNodeType() == Node.ELEMENT_NODE) {
			return domBuilder.build((org.w3c.dom.Element) node);
		} else if (node.getNodeType() == Node.DOCUMENT_NODE) {
			final Document document = domBuilder.build((org.w3c.dom.Document) node);
			return document.getRootElement();
		}
		throw new RuntimeException("Cannot transform to jdom2 " + node);
	}

	/**
	 * Transforms the given {@link Source} to the given {@link Result}. Creates a
	 * new {@link Transformer} for every call, as transformers are not thread-safe.
	 *
	 * @param source the source to transform from
	 * @param result the result to transform to
	 * @throws TransformerException if thrown by JAXP methods
	 */
	protected final void transform(final Source source, final Result result) throws TransformerException {
		transformerHelper.transform(source, result);
	}
}