Terraform workspace prompt in shell (bash, zsh, etc.)

.terraform-workspace-prompt.zsh

function tf_prompt_info() {
    # dont show 'default' workspace in home dir
    [[ "$PWD" == ~ ]] && return
    # check if in terraform dir
    if [ -d .terraform ]; then
      # Most of scripts uses 'terraform workspace show' but it's too slow in some cases.
      # We can just bring out the environment file to bash
      if [ -f .terraform/environment ]; then
        workspace=$(cat .terraform/environment 2> /dev/null) || return
        echo "[${workspace}]"
      else
        echo "[default]"
      fi
    fi
}