Miguel Palhas
65 lines
2.4 KiB
Bash
65 lines
2.4 KiB
Bash
#!/usr/bin/env bash
|
|
# NixOS has no /bin/bash; use env so the shebang resolves on all hosts.
|
|
input=$(cat)
|
|
|
|
# Extract values from JSON input
|
|
MODEL=$(echo "$input" | jq -r '.model.display_name // "Claude"')
|
|
DIR=$(echo "$input" | jq -r '.workspace.current_dir // "~"')
|
|
CONTEXT_USED=$(echo "$input" | jq -r '.context_window.used_percentage // 0')
|
|
COST=$(echo "$input" | jq -r '.cost.total_cost_usd // 0')
|
|
|
|
# Machine hostname (bash sets HOSTNAME; fall back to /etc/hostname)
|
|
HOST="${HOSTNAME:-$(cat /etc/hostname 2>/dev/null || echo host)}"
|
|
|
|
# Git repo root (abbreviated ~/projects -> p/) and branch/PR info
|
|
GIT_INFO=""
|
|
REPO_PATH=""
|
|
if git -C "$DIR" rev-parse --git-dir > /dev/null 2>&1; then
|
|
REPO_ROOT=$(git -C "$DIR" rev-parse --show-toplevel 2>/dev/null)
|
|
# For worktrees, use the main repo root instead of the worktree path
|
|
MAIN_GIT_DIR=$(git -C "$DIR" rev-parse --git-common-dir 2>/dev/null)
|
|
if [ -n "$MAIN_GIT_DIR" ] && [[ "$MAIN_GIT_DIR" == /* ]]; then
|
|
REPO_ROOT=$(dirname "$MAIN_GIT_DIR")
|
|
fi
|
|
REPO_PATH="${REPO_ROOT/#$HOME\/projects\//p/}"
|
|
BRANCH=$(git -C "$DIR" branch --show-current 2>/dev/null)
|
|
if [ -n "$BRANCH" ]; then
|
|
PR_NUMBER=$(GIT_DIR="$DIR/.git" GIT_WORK_TREE="$DIR" gh pr view --json number -q '.number' 2>/dev/null)
|
|
if [ -n "$PR_NUMBER" ]; then
|
|
GIT_INFO="$BRANCH (#$PR_NUMBER)"
|
|
else
|
|
GIT_INFO="$BRANCH"
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
# Format cost (force C locale so the decimal separator is always ".")
|
|
COST_FMT=$(LC_NUMERIC=C printf "%.2f" "$COST")
|
|
|
|
# Colors
|
|
RESET=$'\033[0m'
|
|
C_HOST=$'\033[1;36m' # bold cyan — distinguishes the machine
|
|
C_PATH=$'\033[32m' # green
|
|
C_BRANCH=$'\033[33m' # yellow
|
|
C_MODEL=$'\033[35m' # magenta
|
|
C_COST=$'\033[32m' # green
|
|
SEP=$'\033[90m | \033[0m' # dim separator
|
|
|
|
# Context % colored by fill level: green < 50, yellow < 80, red >= 80
|
|
CTX_INT=${CONTEXT_USED%.*}; CTX_INT=${CTX_INT:-0}
|
|
if [ "$CTX_INT" -ge 80 ] 2>/dev/null; then
|
|
C_CTX=$'\033[31m'
|
|
elif [ "$CTX_INT" -ge 50 ] 2>/dev/null; then
|
|
C_CTX=$'\033[33m'
|
|
else
|
|
C_CTX=$'\033[32m'
|
|
fi
|
|
|
|
out="${C_HOST}${HOST}${RESET}"
|
|
[ -n "$REPO_PATH" ] && out="${out}${SEP}${C_PATH}${REPO_PATH}${RESET}"
|
|
[ -n "$GIT_INFO" ] && out="${out}${SEP}${C_BRANCH}${GIT_INFO}${RESET}"
|
|
out="${out}${SEP}${C_MODEL}[${MODEL}]${RESET}"
|
|
out="${out}${SEP}${C_COST}\$${COST_FMT}${RESET}"
|
|
out="${out}${SEP}${C_CTX}${CONTEXT_USED}%${RESET}"
|
|
printf '%s\n' "$out"
|