#!/bin/bash
set -euo pipefail

SCCACHE="${SCCACHE_PATH:-/sccache/target/debug/sccache}"

echo "==================================================================="
echo "Testing: Basedirs with all backends, autotools + headers + __FILE__"
echo "==================================================================="

autotools() (
    cd "$1"
    autoreconf -i >/dev/null 2>&1 || true
    automake --add-missing >/dev/null 2>&1 || true
    ./configure CXX="$SCCACHE g++" >/dev/null 2>&1
    make >/dev/null 2>&1
)

# Function to test a backend
test_backend() {
    local backend_name="$1"
    shift
    cp -r /sccache/tests/integration/basedirs-autotools /build/dir1
    cp -r /sccache/tests/integration/basedirs-autotools /build/dir2

    echo ""
    echo "=========================================="
    echo "Testing backend: $backend_name"
    echo "=========================================="

    # Stop any running sccache server
    "$SCCACHE" --stop-server 2>/dev/null || true

    # Set backend-specific environment variables (passed as arguments)
    for env_var in "$@"; do
        export "${env_var?}"
    done

    # Configure basedirs - should strip /build/dir1 and /build/dir2 prefixes
    export SCCACHE_BASEDIRS="/build/dir1:/build/dir2"

    # Start sccache server
    "$SCCACHE" --start-server

    echo "Test 1: Compile from first directory (cache miss)"
    autotools /build/dir1

    echo "Checking stats after first build..."
    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    CACHE_LOCATION=$(echo "$STATS_JSON" | python3 -c "import sys, json; print(json.load(sys.stdin).get('cache_location', ''))" || echo "unknown")
    echo "Backend detected: $CACHE_LOCATION"

    # Verify backend is being used
    if ! echo "$CACHE_LOCATION" | grep -qi "$backend_name"; then
        echo "WARNING: Expected backend '$backend_name' not found in cache_location: $CACHE_LOCATION"
    fi

    FIRST_MISSES=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_misses', {}).get('counts', {}).get('C/C++', 0))")
    echo "Cache misses after first build: $FIRST_MISSES"

    echo ""
    echo "Test 2: Compile from second directory with same relative path (cache hit expected)"
    autotools /build/dir2

    echo "Verifying cache hits..."
    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    CACHE_HITS=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_hits', {}).get('counts', {}).get('C/C++', 0))")

    echo "Cache hits for $backend_name: $CACHE_HITS"

    SECOND_MISSES=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_misses', {}).get('counts', {}).get('C/C++', 0))")

    if [ "$FIRST_MISSES" == "$SECOND_MISSES" ]; then
        echo "✓ PASS: $backend_name - Basedir test successful"
    else
        echo "✗ FAIL: $backend_name - Basedir test failed, cache misses did not remain the same: $FIRST_MISSES != $SECOND_MISSES"
        echo "$STATS_JSON" | python3 -m json.tool
        exit 1
    fi

    # Verify basedirs are shown in stats
    BASEDIRS_IN_STATS=$(echo "$STATS_JSON" | python3 -c "import sys, json; bd = json.load(sys.stdin).get('basedirs', []); print(':'.join(bd) if bd else 'none')")
    echo "Basedirs in stats: $BASEDIRS_IN_STATS"

    if [ "$BASEDIRS_IN_STATS" = "none" ]; then
        echo "WARNING: Basedirs not shown in stats output: $STATS_JSON"
    fi

    # Clean up for next backend test
    rm -rf /build/dir1 /build/dir2
    "$SCCACHE" --stop-server &>/dev/null || true

    # Unset environment variables
    for env_var in "$@"; do
        VAR_NAME="${env_var%%=*}"
        unset "$VAR_NAME"
    done
    unset SCCACHE_BASEDIRS
}

# Test each backend
test_backend "local disk" "SCCACHE_DIR=/build/sccache"
test_backend "redis" "SCCACHE_REDIS_ENDPOINT=tcp://redis:6379"
test_backend "memcached" "SCCACHE_MEMCACHED_ENDPOINT=tcp://memcached:11211"
test_backend "s3" \
    "SCCACHE_BUCKET=test" \
    "SCCACHE_ENDPOINT=http://minio:9000/" \
    "SCCACHE_REGION=us-east-1" \
    "AWS_ACCESS_KEY_ID=minioadmin" \
    "AWS_SECRET_ACCESS_KEY=minioadmin" \
    "AWS_EC2_METADATA_DISABLED=true"
test_backend "azblob" \
    "SCCACHE_AZURE_BLOB_CONTAINER=test" \
    "SCCACHE_AZURE_CONNECTION_STRING=DefaultEndpointsProtocol=http;AccountName=devstoreaccount1;AccountKey=Eby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsuFq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==;BlobEndpoint=http://azurite:10000/devstoreaccount1;"
test_backend "webdav" \
    "SCCACHE_WEBDAV_ENDPOINT=http://webdav:8080" \
    "SCCACHE_WEBDAV_USERNAME=bar" \
    "SCCACHE_WEBDAV_PASSWORD=baz"

