prompt.zsh

# prompt
setopt prompt_subst

# mode-aware arrow

function p_arrow {
  if [[ $KEYMAP = "vicmd" ]]; then
    echo "%F{magenta}»%f"
  else
    echo "%F{cyan}»%f"
  fi
}

# colored path

function p_colored_path {
  local slash="%F{cyan}/%f"
  echo "${${PWD/#$HOME/~}//\//$slash}"
}

# git info

function p_vcs {
  vcs_info
  echo $vcs_info_msg_0_
}

# environments:
#  - ssh
#  - virtualenv
#  - cabal sandbox

export VIRTUAL_ENV_DISABLE_PROMPT=1

function p_hostname {
  [[ -n $SSH_CLIENT || -n $SHOW_HOSTNAME_IN_PROMPT ]] && echo " %F{cyan}@%f %U%m%u"
}

function p_envs {
  local envs

  if [[ -n $VIRTUAL_ENV ]] && envs+="PY"

  [[ -n $envs ]] && echo " %F{green}[%f$envs%F{green}]%f"
}

function p_exit_code {
  echo "%(?..%B%F{red}!%f%b )"
}

PS1='
%{$(iterm2_prompt_mark)%}%F{blue}λ%f $(p_colored_path)$(p_hostname)$(p_envs)$(p_vcs)
$(p_exit_code)$(p_arrow) '

function p_continuation {
  echo "%1(_.%_.contd)"
}

PS2='$(p_continuation) $(p_arrow) '