Subversion Repositories XServices

Rev

Rev 177 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
114 brianR 1
/*
2
 *   Copyright 2013 Brian Rosenberger (Brutex Network)
3
 *
4
 *   Licensed under the Apache License, Version 2.0 (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
7
 *
8
 *       http://www.apache.org/licenses/LICENSE-2.0
9
 *
10
 *   Unless required by applicable law or agreed to in writing, software
11
 *   distributed under the License is distributed on an "AS IS" BASIS,
12
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
 *   See the License for the specific language governing permissions and
14
 *   limitations under the License.
15
 */
16
 
17
package net.brutex.xservices.ws.impl;
18
 
19
import java.io.ByteArrayInputStream;
20
import java.io.IOException;
21
import java.io.InputStream;
22
import java.io.StringReader;
23
import java.io.StringWriter;
127 brianR 24
import java.io.UnsupportedEncodingException;
25
import java.nio.charset.Charset;
26
import java.nio.charset.IllegalCharsetNameException;
114 brianR 27
import java.util.Iterator;
28
import java.util.List;
29
import javax.jws.WebService;
127 brianR 30
import javax.xml.namespace.QName;
114 brianR 31
import javax.xml.stream.XMLOutputFactory;
32
import javax.xml.stream.XMLStreamException;
33
import javax.xml.stream.XMLStreamWriter;
127 brianR 34
 
35
import net.brutex.xservices.types.AttributeType;
114 brianR 36
import net.brutex.xservices.types.NamespaceListType;
37
import net.brutex.xservices.types.NamespaceType;
127 brianR 38
import net.brutex.xservices.types.StringSplitType;
114 brianR 39
import net.brutex.xservices.types.ant.FileResource;
40
import net.brutex.xservices.ws.XServicesFault;
41
import net.brutex.xservices.ws.XmlService;
158 brianR 42
 
114 brianR 43
import org.apache.axiom.om.OMAbstractFactory;
127 brianR 44
import org.apache.axiom.om.OMAttribute;
114 brianR 45
import org.apache.axiom.om.OMCloneOptions;
127 brianR 46
import org.apache.axiom.om.OMComment;
114 brianR 47
import org.apache.axiom.om.OMContainer;
48
import org.apache.axiom.om.OMDocument;
49
import org.apache.axiom.om.OMElement;
127 brianR 50
import org.apache.axiom.om.OMFactory;
114 brianR 51
import org.apache.axiom.om.OMNode;
127 brianR 52
import org.apache.axiom.om.OMProcessingInstruction;
114 brianR 53
import org.apache.axiom.om.OMText;
54
import org.apache.axiom.om.OMXMLBuilderFactory;
55
import org.apache.axiom.om.xpath.AXIOMXPath;
185 brianR 56
 
57
import org.apache.logging.log4j.LogManager;
58
import org.apache.logging.log4j.Logger;
177 brianR 59
import org.apache.shiro.authz.annotation.RequiresPermissions;
114 brianR 60
import org.jaxen.JaxenException;
61
import org.jaxen.SimpleNamespaceContext;
62
 
63
/**
64
 * @author Brian Rosenberger, bru(at)brutex.de
65
 *
66
 */
67
@WebService(targetNamespace = "http://ws.xservices.brutex.net", endpointInterface = "net.brutex.xservices.ws.XmlService", serviceName = "XmlService")
68
public class XmlServiceImpl implements XmlService {
185 brianR 69
	private static final Logger logger = LogManager.getLogger();
114 brianR 70
 
185 brianR 71
	public String insertNodesFromFile(FileResource res, NamespaceListType nsList, String xpath, String xmlFragment)
72
			throws XServicesFault {
114 brianR 73
		try {
74
			AXIOMXPath axp = new AXIOMXPath(xpath);
75
			InputStream is = res.getAntResource(null).getInputStream();
185 brianR 76
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
114 brianR 77
			OMDocument fragdoc = null;
78
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
185 brianR 79
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
114 brianR 80
						.getDocument();
81
			} else {
82
				throw new XServicesFault("No xmldata to insert.");
83
			}
84
 
85
			// Initialize XPath context
127 brianR 86
			SimpleNamespaceContext context = createContext(nsList);
114 brianR 87
			axp.setNamespaceContext(context);
88
			axp.addNamespaces(fragdoc.getOMDocumentElement());
89
 
90
			OMDocument document = insertNodes(sourcedoc, axp, fragdoc);
91
 
92
			StringWriter sw = new StringWriter();
93
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
94
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
95
			document.serialize(writer);
96
 
97
			this.logger.trace(sw.getBuffer().toString());
98
			return sw.getBuffer().toString();
99
		} catch (JaxenException e) {
100
			e.printStackTrace();
101
			throw new XServicesFault(e);
102
		} catch (IOException e) {
103
			e.printStackTrace();
104
			throw new XServicesFault(e);
105
		} catch (XMLStreamException e) {
106
			e.printStackTrace();
107
			throw new XServicesFault(e);
108
		}
109
	}
