ZSH 提示符中动态条件显示 Git 分支与虚拟环境信息的完整教程

本文详解如何在 zsh 提示符中按需动态显示 git 分支和 python 虚拟环境名称(如 `[git:main env:venv]`),避免冗余渲染,同时保持中性提示符结构不变。

在 ZSH 中实现「仅当存在 Git 仓库时显示分支」+「仅当激活虚拟环境时显示环境名」+「二者合并为统一方括号组」(如 ⟜[git:main env:venv]⊸),关键在于将条件逻辑封装为可内联执行的 shell 函数,而非依赖静态变量(如 ZSH_THEME_GIT_PROMPT_PREFIX)。ZSH 的 %{...%} 和 $() 机制支持在 PROMPT 中安全执行命令并插入结果,这是实现动态提示符的核心能力。

✅ 正确做法:编写组合式提示符函数

首先,定义一个复合函数 prompt_vcs_env_info,它检测当前目录是否为 Git 仓库、是否激活虚拟环境,并返回格式化字符串:

prompt_vcs_env_info() {
  local git_part=""
  local venv_part=""

  # 检测 Git 分支(仅当在 Git 仓库中)
  if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
    local branch=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)
    git_part="git:${branch:-unknown}"
  fi

  # 检测虚拟环境(检查 VIRTUAL_ENV 或 CONDA_DEFAULT_ENV)
  if [[ -n "$VIRTUAL_ENV" ]]; then
    local venv_name="${VIRTUAL_ENV##*/}"
    venv_part="env:${venv_name}"
  elif [[ -n "$CONDA_DEFAULT_ENV" && "$CONDA_DEFAULT_ENV" != "base" ]]; then
    venv_part="env:$CONDA_DEFAULT_ENV"
  fi

  # 合并显示:仅当至少一项存在时才输出 ⟜[...]⊸
  if [[ -n "$git_part" || -n "$venv_part" ]]; then
    local combined=""
    [[ -n "$git_part" ]] && combined="$git_part"
    [[ -n "$venv_part" ]] && combined="${combined:+$combined }$venv_part"
    echo "%{$fg_bold[green]%}⟜[%{$reset_color%}%{$fg[white]%}${combined}%{$reset_color%}%{$fg_bold[green]%}]⊸"
  fi
}
⚠️ 注意事项:使用 $(...) 执行该函数时,ZSH 会实时调用并插入结果,因此无需预定义 git_prompt_info 等外部插件变量;所有颜色转义必须包裹在 %{...%} 中,防止光标位置计算错误;2>/dev/null 避免 Git/conda 命令报错污染提示符;${combined:+$combined } 实现空格分隔的优雅合并(若仅一项存在则无多余空格)。

? 替换原始 PROMPT 定义

将你原有的 PROMPT 行中 $(git_prompt_info) $(virtualenv_prompt_info) 部分,整体替换为 $(prompt_vcs_env_info)

PROMPT='%{$fg_bold[green]%}╭─[%{$reset_color%}%{$fg_bold[white]%}Swidenn %{$reset_color%}%{$fg_bold[green]%}⚡%{$reset_color%}%{$fg_bold[white]%}%M%{$reset_color%}%{$fg_bold[green]%}]$(prompt_vcs_env_info)%{$reset_color%}%{$fg_bold[green]%}▷ %{$reset_color%}%{$fg[white]%}%~
%{$reset_color%}%{$fg_bold[green]%}╰─[%{$reset_color%}%{$fg[white]%}%!%{$reset_color%}%{$fg_bold[green]%}]-> '

✅ 效果验证:

  • 普通目录 → 无 ⟜[...]⊸,保持中性提示;
  • Git 仓库 → 显示 ⟜[git:main]⊸;
  • Git + venv → 显示 ⟜[git:main env:myvenv]⊸;
  • 仅 venv(非 Git 目录)→ 显示 ⟜[env:myvenv]⊸。

? 进阶建议:考虑 Starship(推荐长期方案)

虽然手动函数完全可控,但维护复杂提示符易出错。Starship 是现代、跨 Shell、模块化、零配置开箱即用的提示符工具,原生支持 Git 分支、Python 虚拟环境、多模块合并(自动去重/排序)、异步渲染且性能极佳。只需:

curl -sS https://www./link/7f876c8429a1389bbfbf9c70b0e6e826/install.sh | sh
# 然后在 ~/.zshrc 中添加:
eval "$(starship init zsh)"

其默认行为即满足你的需求([git:main env:venv] 格式),且自动适配主题、图标、状态码等——对追求稳定与可维护性的用户是更优解。

综上,ZSH 动态提示符的本质是「用函数驱动内容生成」,而非拼接静态片段。掌握这一模式,即可自由扩展 SSH 主机、Node.js 版本、Kubernetes 上下文等任意上下文敏感信息。