package org.apache.axiom.om.xpath;
import org.apache.axiom.om.OMAttribute;
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.OMNamespace;
import org.apache.axiom.om.OMNode;
import org.apache.axiom.om.OMProcessingInstruction;
import org.apache.axiom.om.OMText;
import org.apache.axiom.om.util.StAXUtils;
import org.apache.axiom.om.impl.builder.StAXOMBuilder;
import org.apache.axiom.om.impl.llom.OMNamespaceImpl;
import org.jaxen.BaseXPath;
import org.jaxen.DefaultNavigator;
import org.jaxen.FunctionCallException;
import org.jaxen.JaxenConstants;
import org.jaxen.UnsupportedAxisException;
import org.jaxen.XPath;
import org.jaxen.saxpath.SAXPathException;
import org.jaxen.util.SingleObjectIterator;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamReader;
import java.io.FileInputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
public class DocumentNavigator extends DefaultNavigator {
private static final long serialVersionUID = 7325116153349780805L;
/**
* Returns a parsed form of the given xpath string, which will be suitable
* for queries on documents that use the same navigator as this one.
*
* @param xpath the XPath expression
* @return Returns a new XPath expression object.
* @throws SAXPathException if the string is not a syntactically
* correct XPath expression
* @see XPath
*/
public XPath parseXPath(String xpath) throws SAXPathException {
return new BaseXPath(xpath, this);
}
/**
* Retrieves the namespace URI of the given element node.
*
* @param object the context element node
* @return Returns the namespace URI of the element node.
*/
public String getElementNamespaceUri(Object object) {
OMElement attr = (OMElement) object;
return attr.getQName().getNamespaceURI();
}
/**
* Retrieves the name of the given element node.
*
* @param object the context element node
* @return Returns the name of the element node.
*/
public String getElementName(Object object) {
OMElement attr = (OMElement) object;
return attr.getQName().getLocalPart();
}
/**
* Retrieves the QName of the given element node.
*
* @param object the context element node
* @return Returns the QName of the element node.
*/
public String getElementQName(Object object) {
OMElement attr = (OMElement) object;
String prefix = null;
if (attr.getNamespace() != null) {
prefix = attr.getNamespace().getPrefix();
}
if (prefix == null || "".equals(prefix)) {
return attr.getQName().getLocalPart();
}
return prefix + ":" + attr.getNamespace().getName();
}
/**
* Retrieves the namespace URI of the given attribute node.
*
* @param object the context attribute node
* @return Returns the namespace URI of the attribute node.
*/
public String getAttributeNamespaceUri(Object object) {
OMAttribute attr = (OMAttribute) object;
return attr.getQName().getNamespaceURI();
}
/**
* Retrieves the name of the given attribute node.
*
* @param object the context attribute node
* @return Returns the name of the attribute node.
*/
public String getAttributeName(Object object) {
OMAttribute attr = (OMAttribute) object;
return attr.getQName().getLocalPart();
}
/**
* Retrieves the QName of the given attribute node.
*
* @param object the context attribute node
* @return Returns the qualified name of the attribute node.
*/
public String getAttributeQName(Object object) {
OMAttribute attr = (OMAttribute) object;
String prefix = attr.getNamespace().getPrefix();
if (prefix == null || "".equals(prefix)) {
return attr.getQName().getLocalPart();
}
return prefix + ":" + attr.getNamespace().getName();
}
/**
* Returns whether the given object is a document node. A document node
* is the node that is selected by the xpath expression <code>/</code>.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is a document node,
* else <code>false</code> .
*/
public boolean isDocument(Object object) {
return object instanceof OMDocument;
}
/**
* Returns whether the given object is an element node.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is an element node,
* else <code>false</code> .
*/
public boolean isElement(Object object) {
return object instanceof OMElement;
}
/**
* Returns whether the given object is an attribute node.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is an attribute node,
* else <code>false</code> .
*/
public boolean isAttribute(Object object) {
return object instanceof OMAttribute;
}
/**
* Returns whether the given object is a namespace node.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is a namespace node,
* else <code>false</code> .
*/
public boolean isNamespace(Object object) {
return object instanceof OMNamespace;
}
/**
* Returns whether the given object is a comment node.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is a comment node,
* else <code>false</code> .
*/
public boolean isComment(Object object) {
return (object instanceof OMComment);
}
/**
* Returns whether the given object is a text node.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is a text node,
* else <code>false</code> .
*/
public boolean isText(Object object) {
return (object instanceof OMText);
}
/**
* Returns whether the given object is a processing-instruction node.
*
* @param object the object to test
* @return Returns <code>true</code> if the object is a processing-instruction node,
* else <code>false</code> .
*/
public boolean isProcessingInstruction(Object object) {
return (object instanceof OMProcessingInstruction);
}
/**
* Retrieves the string-value of a comment node.
* This may be the empty string if the comment is empty,
* but must not be null.
*
* @param object the comment node
* @return Returns the string-value of the node.
*/
public String getCommentStringValue(Object object) {
return ((OMComment) object).getValue();
}
/**
* Retrieves the string-value of an element node.
* This may be the empty string if the element is empty,
* but must not be null.
*
* @param object the comment node.
* @return Returns the string-value of the node.
*/
public String getElementStringValue(Object object) {
if (isElement(object)) {
return getStringValue((OMElement) object, new StringBuffer())
.toString();
}
return null;
}
private StringBuffer getStringValue(OMNode node, StringBuffer buffer) {
if (isText(node)) {
buffer.append(((OMText) node).getText());
} else if (node instanceof OMElement) {
Iterator children = ((OMElement) node).getChildren();
while (children.hasNext()) {
getStringValue((OMNode) children.next(), buffer);
}
}
return buffer;
}
/**
* Retrieves the string-value of an attribute node.
* This should be the XML 1.0 normalized attribute value.
* This may be the empty string but must not be null.
*
* @param object the attribute node
* @return Returns the string-value of the node.
*/
public String getAttributeStringValue(Object object) {
return ((OMAttribute) object).getAttributeValue();
}
/**
* Retrieves the string-value of a namespace node.
* This is generally the namespace URI.
* This may be the empty string but must not be null.
*
* @param object the namespace node
* @return Returns the string-value of the node.
*/
public String getNamespaceStringValue(Object object) {
return ((OMNamespace) object).getName();
}
/**
* Retrieve the string-value of a text node.
* This must not be null and should not be the empty string.
* The XPath data model does not allow empty text nodes.
*
* @param object the text node
* @return Returns the string-value of the node.
*/
public String getTextStringValue(Object object) {
return ((OMText) object).getText();
}
/**
* Retrieves the namespace prefix of a namespace node.
*
* @param object the namespace node
* @return Returns the prefix associated with the node.
*/
public String getNamespacePrefix(Object object) {
return ((OMNamespace) object).getPrefix();
}
/**
* Retrieves an <code>Iterator</code> matching the <code>child</code>
* XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the child axis are
* not supported by this object model
*/
public Iterator getChildAxisIterator(Object contextNode) throws UnsupportedAxisException {
if (contextNode instanceof OMContainer) {
return ((OMContainer) contextNode).getChildren();
}
return JaxenConstants.EMPTY_ITERATOR;
}
public Iterator getDescendantAxisIterator(Object object) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getDescendantAxisIterator(object);
}
/**
* Retrieves an <code>Iterator</code> matching the <code>attribute</code>
* XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the attribute axis are
* not supported by this object model
*/
public Iterator getAttributeAxisIterator(Object contextNode) throws UnsupportedAxisException {
if (isElement(contextNode)) {
ArrayList attributes = new ArrayList();
Iterator i = ((OMElement) contextNode).getAllAttributes();
while (i != null && i.hasNext()) {
attributes.add(new OMAttributeEx((OMAttribute) i.next(),
(OMContainer) contextNode, ((OMElement) contextNode)
.getOMFactory()));
}
return attributes.iterator();
}
return JaxenConstants.EMPTY_ITERATOR;
}
/**
* Retrieves an <code>Iterator</code> matching the <code>namespace</code>
* XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the namespace axis are
* not supported by this object model
*/
public Iterator getNamespaceAxisIterator(Object contextNode) throws UnsupportedAxisException {
if (!(contextNode instanceof OMContainer &&
contextNode instanceof OMElement)) {
return JaxenConstants.EMPTY_ITERATOR;
}
List nsList = new ArrayList();
HashSet prefixes = new HashSet();
for (OMContainer context = (OMContainer) contextNode;
context != null && !(context instanceof OMDocument);
context = ((OMElement) context).getParent()) {
OMElement element = (OMElement) context;
ArrayList declaredNS = new ArrayList();
Iterator i = element.getAllDeclaredNamespaces();
while (i != null && i.hasNext()) {
declaredNS.add(i.next());
}
declaredNS.add(element.getNamespace());
for (Iterator iter = element.getAllAttributes();
iter != null && iter.hasNext();) {
OMAttribute attr = (OMAttribute) iter.next();
declaredNS.add(attr.getNamespace());
}
for (Iterator iter = declaredNS.iterator();
iter != null && iter.hasNext();) {
OMNamespace namespace = (OMNamespace) iter.next();
if (namespace != null) {
String prefix = namespace.getPrefix();
if (prefix != null && !prefixes.contains(prefix)) {
prefixes.add(prefix);
nsList.add(new OMNamespaceEx(namespace, context));
}
}
}
}
nsList.add(
new OMNamespaceEx(
new OMNamespaceImpl(
"http://www.w3.org/XML/1998/namespace",
"xml"),
(OMContainer) contextNode));
return nsList.iterator();
}
/**
* Retrieves an <code>Iterator</code> matching the <code>self</code> xpath
* axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the self axis are
* not supported by this object model
*/
public Iterator getSelfAxisIterator(Object contextNode) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getSelfAxisIterator(contextNode);
}
/**
* Retrieves an <code>Iterator</code> matching the
* <code>descendant-or-self</code> XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the descendant-or-self axis are
* not supported by this object model
*/
public Iterator getDescendantOrSelfAxisIterator(Object contextNode) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getDescendantOrSelfAxisIterator(contextNode);
}
/**
* Retrieves an <code>Iterator</code> matching the
* <code>ancestor-or-self</code> XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the ancestor-or-self axis are
* not supported by this object model
*/
public Iterator getAncestorOrSelfAxisIterator(Object contextNode) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getAncestorOrSelfAxisIterator(contextNode);
}
/**
* Retrieves an <code>Iterator</code> matching the <code>parent</code> XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the parent axis are
* not supported by this object model
*/
public Iterator getParentAxisIterator(Object contextNode) throws UnsupportedAxisException {
if (contextNode instanceof OMNode) {
return new SingleObjectIterator(((OMNode) contextNode).getParent());
} else if (contextNode instanceof OMNamespaceEx) {
return new SingleObjectIterator(
((OMNamespaceEx) contextNode).getParent());
} else if (contextNode instanceof OMAttributeEx) {
return new SingleObjectIterator(
((OMAttributeEx) contextNode).getParent());
}
return JaxenConstants.EMPTY_ITERATOR;
}
/**
* Retrieves an <code>Iterator</code> matching the <code>ancestor</code>
* XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the ancestor axis are
* not supported by this object model
*/
public Iterator getAncestorAxisIterator(Object contextNode) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getAncestorAxisIterator(contextNode);
}
/**
* Retrieves an <code>Iterator</code> matching the
* <code>following-sibling</code> XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the following-sibling axis are
* not supported by this object model
*/
public Iterator getFollowingSiblingAxisIterator(Object contextNode) throws UnsupportedAxisException {
ArrayList list = new ArrayList();
if (contextNode != null && contextNode instanceof OMNode) {
while (contextNode != null && contextNode instanceof OMNode) {
contextNode = ((OMNode) contextNode).getNextOMSibling();
if (contextNode != null)
list.add(contextNode);
}
}
return list.iterator();
}
/**
* Retrieves an <code>Iterator</code> matching the
* <code>preceding-sibling</code> XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the preceding-sibling axis are
* not supported by this object model
*/
public Iterator getPrecedingSiblingAxisIterator(Object contextNode) throws UnsupportedAxisException {
ArrayList list = new ArrayList();
if (contextNode != null && contextNode instanceof OMNode) {
while (contextNode != null && contextNode instanceof OMNode) {
contextNode = ((OMNode) contextNode).getPreviousOMSibling();
if (contextNode != null)
list.add(contextNode);
}
}
return list.iterator();
}
/**
* Retrieves an <code>Iterator</code> matching the <code>following</code>
* XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the following axis are
* not supported by this object model
*/
public Iterator getFollowingAxisIterator(Object contextNode) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getFollowingAxisIterator(contextNode);
}
/**
* Retrieves an <code>Iterator</code> matching the <code>preceding</code> XPath axis.
*
* @param contextNode the original context node
* @return Returns an Iterator capable of traversing the axis, not null.
* @throws UnsupportedAxisException if the semantics of the preceding axis are
* not supported by this object model
*/
public Iterator getPrecedingAxisIterator(Object contextNode) throws UnsupportedAxisException {
//TODO: Fix this better?
return super.getPrecedingAxisIterator(contextNode);
}
/**
* Loads a document from the given URI.
*
* @param uri the URI of the document to load
* @return Returns the document.
* @throws FunctionCallException if the document could not be loaded
*/
public Object getDocument(String uri)
throws FunctionCallException {
try {
XMLStreamReader parser;
XMLInputFactory xmlInputFactory = StAXUtils.getXMLInputFactory();
Boolean oldValue = (Boolean) xmlInputFactory.getProperty(XMLInputFactory.IS_COALESCING);
try {
xmlInputFactory.setProperty(XMLInputFactory.IS_COALESCING, Boolean.TRUE);
if (uri.indexOf(':') == -1) {
parser = xmlInputFactory.createXMLStreamReader(
new FileInputStream(uri));
} else {
URL url = new URL(uri);
parser = xmlInputFactory.createXMLStreamReader(
url.openStream());
}
} finally {
if (oldValue != null) {
xmlInputFactory.setProperty(XMLInputFactory.IS_COALESCING, oldValue);
}
StAXUtils.releaseXMLInputFactory(xmlInputFactory);
}
StAXOMBuilder builder =
new StAXOMBuilder(parser);
return builder.getDocumentElement().getParent();
} catch (Exception e) {
throw new FunctionCallException(e);
}
}
/**
* Returns the element whose ID is given by elementId.
* If no such element exists, returns null.
* Attributes with the name "ID" are not of type ID unless so defined.
* Implementations that do not know whether attributes are of type ID or
* not are expected to return null.
*
* @param contextNode a node from the document in which to look for the
* id
* @param elementId id to look for
* @return Returns element whose ID is given by elementId, or null if no such
* element exists in the document or if the implementation
* does not know about attribute types.
*/
public Object getElementById(Object contextNode, String elementId) {
//TODO: Fix this better?
return super.getElementById(contextNode, elementId);
}
/**
* Returns the document node that contains the given context node.
*
* @param contextNode the context node
* @return Returns the document of the context node.
* @see #isDocument(Object)
*/
public Object getDocumentNode(Object contextNode) {
if (contextNode instanceof OMDocument) {
return contextNode;
}
return getDocumentNode(((OMNode) contextNode).getParent());
}
/**
* Translates a namespace prefix to a namespace URI, <b>possibly</b>
* considering a particular element node.
* <p/>
* Strictly speaking, prefix-to-URI translation should occur
* irrespective of any element in the document. This method
* is provided to allow a non-conforming ease-of-use enhancement.
* </p>
*
* @param prefix the prefix to translate
* @param element the element to consider during translation
* @return Returns the namespace URI associated with the prefix.
*/
public String translateNamespacePrefixToUri(String prefix, Object element) {
//TODO: Fix this better?
return super.translateNamespacePrefixToUri(prefix, element);
}
/**
* Retrieves the target of a processing-instruction.
*
* @param object the context processing-instruction node
* @return Returns the target of the processing-instruction node.
*/
public String getProcessingInstructionTarget(Object object) {
return ((OMProcessingInstruction)object).getTarget();
}
/**
* Retrieves the data of a processing-instruction.
*
* @param object the context processing-instruction node
* @return Returns the data of the processing-instruction node.
*/
public String getProcessingInstructionData(Object object) {
return ((OMProcessingInstruction)object).getValue();
}
/**
* Returns a number that identifies the type of node that the given
* object represents in this navigator. See org.jaxen.pattern.Pattern
*
* @param node ????
* @return Returns short.
*/
public short getNodeType(Object node) {
//TODO: Fix this better?
return super.getNodeType(node);
}
/**
* Returns the parent of the given context node.
* <p/>
* The parent of any node must either be a document
* node or an element node.
*
* @param contextNode the context node
* @return Returns the parent of the context node, or null if this is a document node.
* @throws UnsupportedAxisException if the parent axis is not
* supported by the model
* @see #isDocument
* @see #isElement
*/
public Object getParentNode(Object contextNode) throws UnsupportedAxisException {
if (contextNode == null ||
contextNode instanceof OMDocument) {
return null;
} else if (contextNode instanceof OMAttributeEx) {
return ((OMAttributeEx) contextNode).getParent();
} else if (contextNode instanceof OMNamespaceEx) {
return ((OMNamespaceEx) contextNode).getParent();
}
return ((OMNode) contextNode).getParent();
}
class OMNamespaceEx implements OMNamespace {
OMNamespace originalNsp = null;
OMContainer parent = null;
OMNamespaceEx(OMNamespace nsp, OMContainer parent) {
originalNsp = nsp;
this.parent = parent;
}
public boolean equals(String uri, String prefix) {
return originalNsp.equals(uri, prefix);
}
public String getPrefix() {
return originalNsp.getPrefix();
}
public String getName() {
return originalNsp.getName();
}
public OMContainer getParent() {
return parent;
}
}
class OMAttributeEx implements OMAttribute {
OMAttribute attribute = null;
OMContainer parent = null;
OMFactory factory;
OMAttributeEx(OMAttribute attribute, OMContainer parent,
OMFactory factory) {
this.attribute = attribute;
this.parent = parent;
}
public String getLocalName() {
return attribute.getLocalName();
}
public void setLocalName(String localName) {
attribute.setLocalName(localName);
}
public String getAttributeValue() {
return attribute.getAttributeValue();
}
public void setAttributeValue(String value) {
attribute.setAttributeValue(value);
}
public void setOMNamespace(OMNamespace omNamespace) {
attribute.setOMNamespace(omNamespace);
}
public OMNamespace getNamespace() {
return attribute.getNamespace();
}
public QName getQName() {
return attribute.getQName();
}
public OMContainer getParent() {
return parent;
}
public OMFactory getOMFactory() {
return this.factory;
}
}
}