#!/bin/sh

# Helper script to install the heartbeat tools. By default, it will only
# install the heartbeat client, hbc. The server is installed when the arg 'server' is passed 
# to the script. The script will install the heartbeat tools in a python 
# virtual environment in ~/venvs/hbd. The hbd and hbc commands will be
# installed from the wheel and symlinked to ~/bin/hbd and ~/bin/hbc,
# respectively. If the virtual environment already exists, it will be
# reused. The script will also remove any existing symlinks for hbd and hbc
# in ~/bin before creating new ones.

set -e
what=$1
on_ha=0
[ -z "$what" ] && what="client"

if [ -d /homeassistant ]; then
    echo "cannot install in HA, running \"docker exec homeassistant $0 $@\""
    docker exec homeassistant $0 $@
    rc=$?
    if [ $rc -ne 0 ]; then
        echo "Failed to install heartbeat in HA, please check the logs for more details"
        exit 1
    fi
    exit 0
fi
if [ -d /config ]; then
    echo "Installing on HA"
    where="/config/bin"
    venv="/config/venvs"
    on_ha=1
else
    if [ ! -d $HOME/.local/bin ] && [ ! -d $HOME/bin ]; then
        echo "No suitable bin directory found in PATH, please add either $HOME/.local/bin or $HOME/bin to your PATH"
        exit 1
    fi
    for where in $HOME/bin $HOME/.local/bin notset ; do
        if echo ":$PATH:" | grep -q ":$where:" ; then
            break
        fi
    done
    if [ "$where" = "notset" ]; then
        echo "No suitable bin directory found in PATH, please add either $HOME/.local/bin or $HOME/bin to your PATH"
        exit 1
    fi
    venv="$HOME/venvs"
fi

echo "Installing heartbeat $what"

if [ ! -d  $venv/hbd ]; then
    set +e
    python3 -m pip --version > /dev/null 2>&1 
    rc=$?
    set -e
    if [ $rc -ne 0 ]; then
        # truenas does not have pip installed by default, so we need to fetch get-pip.py and install pip
        echo "pip is not installed, fetching get-pip.py and installing pip"
        arg="--without-pip"
    fi
    mkdir -p $venv
    have_venv=$(python3 -c "import venv" &> /dev/null && echo "Installed" || echo "Not Installed")
    if [ "$have_venv" = "Not Installed" ]; then
        echo "python venv module not found, installing virtualenv"
        python3 -m pip install --user virtualenv
        python3 -m virtualenv $venv/hbd --system-site-packages $arg
    else
        python3 -m venv $venv/hbd --system-site-packages $arg
    fi
    . $venv/hbd/bin/activate
    if [ -n "$arg" ]; then  
        curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3 get-pip.py
    fi
    deactivate
fi

. $venv/hbd/bin/activate
python3 -mpip install --upgrade --index-url https://git.wrede.ca/api/packages/andreas/pypi/simple/ --extra-index-url https://pypi.org/simple hbd[$what]

if [ "$what" = "server" ]; then
    rm -f $where/hbd
    ln -sf $(which hbd) $where/hbd
    echo "hbd installed, you can run it with \"$where/hbd\" or \"hbd\" if $where is in your PATH"
else
    rm -f $where/hbc
    ln -sf $(which hbc) $where/hbc
    # rm -f $where/hb_install.sh
    cp "$0" $where/hb_install.sh
    chmod +x $where/hb_install.sh
    if [ $on_ha -eq 1 ]; then
        echo "restarting hbc "
        job=$(grep run_hbc configuration.yaml | sed 's/run_hbc://')
        $job
    else
        echo "hbc installed, you can run it with \"$where/hbc\" or \"hbc\" if $where is in your PATH"
    fi  
fi