#!/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
where=""
venv=""
[ "$2" = "HA" ] && on_ha=1
[ -z "$what" ] && what="client"

if [ -d /homeassistant ]; then
    echo "HA, running \"docker exec homeassistant $0 $@\""
    docker exec homeassistant $0 $@ HA
    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 [ $on_ha -eq 1 ]; then
    echo "Installing under docker on Home Assistant OS, using /config/bin for executables and /config/venvs for virtual environments "
    where="/config/bin"
    venv="/config/venvs"
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
    if [ "$what" = "mini" ]; then
        venv=""
    else
        venv="$HOME/venvs"
    fi
fi

echo "Installing heartbeat $what"

if [ "$venv" != "" ] && [ ! -d  $venv/hbd ]; then
    set +e
    python3 -m pip --version > /dev/null 2>&1 
    rc=$?
    set -e
    arg=""
    if [ $rc -ne 0 ]; then
        # some systems do 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

if [ -z "$venv" ]; then
    echo "Installing heartbeat $what globally"
else
    echo "Installing heartbeat $what in virtual environment $venv/hbd"
    . $venv/hbd/bin/activate
fi
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"
elif [ "$what" = "client" ]; then
    hbc_path=$(which hbc)
    if [ -z "$hbc_path" ]; then
        echo "hbc not found in PATH, installation failed"
        exit 1
    fi
    if [ "$hbc_path" != "$where/hbc" ]; then
        rm -f $where/hbc
        ln -sf $(which hbc) $where/hbc
    fi
    if [ "$0" != "$where/hb_install.sh" ]; then
        cp "$0" $where/hb_install.sh
        chmod +x $where/hb_install.sh
    fi
    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  
elif [ "$what" = "mini" ]; then
    hbc_path=$(which hbc_mini)
    if [ "$hbc_path" != "$where/hbc_mini" ]; then
       ln -sf $(which hbc) $where/hbc_mini
    fi
    echo "hbc mini installed, you can run it with \"$where/hbc_mini\" or \"hbc_mini\" if $where is in your PATH"
fi