#!/bin/bash
# tests/cli-mode/test-coordinator-spawning.sh
# Phase 2 :: Validates cfn-v3-coordinator spawning from /cfn-loop-cli (Priority 2)

set -euo pipefail

PROJECT_ROOT=$(git rev-parse --show-toplevel)
source "$PROJECT_ROOT/tests/test-utils.sh"

# Test counters
PASS_COUNT=0
TOTAL_COUNT=0

pass() { echo "✅ PASS: $1"; PASS_COUNT=$((PASS_COUNT + 1)); TOTAL_COUNT=$((TOTAL_COUNT + 1)); return 0; }
fail() { echo "❌ FAIL: $1"; TOTAL_COUNT=$((TOTAL_COUNT + 1)); return 0; }

cleanup() {
  log_info "Cleanup complete - smoke test only, no processes spawned"
}
trap cleanup EXIT

test_coordinator_agent_exists() {
  log_step "GIVEN cfn-v3-coordinator agent file"

  # WHEN checking if coordinator agent exists
  local coordinator_path="$PROJECT_ROOT/.claude/agents/cfn-dev-team/coordinators/cfn-v3-coordinator.md"

  # THEN file should exist and be readable
  if [[ -f "$coordinator_path" ]]; then
    pass "Coordinator agent file exists"
  else
    fail "Coordinator agent file exists"
    return
  fi

  if [[ -r "$coordinator_path" ]]; then
    pass "Coordinator agent file is readable"
  else
    fail "Coordinator agent file is readable"
  fi

  log_info "✅ Coordinator agent file validation passed"
}

test_coordinator_environment_variables() {
  log_step "GIVEN CLI mode spawning context"

  # WHEN checking required environment variables
  local required_vars=(
    "CFN_DOCKER_MODE"
    "TASK_ID"
    "ITERATION"
    "MODE"
    "TASK_DESCRIPTION"
  )

  # THEN verify environment variable documentation exists
  log_info "Validating required environment variables for coordinator spawning"

  for var in "${required_vars[@]}"; do
    # Check if variable is documented in CLI command
    if grep -q "$var" "$PROJECT_ROOT/.claude/commands/cfn/cfn-loop-cli.md" 2>/dev/null; then
      pass "Environment variable $var documented in CLI command"
    else
      fail "Environment variable $var documented in CLI command"
    fi
  done

  log_info "✅ Environment variable validation passed"
}

test_coordinator_spawning_command() {
  log_step "GIVEN CLI mode execution"

  # WHEN checking coordinator spawning command structure
  local cli_command="$PROJECT_ROOT/.claude/commands/cfn/cfn-loop-cli.md"

  # THEN verify spawning command exists and is correct
  if grep -q "cfn-spawn coordinator" "$cli_command" 2>/dev/null; then
    pass "Coordinator spawning command uses cfn-spawn"
  else
    fail "Coordinator spawning command uses cfn-spawn"
  fi

  if grep -q "cfn-v3-coordinator" "$cli_command" 2>/dev/null; then
    pass "Coordinator spawning command targets cfn-v3-coordinator"
  else
    fail "Coordinator spawning command targets cfn-v3-coordinator"
  fi

  log_info "✅ Coordinator spawning command validation passed"
}

