main.sh

#!/usr/bin/env bash
set -euo pipefail
IFS=$'\n\t'

# Usage Information
#/ Usage: sudo setupopenflixr [OPTION]
#/
#/ This is the main OpenFLIXR2 Setup Script script.
#/
#/ Usage: setupopenflixr [OPTION]
#/ NOTE: setupopenflixr shortcut is only available after the first run
#/
#/ For regular usage you can run without providing any options.
#/
#/  -d --devmode <dev_level>
#/      run in devmode. <dev_level> defaults to 1 if no level is provided
#/  -f
#/      run all fixes
#/  -f <fix_name>
#/      run specified fix
#/  -i
#/      run install
#/  -l
#/      submit logs
#/  -s
#/      symlink setupopenflixr
#/  -t --test <test_name>
#/      run tests to check the program
#/  -u --update
#/      update OpenFLIXR2 Setup Script to the latest stable commits
#/  -u --update <branch>
#/      update OpenFLIXR2 Setup Script to the latest commits from the specified branch
##/ -v --verbose
##/     verbose
#/  -x --debug
#/      debug
#/
usage() {
    grep '^#/' "${SCRIPTNAME}" | cut -c4- || echo "Failed to display usage information."
    exit
}

# Command Line Arguments
readonly ARGS=("$@")

# Github Token for Travis CI
if [[ ${CI:-} == true ]] && [[ ${TRAVIS:-} == true ]] && [[ ${TRAVIS_SECURE_ENV_VARS} == true ]]; then
    readonly GH_HEADER="Authorization: token ${GH_TOKEN}"
fi

