org.alfresco.util.XMLUtil.java Source code

Java tutorial

Introduction

Here is the source code for org.alfresco.util.XMLUtil.java

Source

/*
 * #%L
 * Alfresco Repository
 * %%
 * Copyright (C) 2005 - 2016 Alfresco Software Limited
 * %%
 * This file is part of the Alfresco software. 
 * If the software was purchased under a paid Alfresco license, the terms of 
 * the paid license agreement will prevail.  Otherwise, the software is 
 * provided under the following open source license terms:
 * 
 * Alfresco is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * Alfresco is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

package org.alfresco.util;

import java.io.*;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import org.alfresco.model.ContentModel;
import org.alfresco.service.cmr.repository.ContentReader;
import org.alfresco.service.cmr.repository.ContentService;
import org.alfresco.service.cmr.repository.NodeRef;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.w3c.dom.*;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

/**
 * XML utility functions.
 * 
 * @author Ariel Backenroth
 */
public class XMLUtil {
    private static final Log LOGGER = LogFactory.getLog(XMLUtil.class);

    /** utility function for creating a document */
    public static Document newDocument() {
        return XMLUtil.getDocumentBuilder().newDocument();
    }

    /** utility function for serializing a node */
    public static void print(final Node n, final Writer output) {
        XMLUtil.print(n, output, true);
    }

    /** utility function for serializing a node */
    public static void print(final Node n, final Writer output, final boolean indent) {
        try {
            final TransformerFactory tf = TransformerFactory.newInstance();
            final Transformer t = tf.newTransformer();
            t.setOutputProperty(OutputKeys.INDENT, indent ? "yes" : "no");
            t.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
            t.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "no");
            t.setOutputProperty(OutputKeys.METHOD, "xml");
            if (LOGGER.isDebugEnabled()) {
                LOGGER.debug("writing out a document for "
                        + (n instanceof Document ? ((Document) n).getDocumentElement() : n).getNodeName() + " to "
                        + (output instanceof StringWriter ? "string" : output));
            }
            t.transform(new DOMSource(n), new StreamResult(output));
        } catch (TransformerException te) {
            te.printStackTrace();
            assert false : te.getMessage();
        }
    }

    /** utility function for serializing a node */
    public static void print(final Node n, final File output) throws IOException {
        XMLUtil.print(n, new FileWriter(output));
    }

    /** utility function for serializing a node */
    public static String toString(final Node n) {
        return XMLUtil.toString(n, true);
    }

    /** utility function for serializing a node */
    public static String toString(final Node n, final boolean indent) {
        final StringWriter result = new StringWriter();
        XMLUtil.print(n, result, indent);
        return result.toString();
    }

    /** utility function for parsing xml */
    public static Document parse(final String source) throws SAXException, IOException {
        return XMLUtil.parse(new ByteArrayInputStream(source.getBytes("UTF-8")));
    }

    /** utility function for parsing xml */
    public static Document parse(final NodeRef nodeRef, final ContentService contentService)
            throws SAXException, IOException {
        final ContentReader contentReader = contentService.getReader(nodeRef, ContentModel.TYPE_CONTENT);
        final InputStream in = contentReader.getContentInputStream();
        return XMLUtil.parse(in);
    }

    /** utility function for parsing xml */
    public static Document parse(final File source) throws SAXException, IOException {
        return XMLUtil.parse(new FileInputStream(source));
    }

    /** utility function for parsing xml */
    public static Document parse(final InputStream source) throws SAXException, IOException {
        try {
            final DocumentBuilder db = XMLUtil.getDocumentBuilder();
            return db.parse(source);
        } finally {
            source.close();
        }
    }

    /** utility function for parsing xml */
    public static Document parse(final Reader source) throws SAXException, IOException {
        try {
            final DocumentBuilder db = XMLUtil.getDocumentBuilder();
            return db.parse(new InputSource(source));
        } finally {
            source.close();
        }
    }

    /** provides a document builder that is namespace aware but not validating by default */
    public static DocumentBuilder getDocumentBuilder() {
        return XMLUtil.getDocumentBuilder(true, false);
    }

    /**
     * FOR DIAGNOSTIC PURPOSES ONLY - incomplete<br/>
     * Builds a path to the node relative to the to node provided.
     * @param from the node from which to build the xpath
     * @param to an ancestor of <tt>from</tt> which will be the root of the path
     * @return an xpath to <tt>to</tt> rooted at <tt>from</tt>.
     */
    public static String buildXPath(final Node from, final Element to) {
        String result = "";
        Node tmp = from;
        do {
            if (tmp instanceof Attr) {
                assert result.length() == 0;
                result = "@" + tmp.getNodeName();
            } else if (tmp instanceof Element) {
                Node tmp2 = tmp;
                int position = 1;
                while (tmp2.getPreviousSibling() != null) {
                    if (tmp2.getNodeName().equals(tmp.getNodeName())) {
                        position++;
                    }
                    tmp2 = tmp2.getPreviousSibling();
                }
                String part = tmp.getNodeName() + "[" + position + "]";
                result = "/" + part + result;
            } else if (tmp instanceof Text) {
                assert result.length() == 0;
                result = "/text()";
            } else {
                if (LOGGER.isDebugEnabled()) {
                    throw new IllegalArgumentException("unsupported node type " + tmp);
                }
            }
            tmp = tmp.getParentNode();
        } while (tmp != to.getParentNode() && tmp != null);
        return result;
    }

    public static DocumentBuilder getDocumentBuilder(final boolean namespaceAware, final boolean validating) {
        try {
            final DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
            dbf.setNamespaceAware(namespaceAware);
            dbf.setValidating(validating);
            return dbf.newDocumentBuilder();
        } catch (ParserConfigurationException pce) {
            LOGGER.error(pce);
            return null;
        }
    }

    /**
     * Provides a NodeList of multiple nodelists
     */
    public static NodeList combine(final NodeList... nls) {

        return new NodeList() {
            public Node item(final int index) {
                int offset = 0;
                for (int i = 0; i < nls.length; i++) {
                    if (index - offset < nls[i].getLength()) {
                        return nls[i].item(index - offset);
                    } else {
                        offset += nls[i].getLength();
                    }
                }
                return null;
            }

            public int getLength() {
                int result = 0;
                for (int i = 0; i < nls.length; i++) {
                    result += nls[i].getLength();
                }
                return result;
            }
        };
    }
}