#!/usr/bin/env bash
source <(curl -s https://raw.githubusercontent.com/tteck/Proxmox/next/misc/debian.func)
# Copyright (c) 2021-2023 tteck
# Author: tteck (tteckster)
# License: MIT
# https://github.com/tteck/Proxmox/raw/main/LICENSE

function header_info {
clear
cat <<"EOF"

 _____   _       __             ___   __  _______  ____________
/__  /  (_)___ _/ /_  ___  ___ |__ \ /  |/  / __ \/_  __/_  __/
  / /  / / __  / __ \/ _ \/ _ \__/ // /|_/ / / / / / /   / /   
 / /__/ / /_/ / /_/ /  __/  __/ __// /  / / /_/ / / /   / /    
/____/_/\__, /_.___/\___/\___/____/_/  /_/\___\_\/_/   /_/     
       /____/ 🐝
 
EOF
}
header_info
echo -e "Loading..."
APP="Zigbee2MQTT"
var_disk="4"
var_cpu="2"
var_ram="1024"
var_os="debian"
var_version="11"
variables
color
catch_errors

function default_settings() {
  CT_TYPE="1"
  PW=""
  CT_ID=$NEXTID
  HN=$NSAPP
  DISK_SIZE="$var_disk"
  CORE_COUNT="$var_cpu"
  RAM_SIZE="$var_ram"
  BRG="vmbr0"
  NET=dhcp
  GATE=""
  DISABLEIP6="no"
  MTU=""
  SD=""
  NS=""
  MAC=""
  VLAN=""
  SSH="no"
  VERB="no"
  echo_default
}

function update_script() {
header_info
if [[ ! -d /opt/zigbee2mqtt ]]; then msg_error "No ${APP} Installation Found!"; exit; fi
cd /opt/zigbee2mqtt

stop_zigbee2mqtt() {
  if which systemctl 2> /dev/null > /dev/null; then
    echo "Shutting down Zigbee2MQTT..."
    sudo systemctl stop zigbee2mqtt
  else
    echo "Skipped stopping Zigbee2MQTT, no systemctl found"
  fi
}

start_zigbee2mqtt() {
  if which systemctl 2> /dev/null > /dev/null; then
    echo "Starting Zigbee2MQTT..."
    sudo systemctl start zigbee2mqtt
  else
    echo "Skipped starting Zigbee2MQTT, no systemctl found"
  fi
}

set -e

if [ -d data-backup ]; then
  echo "ERROR: Backup directory exists. May be previous restoring was failed?"
  echo "1. Save 'data-backup' and 'data' dirs to safe location to make possibility to restore config later."
  echo "2. Manually delete 'data-backup' dir and try again."
  exit 1
fi

stop_zigbee2mqtt

echo "Generating a backup of the configuration..."
cp -R data data-backup || { echo "Failed to create backup."; exit 1; }

echo "Initiating update"
if ! git pull; then
  echo "Update failed, temporarily storing changes and trying again."
  git stash && git pull || (echo "Update failed even after storing changes. Aborting."; exit 1)
fi

echo "Acquiring necessary components..."
npm ci || { echo "Failed to install necessary components."; exit 1; }

echo "Restoring configuration..."
cp -R data-backup/* data || { echo "Failed to restore configuration."; exit 1; }

rm -rf data-backup || { echo "Failed to remove backup directory."; exit 1; }

start_zigbee2mqtt

echo "Done!"
exit
}

start
build_container
description

msg_ok "Completed Successfully!\n"