From a645ab6f1ac466ced9bec98c368ee651467149e7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Xose=20P=C3=A9rez?= <xose.perez@gmail.com>
Date: Wed, 6 Sep 2017 00:02:01 +0200
Subject: [PATCH] Custom (not general) OTA flasher utility using Avahi

---
 code/flash.sh | 198 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 198 insertions(+)
 create mode 100755 code/flash.sh

diff --git a/code/flash.sh b/code/flash.sh
new file mode 100755
index 00000000..c29e51e3
--- /dev/null
+++ b/code/flash.sh
@@ -0,0 +1,198 @@
+#!/bin/bash
+
+ip=
+board=
+auth=
+flags=
+
+export boards=()
+ips=""
+
+exists() {
+    command -v "$1" >/dev/null 2>&1
+}
+
+echo_pad() {
+    string=$1
+    pad=$2
+    printf '%s' "$string"
+    printf '%*s' $(( $pad - ${#string} ))
+}
+
+useAvahi() {
+
+    echo_pad "#" 4
+    echo_pad "HOSTNAME" 20
+    echo_pad "IP" 20
+    echo_pad "DEVICE" 30
+    echo_pad "VERSION" 10
+    echo
+
+    printf -v line '%*s\n' 84
+    echo ${line// /-}
+
+    counter=0
+
+    ip_file="/tmp/espurna.flash.ips"
+    board_file="/tmp/espurna.flash.boards"
+    count_file="/tmp/espurna.flash.count"
+    echo -n "" > $ip_file
+    echo -n "" > $board_file
+    echo -n "$counter" > $count_file
+
+    avahi-browse -t -r -p  "_arduino._tcp" 2>/dev/null | grep ^= | while read line; do
+
+        (( counter++ ))
+        echo "$counter" > $count_file
+
+        hostname=`echo $line | cut -d ';' -f4`
+        ip=`echo $line | cut -d ';' -f8`
+        txt=`echo $line | cut -d ';' -f10`
+        board=`echo $txt | sed -n "s/.*espurna_board=\([^\"]*\).*/\1/p"`
+        version=`echo $txt | sed -n "s/.*espurna_version=\([^\"]*\).*/\1/p"`
+
+        echo -n "$ip;" >> $ip_file
+        echo -n "$board;" >> $board_file
+
+        echo_pad "$counter" 4
+        echo_pad "$hostname" 20
+        echo_pad "$ip" 20
+        echo_pad "$board" 30
+        echo_pad "$version" 10
+        echo
+
+
+    done
+
+    echo
+    read -p "Choose the board you want to flash (empty if none of these): " num
+
+    # None of these
+    if [ "$num" == "" ]; then
+        return
+    fi
+
+    # Check boundaries
+    counter=`cat $count_file`
+    if [ $num -lt 1 ] || [ $num -gt $counter ]; then
+        echo "Board number must be between 1 and $counter"
+        exit 1
+    fi
+
+    # Fill the fields
+    ip=`cat $ip_file | cut -d ';' -f$num`
+    board=`cat $board_file | cut -d ';' -f$num`
+
+}
+
+getBoard() {
+
+    boards=(`cat espurna/config/hardware.h | grep "defined" | sed "s/.*(\(.*\)).*/\1/" | sort`)
+
+    echo_pad "#" 4
+    echo_pad "DEVICE" 30
+    echo
+
+    printf -v line '%*s\n' 34
+    echo ${line// /-}
+
+    counter=0
+    for board in "${boards[@]}"; do
+        (( counter++ ))
+        echo_pad "$counter" 4
+        echo_pad "$board" 30
+        echo
+    done
+
+    echo
+    read -p "Choose the board you want to flash (empty if none of these): " num
+
+    # None of these
+    if [ "$num" == "" ]; then
+        return
+    fi
+
+    # Check boundaries
+    counter=${#boards[*]}
+    if [ $num -lt 1 ] || [ $num -gt $counter ]; then
+        echo "Board code must be between 1 and $counter"
+        exit 1
+    fi
+
+    # Fill the fields
+    (( num -- ))
+    board=${boards[$num]}
+
+}
+
+# ------------------------------------------------------------------------------
+
+# Welcome
+echo
+echo "--------------------------------------------------------------"
+echo "ESPURNA FIRMWARE OTA FLASHER"
+
+# Get current version
+version=`cat espurna/config/version.h | grep APP_VERSION | awk '{print $3}' | sed 's/"//g'`
+echo "Building for version $version"
+
+echo "--------------------------------------------------------------"
+echo
+
+if exists avahi-browse; then
+    useAvahi
+fi
+
+if [ "$board" == "" ]; then
+    getBoard
+fi
+
+if [ "$board" == "" ]; then
+    read -p "Board type of the device to flash: " -e -i "NODEMCU_LOLIN" board
+fi
+
+if [ "$board" == "" ]; then
+    echo "You must define the board type"
+    exit 2
+fi
+
+if [ "$ip" == "" ]; then
+    read -p "IP of the device to flash: " -e -i 192.168.1.4 ip
+fi
+
+if [ "$ip" == "" ]; then
+    echo "You must define the IP of the device"
+    exit 2
+fi
+
+if [ "$auth" == "" ]; then
+    read -p "Authorization key of the device to flash: " auth
+fi
+
+if [ "$flags" == "" ]; then
+    read -p "Extra flags for the build: " flags
+fi
+
+read -p "Environment to build: " -e -i "esp8266-1m-ota" env
+
+echo
+echo "IP    = $ip"
+echo "BOARD = $board"
+echo "AUTH  = $auth"
+echo "FLAGS = $flags"
+echo "ENV   = $env"
+
+echo
+echo -n "Are these values corrent [y/N]: "
+read response
+
+if [ "$response" != "y" ]; then
+    exit
+fi
+
+export IP=$ip
+export BOARD=$board
+export AUTH=$auth
+export FLAGS=$flags
+
+pio run -e $env -t upload