Currently working exactly on that https://gitlab.com/makapuf/treewalker (even if it could always use some love)