WebSVN – XServices – Diff – /xservices/trunk/src/main/java/net/brutex/xservices/ws/impl/XmlServiceImpl.java


/*
 *   Copyright 2013 Brian Rosenberger (Brutex Network)
 *
 *   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 net.brutex.xservices.ws.impl;
 
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;
import java.nio.charset.IllegalCharsetNameException;
import java.util.Iterator;
import java.util.List;
import javax.jws.WebService;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLOutputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
 
import lombok.extern.slf4j.Slf4j;
import net.brutex.xservices.types.AttributeType;
import net.brutex.xservices.types.NamespaceListType;
import net.brutex.xservices.types.NamespaceType;
import net.brutex.xservices.types.StringSplitType;
import net.brutex.xservices.types.ant.FileResource;
import net.brutex.xservices.ws.XServicesFault;
import net.brutex.xservices.ws.XmlService;
 
import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMAttribute;
import org.apache.axiom.om.OMCloneOptions;
import org.apache.axiom.om.OMComment;
import org.apache.axiom.om.OMContainer;
import org.apache.axiom.om.OMDocument;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMFactory;
import org.apache.axiom.om.OMNode;
import org.apache.axiom.om.OMProcessingInstruction;
import org.apache.axiom.om.OMText;
import org.apache.axiom.om.OMXMLBuilderFactory;
import org.apache.axiom.om.xpath.AXIOMXPath;
 
 
 
 
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.jaxen.JaxenException;
import org.jaxen.SimpleNamespaceContext;
 
/**
 * @author Brian Rosenberger, bru(at)brutex.de
 * 
 */
@Slf4j 
@WebService(targetNamespace = "http://ws.xservices.brutex.net", endpointInterface = "net.brutex.xservices.ws.XmlService", serviceName = "XmlService")
public class XmlServiceImpl implements XmlService {
 
 
 
