#!/bin/bash

SCRIPT_VER="2.1"

APP="NZBGet"
APP_CATEGORY="Torrent Clients and Utilities"
APP_SCOPE="user"
APP_ENABLED="yes"
STORED_PASSWORD_SERVICE="nzbget"
#><#APP_CHECK_VER_CMD="if [ -x /opt/nzbget/nzbget ] ; then /opt/nzbget/nzbget --version | awk \"{print \\$NF}\" ; else if command -v nzbget > /dev/null 2>&1 ; then nzbget --version | awk \"{print \\$NF}\" ; else if [ -x /opt/nzbgetcom/nzbget ] ; then /opt/nzbgetcom/nzbget --version | awk \"{print \\$NF}\" ; else echo \"\" ; fi ; fi ; fi"
#><#APP_UPGRADE_VER_CMD="curl -Ls -m 30 -o /dev/null -w \"%{url_effective}\" https://github.com/nzbgetcom/nzbget/releases/latest | awk -F/ \"{print \\$NF}\" | tr -d \"v\""
SUPPORTED_ON="Ubuntu20.04,Ubuntu22.04,Ubuntu24.04"
DAEMON_PORT=6789
USE_APACHE2="yes"
LIBRARY="functions4.sh"

if [ "${SUPPORT-}" = '' ]; then
    SUPPORT="https://installers.rapidseedbox.com/scripts/whmcs_client_installers/support"
fi

LOGFILE="/var/log/whmcs_client_installers/${0}.log"
[ ! -d "/var/log/whmcs_client_installers" ] && mkdir -p "/var/log/whmcs_client_installers"

wget -4 -nc -q --no-check-certificate --timeout=20 --tries=5 ${SUPPORT}/${LIBRARY} >> "${LOGFILE}" 2>&1
if [ ! -f "./${LIBRARY}" ] ; then
    echo "55"
    echo "Error downloading \"${SUPPORT}/${LIBRARY}\"" >> "${LOGFILE}"
    exit 55
fi

# shellcheck source=support/functions4.sh
. "./${LIBRARY}"
SCRIPTENTRY

VALIDATE_USER

APP_HOME="/opt/nzbget"
APP_DIRECTORY="${USERHOME}/.config/nzbget"
APP_DATA_DIR="${USERHOME}/NZBGet"
APP_SERVICE="nzbget@${USERNAME}"
APP_APACHE_CONF="/etc/apache2/userconf-enabled/nzbget-${USERNAME}.conf"
APP_MONIT_CONF="/etc/monit/conf.d/nzbget@${USERNAME}"
LEGACY_APP_HOME="/opt/nzbgetcom"
LEGACY_APP_DIRECTORY="${USERHOME}/.config/nzbgetcom"
LEGACY_APP_DATA_DIR="${USERHOME}/NZBGetCom"
LEGACY_APP_SERVICE="nzbgetcom@${USERNAME}"
LEGACY_APACHE_CONF="/etc/apache2/userconf-enabled/nzbgetcom-${USERNAME}.conf"
LEGACY_MONIT_CONF="/etc/monit/conf.d/nzbgetcom@${USERNAME}"
REMOVE_LEGACY_DPKG="no"
MOVED_MAIN_DIR="no"

legacy_nzbget_pkg_installed() {
    [ -d "${APP_DIRECTORY}" ] || return 1
    [ -x "${APP_HOME}/nzbget" ] && return 1
    command -v nzbget > /dev/null 2>&1 || DPKG_CHECK "nzbget"
}

legacy_nzbgetcom_installed() {
    [ -d "${LEGACY_APP_DIRECTORY}" ] \
        || [ -x "${LEGACY_APP_HOME}/nzbget" ] \
        || [ -e "${LEGACY_APACHE_CONF}" ] \
        || [ -e "${LEGACY_MONIT_CONF}" ] \
        || [ -e "/etc/systemd/system/nzbgetcom@.service" ]
}

needs_legacy_migration() {
    legacy_nzbget_pkg_installed || legacy_nzbgetcom_installed
}

stop_legacy_services() {
    systemctl is-active "${APP_SERVICE}" > /dev/null 2>&1 \
        && SVC_STOP "${APP_SERVICE}"
    systemctl is-active "${LEGACY_APP_SERVICE}" > /dev/null 2>&1 \
        && SVC_STOP "${LEGACY_APP_SERVICE}"
}

