#!/bin/bash
set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
REGISTRY="$ROOT_DIR/.smbatcher/REGISTRY.md"
BATCH_DIR="$ROOT_DIR/.smbatcher/batches"
SITES_DIR="$ROOT_DIR/sites"
LOCK_FILE="$ROOT_DIR/.smbatcher/REGISTRY.lock"
DEFAULT_LIST="$ROOT_DIR/sites.csv"

VERSION="v1"
BATCH_SIZE=20
INPUT_FILE="$DEFAULT_LIST"

while [[ $# -gt 0 ]]; do
  case "$1" in
    --version)
      VERSION="$2"
      shift 2
      ;;
    --batch-size)
      BATCH_SIZE="$2"
      shift 2
      ;;
    --input-file)
      INPUT_FILE="$2"
      shift 2
      ;;
    *)
      echo "Usage: $0 [--version vX] [--batch-size N] [--input-file PATH]" >&2
      exit 1
      ;;
  esac
done

"$ROOT_DIR/tools/run-python.sh" "$ROOT_DIR/tools/prepare/batch.py" "$ROOT_DIR" "$REGISTRY" "$BATCH_DIR" "$SITES_DIR" "$LOCK_FILE" "$VERSION" "$BATCH_SIZE" "$INPUT_FILE"