#! /bin/bash
#LED stuff
export LEDPATH="/sys/class/leds/"
prefix="apu"
found=false
led1trig="heartbeat"
led2trig="phy0radio"
led2trigalt="cpu"
#led3trig="phy1radio"
led3trig="disk-activity"
led3trigalt="disk-activity"
# Iterate over files in the current directory
# Need to find out which apu version we have
for filename in ${LEDPATH}${prefix}*; do
    # Extract the number using sed
    apu_version=$(echo "$filename" | grep -oE "${prefix}([0-9]+)" | grep -oE '[0-9]+')
    found=true
    # Print the result
    #echo "Filename: $filename, Extracted Number: $apu_version"
    break
done
if [ "$found" = true ]; then
	echo "Found LED(s) for ${prefix}${apu_version}... Setting them up..."
	# LED1 (left)
	echo "Set $led1trig as trigger for LED1"
	echo $led1trig | sudo tee "${LEDPATH}${prefix}${apu_version}:green:led1/trigger" >/dev/null
	# LED2 (middle)
	if grep -q "phy0radio" "${LEDPATH}${prefix}${apu_version}:green:led1/trigger";then
		echo "Set $led2trig as trigger for LED2"
		echo $led2trig | sudo tee "${LEDPATH}${prefix}${apu_version}:green:led2/trigger" >/dev/null
	else
		echo "Set $led2trigalt as trigger for LED2"
		echo $led2trigalt | sudo tee "${LEDPATH}${prefix}${apu_version}:green:led2/trigger" >/dev/null
	fi
	# LED3 (right)
	if grep -q "phy1radio" "${LEDPATH}${prefix}${apu_version}:green:led1/trigger";then
		echo "Set $led3trig as trigger for LED3"
		echo $led3trig | sudo tee "${LEDPATH}${prefix}${apu_version}:green:led3/trigger" >/dev/null
	else
		echo "Set $led3trigalt as trigger for LED3"
		echo $led3trigalt | sudo tee "${LEDPATH}${prefix}${apu_version}:green:led3/trigger" >/dev/null
	fi
else
	echo "Found no LED(s) with the prefix ${prefix}... Stopping LED setup..."
fi