What we also need is a simple API that given a dom element, takes a screenshot of the subtree.

Nothing such is available.