# Function to test basedirs with multi-level cache (disk + remote)
# Tests that basedirs normalization works across cache levels and backfill
test_multilevel_backend() {
    local backend_name="$1"
    local level_name="$backend_name"
    if [ "$backend_name" = "azblob" ]; then
        level_name="azure"
    fi
    shift
    cp -r /sccache/tests/integration/basedirs-autotools /build/dir1
    cp -r /sccache/tests/integration/basedirs-autotools /build/dir2

    echo ""
    echo "=========================================="
    echo "Testing multilevel basedirs: disk + $backend_name"
    echo "=========================================="

    # Stop any running sccache server
    "$SCCACHE" --stop-server 2>/dev/null || true

    # Set backend-specific environment variables (passed as arguments)
    for env_var in "$@"; do
        export "${env_var?}"
    done

    # Configure basedirs and multi-level cache
    export SCCACHE_BASEDIRS="/build/dir1:/build/dir2"
    export SCCACHE_MULTILEVEL_CHAIN="disk,$level_name"
    export SCCACHE_DIR="/build/sccache-ml-basedirs"
    rm -rf /build/sccache-ml-basedirs
    mkdir -p /build/sccache-ml-basedirs

    # Start sccache server
    "$SCCACHE" --start-server

    # Verify multi-level is active
    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    CACHE_LOCATION=$(echo "$STATS_JSON" | python3 -c "import sys, json; print(json.load(sys.stdin).get('cache_location', ''))" || echo "unknown")
    echo "Cache location: $CACHE_LOCATION"

    if ! echo "$CACHE_LOCATION" | grep -qi "Multi-level"; then
        echo "✗ FAIL: Multi-level cache not detected in cache_location: $CACHE_LOCATION"
        exit 1
    fi

    echo "Test 1: Compile from first directory (cache miss, populates L0 disk + L1 $backend_name)"
    autotools /build/dir1

    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    FIRST_MISSES=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_misses', {}).get('counts', {}).get('C/C++', 0))")
    echo "Cache misses after first build: $FIRST_MISSES"

    echo ""
    echo "Test 2: Compile from second directory (cache hit expected via basedirs)"
    autotools /build/dir2

    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    CACHE_HITS=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_hits', {}).get('counts', {}).get('C/C++', 0))")
    SECOND_MISSES=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_misses', {}).get('counts', {}).get('C/C++', 0))")
    echo "Cache hits: $CACHE_HITS, misses: $SECOND_MISSES (first build: $FIRST_MISSES)"

    if [ "$FIRST_MISSES" != "$SECOND_MISSES" ]; then
        echo "✗ FAIL: multilevel disk+$backend_name - Cache misses increased from $FIRST_MISSES to $SECOND_MISSES"
        echo "$STATS_JSON" | python3 -m json.tool
        exit 1
    fi

    echo ""
    echo "Test 3: Clear L0 (disk), rebuild from dir1 (should hit L1 $backend_name and backfill)"
    "$SCCACHE" --stop-server 2>/dev/null || true
    rm -rf /build/sccache-ml-basedirs
    mkdir -p /build/sccache-ml-basedirs
    rm -rf /build/dir1
    cp -r /sccache/tests/integration/basedirs-autotools /build/dir1
    "$SCCACHE" --start-server

    autotools /build/dir1

    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    THIRD_MISSES=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_misses', {}).get('counts', {}).get('C/C++', 0))")
    echo "Cache misses after L0 clear and rebuild: $THIRD_MISSES (should be 0)"

    if [ "$THIRD_MISSES" -gt 0 ]; then
        echo "✗ FAIL: multilevel disk+$backend_name - Misses after L0 clear ($THIRD_MISSES), L1 should have served data"
        echo "$STATS_JSON" | python3 -m json.tool
        exit 1
    fi

    # Give backfill time to complete
    sleep 2

    echo ""
    echo "Test 4: Rebuild from dir2 (should hit backfilled L0 via basedirs)"
    rm -rf /build/dir2
    cp -r /sccache/tests/integration/basedirs-autotools /build/dir2
    autotools /build/dir2

    STATS_JSON=$("$SCCACHE" --show-stats --stats-format=json)
    FOURTH_MISSES=$(echo "$STATS_JSON" | python3 -c "import sys, json; stats = json.load(sys.stdin).get('stats', {}); print(stats.get('cache_misses', {}).get('counts', {}).get('C/C++', 0))")

    if [ "$FOURTH_MISSES" -gt 0 ]; then
        echo "✗ FAIL: multilevel disk+$backend_name - Misses on build 4, basedirs + backfill should provide hits"
        echo "$STATS_JSON" | python3 -m json.tool
        exit 1
    fi

    echo "✓ PASS: multilevel disk+$backend_name - Basedirs + multilevel + backfill all working"

    # Clean up for next test
    rm -rf /build/dir1 /build/dir2 /build/sccache-ml-basedirs
    "$SCCACHE" --stop-server &>/dev/null || true

    # Unset environment variables
    for env_var in "$@"; do
        VAR_NAME="${env_var%%=*}"
        unset "$VAR_NAME"
    done
    unset SCCACHE_BASEDIRS SCCACHE_MULTILEVEL_CHAIN SCCACHE_DIR
}

# Test multilevel basedirs with redis
test_multilevel_backend "redis" "SCCACHE_REDIS_ENDPOINT=tcp://redis:6379"

echo ""
echo "=========================================="
echo "All basedir tests completed successfully!"
echo "=========================================="