]> git.vanrenterghem.biz Git - elgato-keylight-script.git/blobdiff - keylights.sh
Update documentation structure
[elgato-keylight-script.git] / keylights.sh
index 08b9036f56ff2b9786a74369529880e94d970572..cfd20b6a8c541b40a795becae5ad9fc4687bab44 100755 (executable)
@@ -4,7 +4,6 @@ set -Eeuo pipefail
 trap destroy SIGINT SIGTERM ERR EXIT
 
 # Settings
-temp_file="/tmp/elgatokeylights"
 script_dir=$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd -P)
 icon="${script_dir}/assets/elgato.png"
 
@@ -12,10 +11,13 @@ icon="${script_dir}/assets/elgato.png"
 declare -i silent=0
 declare -i pretty=0
 declare action="usage"
-declare target=""
+declare target='.'
+declare limit=""
 declare format="json"
 declare -A lights
 declare lights_json
+declare full_json
+declare simple_json
 declare flat_json
 declare call='curl --silent --show-error --location --header "Accept: application/json" --request'
 declare devices="/elgato/lights"
@@ -37,14 +39,13 @@ die() {
 
 destroy() {
     code=$?
-    rm "$temp_file" 2>/dev/null
 
     exit ${code}
 }
 
 usage() {
     cat <<EOF
-Usage: $(basename "${BASH_SOURCE[0]}") [-h] [-f <value>] [-l] [-p] [-s] [-t <value>][-v] [--<option>] [--<option> <value>] <action>
+Usage: $(basename "${BASH_SOURCE[0]}") [-h] [-f <value>] [-l <value>] [-p] [-s] [-t <value>][-v] [--<option>] [--<option> <value>] <action>
 
 Elgato Lights controller. Works for Key Light and Key Light Air.
 
@@ -53,14 +54,15 @@ Available actions:
     status      Get state of lights
     on          Turn all lights on
     off         Turn all lights off
-    temperature  Set temperature level (260-470)
+    temperature Set temperature level (260-470)
     brightness  Set brightness level (0-100)
     increase    Increases brightness by 10
     decrease    Decreases brightness by 10
 
 Available formats:
     json        Renders output as JSON (default)
-    flat        Renders output as flattened JSON with .(dot) notation JSON (default)
+    simple      Renders output as JSON array of single level objects with subarrays as .(dot) notation JSON
+    flat        Renders output as fully flattened single level JSON with .(dot) notation JSON
     html        Renders output as basic html table
     csv         Renders output as csv
     table       Renders output as a printed table
@@ -69,12 +71,13 @@ Available formats:
 
 Available options:
 
--h, --help      Print this help and exit
--f  --format    Set output format
--p, --pretty    Pretty print console output
--s, --silent    Supress notifications
--t, --target    Only perform action on devices where value matches filter
--v, --verbose   Print script debug info
+-h, --help               Print this help and exit
+-f, --format             Set output format
+-l, --limit <list>       Limit top level output fields to the specified comma separated list
+-p, --pretty             Pretty print console output
+-s, --silent             Supress notifications
+-t, --target <filter>    Only perform action on devices where value matches filter
+-v, --verbose            Print script debug info
 EOF
     exit
 }
@@ -89,6 +92,10 @@ parse_params() {
             format="${2-}"
             shift
             ;;
+        -l | --limit)
+            limit=$(eval echo "\| {${2-}} ")
+            shift
+            ;;
         -p | --pretty) pretty=1 ;;
         -v | --verbose) set -x ;;
         -s | --silent) silent=1 ;;
@@ -141,77 +148,129 @@ default_light_properties() {
 }
 
 produce_json() {
-    t=$(eval echo "'[.[] | select($target)]'")
+    t=$(eval echo "'[.[] $limit| select($target)]'")
+    f=$(eval echo "'[.[] | select($target)]'")
 
     lights_json=$(echo "${lights[@]}" | jq -c -s "$t")
-}
+    full_json=$(echo "${lights[@]}" | jq -c -s "$f")
+    simple_json=$(echo "${lights_json}" | jq -c '.[] | reduce ( tostream | select(length==2) | .[0] |= [join(".")] ) as [$p,$v] ({}; setpath($p; $v)) ')
+    simple_json=$(echo "${simple_json}" | jq -c -s '.') # slurp it to make it an array
+    flat_json=$(echo "${lights_json}" | jq -c -s '.[] | reduce ( tostream | select(length==2) | .[0] |= [join(".")] ) as [$p,$v] ({}; setpath($p; $v)) ')
 