test_coordinator_task_id_format() {
  log_step "GIVEN CLI mode task ID generation"

  # WHEN checking TASK_ID format in CLI command
  local cli_command="$PROJECT_ROOT/.claude/commands/cfn/cfn-loop-cli.md"

  # THEN verify TASK_ID uses proper format
  if grep -q "TASK_ID=\"task-\$(date +%s)\"" "$cli_command" 2>/dev/null; then
    pass "TASK_ID format follows task-* pattern"
  else
    fail "TASK_ID format follows task-* pattern"
  fi

  # Test TASK_ID sanitization logic (from CRITICAL-004 fix)
  local valid_ids=(
    "task-1234567890"
    "test-spawn-agent-001"
    "infra-test-redis"
    "task-coordinator-12345"
  )

  local invalid_ids=(
    "task; rm -rf /"
    "task\$(whoami)"
    "task|cat /etc/passwd"
    "task&background"
  )

  log_info "Testing TASK_ID sanitization patterns"

  for task_id in "${valid_ids[@]}"; do
    if [[ "$task_id" =~ ^[a-zA-Z0-9._-]+$ ]]; then
      pass "Valid TASK_ID accepted: $task_id"
    else
      fail "Valid TASK_ID accepted: $task_id"
    fi
  done

  for task_id in "${invalid_ids[@]}"; do
    if [[ ! "$task_id" =~ ^[a-zA-Z0-9._-]+$ ]]; then
      pass "Invalid TASK_ID rejected: ${task_id//;/SEMICOLON}"
    else
      fail "Invalid TASK_ID rejected: ${task_id//;/SEMICOLON}"
    fi
  done

  log_info "✅ TASK_ID format and sanitization validation passed"
}

test_coordinator_mode_parameter() {
  log_step "GIVEN CLI mode parameter handling"

  # WHEN checking mode parameter options
  local cli_command="$PROJECT_ROOT/.claude/commands/cfn/cfn-loop-cli.md"

  # THEN verify mode parameter is documented
  if grep -q "MODE=" "$cli_command" 2>/dev/null; then
    pass "MODE parameter exists in CLI command"
  else
    fail "MODE parameter exists in CLI command"
  fi

  # Test valid mode values
  local valid_modes=("mvp" "standard" "enterprise")

  for mode in "${valid_modes[@]}"; do
    if grep -q "$mode" "$cli_command" 2>/dev/null; then
      pass "Mode value '$mode' is documented"
    else
      log_info "Mode value '$mode' may not be explicitly documented (acceptable)"
      TOTAL_COUNT=$((TOTAL_COUNT + 1))  # Count as test but don't fail
    fi
  done

  log_info "✅ Mode parameter validation passed"
}

test_cfn_docker_mode_flag() {
  log_step "GIVEN CLI mode vs Task mode detection"

  # WHEN checking CFN_DOCKER_MODE flag
  local cli_command="$PROJECT_ROOT/.claude/commands/cfn/cfn-loop-cli.md"

  # THEN verify CFN_DOCKER_MODE is set for CLI mode
  if grep -q "CFN_DOCKER_MODE=false" "$cli_command" 2>/dev/null; then
    pass "CFN_DOCKER_MODE flag set in CLI command"
  else
    fail "CFN_DOCKER_MODE flag set in CLI command"
  fi

  log_info "✅ CFN_DOCKER_MODE flag validation passed (CLI mode indicator)"
}

# Execute tests
test_coordinator_agent_exists
test_coordinator_environment_variables
test_coordinator_spawning_command
test_coordinator_task_id_format
test_coordinator_mode_parameter
test_cfn_docker_mode_flag

# Test summary
echo ""
log_step "Test Summary"
PASS_RATE=$(awk "BEGIN {printf \"%.0f\", ($PASS_COUNT / $TOTAL_COUNT * 100)}")
echo -e "${GREEN}Total Tests: $TOTAL_COUNT${NC}"
echo -e "${GREEN}Passed: $PASS_COUNT${NC}"
if [[ $TOTAL_COUNT -ne $PASS_COUNT ]]; then
  echo -e "${RED}Failed: $((TOTAL_COUNT - PASS_COUNT))${NC}"
fi
echo -e "${GREEN}Pass Rate: ${PASS_RATE}%${NC}"

if [[ $PASS_COUNT -eq $TOTAL_COUNT ]]; then
  echo ""
  echo -e "${GREEN}✅ All coordinator spawning tests PASSED${NC}"
  echo ""
  log_info "Validation complete: Coordinator spawning configuration is correct"
  log_info "Environment variables properly documented and validated"
  log_info "TASK_ID sanitization prevents command injection (CRITICAL-004 fix)"
  exit 0
else
  echo ""
  echo -e "${RED}❌ Some tests failed${NC}"
  exit 1
fi