110
 
185 brianR 111
	public String replaceNodesFromFile(FileResource res, NamespaceListType nsList, String xpath, String xmlFragment)
112
			throws XServicesFault {
114 brianR 113
		try {
185 brianR 114
			AXIOMXPath axp = new AXIOMXPath(xpath);
115
			InputStream is = res.getAntResource(null).getInputStream();
116
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
117
			OMDocument fragdoc = null;
118
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
119
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
120
						.getDocument();
121
			} else {
122
				throw new XServicesFault("No xmldata to insert.");
123
			}
114 brianR 124
 
185 brianR 125
			// Initialize XPath context
126
			SimpleNamespaceContext context = createContext(nsList);
127
			axp.setNamespaceContext(context);
128
			axp.addNamespaces(fragdoc.getOMDocumentElement());
114 brianR 129
 
185 brianR 130
			OMDocument document = replaceNodes(sourcedoc, axp, fragdoc);
114 brianR 131
 
185 brianR 132
			StringWriter sw = new StringWriter();
133
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
134
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
135
			document.serialize(writer);
114 brianR 136
 
185 brianR 137
			this.logger.trace(sw.getBuffer().toString());
138
			return sw.getBuffer().toString();
139
		} catch (JaxenException e) {
140
			e.printStackTrace();
141
			throw new XServicesFault(e);
142
		} catch (XMLStreamException e) {
143
			e.printStackTrace();
144
			throw new XServicesFault(e);
145
		} catch (IOException e) {
146
			e.printStackTrace();
147
			throw new XServicesFault(e);
148
		}
114 brianR 149
	}
150
 
185 brianR 151
	public String replaceNodes(String source, String encoding, NamespaceListType nsList, String xpath,
152
			String xmlFragment) throws XServicesFault {
127 brianR 153
		encoding = validateEncoding(encoding);
114 brianR 154
		try {
155
			AXIOMXPath axp = new AXIOMXPath(xpath);
127 brianR 156
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
185 brianR 157
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
114 brianR 158
			OMDocument fragdoc = null;
159
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
185 brianR 160
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
114 brianR 161
						.getDocument();
162
			} else {
163
				throw new XServicesFault("No xmldata to insert.");
164
			}
165
 
166
			// Initialize XPath context
127 brianR 167
			SimpleNamespaceContext context = createContext(nsList);
114 brianR 168
			axp.setNamespaceContext(context);
169
			axp.addNamespaces(fragdoc.getOMDocumentElement());
170
 
171
			OMDocument document = replaceNodes(sourcedoc, axp, fragdoc);
172
 
173
			StringWriter sw = new StringWriter();
174
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
175
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
176
			document.serialize(writer);
177
 
178
			this.logger.trace(sw.getBuffer().toString());
179
			return sw.getBuffer().toString();
180
		} catch (JaxenException e) {
181
			e.printStackTrace();
182
			throw new XServicesFault(e);
183
		} catch (XMLStreamException e) {
184
			e.printStackTrace();
185
			throw new XServicesFault(e);
127 brianR 186
		} catch (UnsupportedEncodingException e) {
187
			throw new XServicesFault(e);
114 brianR 188
		}
189
	}
185 brianR 190
 
177 brianR 191
	@RequiresPermissions("insertNodes")
