I am happy to extend the functionality, can you maybe create an issue on github.
I am mostly concerned with how big that functionality will end up being, but if its not too much, ill happily add it or at least leave the issue up for someone to pick up.