nix/home/prompt.nix

{
  lib,
  pkgs,
  theme,
  ...
}:
{
  home.packages = [ pkgs.pure-prompt ];

  programs.zsh.initContent = lib.mkMerge [
    (lib.mkOrder 800 ''
      fpath+=("${pkgs.pure-prompt}/share/zsh/site-functions")
      autoload -Uz promptinit && promptinit

      export PURE_PROMPT_SYMBOL=$'\xe2\x9d\xaf'
      export PURE_PROMPT_VICMD_SYMBOL=$'\xe2\x9d\xae'
      export PURE_GIT_DIRTY=""
      export PURE_GIT_UP_ARROW="^"
      export PURE_GIT_DOWN_ARROW="v"
      export PURE_GIT_STASH_SYMBOL="="
      export PURE_CMD_MAX_EXEC_TIME=5
      export PURE_GIT_PULL=0
      export PURE_GIT_UNTRACKED_DIRTY=1
      zstyle ':prompt:pure:git:stash' show yes

      ${theme.renderPurePrompt "dark"}

      typeset -g prompt_newline=' '
      prompt pure
      prompt_pure_state_setup() {
        setopt localoptions noshwordsplit

        local ssh_connection=''${SSH_CONNECTION:-$PROMPT_PURE_SSH_CONNECTION}
        local username hostname
        local pure_version=''${prompt_pure_state[version]:-1.27.0}

        if [[ -z $ssh_connection ]] && (( $+commands[who] )); then
          local who_out
          who_out=$(who -m 2>/dev/null)

          if (( $? )); then
            local -a who_in
            who_in=(''${(f)"$(who 2>/dev/null)"})
            who_out="''${(M)who_in:#*[[:space:]]''${TTY#/dev/}[[:space:]]*}"
          fi

          local reIPv6='(([0-9a-fA-F]+:)|:){2,}[0-9a-fA-F]+'
          local reIPv4='([0-9]{1,3}\\.){3}[0-9]+'
          local reHostname='([.][^. ]+){2}'
          local -H MATCH MBEGIN MEND

          if [[ $who_out =~ "\\(?($reIPv4|$reIPv6|$reHostname)\\)?$" ]]; then
            ssh_connection=$MATCH
            export PROMPT_PURE_SSH_CONNECTION=$ssh_connection
          fi

          unset MATCH MBEGIN MEND
        fi

        # Pure's upstream user/host segment stores a deferred color reference,
        # which collapses hex colors to black for this segment over SSH. Resolve
        # the current prompt colors here and rebuild the segment after each
        # prompt_pure_set_colors call so the rendered prompt keeps the theme hex.
        hostname="%F{$prompt_pure_colors[host]}@%m%f"
        [[ -n $ssh_connection ]] && username="%F{$prompt_pure_colors[user]}%n%f""$hostname"
        [[ -z "''${CODESPACES}" ]] && prompt_pure_is_inside_container && username="%F{$prompt_pure_colors[user]}%n%f""$hostname"
        [[ $UID -eq 0 ]] && username="%F{$prompt_pure_colors[user:root]}%n%f""$hostname"

        typeset -gA prompt_pure_state
        prompt_pure_state[version]="$pure_version"
        prompt_pure_state+=(username "$username" prompt "''${PURE_PROMPT_SYMBOL:-❯}")
      }
      prompt_pure_state_setup

      prompt_pure_preprompt_render() {
        setopt localoptions noshwordsplit
        unset prompt_pure_async_render_requested

        prompt_pure_set_colors
        prompt_pure_state_setup
        _codex_pure_default_arrow=$prompt_pure_colors[git:arrow]
        _codex_pure_default_success=$prompt_pure_colors[prompt:success]

        typeset -g prompt_pure_git_branch_color=$prompt_pure_colors[git:branch]
        [[ -n ''${prompt_pure_git_last_dirty_check_timestamp+x} ]] && prompt_pure_git_branch_color=$prompt_pure_colors[git:branch:cached]

        # Branch, arrows, and prompt symbol turn yellow when dirty
        if [[ -n $prompt_pure_git_dirty ]]; then
          prompt_pure_git_branch_color=$prompt_pure_colors[git:dirty]
          prompt_pure_colors[git:arrow]=$prompt_pure_colors[git:dirty]
          prompt_pure_colors[prompt:success]=$prompt_pure_colors[git:dirty]
        else
          prompt_pure_colors[git:arrow]=$_codex_pure_default_arrow
          prompt_pure_colors[prompt:success]=$_codex_pure_default_success
        fi

        psvar[12]=; ((''${(M)#jobstates:#suspended:*} != 0)) && psvar[12]=''${PURE_SUSPENDED_JOBS_SYMBOL:-✦}
        psvar[13]=; [[ -n $prompt_pure_state[username] ]] && psvar[13]=1
        psvar[14]=''${prompt_pure_vcs_info[branch]}
        psvar[15]=
        psvar[16]=''${prompt_pure_vcs_info[action]}
        psvar[17]=''${prompt_pure_git_arrows}
        psvar[18]=; [[ -n $prompt_pure_git_stash ]] && psvar[18]=1
        psvar[19]=''${prompt_pure_cmd_exec_time}

        local expanded_prompt
        expanded_prompt="''${(S%%)PROMPT}"

        if [[ $1 != precmd && $prompt_pure_last_prompt != $expanded_prompt ]]; then
          prompt_pure_reset_prompt
        fi

        typeset -g prompt_pure_last_prompt=$expanded_prompt
      }

      typeset -g _codex_pure_default_arrow=$prompt_pure_colors[git:arrow]
      typeset -g _codex_pure_default_success=$prompt_pure_colors[prompt:success]

      _codex_apply_prompt_theme() {
        local mode="$(_codex_read_theme_mode)"
        [[ "$mode" == "''${_CODEX_LAST_PROMPT_THEME:-}" ]] && return

        if [[ "$mode" == light ]]; then
          ${theme.renderPurePrompt "light"}
        else
          ${theme.renderPurePrompt "dark"}
        fi

        typeset -g _codex_pure_default_arrow=$prompt_pure_colors[git:arrow]
        typeset -g _codex_pure_default_success=$prompt_pure_colors[prompt:success]
        typeset -g _CODEX_LAST_PROMPT_THEME="$mode"
      }
    '')
  ];
}