#!/bin/bash

# tests/docker/redis-validation-test.sh
# Phase 3 :: Redis infrastructure validation for CFN Loop (Loop 3 Investigation)

set -euo pipefail

PROJECT_ROOT=$(git rev-parse --show-toplevel)
source "$PROJECT_ROOT/tests/test-utils.sh" 2>/dev/null || {
  # Fallback if test-utils not available
  log_step() { echo "[$(date +'%Y-%m-%d %H:%M:%S')] $*"; }
  log_info() { echo "  $*"; }
  annotate() { echo "→ $*"; }
}

cleanup() {
  log_step "Cleaning up test artifacts..."
  # Remove test containers if created
  docker ps -a --filter "name=test-redis-" -q 2>/dev/null | xargs docker rm -f 2>/dev/null || true
  # Clean Redis test keys
  redis-cli DEL "cfn:test:*" 2>/dev/null || true
  redis-cli DEL "test:*" 2>/dev/null || true
}

trap cleanup EXIT

log_step "REDIS INFRASTRUCTURE VALIDATION TEST"
echo ""

# Test 1: Host Redis connectivity
test_host_redis() {
  log_step "Test 1: Host Redis (127.0.0.1:6379)"

  if redis-cli -h 127.0.0.1 -p 6379 PING > /dev/null 2>&1; then
    log_info "✅ PASS: Host Redis responding"
    annotate "Version: $(redis-cli INFO server | grep redis_version | cut -d: -f2)"
    return 0
  else
    log_info "❌ FAIL: Host Redis not responding"
    return 1
  fi
}

# Test 2: Docker service connectivity
test_docker_service() {
  log_step "Test 2: Docker Redis Service (redis:6379)"

  # Check if trigger-dev-webapp container exists and is healthy
  local webapp_status=$(docker inspect trigger-dev-webapp --format='{{.State.Health.Status}}' 2>/dev/null || echo "not found")

  if [ "$webapp_status" = "healthy" ]; then
    log_info "✅ PASS: Docker service reachable"
    annotate "Webapp container status: $webapp_status"

    # Verify DNS resolution within network
    if docker exec trigger-dev-webapp sh -c 'getent hosts redis > /dev/null' 2>/dev/null; then
      log_info "✅ DNS resolution works: redis → 172.23.0.x"
    else
      log_info "⚠️  WARNING: DNS resolution check inconclusive"
    fi
    return 0
  else
    log_info "⚠️  WARNING: Webapp container not healthy (status: $webapp_status)"
    annotate "Skipping Docker service test - prerequisite container unhealthy"
    return 0  # Don't fail test on this
  fi
}

# Test 3: Docker network configuration
test_docker_network() {
  log_step "Test 3: Docker Network Configuration"

  local network_name="trigger-dev_trigger-cfn-network"

  if docker network inspect "$network_name" > /dev/null 2>&1; then
    log_info "✅ PASS: Network exists - $network_name"

    # Check connected containers
    local container_count=$(docker network inspect "$network_name" \
      --format='{{range .Containers}}{{.Name}}{{end}}' 2>/dev/null | wc -w)
    annotate "Connected containers: $container_count"

    # Verify Redis container on network
    if docker network inspect "$network_name" --format='{{range .Containers}}{{.Name}}{{end}}' 2>/dev/null | grep -q redis; then
      log_info "✅ Redis container connected to network"
    fi
    return 0
  else
    log_info "❌ FAIL: Network not found"
    return 1
  fi
}

# Test 4: Redis data store health
test_redis_data() {
  log_step "Test 4: Redis Data Store"

  local key_count=$(redis-cli DBSIZE 2>/dev/null | grep -oE '[0-9]+' || echo "0")
  log_info "✅ Redis contains $key_count keys"

  # Test set/get operations
  redis-cli SET "cfn:test:validation" "test-value-$(date +%s)" > /dev/null 2>&1
  local retrieved=$(redis-cli GET "cfn:test:validation" 2>/dev/null || echo "")

  if [ -n "$retrieved" ]; then
    log_info "✅ PASS: Read/write operations functional"
    annotate "Test key: cfn:test:validation = $retrieved"
  else
    log_info "⚠️  WARNING: Test read/write inconclusive"
  fi

  return 0
}

# Test 5: docker-compose.yml configuration
test_compose_config() {
  log_step "Test 5: docker-compose.yml Configuration"

  local compose_file="$PROJECT_ROOT/docker/trigger-dev/docker-compose.yml"

  if [ ! -f "$compose_file" ]; then
    log_info "❌ FAIL: docker-compose.yml not found at $compose_file"
    return 1
  fi

  # Check CFN_REDIS_HOST configuration
  if grep -q "CFN_REDIS_HOST.*redis" "$compose_file"; then
    log_info "✅ PASS: CFN_REDIS_HOST configured"
    annotate "Default: redis (service name)"
  else
    log_info "⚠️  WARNING: CFN_REDIS_HOST not found"
  fi

  # Check CFN_REDIS_PORT configuration
  if grep -q "CFN_REDIS_PORT.*6379" "$compose_file"; then
    log_info "✅ PASS: CFN_REDIS_PORT configured"
    annotate "Default: 6379 (standard Redis port)"
  else
    log_info "⚠️  WARNING: CFN_REDIS_PORT not found"
  fi

  return 0
}

