From 49de8d03378295f262415228578c14de1777c56d Mon Sep 17 00:00:00 2001
From: computermouth <computermouth@crunchbangplusplus.org>
Date: Thu, 17 Mar 2016 12:32:07 -0700
Subject: [PATCH] Added update features

---
 chip-local-flash.sh | 55 ++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 54 insertions(+), 1 deletion(-)

diff --git a/chip-local-flash.sh b/chip-local-flash.sh
index 41187ed..eefc44a 100755
--- a/chip-local-flash.sh
+++ b/chip-local-flash.sh
@@ -5,15 +5,32 @@ set -x
 SCRIPTDIR=$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )
 source $SCRIPTDIR/common.sh
 
+BUILDROOT_OUTPUT_DIR=".new/firmware"
+FIRMWARE_DIR=".new/firmware"
+
+DL_METHOD=fel
+DL_FLAVOR=serv
+DL_DIST=rootfs
+DL_DIR=".dl"
+
 ##############################################################
 #  main
 ##############################################################
-while getopts "fu:" opt; do
+while getopts "fdnu:" opt; do
   case $opt in
     f)
       echo "fastboot enabled"
+      DL_METHOD=fb
       METHOD=fastboot
       ;;
+    d)
+      echo "desktop selected"
+      DL_FLAVOR=desk
+      ;;
+    n)
+      echo "next enabled"
+      DL_DIST=testing-rootfs
+      ;;
     u)
       BUILDROOT_OUTPUT_DIR="${OPTARG}"
       ;;
@@ -26,6 +43,42 @@ done
 
 echo "BUILDROOT_OUTPUT_DIR = $BUILDROOT_OUTPUT_DIR"
 
+function require_directory {
+  if [[ ! -d "${1}" ]]; then
+      mkdir -p "${1}"
+  fi
+}
+
+function dl_check {
+	wget -O $DL_DIR/latest.md5\
+		opensource.nextthing.co/chippian/$DL_DIST/latest.md5
+
+	pushd $DL_DIR
+	if [[ $(cat latest.md5 | grep "`md5sum img-$DL_FLAVOR-$DL_METHOD.tar.gz`")\
+	 && -d "img-$DL_FLAVOR-$DL_METHOD/images" ]]; then
+		echo "Cached files located"
+		echo "Staging for flashing"
+		cp -R img-$DL_FLAVOR-$DL_METHOD/images ../$FIRMWARE_DIR/
+	else
+		echo "New image available"
+		rm -rf img-$DL_FLAVOR-$DL_METHOD*
+		
+		wget opensource.nextthing.co/chippian/$DL_DIST/img-$DL_FLAVOR-$DL_METHOD.tar.gz
+
+		echo "Extracting.."
+		tar -xf img-$DL_FLAVOR-$DL_METHOD.tar.gz
+		echo "Staging for flashing"
+		cp -R img-$DL_FLAVOR-$DL_METHOD/images ../$FIRMWARE_DIR/
+	fi
+	popd
+}
+
+
+require_directory "$FIRMWARE_DIR"
+require_directory "$DL_DIR"
+
+dl_check
+
 FEL=fel
 
 METHOD=${METHOD:-fel}