#!/bin/bash
#set -x

###########################################################
## Script to install the latest WebTitancloud OTG DNS Proxy
###########################################################


### Customer defined variables
# PKG file download link
webURL="https://www.example.com/downloads/otg_setup.pkg"                                           # URL to .pkg setup file

# PKG file local install
localInstall="false"                                                                        # If "false", .pkg file will be downloaded from $webURL. 
localInstallPath="/example/otg_setup.pkg"                                                   # Used when $localInstall is "true"

# WTC Configuration (required)
rpc_url="https://mycloud.webtitancloud.com"                                                 # Required: The URL (and optional port) of your WebTitan Cloud instance in FQDN format.
install_key="00000z0z-0zz0-0000-zz0z-0zz000z0zz0z"                                          # Required: The OTG install key used to register an OTG 2 device on WebTitan Cloud.
location_name="OTG_Mac_Prod"                                                                # Optional: A virtual location is automatically created on WebTitan with a default name. Use location_name to specify a custom location name instead.


###########################################################
### Application variables
appname="WebTitanCloud"
app="WebTitanCloud.app"
logandmetadir="/Library/Logs/Microsoft/IntuneScripts/installWebTitan"
processpath="/Applications/WebTitanCloud.app/Contents/MacOS/WebTitanCloud"
terminateprocess="true"
otgPath="/Library/LaunchAgents/com.titanhq.otg.plist"
tempdir=$(mktemp -d)
log="$logandmetadir/$appname.log"
loggedInUser=$(stat -f '%Su' /dev/console)
loggedInUserId=$(stat -f '%u' /dev/console)

### Functions
waitForProcess () {

    processName=$1
    fixedDelay=$2
    terminate=$3

    echo "$(date) | Waiting for other [$processName] processes to end"
    while ps aux | grep "$processName" | grep -v grep &>/dev/null; do

        if [[ $terminate == "true" ]]; then
            echo "$(date) | + [$appname] running, terminating [$processpath]..."
            pkill -f "$processName"
            bootOutApp
            return
        fi

        # Randomize delay if argument not passed
        if [[ ! $fixedDelay ]]; then
            delay=$(( $RANDOM % 50 + 10 ))
        else
            delay=$fixedDelay
        fi

        echo "$(date) |  + Another instance of $processName is running, waiting [$delay] seconds"
        sleep $delay
    done
    
    echo "$(date) | No instances of [$processName] found, safe to proceed"

}

function downloadApp () {

    echo "$(date) | Starting downlading of [$appname]"

    # Wait for other downloads to complete
    waitForProcess "curl -f"

    # Download the file
    echo "$(date) | Downloading $appname"

    tempfile="$tempdir/otg_setup.pkg"
    curl -L -f -s --connect-timeout 30 --retry 5 --retry-delay 60 --output "$tempfile" "$webURL"

    if [[ -f "$tempfile" ]]; then
        echo "$(date) | Successfully downloaded [$tempfile]"
    else
        echo "$(date) | Failed to download [$webURL] to [$tempfile]"
    fi

}

## Install PKG Function
function installPKG () {

    # Wait if app is running
    waitForProcess "$processpath" "300" "$terminateprocess"
    
    echo "$(date) | Installing $appname"

    # Remove existing files if present
    if [[ -d "/Applications/$app" ]]; then
        rm -rf "/Applications/$app"
    fi
    
    if [[ $localInstall == "true" ]]; then
        installer -pkg "$localInstallPath" -target /
    else
        installer -pkg "$tempfile" -target /
    fi
    
    # Checking if the app was installed successfully
    if [ "$?" = "0" ]; then

        echo "$(date) | $appname Installed"
        echo "$(date) | Cleaning Up"
        rm -rf "$tempdir"
    else
        echo "$(date) | Failed to install $appname"
        rm -rf "$tempdir"
    fi

}

function runOTGProxy() {

    # Params array for WebTitanCloud binary
    PARAMS=(RPC_URL="$rpc_url")
    PARAMS+=(INSTALL_KEY="$install_key")
    
    # Check if parameter null or empty
    if [[ ! -z $location_name ]]; then 
        PARAMS+=(LOCATION_NAME="$location_name")
    fi

    echo "$(date) | Configuring [$appname] for [$loggedInUser]"
    echo "$(date) | Configuring  $processpath ${PARAMS[@]}"
    sudo -u $loggedInUser $processpath "${PARAMS[@]}"

}

function startLog() {

    echo ""
    echo "# $(date) | Logging install of [$appname] to [$log]"
    echo "############################################################"
    echo ""

    if [[ ! -d "$logandmetadir" ]]; then
        echo "$(date) | Creating [$logandmetadir] to store logs"
        mkdir -p "$logandmetadir"
    fi

    exec &> >(tee -a "$log")
    
}

### Main script
# Initiate logging
startLog

# Download app
if [[ $localInstall == "false" ]]; then
    downloadApp
fi

# Kick OTG out of launchCTL
echo "$(date) Booting out app with /bin/launchctl bootout gui/$loggedInUserId $otgPath"
/bin/launchctl bootout gui/$loggedInUserId $otgPath &> /dev/null

# Install WebTitan OTG
if [[ -d "/Applications/$app" ]]; then 
    ### Application health checks
    OTGDNSProxy=$(ps aux | grep -v grep | grep -ci OTGDNSProxy)
    WebTitanCloud=$(ps aux | grep -v grep | grep -ci WebTitanCloud)
    otgHelper=$(ps aux | grep -v grep | grep -ci otgHelper)

    echo "$(date) Existing installation of $appname found, upgrading..."
    installPKG

    if [ ! $OTGDNSProxy -gt 0 ] || [ ! $WebTitanCloud -gt 0 ] || [ ! $otgHelper -gt 0 ]; then
        echo "$(date) Issues detected with previous installation, re-running configuration"
        runOTGProxy
    fi

else 
    echo "$(date) Starting fresh installation of $appname"
    installPKG
    runOTGProxy
fi