#!/bin/bash

# GPIO assignments
LED_RED=17
#LED_GREEN=21                   # Use this line for Rev 1 boards
LED_GREEN=27                    # Use this line for Rev 2 boards
LED_BLUE=22

# Check user ID (run as sudo if not root)
if test "$UID" -ne 0 ; then
    # We are not root, re-run as root
    echo "Not root, re-running as root"
    sudo $0 $@
    exit 0
fi

# Set up GPIO pins as outputs
echo "$LED_RED" > /sys/class/gpio/export
echo "out" > /sys/class/gpio/gpio${LED_RED}/direction
echo "$LED_GREEN" > /sys/class/gpio/export
echo "out" > /sys/class/gpio/gpio${LED_GREEN}/direction
echo "$LED_BLUE" > /sys/class/gpio/export
echo "out" > /sys/class/gpio/gpio${LED_BLUE}/direction

# Determine the user colour choice
red=`echo $1 | cut -c 1`
green=`echo $1 | cut -c 2`
blue=`echo $1 | cut -c 3`

# Change 2s into 1s
if test "$red" -eq 2 ; then
    red=1
fi
if test "$green" -eq 2 ; then
    green=1
fi
if test "$blue" -eq 2 ; then
    blue=1
fi

# Set the colour choice to the GPIO pins
if test "$red" -eq 1 ; then
    echo "1" > /sys/class/gpio/gpio${LED_RED}/value
else
    echo "0" > /sys/class/gpio/gpio${LED_RED}/value
fi
if test "$green" -eq 1 ; then
    echo "1" > /sys/class/gpio/gpio${LED_GREEN}/value
else
    echo "0" > /sys/class/gpio/gpio${LED_GREEN}/value
fi
if test "$blue" -eq 1 ; then
    echo "1" > /sys/class/gpio/gpio${LED_BLUE}/value
else
    echo "0" > /sys/class/gpio/gpio${LED_BLUE}/value
fi

# Clean up GPIO pins
echo "$LED_RED" > /sys/class/gpio/unexport
echo "$LED_GREEN" > /sys/class/gpio/unexport
echo "$LED_BLUE" > /sys/class/gpio/unexport