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

# Trace probe script for Restreamer Core
# Usage: ./scripts/trace_probe.sh <CHANNEL_ID> <SOURCE_URL>
# Reads REACT_APP_CORE_URL from .env if present. Optionally set API_TOKEN env var.

# load .env if exists (simple parser)
if [ -f .env ]; then
  export $(grep -v '^#' .env | xargs)
fi

CORE_URL=${REACT_APP_CORE_URL:-http://localhost:8080}
API_TOKEN=${API_TOKEN:-}

if [ "$#" -lt 2 ]; then
  echo "Usage: $0 <CHANNEL_ID> <SOURCE_URL>"
  echo "Example: $0 channel1 http://example.com/stream.m3u8"
  exit 2
fi

CHANNEL_ID=$1
SOURCE_URL=$2
PROBE_ID="${CHANNEL_ID}_probe_$(date +%s)"

# Build JSON payload for the probe process
read -r -d '' PAYLOAD <<EOF
{
  "type": "ffmpeg",
  "id": "${PROBE_ID}",
  "reference": "${CHANNEL_ID}",
  "input": [
    {
      "id": "input_0",
      "address": "${SOURCE_URL}",
      "options": ["-fflags", "+genpts", "-thread_queue_size", "512", "-probesize", "5000000", "-analyzeduration", "20000000", "-re"]
    }
  ],
  "output": [
    {
      "id": "output_0",
      "address": "-",
      "options": ["-dn", "-sn", "-codec", "copy", "-f", "null"]
    }
  ],
  "options": [],
  "autostart": false,
  "reconnect": false
}
EOF

AUTH_HEADER=()
if [ -n "$API_TOKEN" ]; then
  AUTH_HEADER+=( -H "Authorization: Bearer ${API_TOKEN}" )
fi

echo "Core URL: ${CORE_URL}"
echo "Probe ID: ${PROBE_ID}"

echo "1) Creating probe process..."
curl -s -X POST "${CORE_URL}/api/v3/process" -H "Content-Type: application/json" "${AUTH_HEADER[@]}" -d "$PAYLOAD" | jq '.' || true

echo "\n2) Running probe (this may take a few seconds)..."
# Probe endpoint returns JSON result when finished
curl -s "${CORE_URL}/api/v3/process/${PROBE_ID}/probe" "${AUTH_HEADER[@]}" | jq '.' || true

echo "\n3) Deleting probe process..."
curl -s -X DELETE "${CORE_URL}/api/v3/process/${PROBE_ID}" "${AUTH_HEADER[@]}" | jq '.' || true

echo "\nDone. Check Core logs or /api/v3/process/<id>/report if you need more details."