7f220dc8e4
* test: cover sidebar PR explicit branch fallback * fix: restore sidebar PR badges for workspace branches * test: preserve sidebar PR badge on first prompt * fix: keep sidebar PR badges through first prompt
658 lines
22 KiB
Bash
658 lines
22 KiB
Bash
# cmux shell integration for zsh
|
|
# Injected automatically — do not source manually
|
|
|
|
_cmux_send() {
|
|
local payload="$1"
|
|
if command -v ncat >/dev/null 2>&1; then
|
|
print -r -- "$payload" | ncat -w 1 -U "$CMUX_SOCKET_PATH" --send-only
|
|
elif command -v socat >/dev/null 2>&1; then
|
|
print -r -- "$payload" | socat -T 1 - "UNIX-CONNECT:$CMUX_SOCKET_PATH"
|
|
elif command -v nc >/dev/null 2>&1; then
|
|
# Some nc builds don't support unix sockets, but keep as a last-ditch fallback.
|
|
#
|
|
# Important: macOS/BSD nc will often wait for the peer to close the socket
|
|
# after it has finished writing. cmux keeps the connection open, so
|
|
# a plain `nc -U` can hang indefinitely and leak background processes.
|
|
#
|
|
# Prefer flags that guarantee we exit after sending, and fall back to a
|
|
# short timeout so we never block sidebar updates.
|
|
if print -r -- "$payload" | nc -N -U "$CMUX_SOCKET_PATH" >/dev/null 2>&1; then
|
|
:
|
|
else
|
|
print -r -- "$payload" | nc -w 1 -U "$CMUX_SOCKET_PATH" >/dev/null 2>&1 || true
|
|
fi
|
|
fi
|
|
}
|
|
|
|
_cmux_restore_scrollback_once() {
|
|
local path="${CMUX_RESTORE_SCROLLBACK_FILE:-}"
|
|
[[ -n "$path" ]] || return 0
|
|
unset CMUX_RESTORE_SCROLLBACK_FILE
|
|
|
|
if [[ -r "$path" ]]; then
|
|
/bin/cat -- "$path" 2>/dev/null || true
|
|
/bin/rm -f -- "$path" >/dev/null 2>&1 || true
|
|
fi
|
|
}
|
|
_cmux_restore_scrollback_once
|
|
|
|
# Throttle heavy work to avoid prompt latency.
|
|
typeset -g _CMUX_PWD_LAST_PWD=""
|
|
typeset -g _CMUX_GIT_LAST_PWD=""
|
|
typeset -g _CMUX_GIT_LAST_RUN=0
|
|
typeset -g _CMUX_GIT_JOB_PID=""
|
|
typeset -g _CMUX_GIT_JOB_STARTED_AT=0
|
|
typeset -g _CMUX_GIT_FORCE=0
|
|
typeset -g _CMUX_GIT_HEAD_LAST_PWD=""
|
|
typeset -g _CMUX_GIT_HEAD_PATH=""
|
|
typeset -g _CMUX_GIT_HEAD_SIGNATURE=""
|
|
typeset -g _CMUX_GIT_HEAD_WATCH_PID=""
|
|
typeset -g _CMUX_PR_POLL_PID=""
|
|
typeset -g _CMUX_PR_POLL_PWD=""
|
|
typeset -g _CMUX_PR_POLL_INTERVAL=45
|
|
typeset -g _CMUX_PR_FORCE=0
|
|
typeset -g _CMUX_ASYNC_JOB_TIMEOUT=20
|
|
|
|
typeset -g _CMUX_PORTS_LAST_RUN=0
|
|
typeset -g _CMUX_CMD_START=0
|
|
typeset -g _CMUX_SHELL_ACTIVITY_LAST=""
|
|
typeset -g _CMUX_TTY_NAME=""
|
|
typeset -g _CMUX_TTY_REPORTED=0
|
|
|
|
_cmux_git_resolve_head_path() {
|
|
# Resolve the HEAD file path without invoking git (fast; works for worktrees).
|
|
local dir="$PWD"
|
|
while true; do
|
|
if [[ -d "$dir/.git" ]]; then
|
|
print -r -- "$dir/.git/HEAD"
|
|
return 0
|
|
fi
|
|
if [[ -f "$dir/.git" ]]; then
|
|
local line gitdir
|
|
line="$(<"$dir/.git")"
|
|
if [[ "$line" == gitdir:* ]]; then
|
|
gitdir="${line#gitdir:}"
|
|
gitdir="${gitdir## }"
|
|
gitdir="${gitdir%% }"
|
|
[[ -n "$gitdir" ]] || return 1
|
|
[[ "$gitdir" != /* ]] && gitdir="$dir/$gitdir"
|
|
print -r -- "$gitdir/HEAD"
|
|
return 0
|
|
fi
|
|
fi
|
|
[[ "$dir" == "/" || -z "$dir" ]] && break
|
|
dir="${dir:h}"
|
|
done
|
|
return 1
|
|
}
|
|
|
|
_cmux_git_head_signature() {
|
|
local head_path="$1"
|
|
[[ -n "$head_path" && -r "$head_path" ]] || return 1
|
|
local line=""
|
|
if IFS= read -r line < "$head_path"; then
|
|
print -r -- "$line"
|
|
return 0
|
|
fi
|
|
return 1
|
|
}
|
|
|
|
_cmux_report_tty_once() {
|
|
# Send the TTY name to the app once per session so the batched port scanner
|
|
# knows which TTY belongs to this panel.
|
|
(( _CMUX_TTY_REPORTED )) && return 0
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
[[ -n "$_CMUX_TTY_NAME" ]] || return 0
|
|
_CMUX_TTY_REPORTED=1
|
|
{
|
|
_cmux_send "report_tty $_CMUX_TTY_NAME --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
} >/dev/null 2>&1 &!
|
|
}
|
|
|
|
_cmux_report_shell_activity_state() {
|
|
local state="$1"
|
|
[[ -n "$state" ]] || return 0
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
[[ "$_CMUX_SHELL_ACTIVITY_LAST" == "$state" ]] && return 0
|
|
_CMUX_SHELL_ACTIVITY_LAST="$state"
|
|
{
|
|
_cmux_send "report_shell_state $state --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
} >/dev/null 2>&1 &!
|
|
}
|
|
|
|
_cmux_ports_kick() {
|
|
# Lightweight: just tell the app to run a batched scan for this panel.
|
|
# The app coalesces kicks across all panels and runs a single ps+lsof.
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
_CMUX_PORTS_LAST_RUN=$EPOCHSECONDS
|
|
{
|
|
_cmux_send "ports_kick --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
} >/dev/null 2>&1 &!
|
|
}
|
|
|
|
_cmux_report_git_branch_for_path() {
|
|
local repo_path="$1"
|
|
[[ -n "$repo_path" ]] || return 0
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
|
|
local branch dirty_opt="" first
|
|
branch="$(git -C "$repo_path" branch --show-current 2>/dev/null)"
|
|
if [[ -n "$branch" ]]; then
|
|
first="$(git -C "$repo_path" status --porcelain -uno 2>/dev/null | head -1)"
|
|
[[ -n "$first" ]] && dirty_opt="--status=dirty"
|
|
_cmux_send "report_git_branch $branch $dirty_opt --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
else
|
|
_cmux_send "clear_git_branch --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
fi
|
|
}
|
|
|
|
_cmux_clear_pr_for_panel() {
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
_cmux_send "clear_pr --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
}
|
|
|
|
_cmux_pr_output_indicates_no_pull_request() {
|
|
local output="${1:l}"
|
|
[[ "$output" == *"no pull requests found"* \
|
|
|| "$output" == *"no pull request found"* \
|
|
|| "$output" == *"no pull requests associated"* \
|
|
|| "$output" == *"no pull request associated"* ]]
|
|
}
|
|
|
|
_cmux_github_repo_slug_for_path() {
|
|
local repo_path="$1"
|
|
local remote_url="" path_part=""
|
|
[[ -n "$repo_path" ]] || return 0
|
|
|
|
remote_url="$(git -C "$repo_path" remote get-url origin 2>/dev/null)"
|
|
[[ -n "$remote_url" ]] || return 0
|
|
|
|
case "$remote_url" in
|
|
git@github.com:*)
|
|
path_part="${remote_url#git@github.com:}"
|
|
;;
|
|
ssh://git@github.com/*)
|
|
path_part="${remote_url#ssh://git@github.com/}"
|
|
;;
|
|
https://github.com/*)
|
|
path_part="${remote_url#https://github.com/}"
|
|
;;
|
|
http://github.com/*)
|
|
path_part="${remote_url#http://github.com/}"
|
|
;;
|
|
git://github.com/*)
|
|
path_part="${remote_url#git://github.com/}"
|
|
;;
|
|
*)
|
|
return 0
|
|
;;
|
|
esac
|
|
|
|
path_part="${path_part%.git}"
|
|
[[ "$path_part" == */* ]] || return 0
|
|
print -r -- "$path_part"
|
|
}
|
|
|
|
_cmux_report_pr_for_path() {
|
|
local repo_path="$1"
|
|
[[ -n "$repo_path" ]] || {
|
|
_cmux_clear_pr_for_panel
|
|
return 0
|
|
}
|
|
[[ -d "$repo_path" ]] || {
|
|
_cmux_clear_pr_for_panel
|
|
return 0
|
|
}
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
|
|
local branch repo_slug="" gh_output="" gh_error="" err_file="" number state url status_opt="" gh_status
|
|
local explicit_branch_output="" explicit_branch_error="" explicit_branch_status=0
|
|
local implicit_probe_indicates_no_pr=0 explicit_probe_indicates_no_pr=0
|
|
local -a gh_repo_args
|
|
gh_repo_args=()
|
|
branch="$(git -C "$repo_path" branch --show-current 2>/dev/null)"
|
|
if [[ -z "$branch" ]] || ! command -v gh >/dev/null 2>&1; then
|
|
_cmux_clear_pr_for_panel
|
|
return 0
|
|
fi
|
|
repo_slug="$(_cmux_github_repo_slug_for_path "$repo_path")"
|
|
if [[ -n "$repo_slug" ]]; then
|
|
gh_repo_args=(--repo "$repo_slug")
|
|
fi
|
|
|
|
err_file="$(/usr/bin/mktemp "${TMPDIR:-/tmp}/cmux-gh-pr-view.XXXXXX" 2>/dev/null || true)"
|
|
[[ -n "$err_file" ]] || return 1
|
|
gh_output="$(
|
|
builtin cd "$repo_path" 2>/dev/null \
|
|
&& gh pr view \
|
|
"${gh_repo_args[@]}" \
|
|
--json number,state,url \
|
|
--jq '[.number, .state, .url] | @tsv' \
|
|
2>"$err_file"
|
|
)"
|
|
gh_status=$?
|
|
if [[ -f "$err_file" ]]; then
|
|
gh_error="$("/bin/cat" -- "$err_file" 2>/dev/null || true)"
|
|
/bin/rm -f -- "$err_file" >/dev/null 2>&1 || true
|
|
fi
|
|
|
|
if (( gh_status == 0 )) && [[ -n "$gh_output" ]]; then
|
|
:
|
|
else
|
|
if (( gh_status == 0 )) && [[ -z "$gh_output" ]]; then
|
|
implicit_probe_indicates_no_pr=1
|
|
elif _cmux_pr_output_indicates_no_pull_request "$gh_error"; then
|
|
implicit_probe_indicates_no_pr=1
|
|
fi
|
|
|
|
# `gh pr view` without an explicit branch can fail to resolve the
|
|
# current worktree branch even when the branch has a PR. Fall back to
|
|
# the explicit branch name before concluding there is no PR.
|
|
err_file="$(/usr/bin/mktemp "${TMPDIR:-/tmp}/cmux-gh-pr-view.XXXXXX" 2>/dev/null || true)"
|
|
[[ -n "$err_file" ]] || return 1
|
|
explicit_branch_output="$(
|
|
builtin cd "$repo_path" 2>/dev/null \
|
|
&& gh pr view "$branch" \
|
|
"${gh_repo_args[@]}" \
|
|
--json number,state,url \
|
|
--jq '[.number, .state, .url] | @tsv' \
|
|
2>"$err_file"
|
|
)"
|
|
explicit_branch_status=$?
|
|
if [[ -f "$err_file" ]]; then
|
|
explicit_branch_error="$("/bin/cat" -- "$err_file" 2>/dev/null || true)"
|
|
/bin/rm -f -- "$err_file" >/dev/null 2>&1 || true
|
|
fi
|
|
|
|
if (( explicit_branch_status == 0 )) && [[ -n "$explicit_branch_output" ]]; then
|
|
gh_output="$explicit_branch_output"
|
|
gh_status=0
|
|
else
|
|
if (( explicit_branch_status == 0 )) && [[ -z "$explicit_branch_output" ]]; then
|
|
explicit_probe_indicates_no_pr=1
|
|
elif _cmux_pr_output_indicates_no_pull_request "$explicit_branch_error"; then
|
|
explicit_probe_indicates_no_pr=1
|
|
fi
|
|
|
|
if (( implicit_probe_indicates_no_pr )) && (( explicit_probe_indicates_no_pr )); then
|
|
_cmux_clear_pr_for_panel
|
|
return 0
|
|
fi
|
|
|
|
# Keep the last-known PR badge on transient gh failures (auth hiccups,
|
|
# API lag after creation, or rate limiting) and retry on the next poll.
|
|
return 1
|
|
fi
|
|
fi
|
|
|
|
local IFS=$'\t'
|
|
read -r number state url <<< "$gh_output"
|
|
if [[ -z "$number" ]] || [[ -z "$url" ]]; then
|
|
return 1
|
|
fi
|
|
|
|
case "$state" in
|
|
MERGED) status_opt="--state=merged" ;;
|
|
OPEN) status_opt="--state=open" ;;
|
|
CLOSED) status_opt="--state=closed" ;;
|
|
*) return 1 ;;
|
|
esac
|
|
|
|
_cmux_send "report_pr $number $url $status_opt --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
}
|
|
|
|
_cmux_child_pids() {
|
|
local parent_pid="$1"
|
|
[[ -n "$parent_pid" ]] || return 0
|
|
/bin/ps -ax -o pid= -o ppid= 2>/dev/null | /usr/bin/awk -v parent="$parent_pid" '$2 == parent { print $1 }'
|
|
}
|
|
|
|
_cmux_kill_process_tree() {
|
|
local pid="$1"
|
|
local signal="${2:-TERM}"
|
|
local child_pid=""
|
|
[[ -n "$pid" ]] || return 0
|
|
|
|
while IFS= read -r child_pid; do
|
|
[[ -n "$child_pid" ]] || continue
|
|
[[ "$child_pid" == "$pid" ]] && continue
|
|
_cmux_kill_process_tree "$child_pid" "$signal"
|
|
done < <(_cmux_child_pids "$pid")
|
|
|
|
kill "-$signal" "$pid" >/dev/null 2>&1 || true
|
|
}
|
|
|
|
_cmux_run_pr_probe_with_timeout() {
|
|
local repo_path="$1"
|
|
local probe_pid=""
|
|
local started_at=$EPOCHSECONDS
|
|
local now=$started_at
|
|
|
|
(
|
|
_cmux_report_pr_for_path "$repo_path"
|
|
) &
|
|
probe_pid=$!
|
|
|
|
while kill -0 "$probe_pid" >/dev/null 2>&1; do
|
|
sleep 1
|
|
now=$EPOCHSECONDS
|
|
if (( _CMUX_ASYNC_JOB_TIMEOUT > 0 )) && (( now - started_at >= _CMUX_ASYNC_JOB_TIMEOUT )); then
|
|
_cmux_kill_process_tree "$probe_pid" TERM
|
|
sleep 0.2
|
|
if kill -0 "$probe_pid" >/dev/null 2>&1; then
|
|
_cmux_kill_process_tree "$probe_pid" KILL
|
|
sleep 0.2
|
|
fi
|
|
if ! kill -0 "$probe_pid" >/dev/null 2>&1; then
|
|
wait "$probe_pid" >/dev/null 2>&1 || true
|
|
fi
|
|
return 1
|
|
fi
|
|
done
|
|
|
|
wait "$probe_pid"
|
|
}
|
|
|
|
_cmux_stop_pr_poll_loop() {
|
|
if [[ -n "$_CMUX_PR_POLL_PID" ]]; then
|
|
# Use SIGKILL directly to avoid blocking sleep in preexec.
|
|
# The poll loop is lightweight and safe to kill abruptly.
|
|
_cmux_kill_process_tree "$_CMUX_PR_POLL_PID" KILL
|
|
_CMUX_PR_POLL_PID=""
|
|
fi
|
|
}
|
|
|
|
_cmux_start_pr_poll_loop() {
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
|
|
local watch_pwd="${1:-$PWD}"
|
|
local force_restart="${2:-0}"
|
|
local watch_shell_pid="$$"
|
|
local interval="${_CMUX_PR_POLL_INTERVAL:-45}"
|
|
|
|
if [[ "$force_restart" != "1" && "$watch_pwd" == "$_CMUX_PR_POLL_PWD" && -n "$_CMUX_PR_POLL_PID" ]] \
|
|
&& kill -0 "$_CMUX_PR_POLL_PID" 2>/dev/null; then
|
|
return 0
|
|
fi
|
|
|
|
_cmux_stop_pr_poll_loop
|
|
_CMUX_PR_POLL_PWD="$watch_pwd"
|
|
|
|
{
|
|
while true; do
|
|
kill -0 "$watch_shell_pid" >/dev/null 2>&1 || break
|
|
_cmux_run_pr_probe_with_timeout "$watch_pwd" || true
|
|
sleep "$interval"
|
|
done
|
|
} >/dev/null 2>&1 &!
|
|
_CMUX_PR_POLL_PID=$!
|
|
}
|
|
|
|
_cmux_stop_git_head_watch() {
|
|
if [[ -n "$_CMUX_GIT_HEAD_WATCH_PID" ]]; then
|
|
kill "$_CMUX_GIT_HEAD_WATCH_PID" >/dev/null 2>&1 || true
|
|
_CMUX_GIT_HEAD_WATCH_PID=""
|
|
fi
|
|
}
|
|
|
|
_cmux_start_git_head_watch() {
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
|
|
local watch_pwd="$PWD"
|
|
local watch_head_path
|
|
watch_head_path="$(_cmux_git_resolve_head_path 2>/dev/null || true)"
|
|
[[ -n "$watch_head_path" ]] || return 0
|
|
|
|
local watch_head_signature
|
|
watch_head_signature="$(_cmux_git_head_signature "$watch_head_path" 2>/dev/null || true)"
|
|
|
|
_CMUX_GIT_HEAD_LAST_PWD="$watch_pwd"
|
|
_CMUX_GIT_HEAD_PATH="$watch_head_path"
|
|
_CMUX_GIT_HEAD_SIGNATURE="$watch_head_signature"
|
|
|
|
_cmux_stop_git_head_watch
|
|
{
|
|
local last_signature="$watch_head_signature"
|
|
while true; do
|
|
sleep 1
|
|
|
|
local signature
|
|
signature="$(_cmux_git_head_signature "$watch_head_path" 2>/dev/null || true)"
|
|
if [[ -n "$signature" && "$signature" != "$last_signature" ]]; then
|
|
last_signature="$signature"
|
|
_cmux_report_git_branch_for_path "$watch_pwd"
|
|
fi
|
|
done
|
|
} >/dev/null 2>&1 &!
|
|
_CMUX_GIT_HEAD_WATCH_PID=$!
|
|
}
|
|
|
|
_cmux_preexec() {
|
|
if [[ -z "$_CMUX_TTY_NAME" ]]; then
|
|
local t
|
|
t="$(tty 2>/dev/null || true)"
|
|
t="${t##*/}"
|
|
[[ -n "$t" && "$t" != "not a tty" ]] && _CMUX_TTY_NAME="$t"
|
|
fi
|
|
|
|
_CMUX_CMD_START=$EPOCHSECONDS
|
|
_cmux_report_shell_activity_state running
|
|
|
|
# Heuristic: commands that may change git branch/dirty state without changing $PWD.
|
|
local cmd="${1## }"
|
|
case "$cmd" in
|
|
git\ *|git|gh\ *|lazygit|lazygit\ *|tig|tig\ *|gitui|gitui\ *|stg\ *|jj\ *)
|
|
_CMUX_GIT_FORCE=1
|
|
_CMUX_PR_FORCE=1 ;;
|
|
esac
|
|
|
|
# Register TTY + kick batched port scan for foreground commands (servers).
|
|
_cmux_report_tty_once
|
|
_cmux_ports_kick
|
|
_cmux_stop_pr_poll_loop
|
|
_cmux_start_git_head_watch
|
|
}
|
|
|
|
_cmux_precmd() {
|
|
_cmux_stop_git_head_watch
|
|
|
|
# Skip if socket doesn't exist yet
|
|
[[ -S "$CMUX_SOCKET_PATH" ]] || return 0
|
|
[[ -n "$CMUX_TAB_ID" ]] || return 0
|
|
[[ -n "$CMUX_PANEL_ID" ]] || return 0
|
|
_cmux_report_shell_activity_state prompt
|
|
|
|
if [[ -z "$_CMUX_TTY_NAME" ]]; then
|
|
local t
|
|
t="$(tty 2>/dev/null || true)"
|
|
t="${t##*/}"
|
|
[[ -n "$t" && "$t" != "not a tty" ]] && _CMUX_TTY_NAME="$t"
|
|
fi
|
|
|
|
_cmux_report_tty_once
|
|
|
|
local now=$EPOCHSECONDS
|
|
local pwd="$PWD"
|
|
local cmd_start="$_CMUX_CMD_START"
|
|
_CMUX_CMD_START=0
|
|
|
|
# Post-wake socket writes can occasionally leave a probe process wedged.
|
|
# If one probe is stale, clear the guard so fresh async probes can resume.
|
|
if [[ -n "$_CMUX_GIT_JOB_PID" ]]; then
|
|
if ! kill -0 "$_CMUX_GIT_JOB_PID" 2>/dev/null; then
|
|
_CMUX_GIT_JOB_PID=""
|
|
_CMUX_GIT_JOB_STARTED_AT=0
|
|
elif (( _CMUX_GIT_JOB_STARTED_AT > 0 )) && (( now - _CMUX_GIT_JOB_STARTED_AT >= _CMUX_ASYNC_JOB_TIMEOUT )); then
|
|
_CMUX_GIT_JOB_PID=""
|
|
_CMUX_GIT_JOB_STARTED_AT=0
|
|
_CMUX_GIT_FORCE=1
|
|
fi
|
|
fi
|
|
|
|
# CWD: keep the app in sync with the actual shell directory.
|
|
# This is also the simplest way to test sidebar directory behavior end-to-end.
|
|
if [[ "$pwd" != "$_CMUX_PWD_LAST_PWD" ]]; then
|
|
_CMUX_PWD_LAST_PWD="$pwd"
|
|
{
|
|
# Quote to preserve spaces.
|
|
local qpwd="${pwd//\"/\\\"}"
|
|
_cmux_send "report_pwd \"${qpwd}\" --tab=$CMUX_TAB_ID --panel=$CMUX_PANEL_ID"
|
|
} >/dev/null 2>&1 &!
|
|
fi
|
|
|
|
# Git branch/dirty: update immediately on directory change, otherwise every ~3s.
|
|
# While a foreground command is running, _cmux_start_git_head_watch probes HEAD
|
|
# once per second so agent-initiated git checkouts still surface quickly.
|
|
local should_git=0
|
|
local git_head_changed=0
|
|
|
|
# Git branch can change without a `git ...`-prefixed command (aliases like `gco`,
|
|
# tools like `gh pr checkout`, etc.). Detect HEAD changes and force a refresh.
|
|
if [[ "$pwd" != "$_CMUX_GIT_HEAD_LAST_PWD" ]]; then
|
|
_CMUX_GIT_HEAD_LAST_PWD="$pwd"
|
|
_CMUX_GIT_HEAD_PATH="$(_cmux_git_resolve_head_path 2>/dev/null || true)"
|
|
_CMUX_GIT_HEAD_SIGNATURE=""
|
|
fi
|
|
if [[ -n "$_CMUX_GIT_HEAD_PATH" ]]; then
|
|
local head_signature
|
|
head_signature="$(_cmux_git_head_signature "$_CMUX_GIT_HEAD_PATH" 2>/dev/null || true)"
|
|
if [[ -n "$head_signature" ]]; then
|
|
if [[ -z "$_CMUX_GIT_HEAD_SIGNATURE" ]]; then
|
|
# The first observed HEAD value establishes the baseline for this
|
|
# shell session. Don't treat it as a branch change or we'll clear
|
|
# restore-seeded PR badges before the first background probe runs.
|
|
_CMUX_GIT_HEAD_SIGNATURE="$head_signature"
|
|
elif [[ "$head_signature" != "$_CMUX_GIT_HEAD_SIGNATURE" ]]; then
|
|
_CMUX_GIT_HEAD_SIGNATURE="$head_signature"
|
|
git_head_changed=1
|
|
# Treat HEAD file change like a git command — force-replace any
|
|
# running probe so the sidebar picks up the new branch immediately.
|
|
_CMUX_GIT_FORCE=1
|
|
_CMUX_PR_FORCE=1
|
|
should_git=1
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
if [[ "$pwd" != "$_CMUX_GIT_LAST_PWD" ]]; then
|
|
should_git=1
|
|
elif (( _CMUX_GIT_FORCE )); then
|
|
should_git=1
|
|
elif (( now - _CMUX_GIT_LAST_RUN >= 3 )); then
|
|
should_git=1
|
|
fi
|
|
|
|
if (( should_git )); then
|
|
local can_launch_git=1
|
|
if [[ -n "$_CMUX_GIT_JOB_PID" ]] && kill -0 "$_CMUX_GIT_JOB_PID" 2>/dev/null; then
|
|
# If a stale probe is still running but the cwd changed (or we just ran
|
|
# a git command), restart immediately so branch state isn't delayed
|
|
# until the next user command/prompt.
|
|
# Note: this repeats the cwd check above on purpose. The first check
|
|
# decides whether we should refresh at all; this one decides whether
|
|
# an in-flight older probe can be reused vs. replaced.
|
|
if [[ "$pwd" != "$_CMUX_GIT_LAST_PWD" ]] || (( _CMUX_GIT_FORCE )); then
|
|
kill "$_CMUX_GIT_JOB_PID" >/dev/null 2>&1 || true
|
|
_CMUX_GIT_JOB_PID=""
|
|
_CMUX_GIT_JOB_STARTED_AT=0
|
|
else
|
|
can_launch_git=0
|
|
fi
|
|
fi
|
|
|
|
if (( can_launch_git )); then
|
|
_CMUX_GIT_FORCE=0
|
|
_CMUX_GIT_LAST_PWD="$pwd"
|
|
_CMUX_GIT_LAST_RUN=$now
|
|
{
|
|
_cmux_report_git_branch_for_path "$pwd"
|
|
} >/dev/null 2>&1 &!
|
|
_CMUX_GIT_JOB_PID=$!
|
|
_CMUX_GIT_JOB_STARTED_AT=$now
|
|
fi
|
|
fi
|
|
|
|
# Pull request metadata is remote state. Keep a lightweight background poll
|
|
# alive while the shell is idle so gh-created PRs and merge status changes
|
|
# appear even without another prompt.
|
|
local should_restart_pr_poll=0
|
|
local pr_context_changed=0
|
|
if [[ -n "$_CMUX_PR_POLL_PWD" && "$pwd" != "$_CMUX_PR_POLL_PWD" ]]; then
|
|
pr_context_changed=1
|
|
elif (( git_head_changed )); then
|
|
pr_context_changed=1
|
|
fi
|
|
if [[ "$pwd" != "$_CMUX_PR_POLL_PWD" ]]; then
|
|
should_restart_pr_poll=1
|
|
elif (( _CMUX_PR_FORCE )); then
|
|
should_restart_pr_poll=1
|
|
elif [[ -z "$_CMUX_PR_POLL_PID" ]] || ! kill -0 "$_CMUX_PR_POLL_PID" 2>/dev/null; then
|
|
should_restart_pr_poll=1
|
|
fi
|
|
|
|
if (( should_restart_pr_poll )); then
|
|
_CMUX_PR_FORCE=0
|
|
if (( pr_context_changed )); then
|
|
_cmux_clear_pr_for_panel
|
|
fi
|
|
_cmux_start_pr_poll_loop "$pwd" 1
|
|
fi
|
|
|
|
# Ports: lightweight kick to the app's batched scanner.
|
|
# - Periodic scan to avoid stale values.
|
|
# - Forced scan when a long-running command returns to the prompt (common when stopping a server).
|
|
local cmd_dur=0
|
|
if [[ -n "$cmd_start" && "$cmd_start" != 0 ]]; then
|
|
cmd_dur=$(( now - cmd_start ))
|
|
fi
|
|
|
|
if (( cmd_dur >= 2 || now - _CMUX_PORTS_LAST_RUN >= 10 )); then
|
|
_cmux_ports_kick
|
|
fi
|
|
}
|
|
|
|
# Ensure Resources/bin is at the front of PATH, and remove the app's
|
|
# Contents/MacOS entry so the GUI cmux binary cannot shadow the CLI cmux.
|
|
# Shell init (.zprofile/.zshrc) may prepend other dirs after launch.
|
|
# We fix this once on first prompt (after all init files have run).
|
|
_cmux_fix_path() {
|
|
if [[ -n "${GHOSTTY_BIN_DIR:-}" ]]; then
|
|
local gui_dir="${GHOSTTY_BIN_DIR%/}"
|
|
local bin_dir="${gui_dir%/MacOS}/Resources/bin"
|
|
if [[ -d "$bin_dir" ]]; then
|
|
# Remove existing entries and re-prepend the CLI bin dir.
|
|
local -a parts=("${(@s/:/)PATH}")
|
|
parts=("${(@)parts:#$bin_dir}")
|
|
parts=("${(@)parts:#$gui_dir}")
|
|
PATH="${bin_dir}:${(j/:/)parts}"
|
|
fi
|
|
fi
|
|
add-zsh-hook -d precmd _cmux_fix_path
|
|
}
|
|
|
|
_cmux_zshexit() {
|
|
_cmux_stop_git_head_watch
|
|
_cmux_stop_pr_poll_loop
|
|
}
|
|
|
|
autoload -Uz add-zsh-hook
|
|
add-zsh-hook preexec _cmux_preexec
|
|
add-zsh-hook precmd _cmux_precmd
|
|
add-zsh-hook precmd _cmux_fix_path
|
|
add-zsh-hook zshexit _cmux_zshexit
|