set -uo pipefail

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

REPO_ROOT="$(cd "${SCRIPT_DIR}/.." && pwd)"

if [[ -f "${REPO_ROOT}/.env" ]]; then

    set -o allexport

    source "${REPO_ROOT}/.env"

    set +o allexport

fi

GREEN="\033[0;32m"

RED="\033[0;31m"

YELLOW="\033[0;33m"

RESET="\033[0m"

PASS=0

FAIL=0

check() {

    local name="$1"

    local url="$2"

    local expected_pattern="${3:-}"

    local timeout="${4:-5}"

    local http_code

    local body

    body=$(curl -sf --max-time "${timeout}" -w "\n%{http_code}" "${url}" 2>/dev/null) || true

    http_code=$(echo "${body}" | tail -1)

    body=$(echo "${body}" | head -n -1)

    local ok=false

    if [[ "${http_code}" =~ ^2 ]]; then

        if [[ -z "${expected_pattern}" ]] || echo "${body}" | grep -q "${expected_pattern}"; then

            ok=true

        fi

    fi

    if "${ok}"; then

        printf "  ${GREEN}[OK]${RESET}   %-30s %s\n" "${name}" "${url}"

        PASS=$((PASS + 1))

    else

        printf "  ${RED}[FAIL]${RESET} %-30s %s (HTTP ${http_code:-timeout})\n" "${name}" "${url}"

        FAIL=$((FAIL + 1))

    fi

}

check_docker() {

    local name="$1"

    local service="$2"

    local status

    status=$(docker compose --env-file "${REPO_ROOT}/.env" \

        -f "${REPO_ROOT}/docker-compose.yml" \

        ps --format json "${service}" 2>/dev/null \

        | python3 -c "import sys,json; d=json.load(sys.stdin); print(d.get('Health','unknown'))" 2>/dev/null \

        || echo "unknown")

    if [[ "${status}" == "healthy" ]]; then

        printf "  ${GREEN}[OK]${RESET}   %-30s docker:%s (%s)\n" "${name}" "${service}" "${status}"

        PASS=$((PASS + 1))

    else

        printf "  ${RED}[FAIL]${RESET} %-30s docker:%s (%s)\n" "${name}" "${service}" "${status}"

        FAIL=$((FAIL + 1))

    fi

}

if curl -sf --max-time 2 "http://localhost:80/" >/dev/null 2>&1; then

    BASE_HOST="localhost"

else

    BASE_HOST="localhost"

fi

echo ""

echo "DropHawk Service Health Check - $(date -u '+%Y-%m-%dT%H:%M:%SZ')"

echo "─────────────────────────────────────────────────────"

check "Caddy (edge)"          "http://${BASE_HOST}:80/"       ""       5

check "Backend /api/health"   "http://${BASE_HOST}:80/api/health"   "ok"  5

check "Backend /api/ready"    "http://${BASE_HOST}:80/api/ready"    "ok"  5

check "ntfy (via Caddy)"      "http://${BASE_HOST}:80/ntfy/v1/health"  "ok"  5

check "Prometheus /-/healthy" "http://${BASE_HOST}:9090/-/healthy"   "Healthy"  5

check "Grafana /api/health"   "http://${BASE_HOST}:3000/api/health"  "ok"  5

check "Loki /ready"           "http://${BASE_HOST}:3100/ready"        "ready"  5

echo ""

echo "Docker container health states:"

check_docker "postgres"    "postgres"

check_docker "redis"       "redis"

check_docker "nats"        "nats"

check_docker "ntfy"        "ntfy"

check_docker "backend"     "backend"

check_docker "frontend"    "frontend"

check_docker "caddy"       "caddy"

check_docker "prometheus"  "prometheus"

check_docker "grafana"     "grafana"

check_docker "loki"        "loki"

echo ""

echo "─────────────────────────────────────────────────────"

if [[ "${FAIL}" -eq 0 ]]; then

    printf "${GREEN}All ${PASS} checks passed.${RESET}\n\n"

    exit 0

else

    printf "${RED}${FAIL} check(s) failed${RESET}, ${PASS} passed.\n\n"

    exit 1

fi