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

########################################
# 🎨 Color Support
########################################
if command -v tput >/dev/null 2>&1 && [ "$(tput colors 2>/dev/null || echo 0)" -ge 8 ]; then
  RED="$(tput setaf 1)"
  GREEN="$(tput setaf 2)"
  YELLOW="$(tput setaf 3)"
  BLUE="$(tput setaf 4)"
  CYAN="$(tput setaf 6)"
  BOLD="$(tput bold)"
  RESET="$(tput sgr0)"
else
  RED=""
  GREEN=""
  YELLOW=""
  BLUE=""
  CYAN=""
  BOLD=""
  RESET=""
fi

info()    { echo "${BLUE}$*${RESET}"; }
ok()      { echo "${GREEN}$*${RESET}"; }
warn()    { echo "${YELLOW}$*${RESET}"; }
err()     { echo "${RED}$*${RESET}" >&2; }
section() { echo; echo "${BOLD}${CYAN}═══════════════════════════════════════${RESET}"; echo "${BOLD}${CYAN}  $*${RESET}"; echo "${BOLD}${CYAN}═══════════════════════════════════════${RESET}"; echo; }
step()    { echo "${BLUE}▶${RESET} ${BOLD}$*${RESET}"; }

########################################
# 🧪 Integration Test Runner
########################################

section "Integration Tests"
step "Reading configuration from .ci/ci_env.toml"

repo_root="$(git rev-parse --show-toplevel)"
CFG_FILE="$repo_root/.ci/ci_env.toml"

if [[ ! -f "$CFG_FILE" ]]; then
  err "❌ ERROR: $CFG_FILE not found"
  exit 1
fi

# Parse TOML config via Python
TOML_STATE="$(python3 - "$CFG_FILE" <<'PY'
import json
import os
import sys

cfg_path = sys.argv[1]

defaults = {
    "engine": "nextest",
    "flags": ["--features", "integration-testing", "--all-features", "--run-ignored", "all"],
    "env": ["MAE_TESTCONTAINERS=1"],
}

cfg = {}
if os.path.exists(cfg_path):
    with open(cfg_path, "rb") as f:
        try:
            import tomllib
            parsed = tomllib.load(f)
        except Exception:
            parsed = {}
    if isinstance(parsed, dict):
        cfg = parsed

engine = cfg.get("engine", defaults["engine"])
flags = cfg.get("flags", defaults["flags"])
env = cfg.get("env", defaults["env"])

if not isinstance(engine, str):
    engine = defaults["engine"]
if not isinstance(flags, list):
    flags = defaults["flags"]
if not isinstance(env, list):
    env = defaults["env"]

flags = [str(x) for x in flags]
env = [str(x) for x in env]

print(json.dumps({"engine": engine.strip(), "flags": flags, "env": env}))
PY
)"

ENGINE="$(python3 -c 'import json,sys;print(json.loads(sys.argv[1])["engine"])' "$TOML_STATE")"
FLAGS="$(python3 -c 'import json,sys;print(" ".join(json.loads(sys.argv[1])["flags"]))' "$TOML_STATE")"
ENV_VARS="$(python3 -c 'import json,sys;print("\n".join(json.loads(sys.argv[1])["env"]))' "$TOML_STATE")"

# Export environment variables
if [[ -n "$ENV_VARS" ]]; then
  step "Exporting environment variables"
  while IFS= read -r kv; do
    [[ -z "$kv" ]] && continue
    if [[ "$kv" != *=* ]]; then
      err "❌ ERROR: invalid env entry '$kv' (expected KEY=VALUE)"
      exit 1
    fi
    export "$kv"
    info "   ${kv%%=*}=${kv#*=}"
  done <<< "$ENV_VARS"
fi

# Build flags array
ARGS=()
if [[ -n "$FLAGS" ]]; then
  # shellcheck disable=SC2206
  ARGS=($FLAGS)
fi

# Run tests based on engine
step "Engine: ${BOLD}${ENGINE}${RESET}"

case "$ENGINE" in
  miri)
    info "▶ Running: cargo miri test ${ARGS[*]}"
    cargo miri test "${ARGS[@]}"
    ;;
  cargo)
    info "▶ Running: cargo test ${ARGS[*]}"
    cargo test "${ARGS[@]}"
    ;;
  nextest)
    info "▶ Running: cargo nextest run ${ARGS[*]}"
    cargo nextest run "${ARGS[@]}"
    ;;
  nothing)
    warn "▶ Skipping tests (engine=nothing)"
    exit 0
    ;;
  *)
    err "❌ ERROR: Unknown test engine '$ENGINE' (expected: miri|cargo|nextest|nothing)"
    exit 1
    ;;
esac

echo
ok "✅  Integration tests completed successfully"