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

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
PID_FILE="$ROOT/.tmp/examples-auto-run.pid"
LOG_DIR="$ROOT/.tmp/examples-start-logs"
DEFAULT_UV_EXTRAS="litellm any-llm sqlalchemy redis blaxel modal runloop temporal"

build_uv_prefix() {
  UV_RUN=(uv run)
  local extras_value
  if [[ -n "${EXAMPLES_UV_EXTRAS+x}" ]]; then
    extras_value="$EXAMPLES_UV_EXTRAS"
  else
    extras_value="$DEFAULT_UV_EXTRAS"
  fi

  local extra
  for extra in $extras_value; do
    UV_RUN+=(--extra "$extra")
  done
  export EXAMPLES_UV_EXTRAS="$extras_value"
}

ensure_dirs() {
  mkdir -p "$LOG_DIR" "$ROOT/.tmp"
}

is_running() {
  local pid="$1"
  [[ -n "$pid" ]] && ps -p "$pid" >/dev/null 2>&1
}

cmd_start() {
  ensure_dirs
  local background=0
  if [[ "${1:-}" == "--background" ]]; then
    background=1
    shift
  fi

  local ts main_log stdout_log
  ts="$(date +%Y%m%d-%H%M%S)"
  main_log="$LOG_DIR/main_${ts}.log"
  stdout_log="$LOG_DIR/stdout_${ts}.log"

  build_uv_prefix
  local run_cmd=(
    "${UV_RUN[@]}" examples/run_examples.py
    --auto-mode
    --main-log "$main_log"
    --logs-dir "$LOG_DIR"
  )

  if [[ "$background" -eq 1 ]]; then
    if [[ -f "$PID_FILE" ]]; then
      local pid
      pid="$(cat "$PID_FILE" 2>/dev/null || true)"
      if is_running "$pid"; then
        echo "examples/run_examples.py already running (pid=$pid)."
        exit 1
      fi
    fi
    (
      trap '' HUP
      export EXAMPLES_INTERACTIVE_MODE="${EXAMPLES_INTERACTIVE_MODE:-auto}"
      export APPLY_PATCH_AUTO_APPROVE="${APPLY_PATCH_AUTO_APPROVE:-1}"
      export SHELL_AUTO_APPROVE="${SHELL_AUTO_APPROVE:-1}"
      export AUTO_APPROVE_MCP="${AUTO_APPROVE_MCP:-1}"
      export EXAMPLES_INCLUDE_INTERACTIVE="${EXAMPLES_INCLUDE_INTERACTIVE:-1}"
      export EXAMPLES_INCLUDE_SERVER="${EXAMPLES_INCLUDE_SERVER:-0}"
      export EXAMPLES_INCLUDE_AUDIO="${EXAMPLES_INCLUDE_AUDIO:-0}"
      export EXAMPLES_INCLUDE_EXTERNAL="${EXAMPLES_INCLUDE_EXTERNAL:-0}"
      cd "$ROOT"
      exec "${run_cmd[@]}" "$@" > >(tee "$stdout_log") 2>&1
    ) &
    local pid=$!
    echo "$pid" >"$PID_FILE"
    echo "Started run_examples.py (pid=$pid)"
    echo "Main log: $main_log"
    echo "Stdout log: $stdout_log"
    echo "After the run completes, use examples-run-analysis to inspect its artifacts."
    return 0
  fi

  export EXAMPLES_INTERACTIVE_MODE="${EXAMPLES_INTERACTIVE_MODE:-auto}"
  export APPLY_PATCH_AUTO_APPROVE="${APPLY_PATCH_AUTO_APPROVE:-1}"
  export SHELL_AUTO_APPROVE="${SHELL_AUTO_APPROVE:-1}"
  export AUTO_APPROVE_MCP="${AUTO_APPROVE_MCP:-1}"
  export EXAMPLES_INCLUDE_INTERACTIVE="${EXAMPLES_INCLUDE_INTERACTIVE:-1}"
  export EXAMPLES_INCLUDE_SERVER="${EXAMPLES_INCLUDE_SERVER:-0}"
  export EXAMPLES_INCLUDE_AUDIO="${EXAMPLES_INCLUDE_AUDIO:-0}"
  export EXAMPLES_INCLUDE_EXTERNAL="${EXAMPLES_INCLUDE_EXTERNAL:-0}"
  cd "$ROOT"
  set +e
  "${run_cmd[@]}" "$@" 2>&1 | tee "$stdout_log"
  local run_status=${PIPESTATUS[0]}
  set -e
  return "$run_status"
}

cmd_stop() {
  if [[ ! -f "$PID_FILE" ]]; then
    echo "No pid file; nothing to stop."
    return 0
  fi
  local pid
  pid="$(cat "$PID_FILE" 2>/dev/null || true)"
  if [[ -z "$pid" ]]; then
    rm -f "$PID_FILE"
    echo "Pid file empty; cleaned."
    return 0
  fi
  if ! is_running "$pid"; then
    rm -f "$PID_FILE"
    echo "Process $pid not running; cleaned pid file."
    return 0
  fi
  echo "Stopping pid $pid ..."
  kill "$pid" 2>/dev/null || true
  sleep 1
  if is_running "$pid"; then
    echo "Sending SIGKILL to $pid ..."
    kill -9 "$pid" 2>/dev/null || true
  fi
  rm -f "$PID_FILE"
  echo "Stopped."
}

cmd_status() {
  if [[ -f "$PID_FILE" ]]; then
    local pid
    pid="$(cat "$PID_FILE" 2>/dev/null || true)"
    if is_running "$pid"; then
      echo "Running (pid=$pid)"
      return 0
    fi
  fi
  echo "Not running."
}

cmd_logs() {
  ensure_dirs
  ls -1t "$LOG_DIR"
}

cmd_tail() {
  ensure_dirs
  local file="${1:-}"
  if [[ -z "$file" ]]; then
    file="$(ls -1t "$LOG_DIR" | head -n1)"
  fi
  if [[ -z "$file" ]]; then
    echo "No log files yet."
    exit 1
  fi
  tail -f "$LOG_DIR/$file"
}

usage() {
  cat <<'EOF'
Usage: run_examples.sh <start|stop|status|logs|tail> [args...]

Commands:
  start [--filter ... | other args]   Run examples in auto mode (foreground). Pass --background to run detached.
  stop                                Kill the running examples job (if any).
  status                              Show whether an examples job is running.
  logs                                List log files (.tmp/examples-start-logs).
  tail [logfile]                      Tail the latest (or specified) log.

Environment overrides:
  EXAMPLES_INTERACTIVE_MODE (default auto)
  EXAMPLES_INCLUDE_SERVER/INTERACTIVE/AUDIO/EXTERNAL (defaults: 0/1/0/0)
  EXAMPLES_UV_EXTRAS (default: litellm any-llm sqlalchemy redis blaxel modal runloop temporal; set empty to disable)
  APPLY_PATCH_AUTO_APPROVE, SHELL_AUTO_APPROVE, AUTO_APPROVE_MCP (default 1 in auto mode)
EOF
}

case "${1:-start}" in
  start) shift || true; cmd_start "$@" ;;
  stop) shift || true; cmd_stop ;;
  status) shift || true; cmd_status ;;
  logs) shift || true; cmd_logs ;;
  tail) shift; cmd_tail "${1:-}" ;;
  help | --help | -h) usage ;;
  *) usage; exit 1 ;;
esac