[rbdr/dotfiles] / zsh / functions / status.zsh

function tool_versions() {
  if [[ -s ".tool-versions" ]]; then
        echo '.tool-versions'
        return
  fi
  echo "${HOME}/.tool-versions"
}

function env_info_provider() { echo "%{%F{green}%}n/a"; }
function env_info {
  echo "\u300c%{%F{black}%} $(env_info_provider)%{%F{black}%} \u300d"
}

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 rust_info {
  rustinfo=`cat "$(tool_versions)" | rg rust | cut -d ' ' -f 2`
  rustinfo=${ASDF_RUST_VERSION:-$rustinfo}
  echo "%{%F{red}%}rs%{%F{black}%} ${rustinfo:-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 git_info {
  branch=$(git branch --show-current 2> /dev/null)

  if [ ! -z $branch ]; then
    echo -n "@%F{cyan}$branch%f"

    status_output=$(git status --short)

    if echo "$status_output" | rg -q '^??'; then
      echo -n "%{%F{magenta}%}?%f"
    fi

    if echo "$status_output" | rg -q '^ M'; then
      echo -n "%{%F{magenta}%}!%f"
    fi

    if echo "$status_output" | rg -q '^ D'; then
      echo -n "%{%F{magenta}%}×%f"
    fi

    if echo "$status_output" | rg -q '^A '; then
      echo -n "%{%F{green}%}+%f"
    fi
  fi
}

function status {
  print -P '%{%F{black}%}$(env_info) $(node_info) $(rust_info) $(python_info)'
}