gistfile1.txt

#!/bin/bash

input=$(cat)
cwd=$(echo "$input" | jq -r '.workspace.current_dir')

# Usage limit: fetch from Anthropic API, cache for 5 minutes
CACHE=/tmp/claude-usage-cache.json
usage_str="?%"

get_token() {
    # macOS: credentials stored in keychain
    if command -v security &>/dev/null; then
        security find-generic-password -s "Claude Code-credentials" -w 2>/dev/null \
            | jq -r '.claudeAiOauth.accessToken // empty'
        return
    fi
    # Linux: credentials stored in a JSON file
    local creds=~/.claude/.credentials.json
    if [ -f "$creds" ]; then
        jq -r '.claudeAiOauth.accessToken // empty' "$creds"
    fi
}

file_age_seconds() {
    if stat -f %m "$1" &>/dev/null; then
        # macOS (BSD stat)
        echo $(( $(date +%s) - $(stat -f %m "$1") ))
    else
        # Linux (GNU stat)
        echo $(( $(date +%s) - $(stat -c %Y "$1") ))
    fi
}

update_cache() {
    if [ -f "$CACHE" ]; then
        [ "$(file_age_seconds "$CACHE")" -gt 300 ] || return
    fi
    TOKEN=$(get_token)
    [ -n "$TOKEN" ] || return
    TMPFILE=$(mktemp)
    curl -s -m 5 \
        -H "Authorization: Bearer $TOKEN" \
        -H "anthropic-beta: oauth-2025-04-20" \
        https://api.anthropic.com/api/oauth/usage > "$TMPFILE" 2>/dev/null \
        && [ -s "$TMPFILE" ] && mv "$TMPFILE" "$CACHE" || rm -f "$TMPFILE"
}

build_usage_str() {
    [ -f "$CACHE" ] || return
    pct=$(jq -r '.five_hour.utilization // empty' "$CACHE" 2>/dev/null)
    [ -n "$pct" ] || return
    pct=$(printf '%.0f' "$pct")
    usage_str="${pct}%"
    resets_at=$(jq -r '.five_hour.resets_at // empty' "$CACHE" 2>/dev/null)
    [ -n "$resets_at" ] || return
    reset_epoch=$(python3 -c "from datetime import datetime; print(int(datetime.fromisoformat('$resets_at'.replace('Z','+00:00')).timestamp()))" 2>/dev/null)
    [ -n "$reset_epoch" ] || return
    now=$(date +%s)
    diff=$(( reset_epoch - now ))
    [ "$diff" -gt 0 ] || return
    h=$(( diff / 3600 ))
    m=$(( (diff % 3600) / 60 ))
    usage_str="${pct}% rst:${h}h${m}m"
}

update_cache
build_usage_str

printf '\033[01;32m%s@%s\033[00m:\033[01;36m%s\033[00m \033[97muse:%s\033[00m' \
    "$(whoami)" "$(hostname -s)" "$cwd" "$usage_str"