#!/bin/bash
set -e

echo "[START] DATA SYNC STARTING — cwd=$(pwd) force=${FORCE_DATA_DOWNLOAD:-0}"

python3 - <<'PYEOF'
import os, sys
from pathlib import Path

MIN_SIZES = {
    "vectordb/index.faiss":     100 * 1024 * 1024,
    "data/chunks/chunks.jsonl":  50 * 1024 * 1024,
}

force  = os.getenv("FORCE_DATA_DOWNLOAD", "0") == "1"
s3_bucket = os.getenv("S3_DATA_BUCKET", "")  # set in AWS ECS task definition

# ── AWS S3 (production) ──────────────────────────────────────────────────────
if s3_bucket:
    import boto3
    s3 = boto3.client("s3")
    for dest_str, min_size in MIN_SIZES.items():
        dest = Path(dest_str)
        size = dest.stat().st_size if dest.exists() else 0
        print(f"[DATA] {dest}: exists={dest.exists()}, size={size/1e6:.1f}MB", flush=True)
        if force or not dest.exists() or size < min_size:
            dest.parent.mkdir(parents=True, exist_ok=True)
            print(f"[DATA] Downloading s3://{s3_bucket}/{dest_str} ...", flush=True)
            s3.download_file(s3_bucket, dest_str, str(dest))
            print(f"[DATA] Done: {dest} ({dest.stat().st_size/1e6:.1f}MB)", flush=True)
        else:
            print(f"[DATA] File OK (S3 already synced): {dest}", flush=True)

# ── Google Drive fallback (local dev / Render) ───────────────────────────────
else:
    GDRIVE_FILES = {
        "vectordb/index.faiss":     "1u4EjWwNRz-tJaI8ifKPvpXxscNGiZ68V",
        "data/chunks/chunks.jsonl": "1fehfdhPCh3jxc3TBWitAfb0dv8I1opUM",
    }
    import gdown
    for path_str, file_id in GDRIVE_FILES.items():
        dest = Path(path_str)
        min_size = MIN_SIZES[path_str]
        size = dest.stat().st_size if dest.exists() else 0
        print(f"[DATA] {dest}: exists={dest.exists()}, size={size/1e6:.1f}MB", flush=True)
        if force or not dest.exists() or size < min_size:
            dest.parent.mkdir(parents=True, exist_ok=True)
            print(f"[DATA] Downloading {dest} via gdown...", flush=True)
            gdown.download(id=file_id, output=str(dest), quiet=False, fuzzy=True)
            print(f"[DATA] Done: {dest} ({dest.stat().st_size/1e6:.1f}MB)", flush=True)
        else:
            print(f"[DATA] File OK: {dest}", flush=True)
PYEOF

echo "[START] DATA SYNC COMPLETE — starting gunicorn"

# Worker count: default 1. Each gunicorn worker independently loads the
# embedding model (~1.3 GB) + FAISS index (~230 MB) + FlashRank (~22 MB)
# = ~1.85 GB baseline per worker. FastAPI/uvicorn handles concurrent requests
# async within a single worker, so 1 worker per task is correct.
# 4 ECS tasks × 1 worker = 4 independent async servers, sufficient for production.
# Override via WORKERS env var in the ECS task definition only if needed.
WORKERS=${WORKERS:-1}

exec gunicorn main:app \
  --worker-class uvicorn.workers.UvicornWorker \
  --workers "${WORKERS}" \
  --bind "0.0.0.0:${PORT:-8080}" \
  --timeout 180 \
  --graceful-timeout 180 \
  --keep-alive 5 \
  --max-requests 1000 \
  --max-requests-jitter 100 \
  --log-level info \
  --access-logfile - \
  --error-logfile -