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

if [[ -z "${PG_DSN:-}" ]]; then
  echo "PG_DSN is required (PostgreSQL connection string)." >&2
  exit 1
fi

MONTHS_AHEAD="${1:-1}"

if ! [[ "${MONTHS_AHEAD}" =~ ^[0-9]+$ ]]; then
  echo "monthsAhead must be a non-negative integer." >&2
  exit 1
fi

psql "${PG_DSN}" \
  --no-psqlrc \
  --set ON_ERROR_STOP=on \
  --quiet \
  --tuples-only \
  --command "SELECT partition_name FROM scanner.ensure_artifact_boms_future_partitions(${MONTHS_AHEAD});"