-# produce_json() {
-#     declare json
-#     t=$(eval echo "'.[] | select($target)'")
-
-#     for l in "${!lights[@]}"; do
-#         json+="${lights[$l]},"
-#     done
+}
 
-#     lights_json=$(echo "[${json%,}]" | jq -c "$t")
-# }
 output() {
-    data=${1-}
-    type=${2-"$format"}
 
     # Mange user requested output format
     case $format in
-    json | flat) print_json "$data" ;;
-    table) print_json "$data" ;;
-    csv) print_csv "$data" ;;
-    pair) print_pair "$data" ;;
-    html) print_html "$data" ;;
-    -?*) die "Unknown output format (-f/--format): $type" ;;
+    json) print_json "$lights_json" ;;
+    simple) print_json "$simple_json" ;;
+    flat) print_json "$flat_json" ;;
+    table) print_structured '@tsv' ;;
+    csv) print_structured '@csv' ;;
+    pair) print_structured 'pairs' ;;
+    html) print_html ;;
+    -?*) die "Unknown output format (-f/--format): $format" ;;
     esac
 }
 
 print_json() {
-    # TODO: Evaluate adding jq filtering as filter argument
-    query=""
-
-    # Deconstruct json and assemble in flattened with .(dot) notation
-    if [[ $format == "flat" ]]; then
-        #query='.[]| . as $in | reduce leaf_paths as $path ({}; . + { ($path | map(tostring) | join(".")): $in | getpath($path) })'
-        #query='. as \$data | [ path(.. | select(scalars|tostring), select($empty_tests)) ] | map({ (map(tostring) | join("$join_char")) : (. as \$path | . = \$data | getpath(\$path)) }) | reduce .[] as \$item ({ }; . + \$item);'
-        #query='[leaf_paths as $path | {"key": $path | join("."), "value": getpath($path)}] | from_entries'
-        query='reduce ( tostream | select(length==2) | .[0] |= [join(".")] ) as [$p,$v] ({}; setpath($p; $v))'
-    else
-        query='.'
-    fi
 
     # Manage pretty printing
     if [[ $pretty -eq 1 ]]; then
-        echo "${1-}" | jq "$query"
+        echo "${1-}" | jq '.'
     else
-        echo "${1-}" | jq -c -M "$query"
+        echo "${1-}" | jq -c -M '.'
     fi
 
     exit 0
 }
 
