I probably listen to ~40 audio books a year and I'd love something that allowed me to review the text of what I had just listened to and re-read and highlight s

I probably listen to ~40 audio books a year and I'd love something that allowed me to review the text of what I had just listened to and re-read and highlight specific parts. Also something that displays diagrams/maps sensibly would be good too! Edit: And by "love" I mean "pay" - I'd happily pay the same amount as my Audible subscription for a service that helped with this.