Simple Xslt payload transformer

org.springframework.integration.xml/src/main/java/org/springframework/integration/xml/transformer/XsltPayloadTransformer.java

@@ -0,0 +1,96 @@
+/*
+ * Copyright 2002-2008 the original author or authors.
+ *
+ * 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.springframework.integration.xml.transformer;
+
+import javax.xml.transform.Source;
+import javax.xml.transform.Templates;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMResult;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.stream.StreamSource;
+
+import org.springframework.core.io.Resource;
+import org.springframework.integration.message.Message;
+import org.springframework.integration.message.MessagingException;
+import org.springframework.integration.transformer.MessageTransformer;
+import org.springframework.xml.transform.StringResult;
+import org.springframework.xml.transform.StringSource;
+import org.w3c.dom.Document;
+
+/**
+ * Simple Xslt transformer implementation which returns a transformed {@link Source} a {@link Document} or a {@link String} 
+ * @author Jonas Partner
+ *
+ */
+public class XsltPayloadTransformer implements MessageTransformer {
+
+	private final Templates templates;
+
+	public XsltPayloadTransformer(Templates templates) {
+		this.templates = templates;
+	}
+
+	public XsltPayloadTransformer(Resource xslResource) throws Exception {
+		this.templates = TransformerFactory.newInstance().newTemplates(new StreamSource(xslResource.getInputStream()));
+	}
+
+	
+	@SuppressWarnings("unchecked")
+	public void transform(Message message) {
+		try{
+			
+			if( (Document.class.isAssignableFrom(message.getPayload().getClass()))){
+				transformDocument(message);
+			} else if(Source.class.isAssignableFrom(message.getPayload().getClass())){
+				transformSource(message);
+			} else if (String.class.equals(message.getPayload().getClass())){
+				transformString(message);
+			} else {
+				throw new MessagingException(message,"Unsupproted payload type for transformation " + message.getPayload().getClass().getName());
+			}
+		
+		} catch (TransformerException transE){
+			throw new MessagingException(message,"XSLT transformation failed", transE);
+		}
+		
+	}
+	
+	@SuppressWarnings("unchecked")
+	protected void transformSource(Message message) throws TransformerException {
+		StringResult result = new StringResult();
+		templates.newTransformer().transform((Source)message.getPayload(), result);
+		message.setPayload(new StringSource(result.toString()));
+	}
+	
+	
+	@SuppressWarnings("unchecked")
+	private void transformString(Message message) throws TransformerException {
+		StringResult result = new StringResult();
+		templates.newTransformer().transform(new StringSource((String)message.getPayload()), result);
+		message.setPayload(result.toString());
+	}
+
+	@SuppressWarnings("unchecked")
+	protected void transformDocument(Message message) throws TransformerException{
+		Document doc = (Document)message.getPayload();
+		DOMResult result = new DOMResult();
+		templates.newTransformer().transform(new DOMSource(doc), result);
+		message.setPayload(result.getNode());
+	}
+
+}