-print_table() {
-    bold=$(tput bold)
-    normal=$(tput sgr0)
-    message='
-    
-'
-    die "To be implemented"
+print_structured() {
+    pp=${2-$pretty}
+
+    # Handle csv and table printing
+    query="(.[0] | keys_unsorted | map(ascii_upcase)), (.[] | [.[]])|${1-@csv}"
+
+    # Handle printing as key value pairs
+    if [[ ${1} == 'pairs' ]]; then
+        query='.[] | "--------------",(to_entries[] | [.key, "=", .value] | @tsv)'
+    fi
+
+    # Manage pretty printing
+    if [[ $pp -eq 1 ]]; then
+        echo "${simple_json}" | jq --raw-output "$query" | column -t -s$'\t' | sed -e 's/"//g'
+    else
+        if [[ ${1} == 'pairs' ]]; then
+            echo "${simple_json}" | jq -r "$query" | sed -e 's/\t//g'
+        else
+            echo "${simple_json}" | jq -r "$query"
+        fi
+    fi
+}
+
+print_html() {
+    data=$(print_structured '@csv' 1)
+
+    html="
+    <table>
+    $(
+        print_header=true
+        while read d; do
+            if $print_header; then
+                echo "<tr><th>${d//,/<\/th><th>}</th></tr>"
+                print_header=false
+                continue
+            fi
+            echo "<tr><td>${d//,/</td><td>}</td></tr>"
+        done <<<"${data}"
+    )
+    </table>"
+    echo "$html"
 }
 
 set_state() {
-    new_state=$1
-    die "To be implemented"
+    declare -a data
+    readarray -t data < <(echo "${full_json}" | jq -c '.[] | {displayName, url, numberOfLights, lights}')
+    declare -a updated
+
+    x=$(echo "${1}" | tr 01 10) # "flip the bit"
+
+    for d in "${data[@]}"; do
+        query_old="[.lights[] | select(.on==${x})] | length"
+        count_found=$(echo "${d}" | jq "$query_old")
+
+        # Don't send to lights already in wanted state
+        if [[ $count_found -eq 0 ]]; then continue; fi
+
+        # Extract relevant data and create new json object
+        url=$(echo "${d}" | jq '.url')
+        dn=$(echo "${d}" | jq -r '.displayName')
+        l=$(echo "${d}" | jq -c 'del(.url, .displayName)' | jq ". | .lights[].on = ${1}")
+
+        # Send command
+        if eval "${call} PUT -d '${l}' ${url}${devices}" >/dev/null; then updated+=("$dn"); fi
+    done
+
+    # Text representation of new state
+    state="ON"
+    [[ $1 -eq 0 ]] && state="OFF"
+
+    # Send notification
+    if [[ ${#updated[*]} -gt 0 ]]; then
+        n="Turned $state ${#updated[@]} lights:\n\n"
+        for i in "${updated[@]}"; do
+            n+=$(echo "$i\n")
+        done
+        notify "$n"
+
+    fi
 }
 
 find_lights() {
     # Scan the network for Elgato devices
-    avahi-browse -d local _elg._tcp --resolve -t | grep -v "^\+" >"$temp_file"
+    declare -a avahi
+    readarray -t avahi < <(avahi-browse -d local _elg._tcp --resolve -t -p | grep -v "^\+")
 
     declare device
     declare hostname
@@ -227,69 +286,63 @@ find_lights() {
     declare light
     default_light_properties
 
-    cat "$temp_file" >tmp
-    while read -r line; do
+    for l in "${avahi[@]}"; do
+        IFS=';' read -ra data <<<"$l" # split line into array
 
         # Gather information about the light
-        if [[ ($line == =*) && ($line =~ IPv[46][[:space:]](.+)[[:space:]]_elg) ]]; then
-            device=$(eval echo "${BASH_REMATCH[1]}") # eval to strip whitespace
-        elif [[ $line =~ hostname.+\[(.+)\] ]]; then
-            hostname=${BASH_REMATCH[1]}
-        elif [[ $line =~ address.+\[(.+)\] ]]; then
-            ip=${BASH_REMATCH[1]}
-            [[ $ip =~ fe80 ]] && ipv6="$ip" || ipv4="$ip"
-            ip=""
-        elif [[ $line =~ port.+\[(.+)\] ]]; then
-            port=${BASH_REMATCH[1]}
-        elif [[ $line =~ txt.+\[(.+)\] ]]; then
-            txt=$(eval echo "${BASH_REMATCH[1]}") # eval to strip single and double quotes
-
-            if [[ $txt =~ mf=([^[[:space:]]*]*) ]]; then manufacturer=${BASH_REMATCH[1]}; fi
-            if [[ $txt =~ id=([^[[:space:]]*]*) ]]; then mac=${BASH_REMATCH[1]}; fi
-            if [[ $txt =~ md=.+[[:space:]]([^[[:space:]]*]*)[[:space:]]id= ]]; then sku=${BASH_REMATCH[1]}; fi
-
-            # Get information from the light
-            url="http://$ipv4:$port"
-
-            declare protocol="--ipv4"
-            if [[ $ipv4 == "N/A" ]]; then
-                # Workaround: Ignoring ipv6 as Elgato miss-announces addressing and is not accepting requests
-                # properly for v6. Will not change to filter only on ipv4 from avahi, as that can cause us to only end
-                # up with an ipv6 address even though it was announced as ipv4, which in turn means we cannot communicate.
-                continue
-                # Remove above and uncomment below if a future update fixes ipv6 announcement and requests
-                #protocol="--ipv6"
-                #url="http://[$ip]:$port"
-            fi
+        device=$(echo "${data[3]}" | sed -e 's/\\032/ /g') # fix avahi output
+        hostname=${data[6]}
+        [[ ${data[7]} =~ fe80 ]] && ipv6=${data[7]} || ipv4=${data[7]}
+        port=${data[8]}
+        txt=$(eval echo "${data[9]}") # eval to strip quotes
+        [[ $txt =~ mf=([^[[:space:]]*]*) ]] && manufacturer=${BASH_REMATCH[1]}
+        [[ $txt =~ id=([^[[:space:]]*]*) ]] && mac=${BASH_REMATCH[1]}
+        [[ $txt =~ md=.+[[:space:]]([^[[:space:]]*]*)[[:space:]]id= ]] && sku=${BASH_REMATCH[1]}
+
+        # Get information from the light
+        url="http://$ipv4:$port"
+
+        declare protocol="--ipv4"
+        if [[ $ipv4 == "N/A" ]]; then
+            # Workaround: Ignoring ipv6 as Elgato miss-announces addressing and is not accepting requests
+            # properly for v6. Will not change to filter only on ipv4 from avahi, as that can cause us to only end
+            # up with an ipv6 address even though it was announced as ipv4, which in turn means we cannot communicate.
+            continue
+            # Remove above and uncomment below if a future update fixes ipv6 announcement and requests
+            #protocol="--ipv6"
+            #url="http://[$ip]:$port"
+        fi
 
-            cfg=$(eval "${call} GET $protocol ${url}${settings}") >/dev/null
-            info=$(eval "${call} GET $protocol ${url}${accessory_info}") >/dev/null
-            light=$(eval "${call} GET $protocol ${url}${devices}") >/dev/null
-
-            # Store the light as json
-            lights["$device"]=$(jq -n \
-                --arg dev "$device" \
-                --arg hn "$hostname" \
-                --arg ipv4 "$ipv4" \
-                --arg ipv6 "$ipv6" \
-                --argjson port "$port" \
-                --arg mf "$manufacturer" \
-                --arg mac "$mac" \
-                --arg sku "$sku" \
-                --arg url "$url" \
-                --argjson light "$light" \
-                --argjson cfg "$cfg" \
-                --argjson info "$info" \
-                '{device: $dev, manufacturer: $mf, hostname: $hn, url: $url, ipv4: $ipv4, ipv6: $ipv6, 
-                    port: $port, mac: $mac, sku: $sku, light: $light, settings: $cfg, info: $info}')
-
-            # Reset for next light as we are processing the last avahi line
-            default_light_properties
+        cfg=$(eval "${call} GET $protocol ${url}${settings}") >/dev/null
+        info=$(eval "${call} GET $protocol ${url}${accessory_info}") >/dev/null
+        light=$(eval "${call} GET $protocol ${url}${devices}") >/dev/null
+
+        json=$(jq -n \
+            --arg dev "$device" \
+            --arg hn "$hostname" \
+            --arg ipv4 "$ipv4" \
+            --arg ipv6 "$ipv6" \
+            --argjson port "$port" \
+            --arg mf "$manufacturer" \
+            --arg mac "$mac" \
+            --arg sku "$sku" \
+            --arg url "$url" \
+            --argjson cfg "$cfg" \
+            '{device: $dev, manufacturer: $mf, hostname: $hn, url: $url, ipv4: $ipv4, ipv6: $ipv6, 
+                port: $port, mac: $mac, sku: $sku, settings: $cfg}')
+
+        # Store the light as json and merge info + light into base object
+        lights["$device"]=$(echo "$info $light $json" | jq -s '. | add')
+
+        # Reset for next light as we are processing the last avahi line
+        default_light_properties
 
-        fi
-    done <"$temp_file"
+    done
 }
 
+# Quit if script is run by root
+[[ "$EUID" -eq 0 ]] && die "Not allowed to run as root"
+
 # Manage user parameters
 parse_params "$@"
 
@@ -299,14 +352,14 @@ dependencies avahi-browse curl notify-send jq
 find_lights
 
 # Fail if we cannot find lights
-[[ ${#lights[@]} -eq 0 ]] && die "No lights found" 2
+[[ ${#lights[@]} -eq 0 ]] && die "No lights found"
 
 produce_json
 
 # Dispatch actions
 case $action in
 usage) usage ;;
-list) output "${lights_json}" ;;
+list) output ;;
 status) status ;;
 on) set_state 1 ;;
 off) set_state 0 ;;