#!/bin/bash

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo "Testing FFmpeg binary cataloger with various versions and architectures"
echo "======================================================================"

# Build syft first
echo -e "${YELLOW}Building syft...${NC}"
make bootstrap
make build

# Check if build succeeded
if [ ! -f "./snapshot/darwin-build_darwin_arm64_v8.0/syft" ]; then
    echo -e "${RED}Error: syft binary not found after build${NC}"
    exit 1
fi

SYFT_BINARY="./snapshot/darwin-build_darwin_arm64_v8.0/syft"

# Get some available tags from Docker Hub
echo -e "${YELLOW}Getting available tags...${NC}"
AVAILABLE_TAGS=$(curl -s "https://registry.hub.docker.com/v2/repositories/linuxserver/ffmpeg/tags/?page_size=20" | jq -r '.results[].name')

echo "Available tags:"
echo "$AVAILABLE_TAGS" | head -10

# Define test cases: tag:expected_version
declare -a TEST_CASES=(
    "version-8.0-cli:8.0"
    "7.1.1:7.1.1"
)

# Function to test a specific container
test_ffmpeg_container() {
    local tag="$1"
    local expected_version="$2"
    local image="docker.io/linuxserver/ffmpeg:${tag}"
    
    echo -e "${YELLOW}Testing $image (expecting version $expected_version)...${NC}"
    
    # Check if tag exists
    if ! echo "$AVAILABLE_TAGS" | grep -q "^${tag}$"; then
        echo -e "${RED}  ✗ Tag $tag not found in available tags${NC}"
        return 1
    fi
    
    echo "  Running syft on container..."
    
    # Run syft on the container image
    local output
    if output=$($SYFT_BINARY "$image" 2>/dev/null); then
        # Check if ffmpeg was detected with correct version
        if echo "$output" | grep -q "ffmpeg.*$expected_version.*binary"; then
            echo -e "${GREEN}  ✓ Success: Detected ffmpeg $expected_version${NC}"
            echo "    $(echo "$output" | grep "ffmpeg.*binary")"
            return 0
        else
            echo -e "${RED}  ✗ Failed: ffmpeg $expected_version not detected or wrong version${NC}"
            echo "    Lines containing 'ffmpeg':"
            echo "$output" | grep -i ffmpeg || echo "    (no ffmpeg found in output)"
            return 1
        fi
    else
        echo -e "${RED}  ✗ Failed: syft execution failed${NC}"
        return 1
    fi
}

# Counter for results
total_tests=0
passed_tests=0

# Test each case
for test_case in "${TEST_CASES[@]}"; do
    IFS=':' read -ra PARTS <<< "$test_case"
    tag="${PARTS[0]}"
    expected_version="${PARTS[1]}"
    
    total_tests=$((total_tests + 1))
    
    if test_ffmpeg_container "$tag" "$expected_version"; then
        passed_tests=$((passed_tests + 1))
    fi
    
    echo ""  # Add blank line between tests
done

# Summary
echo "======================================================================"
echo -e "Results: ${GREEN}$passed_tests${NC} passed, ${RED}$((total_tests - passed_tests))${NC} failed out of ${total_tests} total tests"

if [ $passed_tests -eq $total_tests ]; then
    echo -e "${GREEN}All tests passed! The ffmpeg binary cataloger is working correctly.${NC}"
    exit 0
else
    echo -e "${RED}Some tests failed. Please review the output above.${NC}"
    exit 1
fi