com.threelevers.css.Selector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cssselectors Show documentation

Provides the ability to select elements in an XML document using CSS selectors

The newest version!

package com.threelevers.css;

import static com.google.common.collect.Iterables.concat;
import static com.threelevers.css.CssSelectors.selectors;
import static com.threelevers.css.Nodes.isElement;

import java.util.Collections;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;

import com.google.common.collect.ImmutableSet;

public final class Selector {
    private final Element element;
    
    private Selector(Element element) {
        this.element = element;
    }
    
    public static Selector from(Element element) {
        return new Selector(element);
    }
    
    public static Selector from(Document doc) {
        return new Selector(doc.getDocumentElement());
    }

    public Iterable select(String selector) {
        return select(element, selectors(selector));
    }
    
    static Iterable select(Element element, CssSelector matcher) {
        Iterable matches = ImmutableSet.of();
        if (matcher.matches(element)) {
            matches = concat(matches, Collections.singletonList(element));
        }
        NodeList children = element.getChildNodes();
        for (int i = 0; i < children.getLength(); i++) {
            if (isElement(children.item(i))) {
                matches = concat(matches, select((Element) children.item(i), matcher));
            }
        }
        return matches;
    }
}