#!/bin/sh
cd /tmp/

. /etc/firmware_release

. /etc/community.cfg

BOARD=$(uci get board.model.name)

#decide SOC
case $BOARD in
    tl-wdr4900-v1 )
        SOC="mpc85xx" ;;
    * )
        SOC="ar71xx" ;;
esac
echo -ne "\nHardware: $BOARD\n"

#rewrite BOARD
case $BOARD in
    cpe210 )
        BOARD="cpe210-220-510-520" ;;
esac

wget "${UPGRADE_PATH}/release.nfo"
if [ ! -f release.nfo ]; then
  echo -ne "Latest release information not found. Please try to update manually.\n\n"
  exit 1
fi
VERSION=$(cat release.nfo|awk -F: '/VERSION:/ { print $2 }')
rm -f release.nfo
echo -ne "Firmware found on server: $VERSION\n"

if [ $VERSION -eq $FIRMWARE_VERSION ]; then
  echo -ne "The installed firmware version is already the current version.\n\n"

  if [ "$1" = "--script" ]; then
    exit 1
  fi

  while true; do
    echo -ne "Do you want to reinstall the current version? [y/N]\n"
    read DO_UPDATE
    case $DO_UPDATE in
      [yY]*|[Jj]*)
        break
        ;;
      [nN]* | "")
        exit 1
        ;;
      *)
        echo "Invalid input"
        ;;
    esac
  done
fi

if [ "$FIRMWARE_COMMUNITY" == "franken" ]; then
    FIRMWARE_COMMUNITY="fff"
fi

FILE="${FIRMWARE_COMMUNITY}-${VERSION}-${SOC}-g-${BOARD}-squashfs-sysupgrade.bin"
echo -ne "Downloading $FILE\n\n"
wget "${UPGRADE_PATH}/${FILE}"
wget "${UPGRADE_PATH}/${FILE}.sha256"

echo -ne "\ndone. Comparing sha256 sums: "
sha256sum -c ${FILE}.sha256
ret=$?
echo
if [ $ret -ne 0 ]; then
  echo -ne "sha256 sums do not match. Try restarting this script to redownload the firmware.\n\n"
  rm -f ${FILE}*
  exit 1
else
  if [ "$1" = "--script" ]; then
    echo -ne "\nStarting firmware upgrade. Don't touch me until I reboot.\n\n\n"
    sysupgrade ${FILE}
  fi
  while true; do
    read -p "sha256 sums correct. Should I start upgrading the firmware (y/N)? " yn
    case $yn in
        [Yy]*|[Jj]* ) echo -ne "\nStarting firmware upgrade. Don't touch me until I reboot.\n\n\n"; sysupgrade ${FILE}; break;;
        [Nn]|* ) echo -ne "\nAborting firmware upgrade.\n\n"; rm -f ${FILE}*; exit 0;;
    esac
  done
fi