Real Time Cost Tracker

#!/usr/bin/env bash

# Real-Time Cost Tracker
# Calculate session costs based on token usage

read -r input

model=$(echo "$input" | jq -r '.model // "unknown"')
tokens=$(echo "$input" | jq -r '.session.totalTokens // 0')

# 2025 Pricing (per 1M tokens)
if [[ "$model" == *"claude-sonnet-4"* ]]; then
  price_per_m=3.00
elif [[ "$model" == *"gpt-4"* ]]; then
  price_per_m=5.00
elif [[ "$model" == *"gemini"* ]]; then
  price_per_m=1.25
else
  price_per_m=1.00
fi

# Calculate cost
cost=$(awk "BEGIN {printf \"%.4f\", ($tokens / 1000000) * $price_per_m}")

# Budget alert
if (( $(echo "$cost > 0.50" | bc -l) )); then
  color="\033[31m"  # Red
  icon="⚠️"
elif (( $(echo "$cost > 0.10" | bc -l) )); then
  color="\033[33m"  # Yellow
  icon="💰"
else
  color="\033[32m"  # Green
  icon="💵"
fi

echo -e "${icon} ${color}$${cost}${color}\033[0m │ ${tokens} tokens"

{
  "format": "bash",
  "position": "right",
  "refreshInterval": 1000
}