#!/bin/bash

# WildlifeSystems
#
# This script is part of the WildlifeSystems project. For further information
# please refer to https://docs.wildlife.systems, or for more information on
# the project itself, please refer to https://wildlife.systems.

# Return codes
#
# Further information on WildlifeSystems standard reurn codes can be found
# at https://docs.wildlife.systems/return-codes.html
# 0 - Success
# 1 - Already running
# 11 - Incorrect filename pattern


#Check if this script already running
if [[ "$(pgrep -c ws-run-audio)" -gt 1 ]]; then
  echo "ws-run-audio already running. Exiting."
  exit 1
fi

trap "echo 'Terminating ws-run-audio...'; exit 0" SIGINT SIGTERM

echo "Starting ws-run-audio..."

#Config
CONFIG_FILE="/etc/ws/audio.conf"
if [[ ! -f "$CONFIG_FILE" || ! -r "$CONFIG_FILE" ]]; then
  echo "Error: Config file '$CONFIG_FILE' not found or not readable."
  exit 2
fi
CONFIG=$(cat "$CONFIG_FILE")

#Check config exists
if [[ -z "$CONFIG" ]]; then
  echo "No config file found."
  exit 2
fi

COUNTDOWN=5
FNTYPE=$(echo "$CONFIG" | jq -r '.init .filenaming')

function onOff {
  pi-pwr off "$(echo "$CONFIG" | jq -r '.init .off[]' | tr '\n' ' ')"
  pi-pwr on "$(echo "$CONFIG" | jq -r '.init .on[]' | tr '\n' ' ')"
}

while true;
do

	case $FNTYPE in
			timestamp)
					FN=$(date +%s)
					;;
			*)
					echo "Incorrect filename pattern"
					exit 11;
					;;
	esac

	LATEST_RECORD=""

	N=$(echo "$CONFIG" | jq -r ".run | length")
	I=0
	while [ $I  -lt $N ];
	do
		pi-pwr off $(echo "$CONFIG" | jq -r ".run[$I] .off[]" | tr '\n' ' ')
  		pi-pwr on $(echo "$CONFIG" | jq -r ".run[$I] .on[]" | tr '\n' ' ')
		ACTION=`echo $CONFIG| jq -r ".run[$I] .action"`
		echo $ACTION
		sudo ws-heartbeat $ACTION &
        	case  $ACTION in
				record)
					DURATION=$(echo "$CONFIG" | jq -r ".run[$I] .duration")
					CHANNELS=$(echo "$CONFIG" | jq -r ".run[$I] .channels")
					BITRATE=$(echo "$CONFIG" | jq -r ".run[$I] .bitrate")
					FORMAT=$(echo "$CONFIG" | jq -r ".run[$I] .format")
					ws-indicate record "arecord -c $CHANNELS -r $BITRATE -f $FORMAT -d $DURATION $FN.wav"
					LATEST_RECORD=$FN
					;;
				sleep)
					DURATION=$(echo "$CONFIG" | jq -r ".run[$I] .duration")
					sleep "$DURATION"
					;;
				 upload-delete)
					METHOD=`echo $CONFIG| jq -r ".run[$I] .method"`
					UPPATH=`echo $CONFIG| jq -r ".run[$I] .path"`
					EXTENSION=`echo $CONFIG| jq -r ".run[$I] .extension"`
					ASYNC=`echo $CONFIG| jq -r ".run[$I] .async"`
					case $METHOD in
						s3cmd)
							S3UP=`eval echo "s3cmd put $FN.$EXTENSION s3://$UPPATH"`
							S3UP="$S3UP && rm $FN.$EXTENSION"
							if [[ "$ASYNC" == "true" ]]; then
								S3UP="( $S3UP )&";
							fi
							eval $S3UP
							;;
						mv)
							MVUPPATH=$(eval "echo ${UPPATH}")

							# Check if the destination directory exists
							if [[ ! -d "$MVUPPATH" ]]; then
											mkdir -p "$MVUPPATH"
							fi

							mv $FN.$EXTENSION $MVUPPATH &
							;;
						script)
							SCRIPT=`echo $CONFIG| jq -r ".run[$I] .script"`
							# Check if the script exists and is executable
							if [[ -x "$SCRIPT" ]]; then
									$SCRIPT $FN $EXTENSION &
							else
									echo "Script $SCRIPT not found or not executable."
							fi
							;;
					esac
					;;
				clearup)
					METHOD=`echo $CONFIG| jq -r ".run[$I] .method"`
					case $METHOD in
						script)
							SCRIPT=`echo $CONFIG| jq -r ".run[$I] .script"`
							# Check if the script exists and is executable
							if [[ -x "$SCRIPT" ]]; then
									$SCRIPT &
							else
									echo "Script $SCRIPT not found or not executable."
							fi
							;;
						*)
							echo "Unknown method: $METHOD" 1>&2
							;;
					esac
					;;
				*)
					echo "Unknown action: $ACTION" 1>&2
					;;
			esac

		I=$(($I + 1))
		ws-indicate heartbeat
	onOff
	done
done