# Script Information
# https://stackoverflow.com/a/246128/1384186
get_scriptname() {
    local SOURCE
    local DIR
    SOURCE="${BASH_SOURCE[0]:-$0}" # https://stackoverflow.com/questions/35006457/choosing-between-0-and-bash-source
    while [[ -L ${SOURCE} ]]; do # resolve ${SOURCE} until the file is no longer a symlink
        DIR="$(cd -P "$(dirname "${SOURCE}")" > /dev/null && pwd)"
        SOURCE="$(readlink "${SOURCE}")"
        [[ ${SOURCE} != /* ]] && SOURCE="${DIR}/${SOURCE}" # if ${SOURCE} was a relative symlink, we need to resolve it relative to the path where the symlink file was located
    done
    echo "${SOURCE}"
}
readonly SCRIPTNAME="$(get_scriptname)"
readonly SCRIPTPATH="$(cd -P "$(dirname "${SCRIPTNAME}")" > /dev/null && pwd)"

# Other variables
readonly UBU_VER=$(lsb_release -rs)
readonly PUBLIC_IP=$(dig @ns1-1.akamaitech.net ANY whoami.akamai.net +short)
if [ $? -eq 0 ]; then
    HAS_INTERNET=1
else
    HAS_INTERNET=0
fi
readonly NIC=$(ip -o -4 route show to default | head -1 | awk '{print $5}')
readonly LOCAL_IP=$(ifconfig ${NIC} | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*')
readonly ROUTER_IP=$(ip route show | grep -i 'default via'| awk '{print $3 }')
readonly STORE_PATH="/home/openflixr/openflixr_setup"
readonly CONFIG_FILE="${STORE_PATH}/openflixr_setup.config"
readonly CONFIG_FILES_OLD=("/opt/OpenFLIXR2.SetupScript/openflixr_setup.config", "/home/openflixr/openflixr_setup.config")
readonly OPENFLIXR_FOLDERS=(downloads movies series music comics books)
# Init config array
typeset -A config
config=(
    [STEPS_CURRENT]=0
    [CHANGE_PASS]=""
    [NETWORK]=""
    [OPENFLIXR_IP]=${LOCAL_IP}
    [OPENFLIXR_SUBNET]="255.255.255.0"
    [OPENFLIXR_GATEWAY]=""
    [OPENFLIXR_DNS]="127.0.0.1"
    [ACCESS]=""
    [LETSENCRYPT]="off"
    [OPENFLIXR_DOMAIN]=""
    [OPENFLIXR_EMAIL]=""
    [SERIES_MANAGER]='sonarr'
    [MOVIE_MANAGER]='couchpotato'
    [NZB_DOWNLOADER]='sabnzbd'
    [MOUNT_MANAGE]=""
    [HOST_NAME]=""
    [FSTAB_BACKUP]=0
    [FSTAB_MODIFIED]=0
    [OPENFLIXR_PLEX_CLAIM]="claim-YYYYYYYYY"
    [OPENFLIXR_TIMEZONE]=""
    [OPENFLIXR_TIMEZONE_SHORT]=""
    [OPENFLIXR_READY]=""
    [ALWAYS_SUBMIT_LOGS]=""
    [DISCORD_USERNAME]=""
    [SUBMITTED_LOGS]=""
    [SUBMITTED_LOGS_VERSION]=""
    [SETUP_COMPLETED]=""
    [BRANCH]="origin/master"
)
#for FOLDER in ${OPENFLIXR_FOLDERS[@]}; do
#    config[MOUNT_TYPE_$FOLDER]=""
#done
# Init services array
declare -A SERVICES
SERVICES=(
    # System Processes
    [monit]=monit
    [htpcmanager]=htpcmanager
    # Apps
    [couchpotato]=couchpotato
    [sickrage]=sickrage
    [headphones]=headphones
    [mylar]=mylar
    [sabnzbd]=sabnzbdplus
    [jackett]=jackett
    [sonarr]=sonarr
    [radarr]=radarr
    [plexpy]=plexpy
    # Apps - other
    [ombi]=ombi
    [lidarr]=lidarr
    [lazylibrarian]=lazylibrarian
    [mopidy]=mopidy
    [nzbhydra2]=nzbhydra2
)
# init API keys array
declare -A API_KEYS
# OpenFLIXT password variables
OPENFLIXR_PASSWORD_NEW=""
OPENFLIXR_PASSWORD_OLD=""

# User/Group Information
readonly DETECTED_PUID=${SUDO_UID:-$UID}
readonly DETECTED_UNAME=$(id -un "${DETECTED_PUID}" 2> /dev/null || true)
readonly DETECTED_PGID=$(id -g "${DETECTED_PUID}" 2> /dev/null || true)
readonly DETECTED_UGROUP=$(id -gn "${DETECTED_PUID}" 2> /dev/null || true)
readonly DETECTED_HOMEDIR=$(eval echo "~${DETECTED_UNAME}" 2> /dev/null || true)

# Colors
# https://misc.flogisoft.com/bash/tip_colors_and_formatting
readonly BLU='\e[34m'
readonly GRN='\e[32m'
readonly RED='\e[31m'
readonly YLW='\e[33m'
readonly NC='\e[0m'

# Log Functions
readonly LOG_FILE="/var/log/openflixr_setup.log"
sudo chown "${DETECTED_PUID:-$DETECTED_UNAME}":"${DETECTED_PGID:-$DETECTED_UGROUP}" "${LOG_FILE}" > /dev/null 2>&1 || true # This line should always use sudo
log() {
    if [[ -v DEBUG && $DEBUG == 1 ]] || [[ -v VERBOSE && $VERBOSE == 1 ]] || [[ -v DEVMODE && $DEVMODE == 1 ]]; then
        echo -e "${NC}$(date +"%F %T") ${BLU}[LOG]${NC}        $*${NC}" | tee -a "${LOG_FILE}";
    else
        echo -e "${NC}$(date +"%F %T") ${BLU}[LOG]${NC}        $*${NC}" | tee -a "${LOG_FILE}" > /dev/null;
    fi
}
info() { echo -e "${NC}$(date +"%F %T") ${BLU}[INFO]${NC}       $*${NC}" | tee -a "${LOG_FILE}"; }
warning() { echo -e "${NC}$(date +"%F %T") ${YLW}[WARNING]${NC}    $*${NC}" | tee -a "${LOG_FILE}"; }
error() { echo -e "${NC}$(date +"%F %T") ${RED}[ERROR]${NC}      $*${NC}" | tee -a "${LOG_FILE}"; }
fatal() {
    echo -e "${NC}$(date +"%F %T") ${RED}[FATAL]${NC}      $*${NC}" | tee -a "${LOG_FILE}"
    exit 1
}
debug() {
    if [[ -v DEBUG && $DEBUG == 1 ]] || [[ -v VERBOSE && $VERBOSE == 1 ]] || [[ -v DEVMODE && $DEVMODE == 1 ]]; then
        echo -e "${NC}$(date +"%F %T") ${GRN}[DEBUG]${NC}      $*${NC}" | tee -a "${LOG_FILE}"
    fi
}

# Script Runner Function
run_script() {
    local SCRIPTSNAME="${1:-}"
    shift
    if [[ -f "/opt/OpenFLIXR2.SetupScript/.scripts/${SCRIPTSNAME}.sh" ]]; then
        source "/opt/OpenFLIXR2.SetupScript/.scripts/${SCRIPTSNAME}.sh"
        if [[ "${SCRIPTSNAME}" == setup_configure* ]]; then
            ${SCRIPTSNAME} "$@" 2> >(tee -a "${LOG_FILE}")
        else
            ${SCRIPTSNAME} "$@"
        fi
    else
        fatal "/opt/OpenFLIXR2.SetupScript/.scripts/${SCRIPTSNAME}.sh not found."
    fi
}

# Test Runner Function
run_test() {
    local TESTSNAME="${1:-}"
    shift
    if [[ -f ${SCRIPTPATH}/.tests/${TESTSNAME}.sh ]]; then
        # shellcheck source=/dev/null
        source "${SCRIPTPATH}/.tests/${TESTSNAME}.sh"
        ${TESTSNAME} "$@"
    else
        fatal "${SCRIPTPATH}/.tests/${TESTSNAME}.sh not found."
    fi
}

# Root Check
root_check() {
    if [[ ${DETECTED_PUID} == "0" ]] || [[ ${DETECTED_HOMEDIR} == "/root" ]]; then
        fatal "Running as root is not supported. Please run as openflixr user using sudo."
    fi
}

# Cleanup Function
cleanup() {
    if [[ $? = 1 ]]; then
        run_script 'save_config'
        run_script 'load_config'
        if [[ ${LOG_SUBMISSION:-} != "DISABLED" ]]; then
            SUBMIT_MESSAGE="It appears and error has occurred.\nDo you want to submit the logs now?"
            run_script 'submit_logs'
        fi
    fi
    #if [[ ${SCRIPTPATH} == "/opt/OpenFLIXR2.SetupScript" ]]; then
    #    chmod +x "${SCRIPTNAME}" > /dev/null 2>&1 || fatal "${SCRIPTNAME} must be executable."
    #fi
    if [[ ${CI:-} == true ]] && [[ ${TRAVIS:-} == true ]] && [[ ${TRAVIS_SECURE_ENV_VARS} == false ]]; then
        warning "TRAVIS_SECURE_ENV_VARS is false for Pull Requests from remote branches. Please retry failed builds!"
    fi
}
trap 'cleanup' 0 1 2 3 6 14 15

# Main Function
main() {
    # Arch Check
    readonly ARCH=$(uname -m)
    if [[ ${ARCH} != "aarch64" ]] && [[ ${ARCH} != "armv7l" ]] && [[ ${ARCH} != "x86_64" ]]; then
        fatal "Unsupported architecture."
    fi
    # Terminal Check
    if [[ -n ${PS1:-} ]] || [[ ${-} == *"i"* ]]; then
        root_check
    fi
    # Sudo Check
    if [[ ${EUID} != "0" ]]; then
        (sudo setupopenflixr "${ARGS[@]:-}") || true
        exit
    fi
    cd "${SCRIPTPATH}" || fatal "Failed to change to ${SCRIPTPATH} directory."
    readonly GIT_REPO=$(git config --get remote.origin.url)
    readonly PROMPT="GUI"

    run_script 'symlink_setupopenflixr'
    run_script 'create_config'
    # shellcheck source=/dev/null
    source "${SCRIPTPATH}/.scripts/cmdline.sh"
    cmdline "${ARGS[@]:-}"
    savelog -n -C -l -t "$LOG_FILE" >> ${LOG_FILE} #Save current log if not empty and rotate logs

    # Ubuntu Version Check
    if [[ ${UBU_VER} != "18.04" ]]; then
        error "Unsupported Ubuntu Version - ${UBU_VER}. This setup can only be run for OpenFLIXR running Ubuntu 18.04"
        error "Make sure you have completed the steps found here: https://github.com/cyberlooper/Docs/wiki/Setup#getting-set-up"
        error "If you have, check the 'Issues & Troubleshooting' section on that same page."
        exit 0
    fi

    run_script 'load_config'
    readonly LOCAL_COMMIT=$(git rev-parse --short ${config[BRANCH]})
    readonly OF_BACKTITLE="OpenFLIXR Setup - $LOCAL_COMMIT"
    info "${OF_BACKTITLE}"
    run_script 'ombi_check'
    echo "Ombi Setup completed"
    run_script 'check_version'
    run_script 'check_dependencies'
    run_script 'save_config'

    #run_script 'unmaintained'

    if [[ ${config[SETUP_COMPLETED]} == "Y" ]]; then
        run_script 'menu_main'
    else
        run_script 'run_steps'
    fi
}
main