Or just use https://starship.rs/ for any shell