#!/usr/bin/env bash

set -euo pipefail

# 先编译并拉起临时 api-server 进程，再校验 /healthz 的裸响应、envelope 和关键响应头契约。

usage() {
  cat <<'EOF'
用法:
  ./server-rs/scripts/smoke.sh
  GENARRATIVE_SMOKE_PORT=3201 ./server-rs/scripts/smoke.sh

说明:
  1. 先执行 `cargo build -p api-server`
  2. 启动一个临时 `api-server` 进程并等待 `/healthz` 就绪
  3. 校验 raw `/healthz`、envelope `/healthz` 以及 `x-request-id / x-api-version / x-route-version / x-response-time-ms`
  4. 冒烟结束后自动回收临时进程
EOF
}

if [[ "${1:-}" == "-h" || "${1:-}" == "--help" ]]; then
  usage
  exit 0
fi

SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
SERVER_RS_DIR="$(cd -- "${SCRIPT_DIR}/.." && pwd)"
MANIFEST_PATH="${SERVER_RS_DIR}/Cargo.toml"
SERVER_BIN="${SERVER_RS_DIR}/target/debug/api-server"
API_HOST="${GENARRATIVE_SMOKE_HOST:-127.0.0.1}"
API_PORT="${GENARRATIVE_SMOKE_PORT:-3101}"
API_LOG="${GENARRATIVE_SMOKE_LOG:-warn,tower_http=warn}"
STARTUP_TIMEOUT_SECONDS="${GENARRATIVE_SMOKE_TIMEOUT_SECONDS:-30}"
BASE_URL="http://${API_HOST}:${API_PORT}"
HEALTHZ_URL="${BASE_URL}/healthz"
RUN_ID="$(date +%s)-$$"
STDOUT_LOG="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.stdout.log"
STDERR_LOG="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.stderr.log"
KEEP_LOGS=0
SERVER_PID=""

assert_contains() {
  local haystack="$1"
  local needle="$2"
  local message="$3"

  if [[ "${haystack}" != *"${needle}"* ]]; then
    echo "[server-rs:smoke] ${message}" >&2
    exit 1
  fi
}

print_logs() {
  if [[ -f "${STDOUT_LOG}" ]]; then
    echo "[server-rs:smoke] stdout:"
    cat "${STDOUT_LOG}"
  fi

  if [[ -f "${STDERR_LOG}" ]]; then
    echo "[server-rs:smoke] stderr:"
    cat "${STDERR_LOG}"
  fi
}

cleanup() {
  if [[ -n "${SERVER_PID}" ]] && kill -0 "${SERVER_PID}" 2>/dev/null; then
    kill "${SERVER_PID}" 2>/dev/null || true
    wait "${SERVER_PID}" 2>/dev/null || true
  fi

  if [[ "${KEEP_LOGS}" -eq 0 ]]; then
    rm -f "${STDOUT_LOG}" "${STDERR_LOG}"
  fi
}

trap cleanup EXIT

if [[ ! -f "${MANIFEST_PATH}" ]]; then
  echo "[server-rs:smoke] 未找到 ${MANIFEST_PATH}，无法启动冒烟脚本。" >&2
  exit 1
fi

cd "${SERVER_RS_DIR}"

echo "[server-rs:smoke] 步骤: cargo build -p api-server"
cargo build -p api-server --manifest-path "${MANIFEST_PATH}"

if [[ ! -x "${SERVER_BIN}" ]]; then
  echo "[server-rs:smoke] 未找到 ${SERVER_BIN}，构建结果异常。" >&2
  exit 1
fi

echo "[server-rs:smoke] 步骤: start api-server binary"
GENARRATIVE_API_HOST="${API_HOST}" \
GENARRATIVE_API_PORT="${API_PORT}" \
GENARRATIVE_API_LOG="${API_LOG}" \
"${SERVER_BIN}" >"${STDOUT_LOG}" 2>"${STDERR_LOG}" &
SERVER_PID=$!

echo "[server-rs:smoke] 步骤: wait for /healthz readiness"
READY=0
for ((i = 0; i < STARTUP_TIMEOUT_SECONDS * 10; i++)); do
  if ! kill -0 "${SERVER_PID}" 2>/dev/null; then
    KEEP_LOGS=1
    print_logs
    echo "[server-rs:smoke] api-server exited before /healthz became ready." >&2
    exit 1
  fi

  if curl -fsS "${HEALTHZ_URL}" >/dev/null 2>&1; then
    READY=1
    break
  fi

  sleep 0.1
done

