Definitely doable with zsh plugins. There are some options for bash like this but ive never used them