无需修改flake.nix，如何在zsh中添加nix shell/develop环境指示器

要实现无需修改任何Nix配置文件，就能在提示符里识别当前Nix Shell/Develop的嵌套层级，核心是利用Nix自动设置的IN_NIX_SHELL环境变量——每次进入新的Nix子shell，这个变量的值会递增（第一次进入为1，第二次嵌套为2，以此类推）。

实现步骤 #

1. 添加层级检测函数
   打开你的zsh配置文件（~/.zshrc），插入以下函数：

# 生成Nix Shell层级指示器
nix_shell_prompt() {
  if [[ -n "$IN_NIX_SHELL" ]]; then
    echo -n "#$IN_NIX_SHELL "
  fi
}

2. 集成到oh-my-zsh提示符
   根据你使用的主题，选择合适的方式修改提示符：

• 直接修改主题PROMPT：找到~/.zshrc中定义PROMPT的行，把$(nix_shell_prompt)插入到提示符开头。比如默认的robbyrussell主题可修改为：

  PROMPT='$(nix_shell_prompt)%{$fg_bold[green]%}%n@%m %{$fg_bold[blue]%}%~ %{$reset_color%}$(git_prompt_info)%{$fg_bold[red]%}% %{$reset_color%}%#'
  

• 用precmd钩子动态追加：若不想改动主题原始配置，可通过precmd钩子在每次命令执行前自动调整提示符：

  precmd() {
    # 避免重复添加指示器
    PROMPT=${PROMPT/#\#[0-9] /}
    if [[ -n "$IN_NIX_SHELL" ]]; then
      PROMPT="#$IN_NIX_SHELL $PROMPT"
    fi
  }
  

  这个方式会自动适配层级变化，退出Nix Shell时指示器也会同步消失。

3. 生效配置
   保存~/.zshrc后，执行以下命令让配置立即生效：

source ~/.zshrc

测试效果 #

• 执行nix shell nixpkgs#git，提示符开头会显示#1 ；
• 在当前Nix Shell中再执行nix shell nixpkgs#python，提示符会变成#2 ；
• 退出子shell时，层级自动递减，提示符恢复对应层级标识。

内容的提问来源于stack exchange，提问作者Kai Walter