Это почти решает вашу задачу:
А именно этот пост, который я недавно добавил и который, возможно, уже делает именно то, что вам нужно.
Сделайте форк и расширьте его по мере необходимости.