185 brianR 192
	public String insertNodes(String source, String encoding, NamespaceListType nsList, String xpath,
193
			String xmlFragment) throws XServicesFault {
127 brianR 194
		encoding = validateEncoding(encoding);
114 brianR 195
		try {
196
			AXIOMXPath axp = new AXIOMXPath(xpath);
127 brianR 197
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
185 brianR 198
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
114 brianR 199
			OMDocument fragdoc = null;
200
			if ((xmlFragment != null) && (new String(xmlFragment).length() > 0)) {
185 brianR 201
				fragdoc = OMXMLBuilderFactory.createOMBuilder(new StringReader("<XS>" + xmlFragment + "</XS>"))
114 brianR 202
						.getDocument();
203
			} else {
204
				throw new XServicesFault("No xmldata to insert.");
205
			}
206
 
207
			// Initialize XPath context
127 brianR 208
			SimpleNamespaceContext context = createContext(nsList);
114 brianR 209
			axp.setNamespaceContext(context);
210
			axp.addNamespaces(fragdoc.getOMDocumentElement());
211
 
212
			OMDocument document = insertNodes(sourcedoc, axp, fragdoc);
213
 
214
			StringWriter sw = new StringWriter();
215
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
216
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
217
			document.serialize(writer);
218
 
219
			this.logger.trace(sw.getBuffer().toString());
220
			return sw.getBuffer().toString();
221
		} catch (JaxenException e) {
222
			e.printStackTrace();
223
			throw new XServicesFault(e);
224
		} catch (XMLStreamException e) {
225
			e.printStackTrace();
226
			throw new XServicesFault(e);
127 brianR 227
		} catch (UnsupportedEncodingException e) {
228
			throw new XServicesFault(e);
114 brianR 229
		}
230
	}
231
 
232
	public String wrapInCDATA(String data) throws XServicesFault {
185 brianR 233
		String result = "";
114 brianR 234
		String[] tokens = data.split("\\]\\]>", -1);
185 brianR 235
 
236
		for (int i = 0; i < tokens.length; i++) {
237
			result += tokens[i];
238
			if (i + 1 < tokens.length)
239
				result += "]]]]><![CDATA[>";
114 brianR 240
		}
185 brianR 241
 
114 brianR 242
		result = "<![CDATA[" + result + "]]>";
243
		return result;
244
	}
127 brianR 245
 
185 brianR 246
	public StringSplitType selectXPath(String source, String encoding, NamespaceListType nsList, String xpath)
247
			throws XServicesFault {
127 brianR 248
		encoding = validateEncoding(encoding);
249
		try {
250
			StringSplitType rarray = new StringSplitType();
251
			AXIOMXPath axp = new AXIOMXPath(xpath);
252
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
253
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
185 brianR 254
 
127 brianR 255
			// Initialize XPath context
256
			SimpleNamespaceContext context = createContext(nsList);
185 brianR 257
 
127 brianR 258
			axp.setNamespaceContext(context);
259
			List results = axp.selectNodes(sourcedoc);
185 brianR 260
			for (Object o : results) {
127 brianR 261
				String text = null;
185 brianR 262
 
263
				if (o instanceof OMNode) {
264
					switch (((OMNode) o).getType()) {
265
					case OMNode.TEXT_NODE:
266
						text = ((OMText) o).getText();
267
						break;
268
					case OMNode.COMMENT_NODE:
269
						text = ((OMComment) o).getValue();
270
						break;
271
					case OMNode.PI_NODE:
272
						text = ((OMProcessingInstruction) o).getValue();
273
						break;
274
					default:
275
						StringWriter sw = new StringWriter();
276
						XMLOutputFactory xof = XMLOutputFactory.newInstance();
277
						XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
278
						((OMNode) o).serialize(writer);
279
						writer.flush();
280
						text = sw.toString();
281
					}
282
				} else if (o instanceof OMAttribute) {
283
					text = ((OMAttribute) o).getAttributeValue();
127 brianR 284
				} else {
285
					text = String.valueOf(o);
286
				}
287
				rarray.addStringMatch(text);
288
			}
289
 
290
			return rarray;
185 brianR 291
		} catch (JaxenException e) {
127 brianR 292
			e.printStackTrace();
293
			throw new XServicesFault(e);
294
		} catch (XMLStreamException e) {
185 brianR 295
			// TODO Auto-generated catch block
127 brianR 296
			throw new XServicesFault(e.getMessage());
185 brianR 297
		} catch (UnsupportedEncodingException e) {
298
			throw new XServicesFault(e);
127 brianR 299
		}
300
	}
185 brianR 301
 
