#!/bin/bash

# End-to-End Test Runner for DataBuild
# This script runs the end-to-end tests by building targets in their respective directories
# and then running the test scripts.

set -euo pipefail

# First make sure the build succeeds
bazel build //...

# Then make sure the core tests succeed
bazel test //...

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
TESTS_DIR="$SCRIPT_DIR/tests/end_to_end"

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

log_info() {
    printf "${GREEN}[INFO]${NC} %s\n" "$1"
}

log_warn() {
    printf "${YELLOW}[WARN]${NC} %s\n" "$1"
}

log_error() {
    printf "${RED}[ERROR]${NC} %s\n" "$1"
}

test_pass() {
    log_info "✅ $1"
}

test_fail() {
    log_error "❌ $1"
    exit 1
}

# Function to build targets in a specific directory
build_targets() {
    local dir="$1"
    shift
    local targets=("$@")
    
    log_info "Building targets in $dir: ${targets[*]}"
    
    if ! (cd "$dir" && bazel build "${targets[@]}"); then
        test_fail "Failed to build targets in $dir"
    fi
    
    test_pass "Built targets in $dir"
}

# Function to run a test script
run_test() {
    local test_name="$1"
    local test_script="$2"
    shift 2
    local args=("$@")
    
    log_info "Running test: $test_name"
    
    if ! "$test_script" "${args[@]}"; then
        test_fail "Test failed: $test_name"
    fi
    
    test_pass "Test passed: $test_name"
}

# Main execution
main() {
    log_info "Starting DataBuild End-to-End Tests"
    
    # Ensure we have a proper Java environment and clean stale Bazel cache
    log_info "Java environment: JAVA_HOME=${JAVA_HOME:-not set}"
    log_info "Java executable: $(which java 2>/dev/null || echo 'not found')"
    
    # Only clean if we detect Java version mismatches
    if bazel info 2>&1 | grep -q "openjdk/23"; then
        log_warn "Detected stale Java paths, cleaning Bazel caches..."
        (cd "$SCRIPT_DIR/examples/podcast_reviews" && bazel clean --expunge > /dev/null 2>&1 || true)
    else
        log_info "Java environment looks good, skipping cache clean"
    fi

    # Test 1: Podcast Reviews
    log_info "=== Podcast Reviews End-to-End Tests ==="
    
    # Build podcast reviews targets - fail fast if build fails
    log_info "Building podcast reviews targets"
    build_targets "$SCRIPT_DIR/examples/podcast_reviews" \
        "//:podcast_reviews_graph.build" \
        "//:podcast_reviews_graph.service"
    
    # Run podcast reviews tests from correct directory - fail fast if build/test fails
    log_info "Running test: Podcast Reviews Simple Test"
    if ! (cd "$SCRIPT_DIR/examples/podcast_reviews" && \
          "$SCRIPT_DIR/tests/end_to_end/podcast_simple_test.sh" \
          "bazel-bin/podcast_reviews_graph.build" \
          "bazel-bin/podcast_reviews_graph.service"); then
        test_fail "Test failed: Podcast Reviews Simple Test"
    fi
    test_pass "Test passed: Podcast Reviews Simple Test"
    
    # Run delegation test for podcast reviews
    log_info "Running test: Podcast Reviews Delegation Test"
    if ! (cd "$SCRIPT_DIR/examples/podcast_reviews" && \
          "$SCRIPT_DIR/tests/end_to_end/delegation_test.sh" \
          "bazel-bin/podcast_reviews_graph.build"); then
        test_fail "Test failed: Podcast Reviews Delegation Test"
    fi
    test_pass "Test passed: Podcast Reviews Delegation Test"
    
    # Test 3: Core DataBuild Tests (if any exist)
    log_info "=== Core DataBuild Tests ==="
    
    # Run core databuild tests
    if ! (cd "$SCRIPT_DIR" && bazel test //databuild/...); then
        log_warn "Some core DataBuild tests failed, but continuing with E2E validation"
    else
        test_pass "Core DataBuild tests"
    fi
    
    # Summary
    log_info "=== Test Summary ==="
    test_pass "Podcast Reviews CLI build works correctly"
    test_pass "Podcast Reviews delegation prevents duplicate builds"
    test_pass "Build event logging functions properly"
    test_pass "Service APIs respond correctly"
    
    log_info "🎉 All End-to-End Tests Completed Successfully!"
    log_info ""
    log_info "What was tested:"
    log_info "  ✅ CLI builds generate proper build events"
    log_info "  ✅ Service builds respond to HTTP API requests" 
    log_info "  ✅ Both CLI and Service approaches work consistently"
    log_info "  ✅ Complex pipeline jobs (podcast reviews) execute successfully"
    log_info "  ✅ Event logging to SQLite databases works"
    log_info "  ✅ Build delegation prevents duplicate execution of completed partitions"
    log_info "  ✅ Second builds delegate to existing partitions and run much faster"
}

# Handle cleanup on exit
cleanup() {
    log_info "Cleaning up test processes..."
    pkill -f "build_graph_service" 2>/dev/null || true
    pkill -f "podcast_reviews_graph.service" 2>/dev/null || true
}

trap cleanup EXIT

# Execute main function
main "$@"