/*
* Copyright 2014 Attila Szegedi, Daniel Dekany, Jonathan Revusky
*
* 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 freemarker.ext.xml;
import java.io.StringWriter;
import java.util.Iterator;
import java.util.List;
import org.dom4j.Attribute;
import org.dom4j.Branch;
import org.dom4j.Document;
import org.dom4j.DocumentType;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.ProcessingInstruction;
import org.dom4j.tree.DefaultAttribute;
import org.jaxen.Context;
import org.jaxen.NamespaceContext;
import org.jaxen.dom4j.Dom4jXPath;
import freemarker.template.TemplateModelException;
/**
* Don't use this class; it's only public to work around Google App Engine Java
* compliance issues. FreeMarker developers only: treat this class as package-visible.
*/
public class _Dom4jNavigator extends Navigator {
public _Dom4jNavigator() {
}
@Override
void getAsString(Object node, StringWriter sw) {
sw.getBuffer().append(((Node) node).asXML());
}
@Override
void getChildren(Object node, String localName, String namespaceUri, List result) {
if (node instanceof Element) {
Element e = (Element) node;
if (localName == null) {
result.addAll(e.elements());
} else {
result.addAll(e.elements(e.getQName().getDocumentFactory().createQName(localName, "", namespaceUri)));
}
} else if (node instanceof Document) {
Element root = ((Document) node).getRootElement();
if (localName == null || (equal(root.getName(), localName) && equal(root.getNamespaceURI(), namespaceUri))) {
result.add(root);
}
}
}
@Override
void getAttributes(Object node, String localName, String namespaceUri, List result) {
if (node instanceof Element) {
Element e = (Element) node;
if (localName == null) {
result.addAll(e.attributes());
} else {
Attribute attr = e.attribute(e.getQName().getDocumentFactory().createQName(localName, "", namespaceUri));
if (attr != null) {
result.add(attr);
}
}
} else if (node instanceof ProcessingInstruction) {
ProcessingInstruction pi = (ProcessingInstruction) node;
if ("target".equals(localName)) {
result.add(new DefaultAttribute("target", pi.getTarget()));
} else if ("data".equals(localName)) {
result.add(new DefaultAttribute("data", pi.getText()));
} else {
result.add(new DefaultAttribute(localName, pi.getValue(localName)));
}
} else if (node instanceof DocumentType) {
DocumentType doctype = (DocumentType) node;
if ("publicId".equals(localName)) {
result.add(new DefaultAttribute("publicId", doctype.getPublicID()));
} else if ("systemId".equals(localName)) {
result.add(new DefaultAttribute("systemId", doctype.getSystemID()));
} else if ("elementName".equals(localName)) {
result.add(new DefaultAttribute("elementName", doctype.getElementName()));
}
}
}
@Override
void getDescendants(Object node, List result) {
if (node instanceof Branch) {
getDescendants((Branch) node, result);
}
}
private void getDescendants(Branch node, List result) {
List content = node.content();
for (Iterator iter = content.iterator(); iter.hasNext(); ) {
Node subnode = (Node) iter.next();
if (subnode instanceof Element) {
result.add(subnode);
getDescendants(subnode, result);
}
}
}
@Override
Object getParent(Object node) {
return ((Node) node).getParent();
}
@Override
Object getDocument(Object node) {
return ((Node) node).getDocument();
}
@Override
Object getDocumentType(Object node) {
return
node instanceof Document
? ((Document) node).getDocType()
: null;
}
@Override
void getContent(Object node, List result) {
if (node instanceof Branch) {
result.addAll(((Branch) node).content());
}
}
@Override
String getText(Object node) {
return ((Node) node).getText();
}
@Override
String getLocalName(Object node) {
return ((Node) node).getName();
}
@Override
String getNamespacePrefix(Object node) {
if (node instanceof Element) {
return ((Element) node).getNamespacePrefix();
}
if (node instanceof Attribute) {
return ((Attribute) node).getNamespacePrefix();
}
return null;
}
@Override
String getNamespaceUri(Object node) {
if (node instanceof Element) {
return ((Element) node).getNamespaceURI();
}
if (node instanceof Attribute) {
return ((Attribute) node).getNamespaceURI();
}
return null;
}
@Override
String getType(Object node) {
switch(((Node) node).getNodeType()) {
case Node.ATTRIBUTE_NODE: {
return "attribute";
}
case Node.CDATA_SECTION_NODE: {
return "cdata";
}
case Node.COMMENT_NODE: {
return "comment";
}
case Node.DOCUMENT_NODE: {
return "document";
}
case Node.DOCUMENT_TYPE_NODE: {
return "documentType";
}
case Node.ELEMENT_NODE: {
return "element";
}
case Node.ENTITY_REFERENCE_NODE: {
return "entityReference";
}
case Node.NAMESPACE_NODE: {
return "namespace";
}
case Node.PROCESSING_INSTRUCTION_NODE: {
return "processingInstruction";
}
case Node.TEXT_NODE: {
return "text";
}
}
return "unknown";
}
@Override
XPathEx createXPathEx(String xpathString) throws TemplateModelException {
try {
return new Dom4jXPathEx(xpathString);
} catch (Exception e) {
throw new TemplateModelException(e);
}
}
private static final class Dom4jXPathEx
extends
Dom4jXPath
implements
XPathEx {
Dom4jXPathEx(String path)
throws Exception {
super(path);
}
public List selectNodes(Object object, NamespaceContext namespaces)
throws TemplateModelException {
Context context = getContext(object);
context.getContextSupport().setNamespaceContext(namespaces);
try {
return selectNodesForContext(context);
} catch (Exception e) {
throw new TemplateModelException(e);
}
}
}
}