Andreas Wrede
c5ce41762e
Server now sends a bare UPD command; client runs hb_install.sh to reinstall from the package registry, then restarts. hb_install.sh also copies itself alongside hbc on client installs. Co-Authored-By: Claude Sonnet 4.6 (1M context) <noreply@anthropic.com>
98 lines
3.3 KiB
Bash
Executable File
98 lines
3.3 KiB
Bash
Executable File
#!/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
|