XPathLookups.java

package com.googlecode.totallylazy.xml;

import com.googlecode.totallylazy.functions.Function1;
import com.googlecode.totallylazy.Callers;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.w3c.dom.Text;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class XPathLookups {
    private static final Map<String, Function1<String, String>> lookups = new ConcurrentHashMap<String, Function1<String, String>>();

    @XPathFunction("lookup")
    public static NodeArrayList lookup(String name, NodeList nodes) {
        return new NodeArrayList<Text>(Xml.sequence(nodes).map(lookup(lookups.get(name))));
    }

    public static String lookup(String name, String text){
        return Callers.call(lookups.get(name), text);
    }

    private static Function1<Node, Text> lookup(final Function1<String, String> data) {
        return node -> XPathFunctions.createText(node, data.call(node.getTextContent()));
    }

    public static void setLookup(String name, Function1<String, String> lookup){
        lookups.put(name, lookup);
    }

}