# Test 6: Environment variable configuration
test_env_config() {
  log_step "Test 6: Environment Variable Configuration (.env)"

  local env_file="$PROJECT_ROOT/docker/trigger-dev/.env"

  if [ ! -f "$env_file" ]; then
    log_info "⚠️  WARNING: .env file not found"
    return 0
  fi

  if grep -q "^CFN_REDIS_HOST=" "$env_file"; then
    log_info "✅ CFN_REDIS_HOST explicitly set in .env"
    annotate "Value: $(grep '^CFN_REDIS_HOST=' "$env_file" | cut -d= -f2)"
  else
    log_info "ℹ️  INFO: CFN_REDIS_HOST not explicitly set (will use docker-compose default)"
  fi

  if grep -q "^CFN_REDIS_PORT=" "$env_file"; then
    log_info "✅ CFN_REDIS_PORT explicitly set in .env"
    annotate "Value: $(grep '^CFN_REDIS_PORT=' "$env_file" | cut -d= -f2)"
  else
    log_info "ℹ️  INFO: CFN_REDIS_PORT not explicitly set (will use docker-compose default)"
  fi

  return 0
}

# Test 7: CLI agent spawn simulation
test_agent_spawn() {
  log_step "Test 7: CLI Agent Spawn Simulation"

  # Create a simple test container that checks Redis connectivity
  log_info "Spawning test agent container..."

  if docker run --rm \
    --name test-redis-agent \
    --network trigger-dev_trigger-cfn-network \
    -e CFN_REDIS_HOST=redis \
    -e CFN_REDIS_PORT=6379 \
    redis:7-alpine \
    sh -c 'redis-cli -h redis -p 6379 PING' > /dev/null 2>&1; then

    log_info "✅ PASS: Agent can connect to Redis service"
    annotate "Container: test-redis-agent"
    annotate "Network: trigger-dev_trigger-cfn-network"
    annotate "Redis endpoint: redis:6379"
    return 0
  else
    log_info "❌ FAIL: Agent cannot connect to Redis service"
    return 1
  fi
}

# Test 8: Task queue operations
test_task_queue() {
  log_step "Test 8: Task Queue Operations"

  # Clear test queue
  redis-cli DEL "cfn:test:queue" > /dev/null 2>&1

  # Push test task
  if redis-cli LPUSH "cfn:test:queue" "test-task-001" "test-task-002" "test-task-003" > /dev/null 2>&1; then
    log_info "✅ PASS: Push to queue successful"
  else
    log_info "❌ FAIL: Cannot push to queue"
    return 1
  fi

  # Check queue length
  local queue_len=$(redis-cli LLEN "cfn:test:queue" 2>/dev/null || echo "0")
  annotate "Queue length: $queue_len (expected: 3)"

  if [ "$queue_len" = "3" ]; then
    log_info "✅ PASS: Queue length verified"
  else
    log_info "❌ FAIL: Queue length mismatch"
    return 1
  fi

  # Test atomic RPOP (agent task claim)
  local claimed_task=$(redis-cli RPOP "cfn:test:queue" 2>/dev/null || echo "")

  if [ "$claimed_task" = "test-task-001" ]; then
    log_info "✅ PASS: Atomic task claim (RPOP) successful"
    annotate "Claimed task: $claimed_task"
  else
    log_info "❌ FAIL: Task claim failed (got: $claimed_task)"
    return 1
  fi

  return 0
}

# Run all tests
echo ""
run_tests=true
failed_tests=0

test_host_redis || ((failed_tests++))
echo ""

test_docker_service || ((failed_tests++))
echo ""

test_docker_network || ((failed_tests++))
echo ""

test_redis_data || ((failed_tests++))
echo ""

test_compose_config || ((failed_tests++))
echo ""

test_env_config || ((failed_tests++))
echo ""

test_agent_spawn || ((failed_tests++))
echo ""

test_task_queue || ((failed_tests++))
echo ""

# Print summary
echo ""
log_step "TEST SUMMARY"
echo ""

if [ $failed_tests -eq 0 ]; then
  log_info "✅ ALL TESTS PASSED"
  log_info ""
  log_info "Redis infrastructure is ready for CFN Loop agent spawning:"
  log_info "  - Host Redis: 127.0.0.1:6379 (OPERATIONAL)"
  log_info "  - Docker Redis Service: redis:6379 (OPERATIONAL)"
  log_info "  - Network: trigger-dev_trigger-cfn-network (OPERATIONAL)"
  log_info "  - Task Queue: FUNCTIONAL"
  echo ""
  exit 0
else
  log_info "⚠️  $failed_tests TEST(S) FAILED"
  log_info ""
  log_info "Review failures above and troubleshoot:"
  log_info "  1. Verify Redis service is running"
  log_info "  2. Check Docker network connectivity"
  log_info "  3. Review docker-compose.yml configuration"
  echo ""
  exit 1
fi