#!/bin/sh
#
# Turn WLED installations in the C4 hackspace on/off, change brightness or
# switch between presets.
#
# Author: Shy
# License: CC0

NAME="wled.sh"
COMMAND="wled"
VERSION="0.1.2"

# Make the exit status of command pipelines the status of the last command
# with non-zero exit status - if supported by this shell.
(set -o pipefail 2>/dev/null) && set -o pipefail

print_usage() {
    echo "\
Usage: $COMMAND -f(norcenter)|-w(wohnzimmer)|-a(ll) [on|off]
       $COMMAND -f|-w|-a [1-17|list]
       $COMMAND -f|-w|-a -b [1-255]"
}

print_help() {
    print_usage
    echo "
instance selection:
    -f,--fnordcenter        select fnordcenter
    -w,--wohnzimmer         select wohnzimmer
    -a,--all                select all known instances

commands:
    on|off                  switch on/off
    [1-17]                  switch preset
    l, list                 list presets
    -b,--brightness [1-255] set brightness

exit status:
    0   no failure
    1   operating error
    2   child command error
    4   unexpected API response"
}

curl_fetch() {
    curl --silent --show-error --fail --ipv4 --max-filesize 32768 \
         --max-redirs 0 --header 'Accept: application/json' "$1"
}

curl_send() {
    curl --silent --show-error --fail --ipv4 --max-filesize 1024 \
         --max-redirs 0 --header 'Accept: application/json' \
         --header 'Content-Type: application/json' --data "$1" "$2"
}

active_preset() {
    # Find number of active preset. API returns -1 if none is active, we return
    # an empty string.
    curl_fetch "http://$1/json" | \
        # -1 will be a no-match.
        sed -n 's/.*"ps"[[:space:]]*:[[:space:]]*\(1\?[0-9]\).*/\1/p'
}

arg_inst="$1"
arg_cmnd="$2"
arg_param="$3"
exit_code=0

# Clear positional arguments.
set --

# Parse instance argument and store hostnames as positional arguments.
case "$arg_inst" in
    ""|-h|--help) # Missing instance parameter.
        print_help
        exit 0
    ;;
    -v|--version)
        echo "$NAME version $VERSION"
        exit 0
    ;;
    -a|--all)
        set -- 'wled-fnordcenter.local' \
               'wled-wohnzimmer.local'
    ;;
    -f|--fnordcenter)
        set -- 'wled-fnordcenter.local'
    ;;
    -w|--wohnzimmer)
        set -- 'wled-wohnzimmer.local'
    ;;
    *) # Unknown instance parameter.
        echo "Error: unknown instance \"$arg_inst\"." >&2
        print_usage
        exit 1
    ;;
esac

# Parse and execute command argument against given instances.
for wled_host; do
    case "$arg_cmnd" in
        on) # Switch on.
            api_resp=$(curl_send '{"on":true}' "http://$wled_host/json")
        ;;
        off) # Switch off.
            api_resp=$(curl_send '{"on":false}' "http://$wled_host/json")
        ;;
        -b|--brightness) # Set brightness.
            if test -z "$arg_param"; then
                echo "Error: missing parameter for brightness." >&2
                exit 1
            fi
            # Try to convert 3rd parameter to an integer between 1 an 255.
            bright=$(printf '%u' "$arg_param" 2>/dev/null)
            if test $? -gt 0; then # Conversion failed.
                echo "Error: unable to parse parameter for brightness." >&2
                exit 1
            elif test "$bright" -lt 1 -o "$bright" -gt 255; then
                echo "Error: parameter for brightness is out of range." >&2
                exit 1
            fi
            api_resp=$(curl_send "{'bri':$bright}" "http://$wled_host/json")
        ;;
        [1-9]|1[0-7]) # Switch to preset.
            api_resp=$(curl_send "{'ps':$arg_cmnd}" "http://$wled_host/json")
        ;;
        l|list) # List presets.
            printf "loading ...\r"
            active_ps=$(active_preset "$wled_host")
            curl_fetch "http://$wled_host/presets.json" | \
                # Split into lines - one line per preset.
                sed --sandbox 's/"1\?[0-9]"[[:space:]]*:/\n&/g' | \
                # Extract number and name of every preset. Right-align numbers.
                # Drop everything else.
                sed --sandbox -n \
                    '/^"\(1\?[0-9]\)"[[:space:]]*:.*"n"[[:space:]]*:[[:space:]]*"\([[:alnum:] _-]*\)".*$/{
                        s//\1: \2/
                        s/^[0-9]:/ &/
                        p
                    }' | \
                # Sort.
                sort -b -n | \
                # Prepend description and mark active preset.
                sed --sandbox -e "1i\\Presets on $wled_host:" \
                              -e "s/^ \?$active_ps:.*/& */"

        ;;
        "") # Missing command.
            echo "Error: missing command." >&2
            print_usage
            exit 1
        ;;
        *) # Unknown command.
            echo "Error: unknown command \"$arg_cmnd\"." >&2
            print_usage
            exit 1
        ;;
    esac

    # Check curl (or maybe sed) exit status
    child_exit=$?
    if test $child_exit -ne 0; then
        echo "Error: curl or sed exited with exit code $child_exit." >&2
        exit_code=$((exit_code|2))
    fi

    # Check API response.
    if test -n "$api_resp" && echo "$api_resp" | grep -qv '{\s*"success"\s*:\s*true\s*}'; then
        echo "Warning: unexpected response from WLED API." >&2
        exit_code=$((exit_code|4))
    fi
done

exit $exit_code