org.nlab.xml.stream.predicate.AttributePredicate Maven / Gradle / Ivy

Go to download
package org.nlab.xml.stream.predicate;

import java.util.Arrays;

import org.nlab.xml.stream.context.StreamContext;

import jodd.lagarto.dom.Node;

import static javax.xml.stream.XMLStreamConstants.END_ELEMENT;
import static javax.xml.stream.XMLStreamConstants.START_ELEMENT;

/**
 * Created by nlabrot on 15/03/15.
 */
public class AttributePredicate implements XmlPredicate {

    private final String[] attributes;
    private final boolean matchStart;
    private final boolean matchEnd;

    public AttributePredicate(boolean matchStart, boolean matchEnd, String... attributes) {
        this.matchStart = matchStart;
        this.matchEnd = matchEnd;
        this.attributes = attributes;
    }

    @Override
    public boolean test(StreamContext staxContext) {
        if ((!matchStart || START_ELEMENT != staxContext.getEvent()) && (!matchEnd || END_ELEMENT != staxContext.getEvent())) {
            return false;
        }

        Node node = staxContext.getNode();

        for (String attribute : attributes) {
            if (!node.hasAttribute(attribute)){
                return false;
            }
        }
        return true;
    }

    @Override
    public String toString() {
        return "AttributePredicate{" +
                "attributes=" + Arrays.toString(attributes) +
                ", matchStart=" + matchStart +
                ", matchEnd=" + matchEnd +
                '}';
    }
}