Simplify prompt functions

1 files changed, 7 insertions, 11 deletions

diff --git a/zsh/modules/prompt.zsh b/zsh/modules/prompt.zsh
index 2b7454d..a305178 100644
--- a/zsh/modules/prompt.zsh
+++ b/zsh/modules/prompt.zsh
@@ -22,23 +22,19 @@ function env_info {
 }
 
 function python_info {
-  pythoninfo=`asdf current python | sed 's/  */ /g' | cut -d ' ' -f 2`
-  if [[ -z "$VIRTUAL_ENV" ]]; then
-    venv="default"
-  else
-    venv="${VIRTUAL_ENV##*/}"
-  fi
-  echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo}@${venv}"
+  pythoninfo=`cat ~/.tool-versions | rg python | cut -d ' ' -f 2`
+  venv="${VIRTUAL_ENV##*/}"
+  echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo:-no}@${venv:-default}"
 }
 
 function ruby_info {
-  rubyinfo=`asdf current ruby | sed 's/  */ /g' | cut -d ' ' -f 2`
-  echo "%{%F{red}%}rb%{%F{black}%} ${rubyinfo}"
+  rubyinfo=`cat ~/.tool-versions | rg ruby | cut -d ' ' -f 2`
+  echo "%{%F{red}%}rb%{%F{black}%} ${rubyinfo:-no}"
 }
 
 function node_info {
-  nodeinfo=`asdf current nodejs | sed 's/  */ /g' | cut -d ' ' -f 2`
-  echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo}"
+  nodeinfo=`cat ~/.tool-versions | rg nodejs | cut -d ' ' -f 2`
+  echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo:-no}"
 }
 
 function box_name {