#!/bin/bash

# Exit immediately if any command fails
set -e

# RTSP Stream Manager
# Get camera device from first argument, default to device 0 if not provided
DEVICE=${1:-0}
# Variables to store process IDs for cleanup
MEDIAMTX_PID=""
FFMPEG_PID=""

# Cleanup function to gracefully stop both services
cleanup() {
    echo "Stopping services..."
    # Kill FFmpeg process if it's running
    if [ ! -z "$FFMPEG_PID" ]; then
        kill -9 $FFMPEG_PID 2>/dev/null
        echo "FFmpeg stopped"
    fi
    # Kill MediaMTX process if it's running
    if [ ! -z "$MEDIAMTX_PID" ]; then
        kill -9 $MEDIAMTX_PID 2>/dev/null
        echo "MediaMTX stopped"
    fi
    exit 0
}

# Set up signal traps to run cleanup on script interruption or exit
trap cleanup SIGINT SIGTERM EXIT

# Start MediaMTX streaming server in background
echo "Starting MediaMTX server..."
mediamtx mediamtx.yml &
# Store the process ID for cleanup
MEDIAMTX_PID=$!

# Give MediaMTX time to initialize before starting FFmpeg
sleep 2

# Display connection information to user
echo "Starting camera stream from device $DEVICE..."
echo "View RTSP: rtsp://localhost:8554/live"
echo "View Web: http://localhost:8889/live"
echo "Press Ctrl+C to stop both services"

# Start FFmpeg to capture camera and stream to MediaMTX
ffmpeg -f avfoundation \           # Use macOS AVFoundation for camera access
       -video_size 1280x720 \      # Set capture resolution
       -framerate 30 \             # Set capture frame rate
       -i "$DEVICE" \              # Input device (camera number)
       -c:v libx264 \              # Use x264 video codec
       -preset ultrafast \         # Use fastest encoding preset (lower CPU usage)
       -tune zerolatency \         # Optimize for low latency streaming
       -pix_fmt yuv420p \          # Set pixel format for compatibility
       -f flv \                    # Output format (FLV for RTMP)
       "rtmp://localhost:1935/live" &  # Send to MediaMTX via RTMP
# Store FFmpeg process ID for cleanup
FFMPEG_PID=$!

# Wait for FFmpeg process to complete or be terminated
wait $FFMPEG_PID

# Minimal MediaMTX config
paths:
  all: {}