if [[ "${READY}" -ne 1 ]]; then
  KEEP_LOGS=1
  print_logs
  echo "[server-rs:smoke] /healthz readiness timed out." >&2
  exit 1
fi

RAW_BODY_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.raw.body.json"
RAW_HEADER_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.raw.headers.txt"
ENVELOPE_BODY_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.envelope.body.json"
ENVELOPE_HEADER_FILE="${TMPDIR:-/tmp}/genarrative-server-rs-smoke-${RUN_ID}.envelope.headers.txt"

trap 'rm -f "${RAW_BODY_FILE}" "${RAW_HEADER_FILE}" "${ENVELOPE_BODY_FILE}" "${ENVELOPE_HEADER_FILE}"; cleanup' EXIT

RAW_REQUEST_ID="smoke-healthz-raw"
echo "[server-rs:smoke] 步骤: verify raw /healthz contract"
curl -fsS \
  -H "x-request-id: ${RAW_REQUEST_ID}" \
  -D "${RAW_HEADER_FILE}" \
  -o "${RAW_BODY_FILE}" \
  "${HEALTHZ_URL}"

RAW_BODY="$(tr -d '\n' <"${RAW_BODY_FILE}")"
RAW_REQUEST_ID_HEADER="$(grep -i '^x-request-id:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
API_VERSION_HEADER="$(grep -i '^x-api-version:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
ROUTE_VERSION_HEADER="$(grep -i '^x-route-version:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"
RESPONSE_TIME_HEADER="$(grep -i '^x-response-time-ms:' "${RAW_HEADER_FILE}" | tail -n 1 | cut -d':' -f2- | tr -d '\r' | xargs)"

assert_contains "${RAW_BODY}" '"ok":true' "raw /healthz body 缺少 ok=true。"
assert_contains "${RAW_BODY}" '"service":"genarrative-node-server"' "raw /healthz body 返回了意外的 service。"

if [[ "${RAW_REQUEST_ID_HEADER}" != "${RAW_REQUEST_ID}" ]]; then
  echo "[server-rs:smoke] raw /healthz 没有正确回写 x-request-id。" >&2
  exit 1
fi

if [[ -z "${API_VERSION_HEADER}" ]]; then
  echo "[server-rs:smoke] raw /healthz 缺少 x-api-version。" >&2
  exit 1
fi

if [[ "${ROUTE_VERSION_HEADER}" != "${API_VERSION_HEADER}" ]]; then
  echo "[server-rs:smoke] raw /healthz 的 x-route-version 与 x-api-version 不一致。" >&2
  exit 1
fi

if [[ ! "${RESPONSE_TIME_HEADER}" =~ ^[0-9]+$ ]]; then
  echo "[server-rs:smoke] raw /healthz 的 x-response-time-ms 不是合法整数。" >&2
  exit 1
fi

ENVELOPE_REQUEST_ID="smoke-healthz-envelope"
echo "[server-rs:smoke] 步骤: verify envelope /healthz contract"
curl -fsS \
  -H "x-request-id: ${ENVELOPE_REQUEST_ID}" \
  -H "x-genarrative-response-envelope: 1" \
  -D "${ENVELOPE_HEADER_FILE}" \
  -o "${ENVELOPE_BODY_FILE}" \
  "${HEALTHZ_URL}"

ENVELOPE_BODY="$(tr -d '\n' <"${ENVELOPE_BODY_FILE}")"

assert_contains "${ENVELOPE_BODY}" '"ok":true' "envelope /healthz body 缺少 ok=true。"
assert_contains "${ENVELOPE_BODY}" '"error":null' "envelope /healthz body 缺少 error=null。"
assert_contains "${ENVELOPE_BODY}" '"data":{"ok":true,"service":"genarrative-node-server"}' "envelope /healthz data 返回异常。"
assert_contains "${ENVELOPE_BODY}" "\"requestId\":\"${ENVELOPE_REQUEST_ID}\"" "envelope /healthz meta.requestId 没有回写请求 ID。"
assert_contains "${ENVELOPE_BODY}" "\"apiVersion\":\"${API_VERSION_HEADER}\"" "envelope /healthz meta.apiVersion 与响应头不一致。"
assert_contains "${ENVELOPE_BODY}" "\"routeVersion\":\"${ROUTE_VERSION_HEADER}\"" "envelope /healthz meta.routeVersion 与响应头不一致。"
assert_contains "${ENVELOPE_BODY}" '"operation":"GET /healthz"' "envelope /healthz meta.operation 异常。"

echo "[server-rs:smoke] all checks passed"