import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;
import java.util.ArrayList;
import java.util.List;
import jdk.dynalink.CallSiteDescriptor;
import jdk.dynalink.NamedOperation;
import jdk.dynalink.NamespaceOperation;
import jdk.dynalink.Operation;
import jdk.dynalink.StandardNamespace;
import jdk.dynalink.StandardOperation;
import jdk.dynalink.linker.GuardedInvocation;
import jdk.dynalink.linker.GuardingDynamicLinker;
import jdk.dynalink.linker.GuardingDynamicLinkerExporter;
import jdk.dynalink.linker.LinkRequest;
import jdk.dynalink.linker.LinkerServices;
import jdk.dynalink.linker.TypeBasedGuardingDynamicLinker;
import jdk.dynalink.linker.support.Guards;
import jdk.dynalink.linker.support.Lookup;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
public final class DOMLinkerExporter extends GuardingDynamicLinkerExporter {
static {
System.out.println("pluggable dynalink DOM linker loaded");
}
private static List<Element> getChildElements(final Element elem, final String name) {
final NodeList nodeList = elem.getChildNodes();
final List<Element> childElems = new ArrayList<>();
final int len = nodeList.getLength();
for (int i = 0; i < len; i++) {
final Node node = nodeList.item(i);
if (node.getNodeType() == Node.ELEMENT_NODE &&
((Element)node).getTagName().equals(name)) {
childElems.add((Element)node);
}
}
return childElems;
}
public static Object getElementsByName(final Object elem, final String name) {
final List<Element> elems = getChildElements((Element)elem, name);
return elems.size() == 1? elems.get(0) : elems;
}
public static Object getElementText(final Object elem) {
final NodeList nodeList = ((Element)elem).getChildNodes();
final int len = nodeList.getLength();
final StringBuilder text = new StringBuilder();
for (int i = 0; i < len; i++) {
final Node node = nodeList.item(i);
if (node.getNodeType() == Node.TEXT_NODE) {
text.append(node.getNodeValue());
}
}
return text.toString();
}
private static final MethodHandle ELEMENTS_BY_NAME;
private static final MethodHandle ELEMENT_TEXT;
private static final MethodHandle IS_ELEMENT;
static {
ELEMENTS_BY_NAME = Lookup.PUBLIC.findStatic(DOMLinkerExporter.class,
"getElementsByName",
MethodType.methodType(Object.class, Object.class, String.class));
ELEMENT_TEXT = Lookup.PUBLIC.findStatic(DOMLinkerExporter.class,
"getElementText",
MethodType.methodType(Object.class, Object.class));
IS_ELEMENT = Guards.isInstance(Element.class, MethodType.methodType(Boolean.TYPE, Object.class));
}
@Override
public List<GuardingDynamicLinker> get() {
final ArrayList<GuardingDynamicLinker> linkers = new ArrayList<>();
linkers.add(new TypeBasedGuardingDynamicLinker() {
@Override
public boolean canLinkType(final Class<?> type) {
return Element.class.isAssignableFrom(type);
}
@Override
public GuardedInvocation getGuardedInvocation(final LinkRequest request,
final LinkerServices linkerServices) throws Exception {
final Object self = request.getReceiver();
if (! (self instanceof Element)) {
return null;
}
final CallSiteDescriptor desc = request.getCallSiteDescriptor();
final Operation op = desc.getOperation();
final Object name = NamedOperation.getName(op);
final boolean getProp = NamespaceOperation.contains(
NamedOperation.getBaseOperation(op),
StandardOperation.GET, StandardNamespace.PROPERTY);
if (getProp && name instanceof String) {
final String nameStr = (String)name;