302
	public String setAttribute(String source, String encoding, NamespaceListType nsList, String xpath,
303
			AttributeType attr) throws XServicesFault {
127 brianR 304
		encoding = validateEncoding(encoding);
305
		try {
306
			StringSplitType rarray = new StringSplitType();
307
			AXIOMXPath axp = new AXIOMXPath(xpath);
308
			InputStream is = new ByteArrayInputStream(source.getBytes(encoding));
309
			OMDocument sourcedoc = OMXMLBuilderFactory.createOMBuilder(is).getDocument();
310
			OMFactory fac = OMAbstractFactory.getOMFactory();
185 brianR 311
 
127 brianR 312
			// Initialize XPath context
313
			SimpleNamespaceContext context = createContext(nsList);
185 brianR 314
 
127 brianR 315
			axp.setNamespaceContext(context);
316
			List results = axp.selectNodes(sourcedoc);
185 brianR 317
			for (Object o : results) {
127 brianR 318
				String text = null;
185 brianR 319
 
320
				if (o instanceof OMNode) {
321
					switch (((OMNode) o).getType()) {
322
					case OMNode.ELEMENT_NODE:
323
						OMElement node = ((OMElement) o);
324
						if (attr.value == null) {
325
							node.removeAttribute(node.getAttribute(new QName(attr.name)));
326
						} else {
327
							node.addAttribute(attr.name, attr.value, node.getNamespace());
328
						}
329
						break;
330
					default:
331
						throw new XServicesFault("XPath expression did not match an element node.");
127 brianR 332
					}
333
				} else {
334
					throw new XServicesFault("XPath expression did not match a node.");
335
				}
336
			}
185 brianR 337
 
127 brianR 338
			StringWriter sw = new StringWriter();
339
			XMLOutputFactory xof = XMLOutputFactory.newInstance();
340
			XMLStreamWriter writer = xof.createXMLStreamWriter(sw);
341
			sourcedoc.serialize(writer);
342
			writer.flush();
185 brianR 343
			return sw.toString();
344
		} catch (JaxenException e) {
127 brianR 345
			e.printStackTrace();
346
			throw new XServicesFault(e);
347
		} catch (XMLStreamException e) {
185 brianR 348
			// TODO Auto-generated catch block
127 brianR 349
			throw new XServicesFault(e.getMessage());
185 brianR 350
		} catch (UnsupportedEncodingException e) {
351
			throw new XServicesFault(e);
127 brianR 352
		}
353
	}
185 brianR 354
 
355
	private OMDocument insertNodes(OMDocument xmldocument, AXIOMXPath axp, OMDocument xmlfragment)
356
			throws XServicesFault {
114 brianR 357
		List<?> olist = null;
358
		try {
359
			olist = axp.selectNodes(xmldocument.getOMDocumentElement());
185 brianR 360
			this.logger.debug("XPath '" + axp.toString() + "' has " + olist.size() + " matches.");
361
			this.logger.trace("XPath root expression is: '" + axp.debug() + "'.");
114 brianR 362
		} catch (JaxenException e) {
363
			throw new XServicesFault(e.getMessage(), e);
364
		}
365
		if (olist.size() == 0)
185 brianR 366
			throw new XServicesFault(Messages.getString("XmlService.no_match", new Object[] { axp.toString() }));
114 brianR 367
 
368
		// Prepare children to insert
369
		xmlfragment.build();
370
 
371
		// Determine what has been matched
372
		OMContainer match = null;
373
		for (Object o : olist) {
185 brianR 374
			Iterator<?> children = xmlfragment.getOMDocumentElement().getChildren();
114 brianR 375
			if ((o instanceof OMNode)) {
376
				OMNode node = (OMNode) o;
377
				switch (node.getType()) {
378
				case OMNode.ELEMENT_NODE:
379
					if ((o instanceof OMElement))
380
						match = (OMElement) o;
381
					if ((o instanceof OMDocument))
382
						match = ((OMDocument) o).getOMDocumentElement();
383
					this.logger.debug(Messages.getString("XmlService.8"));
384
					break;
385
				case OMNode.TEXT_NODE:
386
					match = ((OMText) o).getParent();
387
					this.logger.debug(Messages.getString("XmlService.9"));
388
					break;
389
				case OMNode.COMMENT_NODE:
390
					// match = node.getParent();
391
					match = (OMContainer) node;
392
					this.logger.debug(Messages.getString("XmlService.10"));
393
					break;
394
				default:
185 brianR 395
					this.logger
396
							.error("XPath matched " + o.getClass().getCanonicalName() + " Node Type:" + node.getType());
114 brianR 397
					this.logger.error(Messages.getString("XmlService.11"));
185 brianR 398
					throw new XServicesFault(Messages.getString("XmlService.12"));
114 brianR 399
				}
400
			} else {
185 brianR 401
				this.logger.error("XPath matched " + o.getClass().getCanonicalName());
114 brianR 402
				this.logger.error(Messages.getString("XmlService.11"));
403
				throw new XServicesFault(Messages.getString("XmlService.12"));
404
			}
405
 
406
			while (children.hasNext()) {
407
				OMNode container = (OMNode) children.next();
408
				match.addChild((OMNode) container.clone(new OMCloneOptions()));
409
			}
410
		}
411
 
412
		xmldocument.build();
413
		return xmldocument;
414
	}
