#!/usr/bin/env bash
source <(curl -s https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/misc/build.func)
# Copyright (c) 2021-2024 tteck
# Author: tteck (tteckster)
# Co-Author: MickLesk (Canbiz)
# License: MIT
# https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE

function header_info {
clear
cat <<"EOF"
    __  __
   / / / /___  ____ ___  ____ ___________
  / /_/ / __ \/ __ `__ \/ __ `/ ___/ ___/
 / __  / /_/ / / / / / / /_/ / /  / /
/_/ /_/\____/_/ /_/ /_/\__,_/_/  /_/

EOF
}
header_info
echo -e "Loading..."
APP="Homarr"
var_disk="8"
var_cpu="2"
var_ram="2048"
var_os="debian"
var_version="12"
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=""
  APT_CACHER=""
  APT_CACHER_IP=""
  DISABLEIP6="no"
  MTU=""
  SD=""
  NS=""
  MAC=""
  VLAN=""
  SSH="no"
  VERB="no"
  echo_default
}

function update_script() {
header_info
if [[ ! -d /opt/homarr ]]; then msg_error "No ${APP} Installation Found!"; exit; fi
if (( $(df /boot | awk 'NR==2{gsub("%","",$5); print $5}') > 80 )); then
  read -r -p "Warning: Storage is dangerously low, continue anyway? <y/N> " prompt
  [[ ${prompt,,} =~ ^(y|yes)$ ]] || exit
fi
RELEASE=$(curl -s https://api.github.com/repos/ajnart/homarr/releases/latest | grep "tag_name" | awk '{print substr($2, 3, length($2)-4) }')
if [[ ! -f /opt/${APP}_version.txt ]] || [[ "${RELEASE}" != "$(cat /opt/${APP}_version.txt)" ]]; then
  msg_info "Stopping Services"
  systemctl stop homarr
  msg_ok "Services Stopped"

  msg_info "Backing up Data"
  mkdir -p /opt/homarr-data-backup
  cp /opt/homarr/.env /opt/homarr-data-backup/.env
  cp /opt/homarr/database/db.sqlite /opt/homarr-data-backup/db.sqlite
  cp -r /opt/homarr/data/configs /opt/homarr-data-backup/configs
  msg_ok "Backed up Data"

  msg_info "Updating ${APP} to ${RELEASE}"
  wget -q "https://github.com/ajnart/homarr/archive/refs/tags/v${RELEASE}.zip"
  unzip -q v${RELEASE}.zip
  rm -rf v${RELEASE}.zip
  rm -rf /opt/homarr
  mv homarr-${RELEASE} /opt/homarr
  mv /opt/homarr-data-backup/.env /opt/homarr/.env
  cd /opt/homarr
  yarn install &>/dev/null
  yarn build &>/dev/null
  echo "${RELEASE}" >/opt/${APP}_version.txt
  msg_ok "Updated ${APP}"

  msg_info "Restoring Data"
  rm -rf /opt/homarr/data/configs
  mv /opt/homarr-data-backup/configs /opt/homarr/data/configs
  mv /opt/homarr-data-backup/db.sqlite /opt/homarr/database/db.sqlite
  yarn db:migrate &>/dev/null
  rm -rf /opt/homarr-data-backup
  msg_ok "Restored Data"

  msg_info "Starting Services"
  systemctl start homarr
  msg_ok "Started Services"
  msg_ok "Updated Successfully"
else
  msg_ok "No update required. ${APP} is already at ${RELEASE}"
fi
exit
}

start
build_container
description

msg_ok "Completed Successfully!\n"
echo -e "${APP} should be reachable by going to the following URL.
         ${BL}http://${IP}:3000${CL} \n"