#!/usr/bin/env bash
source <(curl -s https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/misc/build.func)
# Copyright (c) 2021-2024 community-scripts ORG
# Author: bvdberg01
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
# Source: https://netboxlabs.com/

# App Default Values
APP="NetBox"
var_tags="network"
var_cpu="2"
var_ram="2048"
var_disk="4"
var_os="debian"
var_version="12"
var_unprivileged="1"

# App Output & Base Settings
header_info "$APP"
base_settings

# Core
variables
color
catch_errors

function update_script() {
  header_info
  check_container_storage
  check_container_resources
  if [[ ! -f /etc/systemd/system/netbox.service ]]; then
    msg_error "No ${APP} Installation Found!"
    exit
  fi

  RELEASE=$(curl -s https://api.github.com/repos/netbox-community/netbox/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 ${APP}"
    systemctl stop netbox netbox-rq
    msg_ok "Stopped ${APP}"

    msg_info "Updating $APP to v${RELEASE}"
    mv /opt/netbox/ /opt/netbox-backup
    cd /opt
    wget -q "https://github.com/netbox-community/netbox/archive/refs/tags/v${RELEASE}.zip"
    unzip -q "v${RELEASE}.zip"
    mv /opt/netbox-${RELEASE}/ /opt/netbox/

    cp -r /opt/netbox-backup/netbox/netbox/configuration.py /opt/netbox/netbox/netbox/
    cp -r /opt/netbox-backup/netbox/media/ /opt/netbox/netbox/
    cp -r /opt/netbox-backup/netbox/scripts /opt/netbox/netbox/
    cp -r /opt/netbox-backup/netbox/reports /opt/netbox/netbox/
    cp -r /opt/netbox-backup/gunicorn.py /opt/netbox/

    if [ -f /opt/netbox-backup/local_requirements.txt ]; then
      cp -r /opt/netbox-backup/local_requirements.txt /opt/netbox/
    fi

    if [ -f /opt/netbox-backup/netbox/netbox/ldap_config.py ]; then
      cp -r /opt/netbox-backup/netbox/netbox/ldap_config.py /opt/netbox/netbox/netbox/
    fi

    /opt/netbox/upgrade.sh &>/dev/null
    echo "${RELEASE}" >/opt/${APP}_version.txt
    msg_ok "Updated $APP to v${RELEASE}"

    msg_info "Starting ${APP}"
    systemctl start netbox netbox-rq
    msg_ok "Started ${APP}"

    msg_info "Cleaning up"
    rm -r "/opt/v${RELEASE}.zip"
    rm -r /opt/netbox-backup
    msg_ok "Cleaned"
    msg_ok "Updated Successfully"
  else
    msg_ok "No update required. ${APP} is already at v${RELEASE}"
  fi
  exit
}

start
build_container
description

msg_ok "Completed Successfully!\n"
echo -e "${CREATING}${GN}${APP} setup has been successfully initialized!${CL}"
echo -e "${INFO}${YW} Access it using the following URL:${CL}"
echo -e "${TAB}${GATEWAY}${BGN}https://${IP}${CL}"