415
 
185 brianR 416
	private OMDocument replaceNodes(OMDocument xmldocument, AXIOMXPath axp, OMDocument xmlfragment)
417
			throws XServicesFault {
114 brianR 418
 
419
		List<?> olist = null;
420
		try {
421
			olist = axp.selectNodes(xmldocument.getOMDocumentElement());
185 brianR 422
			this.logger.debug("XPath '" + axp.toString() + "' has " + olist.size() + " matches.");
423
			this.logger.trace("XPath root expression is: '" + axp.debug() + "'.");
114 brianR 424
		} catch (JaxenException e) {
425
			throw new XServicesFault(e.getMessage(), e);
426
		}
427
		if (olist.size() == 0)
185 brianR 428
			throw new XServicesFault(Messages.getString("XmlService.no_match", new Object[] { axp.toString() }));
114 brianR 429
 
430
		// Prepare children to insert
431
		xmlfragment.build();
432
 
433
		// Determine what has been matched
434
		OMNode match = null;
435
		for (Object o : olist) {
185 brianR 436
			Iterator<?> children = xmlfragment.getOMDocumentElement().getChildren();
114 brianR 437
			if ((o instanceof OMNode)) {
438
				OMNode node = (OMNode) o;
439
				switch (node.getType()) {
440
				case OMNode.ELEMENT_NODE:
441
					if ((o instanceof OMElement))
442
						match = (OMElement) o;
443
					if ((o instanceof OMDocument))
444
						match = ((OMDocument) o).getOMDocumentElement();
445
					this.logger.debug(Messages.getString("XmlService.8"));
446
					break;
447
				default:
185 brianR 448
					this.logger
449
							.error("XPath matched " + o.getClass().getCanonicalName() + " Node Type:" + node.getType());
114 brianR 450
					this.logger.error(Messages.getString("XmlService.11"));
185 brianR 451
					throw new XServicesFault(Messages.getString("XmlService.12"));
114 brianR 452
				}
453
			} else {
185 brianR 454
				this.logger.error("XPath matched " + o.getClass().getCanonicalName());
114 brianR 455
				this.logger.error(Messages.getString("XmlService.11"));
456
				throw new XServicesFault(Messages.getString("XmlService.12"));
457
			}
458
 
459
			while (children.hasNext()) {
460
				OMNode container = (OMNode) children.next();
158 brianR 461
				match.insertSiblingBefore((OMNode) container.clone(new OMCloneOptions()));
114 brianR 462
			}
463
			match.detach();
464
		}
465
		xmldocument.build();
466
		return xmldocument;
467
	}
127 brianR 468
 
469
	private SimpleNamespaceContext createContext(NamespaceListType nsList) {
470
		// Initialize XPath context
471
		SimpleNamespaceContext context = new SimpleNamespaceContext();
185 brianR 472
		if (nsList != null) {
127 brianR 473
			for (NamespaceType ns : nsList.getNamespaces()) {
474
				context.addNamespace(ns.getPrefix(), ns.getUri().toString());
185 brianR 475
				this.logger.debug(
476
						Messages.getString("XmlService.0") + ns.getPrefix() + "=\"" + ns.getUri().toString() + "\"'");
127 brianR 477
			}
478
		} else {
479
			logger.debug("No namespaces defined.");
480
		}
481
		return context;
482
	}
483
 
484
	private String validateEncoding(String encoding) throws XServicesFault {
185 brianR 485
		if (encoding == null || encoding.equals("")) {
486
			encoding = Charset.defaultCharset().displayName();
487
		}
488
		try {
127 brianR 489
			Charset.isSupported(encoding);
490
		} catch (IllegalCharsetNameException e) {
185 brianR 491
			throw new XServicesFault("Endcoding '" + encoding + "' is not supported by this JRE.");
127 brianR 492
		}
185 brianR 493
		logger.debug("Setting source xml string encoding to '" + encoding + "'");
127 brianR 494
		return encoding;
495
	}
158 brianR 496
 
114 brianR 497
}