All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.databasesandlife.util.DomParser Maven / Gradle / Ivy

There is a newer version: 21.0.1
Show newest version
package com.databasesandlife.util;

import java.io.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.w3c.dom.*;

import com.databasesandlife.util.gwtsafe.ConfigurationException;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

import javax.annotation.Nonnull;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.*;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.xpath.*;

/**
 * @author This source is copyright Adrian Smith and licensed under the LGPL 3.
 * @see Project on GitHub
 */
public class DomParser {

    /** @param elementNames can be "*" */
    public static List getSubElements(Node container, String... elementNames) {
        var allElementsDesired = "*".equals(elementNames[0]);
        var elementNameSet = new HashSet<>(Arrays.asList(elementNames));
        
        var result = new ArrayList();
        var children = container.getChildNodes();
        for (var i = 0; i < children.getLength(); i++) {
            var child = children.item(i);
            if (child.getNodeType() == Node.ELEMENT_NODE) {
                if (allElementsDesired || elementNameSet.contains(child.getNodeName()))
                    result.add((Element) child);
            }
        }
        return result;
    }

    public static void assertNoOtherElements(Node container, String... elements)
    throws ConfigurationException {
        var elementsSet = new HashSet<>(Arrays.asList(elements));
        var children = container.getChildNodes();
        for (var i = 0; i < children.getLength(); i++) {
            var child = children.item(i);
            if (child.getNodeType() == Node.ELEMENT_NODE)
                if ( ! elementsSet.contains(child.getNodeName()))
                    throw new ConfigurationException("Unexpected element <" + child.getNodeName() + ">");
        }
    }

    public static String getMandatoryAttribute(Element node, String attributeName) throws ConfigurationException {
        var attributeNode = node.getAttributeNode(attributeName); 
        if (attributeNode == null)
            throw new ConfigurationException("<" + node.getNodeName() + "> expects mandatory attribute '" + attributeName + "'");
        return attributeNode.getValue();
    }

    public static String getOptionalAttribute(Element node, String attributeName, String defaultValue) {
        var attributeNode = node.getAttributeNode(attributeName); 
        if (attributeNode == null) return defaultValue;
        return attributeNode.getValue();
    }
    
    /** @return null if the attribute is not defined */
    public static String getOptionalAttribute(Element node, String attributeName) {
        return getOptionalAttribute(node, attributeName, null);
    }
    
    public static double parseMandatoryDoubleAttribute(Element node, String attributeName)
    throws ConfigurationException {
        var str = getMandatoryAttribute(node, attributeName);
        try { return Double.parseDouble(str); }
        catch (NumberFormatException e) { throw new ConfigurationException("<" + node.getNodeName() + " " + attributeName +
                "='" + str + "'>: couldn't parse decimal attribute"); }
    }

    /** @return null if attribute not found */
    public static Integer parseOptionalIntegerAttribute(Element node, String attributeName)
    throws ConfigurationException {
        var str = node.getAttribute(attributeName);
        if (str.equals("")) return null;
        try { return Integer.parseInt(str); }
        catch (NumberFormatException e) { throw new ConfigurationException("<" + node.getNodeName() + " " + attributeName +
                "='" + str + "'>: couldn't parse integer attribute"); }
    }

    public static int parseOptionalIntAttribute(Element node, String attributeName, int defaultValue)
    throws ConfigurationException {
        var result = parseOptionalIntegerAttribute(node, attributeName);
        if (result == null) return defaultValue;
        return result;
    }

    public static int parseMandatoryIntAttribute(Element node, String attributeName)
        throws ConfigurationException {
        var result = parseOptionalIntegerAttribute(node, attributeName);
        if (result == null)
            throw new ConfigurationException("<" + node.getNodeName() + "'>: mandatory attribute '" + attributeName + "' missing");
        return result;
    }

    /** 
     * @param subNodeName can be "*" 
     * @throws ConfigurationException if more than one element found
     */
    public static Element getMandatorySingleSubElement(Node node, String subNodeName) throws ConfigurationException {
        var resultList = getSubElements(node, subNodeName);
        if (resultList.size() != 1) throw new ConfigurationException("<" + node.getNodeName() + ">: found " +
            resultList.size() + ("*".equals(subNodeName) ? " sub-elements" : (" <" + subNodeName + "> sub-elements")));
        return resultList.get(0);
    }

    /** 
     * @param subNodeName can be "*"
     * @return null if element not found 
     * @throws ConfigurationException if more than one element found
     */
    public static Element getOptionalSingleSubElement(Element node, String subNodeName) throws ConfigurationException {
        var resultList = getSubElements(node, subNodeName);
        if (resultList.size() == 0) return null;
        else if (resultList.size() == 1) return resultList.get(0);
        else throw new ConfigurationException("<" + node.getNodeName() + ">: found " +
                resultList.size() + ("*".equals(subNodeName) ? " sub-elements" : (" <" + subNodeName + "> sub-elements")));
    }
    
    /** 
     * @param subNodeName can be "*"
     * @return null if element not found 
     * @throws ConfigurationException if more than one element found
     */
    public static String getOptionalSingleSubElementTextContent(Element node, String subNodeName) throws ConfigurationException {
        var el = getOptionalSingleSubElement(node, subNodeName);
        if (el == null) return null;
        else return el.getTextContent();
    }

