#!/bin/bash

set -e

echo "Running tests and generating coverage..."
go test -coverprofile=coverage.out -v $(go list ./... | grep -v -E 'examples' | grep -v -E 'test')

echo "Generating HTML coverage report..."
go tool cover -html=coverage.out -o coverage.html

echo ""
echo "Generating sorted coverage table..."

# Create a simple approach using grep and sed to extract file coverage
# First, get the total coverage
total_coverage=$(go tool cover -func=coverage.out | grep "^total:" | sed 's/.*([^)]*)[[:space:]]*\([0-9.]*\)%.*/\1/')

# Extract file-level coverage by finding the last occurrence of each file
go tool cover -func=coverage.out | grep -E "\.go:[0-9]+:" | \
sed 's/^\([^:]*\.go\):.*[[:space:]]\([0-9.]*\)%.*/\2 \1/' | \
sort -k2 | \
awk '{file_coverage[$2] = $1} END {for (file in file_coverage) printf "%.2f %s\n", file_coverage[file], file}' | \
sort -nr > coverage_sorted.txt

# Add total coverage to the file
if [[ -n "$total_coverage" && "$total_coverage" != "0" ]]; then
    echo "TOTAL: $total_coverage" >> coverage_sorted.txt
fi

echo ""
echo "Coverage Summary (sorted by percentage - lowest coverage first):"
echo "================================================================="
printf "%-60s %10s %12s\n" "FILE" "COVERAGE" "STATUS"
echo "================================================================="

# Display results with status indicators
tail -n +1 coverage_sorted.txt | while read percent file; do
    if [[ "$file" == "TOTAL:" ]]; then
        echo ""
        printf "%-60s %8s%% %12s\n" "OVERALL PROJECT COVERAGE" "$percent" "📊 TOTAL"
        echo "================================================================="
        continue
    fi
    
    filename=$(basename "$file")
    status=""
    if (( $(echo "$percent < 50" | bc -l 2>/dev/null || echo "0") )); then
        status="🔴 CRITICAL"
    elif (( $(echo "$percent < 70" | bc -l 2>/dev/null || echo "0") )); then
        status="🟡 LOW"
    elif (( $(echo "$percent < 90" | bc -l 2>/dev/null || echo "0") )); then
        status="🟢 GOOD"
    else
        status="✅ EXCELLENT"
    fi
    
    printf "%-60s %8s%% %12s\n" "$filename" "$percent" "$status"
done

echo ""
echo "Top 10 files needing attention (lowest coverage):"
echo "================================================="
grep -v "TOTAL:" coverage_sorted.txt | tail -10 | while read percent file; do
    filename=$(basename "$file")
    printf "%-60s %8.1f%%\n" "$filename" "$percent"
done

echo ""
echo "Coverage reports generated:"
echo "- HTML report: coverage.html (detailed line-by-line coverage)"
echo "- Sorted table: coverage_sorted.txt"
echo "- Use 'go tool cover -func=coverage.out' for function-level details"