Apaches DTM to Plaintext for css-querying

Question

I am writing a XPath function expression to find elements by CSS query.

This is what I have:

import javax.xml.transform.TransformerException;    
import org.apache.xml.dtm.ref.dom2dtm.DOM2DTM;
import org.apache.xpath.XPathContext;
import org.apache.xpath.functions.FunctionOneArg;
import org.apache.xpath.objects.XObject;
import org.apache.xpath.objects.XString;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

public class CSSFinder extends FunctionOneArg {
    @Override
    public XObject execute(XPathContext c) throws TransformerException {
        final DOM2DTM dtm = (DOM2DTM) c.getDTM(c.getContextNode());
        final Document parse = Jsoup.parse(dtm.toString()); // toString returns not html.
        final Elements parents = parse.select(getArg0().toString()).first().parents();
        String xpath = "/";
        for (final Element parent : parents) {
            xpath += "/*[" + parent.elementSiblingIndex() + "]";
        }
        // replace this later...
        return new XString("sc-login");
    }
}

The problem is that toString does not return html. How to get the complete parsed html?

Answer 1

Found the solution:

import java.io.IOException;
import java.util.Collections;

import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.apache.commons.io.output.ByteArrayOutputStream;
import org.apache.xml.dtm.ref.dom2dtm.DOM2DTM;
import org.apache.xpath.XPathContext;
import org.apache.xpath.functions.FunctionOneArg;
import org.apache.xpath.objects.XNodeSetForDOM;
import org.apache.xpath.objects.XObject;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import org.w3c.dom.Node;

import com.gargoylesoftware.htmlunit.html.xpath.XPathUtils;

public class CSSFinder extends FunctionOneArg {
    @Override
    public XObject execute(XPathContext c) throws TransformerException {
        final DOM2DTM currentNode = (DOM2DTM) c.getDTM(c.getContextNode());
        final Node root = currentNode.getNode(currentNode.getDocumentRoot(c.getContextNode()));
        final Transformer transformer = TransformerFactory.newInstance().newTransformer();
        final ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        transformer.transform(new DOMSource(root), new StreamResult(outputStream));
        try {
            outputStream.close();
        } catch (final IOException e) {
            e.printStackTrace();
        }
        final Document parse = Jsoup.parse(outputStream.toString());
        final Element current = parse.select(getArg0().toString()).first();
        final Elements parents = current.parents();
        Collections.reverse(parents);
        String xpath = "";
        for (final Element parent : parents) {
            final int index = parent.elementSiblingIndex() + 1;
            xpath += "/*[" + index + "]";
        }
        xpath += "/*[" + (current.elementSiblingIndex()+1) + "]";
        return new XNodeSetForDOM((Node) XPathUtils.getByXPath(root, xpath).get(0), c.getDTMManager());
    }
}

I must register the function

XPathHelper.registerGlobalFunction("http://webtest.canoo.com", "css", CSSFinder.class);

And here we go in canoo webtest:

<clickElement xpath="wt:css('button.sc-login')" />