Have a look at liquid prompt https://github.com/nojhan/liquidprompt