#!/bin/bash

MQTTSERVER=$1
MQTTPREFIX=$2
CUSTOMFILE=$3

FLASHTOOL=./tools/tcpFlash.py
TOOLDIR=tools/
DIET=bin/luasrcdiet

UPGRADEPREP=/tmp/upgradeCMD4clock.txt

if [ ! -f $FLASHTOOL ]; then
 echo "Execute the script in root folder of the project"
 exit 2
fi

if [[ "$MQTTPREFIX" == "" ]] || [[ "$MQTTSERVER" == "" ]]; then
 echo "MQTTSERVER: ip address to mqtt server"
 echo "MQTTPREFIX: configured prefex in MQTT of ESP required"
 echo "usage:"
 echo "$0 <MQTTSERVER> <MQTTPREFIX>"
 echo "$0 192.168.0.2 basetopic"
 exit 1
fi

# Prepare all files on host
if [[ "$CUSTOMFILE" == "" ]]; then
	FILES="displayword.lua main.lua timecore.lua webpage.html webserver.lua wordclock.lua init.lua"
	echo "Start Flasing ..."
else
	FILES=$CUSTOMFILE
	echo "Start Flasing $FILES ..."
fi


# Convert files, if necessary
if [ "$FILES" != "config.lua" ]; then
        echo "Generate DIET version of the files"
        OUTFILES=""
        ROOTDIR=$PWD
        cd $TOOLDIR
        for f in $FILES; do
                if [[ "$f" == *.lua ]] && [[ "$f" != init.lua ]]; then
                        echo "Compress $f ..."
                        out=$(echo "$f" | sed 's/.lua/_diet.lua/g')
                        $DIET ../$f -o ../diet/$out >> /dev/null
                        OUTFILES="$OUTFILES diet/$out"
                else
                        OUTFILES="$OUTFILES $f"
                fi
        done
        FILES=$OUTFILES
        cd $ROOTDIR
fi


# check the connection
echo "Searching $MQTTPREFIX ..."
mosquitto_sub -h $MQTTSERVER -t "$MQTTPREFIX/#" -C 1 -v
if [ $? -ne 0 ]; then
 echo "Entered Wordclock address: $MQTTPREFIX on $MQTTSERVER is NOT online"
 exit 2
fi
echo "Activate Telnet server ..."
TELNETIP="empty"
while [ "$TELNETIP" = "empty" ]; do
 date
 mosquitto_pub -h $MQTTSERVER -t "$MQTTPREFIX/cmd/telnet" -m "a"
 TELNETIP=$(mosquitto_sub -h $MQTTSERVER -t "$MQTTPREFIX/telnet" -C 1 -W 30)
done

echo "Upgrading $MQTTPREFIX via telenet on $TELNETIP ..."
sleep 1
echo "if (mlt ~= nil) then mlt:unregister() end" > $UPGRADEPREP
echo "uart.write(0, tostring(node.heap())" >> $UPGRADEPREP
echo "collectgarbage()"  >> $UPGRADEPREP
echo "" >> $UPGRADEPREP
echo "download = string.char(0,0,64)"  >> $UPGRADEPREP
echo "w = string.char(0,0,0)"  >> $UPGRADEPREP
echo "ws2812.write(w:rep(4) .. download .. w:rep(15) .. download .. w:rep(9) .. download .. w:rep(30) .. download .. w:rep(41) .. download )"  >> $UPGRADEPREP
echo "collectgarbage()"  >> $UPGRADEPREP
$FLASHTOOL -f $UPGRADEPREP -t $TELNETIP -v

for f in $FILES; do
    if [ ! -f $f ]; then
        echo "Cannot find $f"
        echo "place the terminal into the folder where the lua files are present"
        exit 1
    fi
    espFile=$(echo "$f" | sed 's;diet/;;g')
    echo "------------- $espFile ------------"
    $FLASHTOOL -t $TELNETIP -f $f -o $espFile
    if [ $? -ne 0 ]; then
        echo "STOOOOP"
        exit 1
    fi
done

echo "TODO: Reboot the ESP"
#echo "node.restart()" | nc $TELNETIP 23

exit 0