Add sway to installation

[rbdr/dotfiles] / zsh / modules / prompt.zsh

diff --git a/zsh/modules/prompt.zsh b/zsh/modules/prompt.zsh
index b21a41f8c963c620dcea4a3c4f20760071947401..beeb85d720a3c8e7b298857161d72ce86e4d03fa 100644 (file)
--- a/zsh/modules/prompt.zsh
+++ b/zsh/modules/prompt.zsh
@@ -1,5 +1,13 @@
 autoload -Uz promptinit && promptinit
 
+function tool_versions() {
+  if [[ -s ".tool-versions" ]]; then
+       echo '.tool-versions'
+       return
+  fi
+  echo "${HOME}/.tool-versions"
+}
+
 if [[ "$TERM" == (dumb|linux|*bsd*) ]] || (( $#prompt_argv < 1 )); then
   prompt 'off'
 fi
@@ -16,31 +24,28 @@ function prompt_precmd {
 # Functions
 #
 
-function virtualenv_info {
-  if [[ -z "$VIRTUAL_ENV" ]]; then
-    venv="default"
-  else
-    venv="${VIRTUAL_ENV##*/}"
-  fi
-  echo "🐍 ${venv}"
+function env_info_provider() { echo "%{%F{green}%}n/a"; }
+function env_info {
+  echo "[%{%F{cyan}%}ENV%{%F{black}%} $(env_info_provider)%{%F{black}%}]"
 }
 
-function rvm_info {
-  if [[ -z "$RUBY_VERSION" ]]; then
-    rvminfo="default"
-  else
-    rvminfo=`rvm-prompt`
-  fi
-  echo "💎 ${rvminfo}"
+function python_info {
+  pythoninfo=`cat "$(tool_versions)" | rg python | cut -d ' ' -f 2`
+  pythoninfo=${ASDF_PYTHON_VERSION:-$pythoninfo}
+  venv="${VIRTUAL_ENV##*/}"
+  echo "%{%F{green}%}py%{%F{black}%} ${pythoninfo:-no}@${venv:-default}"
 }
 
-function nvm_info {
-  if [[ -z "$NVM_BIN" ]]; then
-    nvminfo="default"
-  else
-    nvminfo=`nvm current`
-  fi
-  echo "💛 ${nvminfo}"
+function ruby_info {
+  rubyinfo=`cat "$(tool_versions)" | rg ruby | cut -d ' ' -f 2`
+  rubyinfo=${ASDF_RUBY_VERSION:-$rubyinfo}
+  echo "%{%F{red}%}rb%{%F{black}%} ${rubyinfo:-no}"
+}
+
+function node_info {
+  nodeinfo=`cat "$(tool_versions)" | rg nodejs | cut -d ' ' -f 2`
+  nodeinfo=${ASDF_NODEJS_VERSION:-$nodeinfo}
+  echo "%{%F{yellow}%}js%{%F{black}%} ${nodeinfo:-no}"
 }
 
 function box_name {
@@ -56,13 +61,9 @@ function zle-line-init zle-keymap-select {
     # Couldn't figure out how to get tmux and iTerm to agree on how to render
     # these characters... so there you go.
 
-    local normal_symbol="🅽"
-    local insert_symbol="🅸"
-    local error_symbol="🆇"
-
-    [ ! -z "$TMUX" ] && normal_symbol="%{$normal_symbol%2G%}"
-    [ ! -z "$TMUX" ] && insert_symbol="%{$insert_symbol%2G%}"
-    [ ! -z "$TMUX" ] && error_symbol="%{$error_symbol%2G%}"
+    local normal_symbol="N"
+    local insert_symbol="I"
+    local error_symbol="X"
 
     # Show vi status / return status on the right side
 
@@ -106,7 +107,7 @@ function prompt_setup {
 
   # Define prompts.
   PROMPT='
-%{%F{black}%}$(nvm_info) $(rvm_info) $(virtualenv_info)
+%{%F{black}%}$(env_info) $(node_info) $(ruby_info) $(python_info)
 %{%F{magenta}%}%n%{%f%}@%{%F{yellow}%}$(box_name)%{%f%} %{%B%F{green}%}${PWD/#$HOME/~}%{%f%b%}${git_info[prompt]} %{%f%}
 %(?,,%{${%B%F{white}%}[%?]%{%f%b%} )$ '
 }