install.sh

#!/usr/bin/env bash
set -e

if [ ! "$BASH_VERSION" ] ; then
    echo "You must use bash to run this script. If running this script from curl, make sure the final word is 'bash'" 1>&2
    exit 1
fi

CURRENT_ARCH=$(dpkg --print-architecture)
if [[ "$CURRENT_ARCH" != "x64" && "$CURRENT_ARCH" != "amd64" && "$CURRENT_ARCH" != "arm64" ]]; then
    echo "$CURRENT_ARCH is not a supported cpu architecture for running Companion."
    echo "If you are running on an arm device (such as a Raspberry Pi), make sure to use an arm64 image."
    exit 1
fi

echo "This will attempt to install Companion as a system service on this device."
echo "It is designed to be run on headless servers, but can be used on desktop machines if you are happy to not have the tray icon."
echo "A user called 'companion' will be created to run the service, and various scripts will be installed to manage the service"

if [ $(/usr/bin/id -u) -ne 0 ]; then
    echo "Must be run as root"
    exit 1
fi

# Install a specific stable build. It is advised to not use this, as attempting to install a build that doesn't
# exist can leave your system in a broken state that needs fixing manually
COMPANION_BUILD="${COMPANION_BUILD:-beta}"
# Development only: Allow building using a testing branch of this updater
COMPANIONPI_BRANCH="${COMPANIONPI_BRANCH:-main}"

# install some dependencies
apt-get update
apt-get install -yq git zip unzip curl libusb-1.0-0-dev libudev-dev adduser wget
apt-get clean

# add a system user, if the user doesn't already exist
id -u companion &>/dev/null || adduser --disabled-password companion --gecos ""

# install fnm to manage node version
# we do this to /opt/fnm, so that the companion user can use the same installation
export FNM_DIR=/opt/fnm
echo "export FNM_DIR=/opt/fnm" >> /root/.bashrc
curl -fsSL https://fnm.vercel.app/install | bash -s -- --install-dir /opt/fnm &>/dev/null
export PATH=/opt/fnm:$PATH
eval "`fnm env --shell bash`"

# clone the companionpi repository
rm -R /usr/local/src/companionpi &>/dev/null || true
git clone https://github.com/bitfocus/companion-pi.git -b $COMPANIONPI_BRANCH /usr/local/src/companionpi
cd /usr/local/src/companionpi

# configure git for future updates
git config --global pull.rebase false

# run the update script
if [ "$COMPANION_BUILD" == "beta" ] || [ "$COMPANION_BUILD" == "experimental" ]; then
    ./update.sh beta
else
    ./update.sh stable "$COMPANION_BUILD"
fi

# install update script dependencies, as they were ignored
yarn --cwd "/usr/local/src/companionpi/update-prompt" install

# enable start on boot
systemctl enable companion

# add the fnm node to this users path
# TODO - verify permissions
echo "export PATH=/opt/fnm/aliases/default/bin:\$PATH" >> /home/companion/.bashrc

# check that a build of companion was installed
if [ ! -d "/opt/companion" ] 
then
    echo "No Companion build was installed!\nIt should be possible to recover from this with \"sudo companion-update\"" 
    exit 9999 # die with error code 9999
fi

echo "Companion is installed!"
echo "You can start it with \"sudo systemctl start companion\" or \"sudo companion-update\""