migrate_legacy_layout() {
    if legacy_nzbget_pkg_installed ; then
        REMOVE_LEGACY_DPKG="yes"
    fi

    legacy_nzbgetcom_installed || return 0

    if [ ! -d "${USERHOME}/.config" ] ; then
        MKDIR "${USERHOME}/.config" || return 1
        CHOWN_DIR "${USERNAME}:${USERNAME}" "${USERHOME}/.config" || return 1
    fi

    if [ -d "${LEGACY_APP_DIRECTORY}" ] && [ ! -e "${APP_DIRECTORY}" ] ; then
        MV "${LEGACY_APP_DIRECTORY}" "${APP_DIRECTORY}" || return 1
    elif [ -f "${LEGACY_APP_DIRECTORY}/nzbget.conf" ] && [ ! -f "${APP_DIRECTORY}/nzbget.conf" ] ; then
        MKDIR "${APP_DIRECTORY}" || return 1
        MV "${LEGACY_APP_DIRECTORY}/nzbget.conf" "${APP_DIRECTORY}/" || return 1
    fi

    if [ -d "${LEGACY_APP_DATA_DIR}" ] && [ ! -e "${APP_DATA_DIR}" ] ; then
        MV "${LEGACY_APP_DATA_DIR}" "${APP_DATA_DIR}" || return 1
        MOVED_MAIN_DIR="yes"
    fi

    [ -e "${LEGACY_APACHE_CONF}" ] && RM_FILES "${LEGACY_APACHE_CONF}"
    [ -e "${LEGACY_MONIT_CONF}" ] && RM_FILES "${LEGACY_MONIT_CONF}"

    systemctl is-enabled "${LEGACY_APP_SERVICE}" > /dev/null 2>&1 \
        && SVC_DISABLE "${LEGACY_APP_SERVICE}"

    return 0
}

cleanup_legacy_runtime() {
    local _ret

    if [ "${REMOVE_LEGACY_DPKG}" = "yes" ] && DPKG_CHECK "nzbget" ; then
        APT_REMOVE "nzbget" || return 1
        APT_AUTOREMOVE || return 1
    fi

    _ret="$( find /home -type d -a -iwholename '*/.config/nzbgetcom' 2> /dev/null )"
    if [ -z "${_ret}" ] ; then
        [ -e "/etc/systemd/system/nzbgetcom.slice" ] && RM_FILES "/etc/systemd/system/nzbgetcom.slice"
        [ -e "/etc/systemd/system/nzbgetcom@.service" ] && RM_FILES "/etc/systemd/system/nzbgetcom@.service"
        [ -d "${LEGACY_APP_HOME}" ] && RM_DIRS "${LEGACY_APP_HOME}"
        SYSTEMD_RELOAD
    fi
}

ensure_data_dirs() {
    MKDIR "${APP_DATA_DIR}" || return 1
    for __dir in \
        "Movies" \
        "Series" \
        "Music" \
        "Software" \
        "completed" \
        "intermediate" \
        "nzb" \
        "queue" \
        "tmp" \
    ; do
        MKDIR "${APP_DATA_DIR}/${__dir}" || return 1
    done
    CHOWN_DIR "${USERNAME}:${USERNAME}" "${APP_DATA_DIR}" || return 1
}

prepare_config() {
    MKDIR "${APP_DIRECTORY}" || return 1

    if [ ! -f "${APP_DIRECTORY}/nzbget.conf" ] ; then
        MV "${APP_HOME}/nzbget.conf" "${APP_DIRECTORY}/" || return 1
    elif [ -e "${APP_HOME}/nzbget.conf" ] ; then
        RM_FILES "${APP_HOME}/nzbget.conf"
    fi

    RUN_LOG "echo 's|^ControlIP=.*$|ControlIP=127.0.0.1|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^ControlPort=.*$|ControlPort=${NZBGET_CONTROL_PORT}|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^AuthorizedIP=.*$|AuthorizedIP=127.0.0.1|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^LogFile=.*$|LogFile=${USERHOME}/log/nzbget.log|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^WebDir=.*$|WebDir=/opt/nzbget/webui|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^ConfigTemplate=.*$|ConfigTemplate=/opt/nzbget/webui/nzbget.conf.template|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    if [ "${MOVED_MAIN_DIR}" = "yes" ] || [ ! -d "${LEGACY_APP_DATA_DIR}" ] ; then
        RUN_LOG "echo 's|^MainDir=.*$|MainDir=~/NZBGet|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    fi
    RUN_LOG "echo 's|^DirectUnpack=.*$|DirectUnpack=no|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^DownloadRate=.*$|DownloadRate=30000|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^\\(ParCleanupQueue=.*\\)$|#\\1|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"
    RUN_LOG "echo 's|^\\(DeleteCleanupDisk=.*\\)$|#\\1|' | sed -f - -i '${APP_DIRECTORY}/nzbget.conf'"

    CHOWN_DIR "${USERNAME}:${USERNAME}" "${APP_DIRECTORY}" || return 1
}

