OK. I made a custom plugin. For those interested in using it, see here:
https://plugins.gradle.org/plugin/com.dorongold.task-tree