#!/bin/sh
source /eng/ssb/auto/astroconda/include/sysinfo.sh
porcelain_continuum_url=https://repo.continuum.io/miniconda
porcelain_continuum_script=Miniconda3-latest-${sysinfo_platform}-${sysinfo_arch}.sh
PORCELAIN_ALREADY_DEAD=0
PORCELAIN_SIGNALED=0

function porcelain_init
{
    echo "Porcelain initializing..."
    if [[ -z $HOME ]]; then
        echo "\$HOME is not set, dying."
        exit 1
    fi
    
    export TMPDIR="$HOME/bldtmp"
    
    if [[ ! -d $TMPDIR ]]; then
        mkdir -p $TMPDIR
        if [[ $? != 0 ]]; then
            echo "Cannot create temporary storage directory '$TMPDIR', dying."
            exit 1
        fi
    fi

    export PORCELAIN_PREFIX=`mktemp -d -t porcelain.XXXXXXXXXX` 
    export PORCELAIN_TMPDIR="$PORCELAIN_PREFIX/tmp"
    export PORCELAIN_DESTDIR="$PORCELAIN_PREFIX/porcelain"
    export PATH="$PORCELAIN_DESTDIR/bin:$PATH"
    echo "Prepended $PORCELAIN_DESTDIR/bin to PATH..."

    if [[ ! -d $PORCELAIN_TMPDIR ]]; then
        mkdir -p "$PORCELAIN_TMPDIR"
    fi

    export TMPDIR="$PORCELAIN_TMPDIR"
    echo "Redirected TMPDIR to $PORCELAIN_TMPDIR"
    echo "(Always use \$TMPDIR instead of /tmp for destructible storage)"
    
    echo "Activating signal handler... (will deinit on exit)"
    trap porcelain_signal SIGINT SIGKILL SIGTERM EXIT
}

function porcelain_verify
{
    local override="$1"
    /bin/echo -n "Running safety check... "

    if [[ $override == '-f' ]] || [[ $override == '--force' ]]; then
        echo "done (forced)"
        return 0
    fi

    if [[ -z $TMPDIR ]]; then
        echo "failed"
        echo "TMPDIR='$TMPDIR'; we have lost trust in our environment. Dying."
        exit 1
    fi

    if [[ ! -d $PORCELAIN_PREFIX ]]; then
        echo "failed"
        echo "PORCELAIN_PREFIX='$PORCELAIN_PREFIX'; mktemp must have failed. Dying."
        exit 1
    fi

    # Note: we don't care about PORCELAIN_DESTDIR
    echo "done"
}

function porcelain_get_installer
{
    porcelain_verify

    retries=3
    wtime=5
    success=0
    count=0

    echo "Obtaining $porcelain_continuum_script..."

    while [[ $count != $retries ]]
    do
        $sysinfo_fetch $sysinfo_fetch_args "$porcelain_continuum_url/$porcelain_continuum_script"
        if [[ $? == 0 ]]; then
            success=1
            break
        fi

        echo "Retrying in $wtime second(s)..."
        sleep $wtime
        count=$(( count + 1 ))
    done

    if [[ $success != 1 ]]; then
        echo "Dying..."
        exit 1
    fi
}

function porcelain_run_installer
{
    porcelain_verify

    bash $porcelain_continuum_script -b -p $PORCELAIN_DESTDIR
    if [[ $? > 0 ]]; then
        echo "Dying..."
        exit 1
    fi
}

function porcelain_deinit
{
    if [[ $PORCELAIN_ALREADY_DEAD != 0 ]]; then
        return
    fi

    porcelain_verify

    echo "Deinitializing porcelain..."
    if [[ -d $PORCELAIN_PREFIX ]]; then
        echo "Removing $PORCELAIN_PREFIX"
        rm -rf "$PORCELAIN_PREFIX"
    fi
    export PORCELAIN_ALREADY_DEAD=1
}

# Given the chance this script could be killed in a number of different
# ways, resulting in files left behind... we need to be pretty damn
# thourough.
function porcelain_signal
{
    retval=$?
    if [[ $PORCELAIN_SIGNALED != 0 ]]; then
        exit $?
    fi

    export PORCELAIN_SIGNALED=1
    porcelain_deinit

    exit $?
}