#!/usr/bin/env bash
#
# 로컬 부하테스트(locust) 실행 (대화형). 실행하면 파일/시드/방식을 골라 입력한다.
# loadtest/ 아래 locust 파일이 늘어나도 목록에서 선택만 하면 된다.
#
set -euo pipefail
cd "$(dirname "$0")"   # backend/

VENV=".venv"
LOCUST="$VENV/bin/locust"
HOST="${LOCUST_HOST:-http://localhost:9300}"
# 시드용 DB 접속 (local 기본값, 환경변수로 override 가능)
PGHOST="${PGHOST:-127.0.0.1}"
PGPORT="${PGPORT:-5432}"
PGUSER="${PGUSER:-postgres}"
PGDATABASE="${PGDATABASE:-negosium_db}"

# locust 설치 보장
if [[ ! -x "$LOCUST" ]]; then
  echo "[setup] locust 설치..."
  "$VENV/bin/python" -m pip install -q locust
fi

# 1) locust 파일 선택
FILES=()
while IFS= read -r f; do
  FILES+=("$f")
done < <(ls -1 loadtest/*.py 2>/dev/null | grep -v __pycache__ || true)
if [[ ${#FILES[@]} -eq 0 ]]; then
  echo "[error] loadtest/ 에 locust 파일이 없습니다."
  exit 1
fi
echo "── locust 파일 선택 ──"
i=1
for f in "${FILES[@]}"; do
  echo "  $i) ${f#loadtest/}"
  i=$((i + 1))
done
read -rp "선택 [1]: " fsel
fsel="${fsel:-1}"
FILE="${FILES[$((fsel - 1))]:-}"
if [[ -z "$FILE" ]]; then
  echo "[error] 잘못된 선택: $fsel"
  exit 1
fi

# 2) 부하용 공급사 1건 보장 → supplier_id 를 LOAD_SUPPLIER_ID 로 넘긴다.
#    유저는 사전 시드하지 않고 locust on_start 에서 self-register 한다(tbl 방식).
#    (없으면 만들고, 있으면 그대로 사용 = get-or-create)
LOAD_SUPPLIER_ID="$(psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d "$PGDATABASE" -t -A -c "
WITH ex AS (
  SELECT supplier_id FROM partner.suppliers WHERE name='부하테스트공급사' AND deleted=false LIMIT 1
), ins AS (
  INSERT INTO partner.suppliers (supplier_id, company_id, user_id, name)
  SELECT gen_random_uuid(), gen_random_uuid(), gen_random_uuid(), '부하테스트공급사'
  WHERE NOT EXISTS (SELECT 1 FROM ex)
  RETURNING supplier_id
)
SELECT supplier_id FROM ins UNION ALL SELECT supplier_id FROM ex LIMIT 1;
" 2>/dev/null | tr -d '[:space:]')"
if [[ -z "$LOAD_SUPPLIER_ID" ]]; then
  echo "[error] 부하용 공급사 supplier_id 를 확보하지 못했습니다 (DB 접속/스키마 확인)."
  exit 1
fi

# 이전 실행에서 self-register 로 쌓인 계정/토큰 정리(누적 방지). 공급사 행은 재사용하므로 남긴다.
psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d "$PGDATABASE" -q \
  -c "DELETE FROM supplier.supplier_user_tokens WHERE su_id IN (SELECT su_id FROM supplier.supplier_users WHERE id LIKE 'load_user_%');
      DELETE FROM supplier.supplier_users WHERE id LIKE 'load_user_%';" >/dev/null 2>&1 || true
echo "[info] 부하용 공급사 supplier_id=$LOAD_SUPPLIER_ID (유저는 self-register, 이전 부하계정 정리됨)"

# 3) 실행 방식 선택
echo "── 실행 방식 ──"
echo "  1) 웹 UI (브라우저에서 사용자 수 조절, http://localhost:8089)"
echo "  2) 헤드리스 (값 입력)"
read -rp "선택 [1]: " mode
mode="${mode:-1}"

if [[ "$mode" == "2" ]]; then
  read -rp "동시 사용자 수 [50]: " VU;   VU="${VU:-50}"
  read -rp "초당 증가 수 [10]: " RATE;   RATE="${RATE:-10}"
  read -rp "지속 시간(예 30s/2m) [1m]: " DUR; DUR="${DUR:-1m}"
  echo "[run] $FILE  headless -u $VU -r $RATE -t $DUR  (host=$HOST)"
  exec env LOAD_SUPPLIER_ID="$LOAD_SUPPLIER_ID" "$LOCUST" -f "$FILE" --host "$HOST" --headless -u "$VU" -r "$RATE" -t "$DUR"
else
  echo "[run] $FILE  웹 UI → http://localhost:8089  (host=$HOST)"
  exec env LOAD_SUPPLIER_ID="$LOAD_SUPPLIER_ID" "$LOCUST" -f "$FILE" --host "$HOST"
fi