com.firefly.utils.dom.DefaultDom Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of firefly-common Show documentation
There is a newer version: 5.0.2
package com.firefly.utils.dom;

import com.firefly.utils.StringUtils;
import com.firefly.utils.exception.CommonRuntimeException;
import org.w3c.dom.CharacterData;
import org.w3c.dom.*;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class DefaultDom implements Dom {

    private DocumentBuilderFactory dbf;
    private DocumentBuilder db;

    public DefaultDom() {
        dbf = DocumentBuilderFactory.newInstance();
        try {
            db = dbf.newDocumentBuilder();
        } catch (ParserConfigurationException e) {
            throw new CommonRuntimeException(e);
        }
    }

    @Override
    public Document getDocument(String file) {
        try (InputStream is = DefaultDom.class.getResourceAsStream("/" + file)) {
            if (is == null) {
                return null;
            }
            Document doc = db.parse(is);
            return doc;
        } catch (Exception e) {
            throw new CommonRuntimeException(e);
        }
    }

    @Override
    public Element getRoot(Document doc) {
        return doc.getDocumentElement();
    }

    @Override
    public List elements(Element e) {
        return elements(e, null);
    }

    @Override
    public List elements(Element e, String name) {
        List eList = new ArrayList();

        NodeList nodeList = e.getChildNodes();
        for (int i = 0; i < nodeList.getLength(); ++i) {
            Node node = nodeList.item(i);
            if (node.getNodeType() == Node.ELEMENT_NODE) {
                if (name != null) {
                    if (node.getNodeName().equals(name))
                        eList.add((Element) node);
                } else {
                    eList.add((Element) node);
                }
            }
        }
        return eList;
    }

    @Override
    public Element element(Element e, String name) {
        NodeList element = e.getElementsByTagName(name);
        if (element != null && e.getNodeType() == Node.ELEMENT_NODE) {
            return (Element) element.item(0);
        }
        return null;
    }

    @Override
    public String getTextValue(Element valueElement) {
        if (valueElement != null) {
            StringBuilder sb = new StringBuilder();
            NodeList nl = valueElement.getChildNodes();
            for (int i = 0; i < nl.getLength(); i++) {
                Node item = nl.item(i);
                if ((item instanceof CharacterData && !(item instanceof Comment)) || item instanceof EntityReference) {
                    sb.append(item.getNodeValue());
                }
            }
            return sb.toString().trim();
        }
        return null;
    }

    @Override
    public String getTextValueByTagName(Element e, String name) {
        Element valueElement = element(e, name);
        if (valueElement == null) {
            return null;
        } else {
            String value = getTextValue(valueElement);
            if (StringUtils.hasText(value)) {
                return value;
            } else {
                return null;
            }
        }
    }

    @Override
    public String getTextValueByTagName(Element e, String name, String defaultValue) {
        String value = getTextValueByTagName(e, name);
        return StringUtils.hasText(value) ? value : defaultValue;
    }
}