|
| 1 | +#!/usr/bin/env bash |
| 2 | + |
| 3 | +set -euo pipefail |
| 4 | + |
| 5 | +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
| 6 | +MODULE_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)" |
| 7 | +MM_DIR="$(cd "${MODULE_DIR}/../.." && pwd)" |
| 8 | + |
| 9 | +MEDIAMTX_VERSION="${MEDIAMTX_VERSION:-v1.13.1}" |
| 10 | +UNAME_ARCH="$(uname -m)" |
| 11 | + |
| 12 | +case "${UNAME_ARCH}" in |
| 13 | + x86_64) |
| 14 | + DEFAULT_MEDIAMTX_ARCH="linux_amd64" |
| 15 | + ;; |
| 16 | + aarch64|arm64) |
| 17 | + DEFAULT_MEDIAMTX_ARCH="linux_arm64v8" |
| 18 | + ;; |
| 19 | + armv7l|armv7*) |
| 20 | + DEFAULT_MEDIAMTX_ARCH="linux_armv7" |
| 21 | + ;; |
| 22 | + armv6l|armv6*) |
| 23 | + DEFAULT_MEDIAMTX_ARCH="linux_armv6" |
| 24 | + ;; |
| 25 | + *) |
| 26 | + DEFAULT_MEDIAMTX_ARCH="linux_amd64" |
| 27 | + ;; |
| 28 | +esac |
| 29 | + |
| 30 | +MEDIAMTX_ARCH="${MEDIAMTX_ARCH:-${DEFAULT_MEDIAMTX_ARCH}}" |
| 31 | +MEDIAMTX_ARCHIVE="${SCRIPT_DIR}/mediamtx.tar.gz" |
| 32 | +MEDIAMTX_DIR="${MODULE_DIR}/mediamtx" |
| 33 | +MEDIAMTX_BIN="${MEDIAMTX_DIR}/mediamtx" |
| 34 | +MEDIAMTX_CFG_SRC="${SCRIPT_DIR}/mediamtx.yml" |
| 35 | +MEDIAMTX_CFG_DST="${MEDIAMTX_DIR}/mediamtx.yml" |
| 36 | + |
| 37 | +FFMPEG_PID="" |
| 38 | +MEDIAMTX_PID="" |
| 39 | +STARTED_MEDIAMTX="0" |
| 40 | + |
| 41 | +require_cmd () { |
| 42 | + if ! command -v "$1" >/dev/null 2>&1; then |
| 43 | + echo "Missing required command: $1" >&2 |
| 44 | + exit 1 |
| 45 | + fi |
| 46 | +} |
| 47 | + |
| 48 | +wait_for_http () { |
| 49 | + local url="$1" |
| 50 | + local attempts="${2:-30}" |
| 51 | + local sleep_s="${3:-0.5}" |
| 52 | + local i |
| 53 | + |
| 54 | + for ((i = 1; i <= attempts; i++)); do |
| 55 | + if curl -fsS --max-time 2 "$url" >/dev/null 2>&1; then |
| 56 | + return 0 |
| 57 | + fi |
| 58 | + sleep "$sleep_s" |
| 59 | + done |
| 60 | + |
| 61 | + return 1 |
| 62 | +} |
| 63 | + |
| 64 | +cleanup () { |
| 65 | + set +e |
| 66 | + |
| 67 | + if [[ -n "${FFMPEG_PID}" ]]; then |
| 68 | + kill "${FFMPEG_PID}" >/dev/null 2>&1 || true |
| 69 | + wait "${FFMPEG_PID}" >/dev/null 2>&1 || true |
| 70 | + fi |
| 71 | + |
| 72 | + if [[ "${STARTED_MEDIAMTX}" == "1" && -n "${MEDIAMTX_PID}" ]]; then |
| 73 | + kill "${MEDIAMTX_PID}" >/dev/null 2>&1 || true |
| 74 | + wait "${MEDIAMTX_PID}" >/dev/null 2>&1 || true |
| 75 | + fi |
| 76 | +} |
| 77 | + |
| 78 | +trap cleanup EXIT INT TERM |
| 79 | + |
| 80 | +require_cmd curl |
| 81 | +require_cmd tar |
| 82 | +require_cmd ffmpeg |
| 83 | + |
| 84 | +mkdir -p "${MEDIAMTX_DIR}" |
| 85 | + |
| 86 | +if [[ ! -x "${MEDIAMTX_BIN}" ]]; then |
| 87 | + if [[ ! -f "${MEDIAMTX_ARCHIVE}" ]]; then |
| 88 | + echo "Downloading MediaMTX ${MEDIAMTX_VERSION} (${MEDIAMTX_ARCH})..." |
| 89 | + curl -L -o "${MEDIAMTX_ARCHIVE}" "https://github.com/bluenviron/mediamtx/releases/download/${MEDIAMTX_VERSION}/mediamtx_${MEDIAMTX_VERSION}_${MEDIAMTX_ARCH}.tar.gz" |
| 90 | + fi |
| 91 | + |
| 92 | + echo "Extracting MediaMTX..." |
| 93 | + tar -xzf "${MEDIAMTX_ARCHIVE}" -C "${MEDIAMTX_DIR}" |
| 94 | +fi |
| 95 | + |
| 96 | +cp "${MEDIAMTX_CFG_SRC}" "${MEDIAMTX_CFG_DST}" |
| 97 | + |
| 98 | +if curl -fsS --max-time 2 http://127.0.0.1:9997/v3/config/global/get >/dev/null 2>&1; then |
| 99 | + echo "Using existing MediaMTX instance on 127.0.0.1:9997" |
| 100 | +else |
| 101 | + echo "Starting local MediaMTX..." |
| 102 | + ( |
| 103 | + cd "${MEDIAMTX_DIR}" |
| 104 | + ./mediamtx "${MEDIAMTX_CFG_DST}" |
| 105 | + ) & |
| 106 | + MEDIAMTX_PID="$!" |
| 107 | + STARTED_MEDIAMTX="1" |
| 108 | + |
| 109 | + if ! wait_for_http "http://127.0.0.1:9997/v3/config/global/get" 40 0.5; then |
| 110 | + echo "MediaMTX did not become ready on port 9997" >&2 |
| 111 | + exit 1 |
| 112 | + fi |
| 113 | +fi |
| 114 | + |
| 115 | +echo "Starting FFmpeg test publisher on rtsp://127.0.0.1:8554/test..." |
| 116 | +ffmpeg -nostdin -re \ |
| 117 | + -f lavfi -i testsrc=size=640x480:rate=25 \ |
| 118 | + -f lavfi -i sine=frequency=1000 \ |
| 119 | + -pix_fmt yuv420p \ |
| 120 | + -c:v libx264 -preset ultrafast -tune zerolatency -b:v 500k \ |
| 121 | + -c:a aac -b:a 128k \ |
| 122 | + -f rtsp rtsp://127.0.0.1:8554/test \ |
| 123 | + -loglevel warning >/dev/null 2>&1 & |
| 124 | +FFMPEG_PID="$!" |
| 125 | + |
| 126 | +sleep 1 |
| 127 | +if ! kill -0 "${FFMPEG_PID}" >/dev/null 2>&1; then |
| 128 | + echo "FFmpeg test stream failed to start. Check your ffmpeg installation/logs." >&2 |
| 129 | + exit 1 |
| 130 | +fi |
| 131 | + |
| 132 | +echo "WHEP endpoint expected at http://127.0.0.1:8889/test/whep" |
| 133 | + |
| 134 | +if [[ "${RTSPSTREAM_SKIP_MM:-0}" == "1" ]]; then |
| 135 | + echo "RTSPSTREAM_SKIP_MM=1 set, not starting MagicMirror." |
| 136 | + exit 0 |
| 137 | +fi |
| 138 | + |
| 139 | +cd "${MM_DIR}" |
| 140 | +MM_CONFIG_FILE=modules/MMM-RTSPStream/demo.config.js node --run start:dev |
0 commit comments