    public static List parseList(Element container, String elementName, String attribute)
    throws ConfigurationException {
        var result = new ArrayList();
        for (var e : getSubElements(container, elementName)) result.add(getMandatoryAttribute(e, attribute));
        return result;
    }

    public static Set parseSet(Element container, String elementName, String attribute)
    throws ConfigurationException {
        return new HashSet<>(parseList(container, elementName, attribute));
    }

    public static List parseList(Element container, String elementName) {
        var result = new ArrayList();
        for (var e : getSubElements(container, elementName)) result.add(e.getTextContent().trim());
        return result;
    }

    public static Set parseSet(Element container, String elementName) {
        return new HashSet<>(parseList(container, elementName));
    }

    public static Map parseMap(Element container, String elementName, String keyAttribute, String valueAttribute)
    throws ConfigurationException {
        Map result = new HashMap<>();
        for (var e : getSubElements(container, elementName))
            result.put(getMandatoryAttribute(e, keyAttribute), getMandatoryAttribute(e, valueAttribute));
        return result;
    }

    public static Map parseMap(Element container, String elementName, String keyAttribute)
    throws ConfigurationException {
        Map result = new HashMap<>();
        for (var e : getSubElements(container, elementName))
            result.put(getMandatoryAttribute(e, keyAttribute), e.getTextContent());
        return result;
    }
    
    public static Map parseDoubleMap(Element container, String elementName, String keyAttribute, String valueAttribute)
    throws ConfigurationException {
        Map result = new HashMap<>();
        for (var e : getSubElements(container, elementName))
            result.put(getMandatoryAttribute(e, keyAttribute), parseMandatoryDoubleAttribute(e, valueAttribute));
        return result;
    }

    //========== XPath API ===========

    public static XPathExpression getExpression(String expression)
    throws XPathExpressionException {
        var xpath = XPathFactory.newInstance().newXPath();
        return xpath.compile(expression);
    }

    public static NodeList getNodesFromXPath(Document document, String expression)
    throws XPathExpressionException {
        return (NodeList) getExpression(expression).evaluate(document, XPathConstants.NODESET);
    }

    public static NodeList getNodesFromXPath(Element root, String expression)
    throws XPathExpressionException {
        return getNodesFromXPath(root.getOwnerDocument(), expression);
    }

    public static List getElementsFromXPath(Document document, String expression)
    throws XPathExpressionException {
        return toElementList(getNodesFromXPath(document, expression));
    }

    public static List getElementsFromXPath(Element root, String expression)
    throws XPathExpressionException {
        return getElementsFromXPath(root.getOwnerDocument(), expression);
    }

    public static List toElementList(NodeList nl) {
        var elements = new ArrayList(nl.getLength());
        for (var i = 0; i < nl.getLength(); i++) {
            var n = nl.item(i);
            if (n instanceof Element) {
                elements.add((Element) n);
            }
        }
        return elements;
    }

    public static Element getElementFromXPath(Document document, String expression)
    throws XPathExpressionException {
        return (Element) getExpression(expression).evaluate(document, XPathConstants.NODE);
    }

    public static Element getElementFromXPath(Element root, String expression)
    throws XPathExpressionException {
        return (Element) getExpression(expression).evaluate(root, XPathConstants.NODE);
    }

    public static Element from(File f) throws ConfigurationException {
        try {
            return newDocumentBuilder().parse(f).getDocumentElement();
        }
        catch (IOException e) { throw new RuntimeException(e); }
        catch (SAXException e) { throw new ConfigurationException("File '"+f+"'", e); }
    }

    public static Element from(InputStream f) throws ConfigurationException {
        try {
            return newDocumentBuilder().parse(f).getDocumentElement();
        }
        catch (IOException e) { throw new RuntimeException(e); }
        catch (SAXException e) { throw new ConfigurationException(e); }
    }

    public static @Nonnull Element from(@Nonnull String xmlValue) throws ConfigurationException {
        try {
            var docBuilder = newDocumentBuilder();
            var inputSource = new InputSource(new StringReader(xmlValue));
            var doc = docBuilder.parse(inputSource);
            return doc.getDocumentElement();
        }
        catch (IOException e) { throw new RuntimeException(e); }
        catch (SAXException e) { throw new ConfigurationException(e); }
    }

    public static DocumentBuilder newDocumentBuilder() {
        try {
            var dbf = DocumentBuilderFactory.newInstance();
            dbf.setNamespaceAware(true);     // See https://stackoverflow.com/a/49800040
            return dbf.newDocumentBuilder();
        }
        catch (ParserConfigurationException e) { throw new RuntimeException(e); }
    }

    public static @Nonnull String formatXml(@Nonnull Element element) {
        try {
            var str = new StringWriter();

            var transformer = TransformerFactory.newInstance().newTransformer();
            transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION,"yes");
            transformer.transform(new DOMSource(element), new StreamResult(str));

            return str.toString();
        }
        catch (TransformerException e) { throw new RuntimeException(e); }
    }
    
    public static @Nonnull String formatXmlPretty(@Nonnull Element element) {
        try {
            var str = new StringWriter();
            
            var transformer = TransformerFactory.newInstance().newTransformer();
            transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION,"yes");
            transformer.setOutputProperty(OutputKeys.INDENT, "yes");
            transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "2");
            transformer.transform(new DOMSource(element), new StreamResult(str));

            return str.toString();
        }
        catch (TransformerException e) { throw new RuntimeException(e); }
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy