#!/usr/bin/env bash

input=$(cat)

session_name=$(echo "$input" | jq -r '.session_name // empty')
project_dir=$(echo "$input" | jq -r '.workspace.project_dir')
cwd=$(echo "$input" | jq -r '.workspace.current_dir')
model=$(echo "$input" | jq -r '.model.display_name')

total_in=$(echo "$input" | jq -r '.context_window.total_input_tokens // empty')
total_out=$(echo "$input" | jq -r '.context_window.total_output_tokens // empty')
ctx_size=$(echo "$input" | jq -r '.context_window.context_window_size // empty')
cur_in=$(echo "$input" | jq -r '.context_window.current_usage.input_tokens // empty')
cur_out=$(echo "$input" | jq -r '.context_window.current_usage.output_tokens // empty')
cache_read=$(echo "$input" | jq -r '.context_window.current_usage.cache_read_input_tokens // empty')

git_branch=$(cd "$cwd" 2>/dev/null && git -c core.useBuiltinFSMonitor=false rev-parse --abbrev-ref HEAD 2>/dev/null)
git_stat=$(cd "$cwd" 2>/dev/null && git -c core.useBuiltinFSMonitor=false diff --shortstat HEAD 2>/dev/null | awk '{
    ins=0; del=0
    for (i = 1; i <= NF; i++) {
        if ($i ~ /insertion/) { ins = $(i-1) }
        if ($i ~ /deletion/) { del = $(i-1) }
    }
    if (ins > 0 || del > 0) { printf "+%d/-%d", ins, del }
}')

output=""
[ -n "$session_name" ] && output="[$session_name] "

project_name=$(basename "$project_dir")
output="${output}${project_name}"

if [ -n "$git_branch" ]; then
	git_info="git:$git_branch"
	[ -n "$git_stat" ] && git_info="$git_info $git_stat"
	output="$output ($git_info)"
fi

output="$output | $model"

if [ -n "$total_in" ] && [ -n "$total_out" ] && [ -n "$ctx_size" ]; then
	total_used=$((total_in + total_out))
	used_k=$(awk "BEGIN {printf \"%.1f\", $total_used/1000}")
	output="$output | Session: ${used_k}k"
fi

if [ -n "$cur_in" ] && [ -n "$cur_out" ]; then
	cur_total=$((cur_in + cur_out))
	cur_k=$(awk "BEGIN {printf \"%.1f\", $cur_total/1000}")
	if [ -n "$cache_read" ] && [ "$cache_read" != "0" ]; then
		cache_read_k=$(awk "BEGIN {printf \"%.1f\", $cache_read/1000}")
		output="$output | Cached: ${cache_read_k}k (last:${cur_k}k)"
	else
		output="$output | Last: ${cur_k}k"
	fi
fi

echo "$output"