#!/usr/bin/env bash
# Boot BEAM 10M ingest inside screen. Keep this file free of long uvicorn/celery
# command strings in the screen argv — use this script path only:
#   screen -dmS beam10m /path/to/boot_beam_10m_screen.sh
set +e
SRC="${BRAINAPI_SRC:-$HOME/.brainapi/source}"
UV_LOG=/tmp/uvicorn-beam10m.log
CEL_LOG=/tmp/brainapi-beam-worker.log
UV_PID=/tmp/beam10m-uvicorn.pid
CEL_PID=/tmp/beam10m-celery.pid
ROOT="$(cd "$(dirname "$0")/../.." && pwd)"

start_api() {
  if lsof -iTCP:8000 -sTCP:LISTEN >/dev/null 2>&1; then
    return 0
  fi
  nohup bash -c "cd '$SRC' && exec ./.venv/bin/uvicorn src.services.api.app:app --host 127.0.0.1 --port 8000" \
    >> "$UV_LOG" 2>&1 &
  echo $! > "$UV_PID"
  sleep 5
}

start_worker() {
  if pgrep -f 'celery -A src.workers.app worker' >/dev/null 2>&1; then
    return 0
  fi
  CELERY_C="${CELERY_WORKER_CONCURRENCY:-4}"
  nohup bash -c "cd '$SRC' && exec ./.venv/bin/celery -A src.workers.app worker --loglevel=info --pool=threads --concurrency=${CELERY_C} -Q ingest_data,finalize_ingestion,process_architect_relationships,consolidate_graph,chatbot_memory,ingest_file,ingest_structured_data" \
    >> "$CEL_LOG" 2>&1 &
  echo $! > "$CEL_PID"
  sleep 5
}

start_api
start_worker
exec bash "$ROOT/benchmarks/scripts/run_beam_10m_1_ingest_eval.sh" >> /tmp/beam-10m-1-run.log 2>&1