#!/usr/bin/env bash
set -euo pipefail

IMAGE="${1:?Usage: test.sh <image>}"
CONTAINER_NAME="test-$(echo "$IMAGE" | tr ':/' '-')-$$"
PASSED=0
FAILED=0
TOTAL=0

cleanup() {
  docker rm -f "$CONTAINER_NAME" >/dev/null 2>&1 || true
}
trap cleanup EXIT

assert() {
  local name="$1" expected="$2" actual="$3"
  TOTAL=$((TOTAL + 1))
  if [ "$expected" = "$actual" ]; then
    echo "  PASS: $name"
    PASSED=$((PASSED + 1))
  else
    echo "  FAIL: $name (expected: '$expected', got: '$actual')"
    FAILED=$((FAILED + 1))
  fi
}

assert_in() {
  local name="$1" expected="$2" actual="$3"
  TOTAL=$((TOTAL + 1))
  if [ "$actual" -ge "$expected" ] && [ "$actual" -lt 500 ]; then
    echo "  PASS: $name (got: $actual)"
    PASSED=$((PASSED + 1))
  else
    echo "  FAIL: $name (expected < 500, got: $actual)"
    FAILED=$((FAILED + 1))
  fi
}

assert_match() {
  local name="$1" pattern="$2" actual="$3"
  TOTAL=$((TOTAL + 1))
  if echo "$actual" | grep -qE "$pattern"; then
    echo "  PASS: $name"
    PASSED=$((PASSED + 1))
  else
    echo "  FAIL: $name (pattern: '$pattern', got: '$actual')"
    FAILED=$((FAILED + 1))
  fi
}

echo "Running container: $IMAGE"
docker run -d --name "$CONTAINER_NAME" -p 8080:8080 "$IMAGE" >/dev/null

DOCKER_GW=$(docker network inspect bridge --format '{{range .IPAM.Config}}{{.Gateway}}{{end}}')
BASE_URL="http://${DOCKER_GW}:8080"

echo "Waiting for container on ${DOCKER_GW}:8080..."
for i in $(seq 1 30); do
  if curl -sf -o /dev/null "$BASE_URL/" 2>/dev/null || [ "$(curl -s -o /dev/null -w '%{http_code}' "$BASE_URL/" 2>/dev/null)" != "000" ]; then
    echo "Container ready after ${i}s"
    break
  fi
  if [ "$i" -eq 30 ]; then
    echo "FAIL: Container did not become ready within 30s"
    docker logs "$CONTAINER_NAME"
    exit 1
  fi
  sleep 1
done

echo ""
echo "Test: GET / (Apache responds on port 8080)"
RESPONSE=$(curl -s -D - "$BASE_URL/")
STATUS=$(echo "$RESPONSE" | head -1 | grep -oP '\d{3}')
SERVER=$(echo "$RESPONSE" | grep -i '^server:' | tr -d '\r' | cut -d: -f2- | xargs)

assert_in "HTTP status is valid" 200 "$STATUS"
assert_match "Server is Apache" "[Aa]pache" "$SERVER"

echo ""
echo "Results: $PASSED/$TOTAL passed, $FAILED failed"
[ "$FAILED" -eq 0 ]