| 1 | #!/bin/bash |
| 2 | # Four segments: model, thinking/effort, context used %, rate-limit remaining. |
| 3 | # Meters ramp cool -> hot (blue, cyan, green, yellow, red) as each stat grows big. |
| 4 | input=$(cat) |
| 5 | |
| 6 | sep_color=240 |
| 7 | model_color=176 |
| 8 | thinking_color=108 |
| 9 | |
| 10 | color_for_pct() { |
| 11 | local pct="${1%.*}" |
| 12 | if (( pct >= 85 )); then echo 203 |
| 13 | elif (( pct >= 70 )); then echo 214 |
| 14 | elif (( pct >= 50 )); then echo 114 |
| 15 | elif (( pct >= 25 )); then echo 80 |
| 16 | else echo 111 |
| 17 | fi |
| 18 | } |
| 19 | |
| 20 | humanize() { |
| 21 | local n=$1 |
| 22 | if (( n >= 1000000 )); then |
| 23 | printf '%s' "$(awk -v n="$n" 'BEGIN{printf "%.1fM", n/1000000}' | sed 's/\.0M/M/')" |
| 24 | elif (( n >= 1000 )); then |
| 25 | printf '%dk' $(( n / 1000 )) |
| 26 | else |
| 27 | printf '%d' "$n" |
| 28 | fi |
| 29 | } |
| 30 | |
| 31 | model=$(printf '%s' "$input" | jq -r '.model.display_name // ""') |
| 32 | model_short=$(grep -oE 'Opus|Sonnet|Haiku' <<<"$model" | head -1) |
| 33 | [[ -z "$model_short" ]] && model_short="$model" |
| 34 | |
| 35 | effort=$(printf '%s' "$input" | jq -r '.effort.level // ""') |
| 36 | if [[ -n "$effort" ]]; then |
| 37 | thinking="$(tr '[:lower:]' '[:upper:]' <<<"${effort:0:1}")${effort:1}" |
| 38 | elif [[ "$(printf '%s' "$input" | jq -r '.thinking.enabled // false')" == "true" ]]; then |
| 39 | thinking="On" |
| 40 | else |
| 41 | thinking="Off" |
| 42 | fi |
| 43 | |
| 44 | segments=("$(printf "\033[38;5;%sm%s\033[0m" "$model_color" "$model_short")" "$(printf "\033[38;5;%sm%s\033[0m" "$thinking_color" "$thinking")") |
| 45 | |
| 46 | ctx_pct=$(printf '%s' "$input" | jq -r '.context_window.used_percentage // empty') |
| 47 | if [[ -n "$ctx_pct" ]]; then |
| 48 | read -r ctx_used ctx_size < <(printf '%s' "$input" | jq -r '.context_window | "\(.total_input_tokens // 0) \(.context_window_size // 0)"') |
| 49 | c=$(color_for_pct "$ctx_pct") |
| 50 | if (( ctx_size > 0 )); then |
| 51 | segments+=("$(printf "\033[38;5;%sm%.0f%% ctx (%s/%s)\033[0m" "$c" "$ctx_pct" "$(humanize "$ctx_used")" "$(humanize "$ctx_size")")") |
| 52 | else |
| 53 | segments+=("$(printf "\033[38;5;%sm%.0f%% ctx\033[0m" "$c" "$ctx_pct")") |
| 54 | fi |
| 55 | fi |
| 56 | |
| 57 | five=$(printf '%s' "$input" | jq -r '.rate_limits.five_hour.used_percentage // empty') |
| 58 | week=$(printf '%s' "$input" | jq -r '.rate_limits.seven_day.used_percentage // empty') |
| 59 | if [[ -n "$five" || -n "$week" ]]; then |
| 60 | limits="" |
| 61 | if [[ -n "$five" ]]; then |
| 62 | c=$(color_for_pct "$five") |
| 63 | limits+=$(printf "\033[38;5;%sm%.0f%% 5h\033[0m" "$c" "$(( 100 - ${five%.*} ))") |
| 64 | fi |
| 65 | if [[ -n "$week" ]]; then |
| 66 | [[ -n "$limits" ]] && limits+=$(printf " \033[38;5;%sm/\033[0m " "$sep_color") |
| 67 | c=$(color_for_pct "$week") |
| 68 | limits+=$(printf "\033[38;5;%sm%.0f%% 7d\033[0m" "$c" "$(( 100 - ${week%.*} ))") |
| 69 | fi |
| 70 | segments+=("$limits") |
| 71 | fi |
| 72 | |
| 73 | sep=$(printf " \033[38;5;%sm·\033[0m " "$sep_color") |
| 74 | line="${segments[0]}" |
| 75 | for ((i = 1; i < ${#segments[@]}; i++)); do |
| 76 | line+="${sep}${segments[i]}" |
| 77 | done |
| 78 | printf '%s\n' "$line" |