	public String insertNodesFromFile(FileResource res, NamespaceListType nsList, String xpath, String xmlFragment)
			throws XServicesFault {
		try {
			AXIOMXPath axp = new AXIOMXPath(xpath);
			InputStream is = res.getAntResource(null).getInputStream();
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
			OMDocument fragdoc = null;
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
						.getDocument();
			} else {
				throw new XServicesFault("No xmldata to insert.");
			}
 
			// Initialize XPath context
			SimpleNamespaceContext context = createContext(nsList);
			axp.setNamespaceContext(context);
			axp.addNamespaces(fragdoc.getOMDocumentElement());
 
			OMDocument document = insertNodes(sourcedoc, axp, fragdoc);
 
			StringWriter sw = new StringWriter();
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
			document.serialize(writer);
 
			log.debug(sw.getBuffer().toString());
			return sw.getBuffer().toString();
		} catch (JaxenException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (IOException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (XMLStreamException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		}
	}
 
	public String replaceNodesFromFile(FileResource res, NamespaceListType nsList, String xpath, String xmlFragment)
			throws XServicesFault {
		try {
			AXIOMXPath axp = new AXIOMXPath(xpath);
			InputStream is = res.getAntResource(null).getInputStream();
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
			OMDocument fragdoc = null;
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
						.getDocument();
			} else {
				throw new XServicesFault("No xmldata to insert.");
			}
 
			// Initialize XPath context
			SimpleNamespaceContext context = createContext(nsList);
			axp.setNamespaceContext(context);
			axp.addNamespaces(fragdoc.getOMDocumentElement());
 
			OMDocument document = replaceNodes(sourcedoc, axp, fragdoc);
 
			StringWriter sw = new StringWriter();
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
			document.serialize(writer);
 
			log.debug(sw.getBuffer().toString());
			return sw.getBuffer().toString();
		} catch (JaxenException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (XMLStreamException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (IOException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		}
	}
 
	public String replaceNodes(String source, String encoding, NamespaceListType nsList, String xpath,
			String xmlFragment) throws XServicesFault {
		encoding = validateEncoding(encoding);
		try {
			AXIOMXPath axp = new AXIOMXPath(xpath);
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
			OMDocument fragdoc = null;
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
						.getDocument();
			} else {
				throw new XServicesFault("No xmldata to insert.");
			}
 
			// Initialize XPath context
			SimpleNamespaceContext context = createContext(nsList);
			axp.setNamespaceContext(context);
			axp.addNamespaces(fragdoc.getOMDocumentElement());
 
			OMDocument document = replaceNodes(sourcedoc, axp, fragdoc);
 
			StringWriter sw = new StringWriter();
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
			document.serialize(writer);
 
			log.debug(sw.getBuffer().toString());
			return sw.getBuffer().toString();
		} catch (JaxenException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (XMLStreamException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (UnsupportedEncodingException e) {
			throw new XServicesFault(e);
		}
	}
 
	@RequiresPermissions("insertNodes")
	public String insertNodes(String source, String encoding, NamespaceListType nsList, String xpath,
			String xmlFragment) throws XServicesFault {
		encoding = validateEncoding(encoding);
		try {
			AXIOMXPath axp = new AXIOMXPath(xpath);
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
			OMDocument fragdoc = null;
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
						.getDocument();
			} else {
				throw new XServicesFault("No xmldata to insert.");
			}
 
			// Initialize XPath context
			SimpleNamespaceContext context = createContext(nsList);
			axp.setNamespaceContext(context);
			axp.addNamespaces(fragdoc.getOMDocumentElement());
 
			OMDocument document = insertNodes(sourcedoc, axp, fragdoc);
 
			StringWriter sw = new StringWriter();
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
			document.serialize(writer);
 
			log.debug(sw.getBuffer().toString());
			return sw.getBuffer().toString();
		} catch (JaxenException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (XMLStreamException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (UnsupportedEncodingException e) {
			throw new XServicesFault(e);
		}
	}
 
	public String wrapInCDATA(String data) throws XServicesFault {
		String result = "";
		String[] tokens = data.split("\\]\\]>", -1);
 
		for (int i = 0; i < tokens.length; i++) {
			result += tokens[i];
			if (i + 1 < tokens.length)
				result += "]]]]><![CDATA[>";
		}
 
		result = "<![CDATA[" + result + "]]>";
		return result;
	}
 
	public StringSplitType selectXPath(String source, String encoding, NamespaceListType nsList, String xpath)
			throws XServicesFault {
		encoding = validateEncoding(encoding);
		try {
			StringSplitType rarray = new StringSplitType();
			AXIOMXPath axp = new AXIOMXPath(xpath);
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
 
			// Initialize XPath context
			SimpleNamespaceContext context = createContext(nsList);
 
			axp.setNamespaceContext(context);
			List results = axp.selectNodes(sourcedoc);
			for (Object o : results) {
				String text = null;
 
				if (o instanceof OMNode) {
					switch (((OMNode) o).getType()) {
					case OMNode.TEXT_NODE:
						text = ((OMText) o).getText();
						break;
					case OMNode.COMMENT_NODE:
						text = ((OMComment) o).getValue();
						break;
					case OMNode.PI_NODE:
						text = ((OMProcessingInstruction) o).getValue();
						break;
					default:
						StringWriter sw = new StringWriter();
						XMLOutputFactory xof = XMLOutputFactory.newInstance();
						XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
						((OMNode) o).serialize(writer);
						writer.flush();
						text = sw.toString();
					}
				} else if (o instanceof OMAttribute) {
					text = ((OMAttribute) o).getAttributeValue();
				} else {
					text = String.valueOf(o);
				}
				rarray.addStringMatch(text);
			}
 
			return rarray;
		} catch (JaxenException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (XMLStreamException e) {
			// TODO Auto-generated catch block
			throw new XServicesFault(e.getMessage());
		} catch (UnsupportedEncodingException e) {
			throw new XServicesFault(e);
		}
	}
 
	public String setAttribute(String source, String encoding, NamespaceListType nsList, String xpath,
			AttributeType attr) throws XServicesFault {
		encoding = validateEncoding(encoding);
		try {
			StringSplitType rarray = new StringSplitType();
			AXIOMXPath axp = new AXIOMXPath(xpath);
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
			OMFactory fac = OMAbstractFactory.getOMFactory();
 
			// Initialize XPath context
			SimpleNamespaceContext context = createContext(nsList);
 
			axp.setNamespaceContext(context);
			List results = axp.selectNodes(sourcedoc);
			for (Object o : results) {
				String text = null;
 
				if (o instanceof OMNode) {
					switch (((OMNode) o).getType()) {
					case OMNode.ELEMENT_NODE:
						OMElement node = ((OMElement) o);
						if (attr.value == null) {
							node.removeAttribute(node.getAttribute(new QName(attr.name)));
						} else {
							node.addAttribute(attr.name, attr.value, node.getNamespace());
						}
						break;
					default:
						throw new XServicesFault("XPath expression did not match an element node.");
					}
				} else {
					throw new XServicesFault("XPath expression did not match a node.");
				}
			}
 
			StringWriter sw = new StringWriter();
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
			sourcedoc.serialize(writer);
			writer.flush();
			return sw.toString();
		} catch (JaxenException e) {
			e.printStackTrace();
			throw new XServicesFault(e);
		} catch (XMLStreamException e) {
			// TODO Auto-generated catch block
			throw new XServicesFault(e.getMessage());
		} catch (UnsupportedEncodingException e) {
			throw new XServicesFault(e);
		}
	}
 
	private OMDocument insertNodes(OMDocument xmldocument, AXIOMXPath axp, OMDocument xmlfragment)
			throws XServicesFault {
		List<?> olist = null;
		try {
			olist = axp.selectNodes(xmldocument.getOMDocumentElement());
			log.debug("XPath '" + axp.toString() + "' has " + olist.size() + " matches.");
			log.debug("XPath root expression is: '" + axp.debug() + "'.");
		} catch (JaxenException e) {
			throw new XServicesFault(e.getMessage(), e);
		}
		if (olist.size() == 0)
			throw new XServicesFault(Messages.getString("XmlService.no_match", new Object[] { axp.toString() }));
 
		// Prepare children to insert
		xmlfragment.build();
 
		// Determine what has been matched
		OMContainer match = null;
		for (Object o : olist) {
			Iterator<?> children = xmlfragment.getOMDocumentElement().getChildren();
			if ((o instanceof OMNode)) {
				OMNode node = (OMNode) o;
				switch (node.getType()) {
				case OMNode.ELEMENT_NODE:
					if ((o instanceof OMElement))
						match = (OMElement) o;
					if ((o instanceof OMDocument))
						match = ((OMDocument) o).getOMDocumentElement();
					log.debug(Messages.getString("XmlService.8"));
					break;
				case OMNode.TEXT_NODE:
					match = ((OMText) o).getParent();
					log.debug(Messages.getString("XmlService.9"));
					break;
				case OMNode.COMMENT_NODE:
					// match = node.getParent();
					match = (OMContainer) node;
					log.debug(Messages.getString("XmlService.10"));
					break;
				default:
 
					log.error("XPath matched " + o.getClass().getCanonicalName() + " Node Type:" + node.getType());
					log.error(Messages.getString("XmlService.11"));
					throw new XServicesFault(Messages.getString("XmlService.12"));
				}
			} else {
				log.error("XPath matched " + o.getClass().getCanonicalName());
				log.error(Messages.getString("XmlService.11"));
				throw new XServicesFault(Messages.getString("XmlService.12"));
			}
 
			while (children.hasNext()) {
				OMNode container = (OMNode) children.next();
				match.addChild((OMNode) container.clone(new OMCloneOptions()));
			}
		}
 
		xmldocument.build();
		return xmldocument;
	}
 
	private OMDocument replaceNodes(OMDocument xmldocument, AXIOMXPath axp, OMDocument xmlfragment)
			throws XServicesFault {
 
		List<?> olist = null;
		try {
			olist = axp.selectNodes(xmldocument.getOMDocumentElement());
			log.debug("XPath '" + axp.toString() + "' has " + olist.size() + " matches.");
			log.debug("XPath root expression is: '" + axp.debug() + "'.");
		} catch (JaxenException e) {
			throw new XServicesFault(e.getMessage(), e);
		}
		if (olist.size() == 0)
			throw new XServicesFault(Messages.getString("XmlService.no_match", new Object[] { axp.toString() }));
 
		// Prepare children to insert
		xmlfragment.build();
 
		// Determine what has been matched
		OMNode match = null;
		for (Object o : olist) {
			Iterator<?> children = xmlfragment.getOMDocumentElement().getChildren();
			if ((o instanceof OMNode)) {
				OMNode node = (OMNode) o;
				switch (node.getType()) {
				case OMNode.ELEMENT_NODE:
					if ((o instanceof OMElement))
						match = (OMElement) o;
					if ((o instanceof OMDocument))
						match = ((OMDocument) o).getOMDocumentElement();
					log.debug(Messages.getString("XmlService.8"));
					break;
				default:
 
					log.error("XPath matched " + o.getClass().getCanonicalName() + " Node Type:" + node.getType());
					log.error(Messages.getString("XmlService.11"));
					throw new XServicesFault(Messages.getString("XmlService.12"));
				}
			} else {
				log.error("XPath matched " + o.getClass().getCanonicalName());
				log.error(Messages.getString("XmlService.11"));
				throw new XServicesFault(Messages.getString("XmlService.12"));
			}
 
			while (children.hasNext()) {
				OMNode container = (OMNode) children.next();
				match.insertSiblingBefore((OMNode) container.clone(new OMCloneOptions()));
			}
			match.detach();
		}
		xmldocument.build();
		return xmldocument;
	}
 
	private SimpleNamespaceContext createContext(NamespaceListType nsList) {
		// Initialize XPath context
		SimpleNamespaceContext context = new SimpleNamespaceContext();
		if (nsList != null) {
			for (NamespaceType ns : nsList.getNamespaces()) {
				context.addNamespace(ns.getPrefix(), ns.getUri().toString());
				log.debug(
						Messages.getString("XmlService.0") + ns.getPrefix() + "=\"" + ns.getUri().toString() + "\"'");
			}
		} else {
			log.debug("No namespaces defined.");
		}
		return context;
	}
 
	private String validateEncoding(String encoding) throws XServicesFault {
		if (encoding == null || encoding.equals("")) {
			encoding = Charset.defaultCharset().displayName();
		}
		try {
			Charset.isSupported(encoding);
		} catch (IllegalCharsetNameException e) {
			throw new XServicesFault("Endcoding '" + encoding + "' is not supported by this JRE.");
		}
		log.debug("Setting source xml string encoding to '" + encoding + "'");
		return encoding;
	}
 
}

Subversion Repositories XServices

(root)/xservices/trunk/src/main/java/net/brutex/xservices/ws/impl/XmlServiceImpl.java – Rev 198 → 199

Rev 198		Rev 199
1	`/*`	1	`/*`
2	`* Copyright 2013 Brian Rosenberger (Brutex Network)`	2	`* Copyright 2013 Brian Rosenberger (Brutex Network)`
3	`*`	3	`*`
4	`* Licensed under the Apache License, Version 2.0 (the "License");`	4	`* Licensed under the Apache License, Version 2.0 (the "License");`
5	`* you may not use this file except in compliance with the License.`	5	`* you may not use this file except in compliance with the License.`
6	`* You may obtain a copy of the License at`	6	`* You may obtain a copy of the License at`
7	`*`	7	`*`
8	`* http://www.apache.org/licenses/LICENSE-2.0`	8	`* http://www.apache.org/licenses/LICENSE-2.0`
9	`*`	9	`*`
10	`* Unless required by applicable law or agreed to in writing, software`	10	`* Unless required by applicable law or agreed to in writing, software`
11	`* distributed under the License is distributed on an "AS IS" BASIS,`	11	`* distributed under the License is distributed on an "AS IS" BASIS,`
12	`* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.`	12	`* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.`
13	`* See the License for the specific language governing permissions and`	13	`* See the License for the specific language governing permissions and`
14	`* limitations under the License.`	14	`* limitations under the License.`
15	`*/`	15	`*/`
16		16
17	`package net.brutex.xservices.ws.impl;`	17	`package net.brutex.xservices.ws.impl;`
18		18
19	`import java.io.ByteArrayInputStream;`	19	`import java.io.ByteArrayInputStream;`
20	`import java.io.IOException;`	20	`import java.io.IOException;`
21	`import java.io.InputStream;`	21	`import java.io.InputStream;`
22	`import java.io.StringReader;`	22	`import java.io.StringReader;`
23	`import java.io.StringWriter;`	23	`import java.io.StringWriter;`
24	`import java.io.UnsupportedEncodingException;`	24	`import java.io.UnsupportedEncodingException;`
25	`import java.nio.charset.Charset;`	25	`import java.nio.charset.Charset;`
26	`import java.nio.charset.IllegalCharsetNameException;`	26	`import java.nio.charset.IllegalCharsetNameException;`
27	`import java.util.Iterator;`	27	`import java.util.Iterator;`
28	`import java.util.List;`	28	`import java.util.List;`
29	`import javax.jws.WebService;`	29	`import javax.jws.WebService;`
30	`import javax.xml.namespace.QName;`	30	`import javax.xml.namespace.QName;`
31	`import javax.xml.stream.XMLOutputFactory;`	31	`import javax.xml.stream.XMLOutputFactory;`
32	`import javax.xml.stream.XMLStreamException;`	32	`import javax.xml.stream.XMLStreamException;`
33	`import javax.xml.stream.XMLStreamWriter;`	33	`import javax.xml.stream.XMLStreamWriter;`
-		34
34		35	`import lombok.extern.slf4j.Slf4j;`
35	`import net.brutex.xservices.types.AttributeType;`	36	`import net.brutex.xservices.types.AttributeType;`
36	`import net.brutex.xservices.types.NamespaceListType;`	37	`import net.brutex.xservices.types.NamespaceListType;`
37	`import net.brutex.xservices.types.NamespaceType;`	38	`import net.brutex.xservices.types.NamespaceType;`
38	`import net.brutex.xservices.types.StringSplitType;`	39	`import net.brutex.xservices.types.StringSplitType;`
39	`import net.brutex.xservices.types.ant.FileResource;`	40	`import net.brutex.xservices.types.ant.FileResource;`