owncast/test/automated/tools.sh

#!/bin/bash

set -e

function install_ffmpeg() {
	# install a specific version of ffmpeg

	FFMPEG_VER="4.4.1"
	FFMPEG_PATH="$(pwd)/ffmpeg-$FFMPEG_VER"
	PATH=$FFMPEG_PATH:$PATH

	if ! [[ -d "$FFMPEG_PATH" ]]; then
		mkdir "$FFMPEG_PATH"
	fi

	pushd "$FFMPEG_PATH" >/dev/null

	if [[ -x "$FFMPEG_PATH/ffmpeg" ]]; then

		ffmpeg_version=$("$FFMPEG_PATH/ffmpeg" -version | awk -F 'ffmpeg version' '{print $2}' | awk 'NR==1{print $1}')

		if [[ "$ffmpeg_version" == "$FFMPEG_VER-static" ]]; then
			popd >/dev/null
			return 0
		else
			mv "$FFMPEG_PATH/ffmpeg" "$FFMPEG_PATH/ffmpeg.bk" || rm -f "$FFMPEG_PATH/ffmpeg"
		fi
	fi

	rm -f ffmpeg.zip
	curl -sL --fail https://github.com/ffbinaries/ffbinaries-prebuilt/releases/download/v${FFMPEG_VER}/ffmpeg-${FFMPEG_VER}-linux-64.zip --output ffmpeg.zip >/dev/null
	unzip -o ffmpeg.zip >/dev/null && rm -f ffmpeg.zip
	chmod +x ffmpeg
	PATH=$FFMPEG_PATH:$PATH

	popd >/dev/null
}

function start_owncast() {
	# Build and run owncast from source
	echo "Building owncast..."
	pushd "$(git rev-parse --show-toplevel)" >/dev/null
	go build -o owncast main.go

	if [ -z "$1" ]; then
		echo "Running owncast..."
	else
		echo "Running owncast with flags: $1"
	fi

	./owncast -database "$TEMP_DB" $1 &
	SERVER_PID=$!
	popd >/dev/null

	sleep 5

}

function start_stream() {
	# Start streaming the test file over RTMP to the local owncast instance.
	../../ocTestStream.sh &
	STREAM_PID=$!

	echo "Waiting for stream to start..."
	sleep 12
}

function update_storage_config() {
	echo "Configuring external storage to use ${S3_BUCKET}..."

	# Hard-coded to admin:abc123 for auth
	curl --fail 'http://localhost:8080/api/admin/config/s3' \
		-H 'Authorization: Basic YWRtaW46YWJjMTIz' \
		--data-raw "{\"value\":{\"accessKey\":\"${S3_ACCESS_KEY}\",\"acl\":\"\",\"bucket\":\"${S3_BUCKET}\",\"enabled\":true,\"endpoint\":\"${S3_ENDPOINT}\",\"region\":\"${S3_REGION}\",\"secret\":\"${S3_SECRET}\",\"servingEndpoint\":\"\"}}"
}

function kill_with_kids() {
	# kill a process and all its children (by pid)! return no error.

	if [[ -n $1 ]]; then
		mapfile -t CHILDREN_PID_LIST < <(ps --ppid "$1" -o pid= &>/dev/null || true)
		for child_pid in "${CHILDREN_PID_LIST[@]}"; do
			kill "$child_pid" &>/dev/null || true
			wait "$child_pid" &>/dev/null || true
		done
		kill "$1" &>/dev/null || true
		wait "$1" &>/dev/null || true
	fi
}

function finish() {
	echo "Cleaning up..."
	kill_with_kids "$STREAM_PID"
	kill "$SERVER_PID" &>/dev/null || true
	wait "$SERVER_PID" &>/dev/null || true
	rm -fr "$TEMP_DB"
}

trap finish EXIT

TEMP_DB=$(mktemp)