#!/usr/bin/env bash
# STATUS: LIVE_VERIFIED 2026-09-18 — same sequence run via scripts/live.py (create → retrieve → list → poll → results → delete; cancel path).
# Model: claude-haiku-4-5-20251001, 2 requests of 8 max_tokens (~$0.0001 at batch prices).
# Usage: source scripts/lib.sh && bash examples/anthropic/batch/batch_lifecycle.sh
# Requires: jq
set -eo pipefail
cd "$(dirname "$0")/../../.."
# shellcheck disable=SC1091
source scripts/lib.sh
OUTF=$(mktemp); trap 'rm -f "$OUTF"' EXIT
call() { ant "$@" > "$OUTF"; OUT="$(cat "$OUTF")"; }   # runs a request, keeps body in $OUT and status in $ANT_STATUS

MODEL="claude-haiku-4-5-20251001"
body=$(cat <<JSON
{"requests":[
 {"custom_id":"ok-1","params":{"model":"$MODEL","max_tokens":8,"messages":[{"role":"user","content":"Reply with OK."}]}},
 {"custom_id":"bad-model-1","params":{"model":"claude-does-not-exist-1","max_tokens":8,"messages":[{"role":"user","content":"Reply with OK."}]}}
]}
JSON
)

# 1. create
created=$(ant POST /v1/messages/batches -d "$body"); log_request anthropic POST /v1/messages/batches "$ANT_STATUS" 0.0001 "batch example create"
bid=$(echo "$created" | jq -r .id); echo "created $bid status=$(echo "$created" | jq -r .processing_status)"

# 2. retrieve + list
ant GET "/v1/messages/batches/$bid" | jq -c '{processing_status, request_counts}'; log_request anthropic GET "/v1/messages/batches/{id}" "$ANT_STATUS" 0 "batch example retrieve"
ant GET "/v1/messages/batches?limit=1" | jq -c '{n: (.data|length), has_more, first_id}'; log_request anthropic GET /v1/messages/batches "$ANT_STATUS" 0 "batch example list"

# 3. poll until ended (≤ 5 min, every 20 s). Observed: ~100 s for 2 requests.
for _ in $(seq 1 15); do
  st=$(ant GET "/v1/messages/batches/$bid" | jq -r .processing_status); log_request anthropic GET "/v1/messages/batches/{id}" "$ANT_STATUS" 0 "batch example poll"
  echo "  processing_status=$st"; [ "$st" = "ended" ] && break; sleep 20
done

# 4. results (.jsonl — one line per request, unordered; match on custom_id)
ant GET "/v1/messages/batches/$bid/results" | while IFS= read -r line; do
  echo "$line" | jq -c '{custom_id, type: .result.type, text: (.result.message.content[0].text // null), error: (.result.error.error.message // null)}'
done; log_request anthropic GET "/v1/messages/batches/{id}/results" "$ANT_STATUS" 0 "batch example results"

# 5. delete (only allowed once ended; 400 while in_progress/canceling)
ant DELETE "/v1/messages/batches/$bid" | jq -c .; log_request anthropic DELETE "/v1/messages/batches/{id}" "$ANT_STATUS" 0 "batch example delete"

# 6. cancel path: create a second batch and cancel immediately → canceling → ended with request_counts.canceled=1
c2=$(ant POST /v1/messages/batches -d "{\"requests\":[{\"custom_id\":\"cancel-1\",\"params\":{\"model\":\"$MODEL\",\"max_tokens\":8,\"messages\":[{\"role\":\"user\",\"content\":\"Reply with OK.\"}]}}]}")
log_request anthropic POST /v1/messages/batches "$ANT_STATUS" 0.0001 "batch example create #2"
bid2=$(echo "$c2" | jq -r .id)
ant POST "/v1/messages/batches/$bid2/cancel" | jq -c '{processing_status, cancel_initiated_at}'; log_request anthropic POST "/v1/messages/batches/{id}/cancel" "$ANT_STATUS" 0 "batch example cancel"
for _ in $(seq 1 12); do
  st=$(ant GET "/v1/messages/batches/$bid2" | jq -r .processing_status); log_request anthropic GET "/v1/messages/batches/{id}" "$ANT_STATUS" 0 "batch example poll #2"
  [ "$st" = "ended" ] && break; sleep 15
done
ant GET "/v1/messages/batches/$bid2" | jq -c .request_counts
ant DELETE "/v1/messages/batches/$bid2" | jq -c .; log_request anthropic DELETE "/v1/messages/batches/{id}" "$ANT_STATUS" 0 "batch example delete #2"