linux/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2026-01-30 02:44:42 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Refactor tofu_prompt_info, only show if in an OpenTofu project and use tofu workspace show

Browse source
This commit is contained in:
Marc Cornellà 2024-07-13 19:15:42 +02:00
commit e87e913d1a
No known key found for this signature in database
GPG key ID: 0314585E776A9C1B
1 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
plugins/opentofu/opentofu.plugin.zsh
View file

@ -2,14 +2,20 @@
autoload -Uz bashcompinit && bashcompinit
complete -C tofu tofu
# tofu prompt functions
# tofu workspace prompt function
function tofu_prompt_info() {
# dont show 'default' workspace in home dir
[[ "$PWD" != ~ ]] || return
# to keep compatibility with opentofu, the data dir names .terraform in OpenTofu
[[ -d .terraform && -r .terraform/environment ]] || return
# only show the workspace name if we're in an opentofu project
# i.e. if a .terraform directory exists within the hierarchy
local dir="$PWD"
while [[ ! -d "${dir}/.terraform" ]]; do
[[ "$dir" != / ]] || return 0 # stop at the root directory
dir="${dir:h}" # get the parent directory
done
local workspace="$(< .terraform/environment)"
# workspace might be different than the .terraform/environment file
# for example, if set with the TF_WORKSPACE environment variable
local workspace="$(tofu workspace show)"
# make sure to escape % signs in the workspace name to prevent command injection
echo "${ZSH_THEME_TOFU_PROMPT_PREFIX-[}${workspace:gs/%/%%}${ZSH_THEME_TOFU_PROMPT_SUFFIX-]}"
}