From d36ecb90e03c6a2f1a2d0af4d5905886ffd1a564 Mon Sep 17 00:00:00 2001
From: tteckster <tteckster@gmail.com>
Date: Wed, 9 Feb 2022 01:44:40 -0500
Subject: [PATCH] Update ha_setup.sh

---
 setup/ha_setup.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/setup/ha_setup.sh b/setup/ha_setup.sh
index b3a74f0c..9b8387e3 100644
--- a/setup/ha_setup.sh
+++ b/setup/ha_setup.sh
@@ -92,11 +92,17 @@ docker run -d \
   -v /etc/timezone:/etc/timezone:ro\
   --net=host \
   homeassistant/home-assistant:stable &>/dev/null
+
+echo -e "${CHECKMARK} \e[1;92m Creating Update Script... \e[0m"
 file_path="/root/update.sh"
 echo "#!/bin/bash
+echo -e '\e[1;33m Pulling New Stable Version... \e[0m'
 docker pull homeassistant/home-assistant:stable
+echo -e '\e[1;33m Stopping Home Assistant... \e[0m'
 docker stop homeassistant
+echo -e '\e[1;33m Removing Home Assistant... \e[0m'
 docker rm homeassistant
+echo -e '\e[1;33m Starting Home Assistant... \e[0m'
 docker run -d \
   --name homeassistant \
   --privileged \
@@ -108,7 +114,9 @@ docker run -d \
   -v /etc/timezone:/etc/timezone:ro\
   --net=host \
   homeassistant/home-assistant:stable
-docker image prune -f" > $file_path
+echo -e '\e[1;33m Removing Old Image... \e[0m'
+docker image prune -f
+echo -e '\e[1;33m Finished Update! \e[0m'" > $file_path
 sudo chmod +x /root/update.sh
 
 echo -e "${CHECKMARK} \e[1;92m Customizing LXC... \e[0m"