write_systemd_units() {
    cat <<-EOT > "/etc/systemd/system/nzbget.slice" 2>> "${LOGFILE}"
	[Unit]
	Description=NZBGet resource limiting
	Before=slices.target

	[Slice]
	CPUAccounting=yes
	CPUQuota=45%
	IOAccounting=yes
	IOReadBandwidthMax=/ 20M
	IOWriteBandwidthMax=/ 20M
	EOT

    cat <<-EOT > "/etc/systemd/system/nzbget@.service" 2>> "${LOGFILE}"
	[Unit]
	Description=NZBGet Daemon
	Documentation=http://nzbget.com/documentation
	After=network.target

	[Service]
	User=%i
	Group=%i
	Type=forking
	ExecStart=/opt/nzbget/nzbget -c /home/%i/.config/nzbget/nzbget.conf -D
	ExecStartPost=/bin/sh -c "echo \$MAINPID > /home/%i/.config/nzbget/nzbget.pid"
	ExecStop=/opt/nzbget/nzbget -Q
	ExecReload=/opt/nzbget/nzbget -O
	KillMode=process
	Restart=on-failure
	Slice=nzbget.slice

	[Install]
	WantedBy=multi-user.target
	EOT

    SYSTEMD_RELOAD
}

write_apache_conf() {
    [ -d "/etc/apache2/userconf-enabled" ] || return 0

    if [ "${USERUID}" = "${FIRST_UID}" ] ; then
        cat <<-EOT > "${APP_APACHE_CONF}" 2>> "${LOGFILE}"
		RewriteEngine on
		RewriteRule ^/nzbget\$ /nzbget/ [R=permanent]
		ProxyPass /nzbget http://localhost:${NZBGET_CONTROL_PORT}
		ProxyPassReverse /nzbget http://localhost:${NZBGET_CONTROL_PORT}

		<Location /nzbget/>
		  AuthType Digest
		  AuthName "NZBGet"
		  AuthDigestDomain "/nzbget/"
		  AuthDigestProvider file
		  AuthUserFile "/etc/apache2/nzbget.users.password"
		  Require user ${USERNAME}
		</Location>
		EOT
    else
        cat <<-EOT > "${APP_APACHE_CONF}" 2>> "${LOGFILE}"
		RewriteEngine on
		RewriteRule ^/nzbget-${USERNAME}\$ /nzbget-${USERNAME}/ [R=permanent]
		ProxyPass /nzbget-${USERNAME} http://localhost:${NZBGET_CONTROL_PORT}
		ProxyPassReverse /nzbget-${USERNAME} http://localhost:${NZBGET_CONTROL_PORT}

		<Location /nzbget-${USERNAME}/>
		  AuthType Digest
		  AuthName "NZBGet"
		  AuthDigestDomain "/nzbget-${USERNAME}/"
		  AuthDigestProvider file
		  AuthUserFile "/etc/apache2/nzbget.users.password"
		  Require user ${USERNAME}
		</Location>
		EOT
    fi

    SVC_RESTART "apache2"
}

write_password_file() {
    local passwd_file
    local pline

    passwd_file="/etc/apache2/nzbget.users.password"
    pline="$( ( echo -n "${USERNAME}:NZBGet:" && echo -n "${USERNAME}:NZBGet:${USERPASSWORD}" | md5sum | awk '{print $1}' ) )"
    if [ -e "${passwd_file}" ] && grep -q -e "^${USERNAME}:NZBGet:" "${passwd_file}" ; then
        RUN_LOG "echo 's#^${USERNAME}:NZBGet:.*#${pline}#' | sed -f - -i '${passwd_file}'"
    else
        echo "${pline}" >> "${passwd_file}"
    fi
}

write_monit_conf() {
    [ -d "/etc/monit/conf.d" ] || return 0

    cat <<-EOT > "${APP_MONIT_CONF}" 2>> "${LOGFILE}"
		check process nzbget-${USERNAME} with pidfile ${APP_DIRECTORY}/nzbget.pid
		  start program = "/usr/sbin/service nzbget@${USERNAME} restart"
		  stop program = "/usr/sbin/service nzbget@${USERNAME} stop"
		EOT
    MONIT_RELOAD
}

while : ; do

    main_rc=1

    _legacy_migration="no"
    needs_legacy_migration && _legacy_migration="yes"

    CHECK_APP ; _rc=$?
    case "${_rc}" in
      111)
          if [ "${_legacy_migration}" = "yes" ] ; then
              INFO "Legacy NZBGet installation detected. Continuing with migration."
          else
              main_rc=${_rc}
              break
          fi
          ;;
      0)
          ;;
      *)
          main_rc=${_rc}
          break
          ;;
    esac

    stop_legacy_services
    migrate_legacy_layout || break

    RM_FILES "${TMP_DIR}/nzbget-latest-bin-linux.run"
    WGET "https://nzbget.com/download/nzbget-latest-bin-linux.run" || break
    INFO "Executing ${APP} installer"
    sh "${TMP_DIR}/nzbget-latest-bin-linux.run" --destdir "${APP_HOME}" || break
    INFO "Execution finished"

    ensure_data_dirs || break

    NZBGET_CONTROL_PORT="$(( USERUID + 6789 - FIRST_UID ))"

    prepare_config || break
    write_systemd_units || break

    SVC_ENABLE "${APP_SERVICE}"
    SVC_START  "${APP_SERVICE}"

    write_apache_conf || break
    write_password_file || break
    write_monit_conf || break
    cleanup_legacy_runtime || break

    main_rc=0
    break

done

SCRIPTEXIT