I can work on a PR for this feature but do you have any idea on the best way I should go about implementing this?