stretch_new_robot_install.sh

#!/bin/bash
set -o pipefail

do_factory_install='false'
do_update='false'
while getopts ":fu" opt; do
    case $opt in
        f)
            do_factory_install='true'
            ;;
        u)
            do_update='true'
            ;;
    esac
done
if $do_factory_install && $do_update; then
    echo "Cannot both do_factory_install and do_update."
    exit 1
fi

source /etc/os-release
factory_osdir="$VERSION_ID"
if [[ ! $factory_osdir =~ ^(18.04|20.04|22.04)$ ]]; then
    echo "Could not identify OS. Please contact Hello Robot Support."
    exit 1
fi


sudo echo "#############################################"
echo "STARTING NEW ROBOT INSTALL"
echo "#############################################"

timestamp='stretch_install_'`date '+%Y%m%d%H%M'`;
logdir="$HOME/stretch_user/log/$timestamp"
logfile_initial="$logdir/stretch_initial_configuration.txt"
logfile_system="$logdir/stretch_system_install.txt"
logfile_user="$logdir/stretch_user_install.txt"
logfile_dev_tools="$logdir/stretch_dev_tools_install.txt"
logfile_firmware="$logdir/stretch_firmware_install.txt"
logzip="$logdir/stretch_robot_install_logs.zip"
mkdir -p $logdir

function echo_failure_help {
    zip -r $logzip $logdir/ > /dev/null
    echo ""
    echo "#############################################"
    echo "FAILURE. INSTALLATION DID NOT COMPLETE."
    echo "Look at the troubleshooting guide for solutions to common issues: https://docs.hello-robot.com/0.3/installation/robot_install/#troubleshooting"
    echo "or contact Hello Robot support and include $logzip"
    echo "#############################################"
    echo ""
    exit 1
}

if ! $do_update; then
    cd $HOME/stretch_install/factory/$factory_osdir
    ./stretch_initial_setup.sh $do_factory_install |& tee $logfile_initial
    if [ $? -ne 0 ]; then
        echo_failure_help
    fi
fi

echo ""
cd $HOME/stretch_install/factory/$factory_osdir
./stretch_install_system.sh -l $logdir |& tee $logfile_system
if [ $? -ne 0 ]; then
    echo_failure_help
fi

echo ""
cd $HOME/stretch_install
./stretch_new_user_install.sh -l $logdir |& tee $logfile_user
if [ $? -ne 0 ]; then
    echo_failure_help
fi

if $do_factory_install; then
    echo ""
    cd $HOME/stretch_install/factory/$factory_osdir
    ./stretch_install_dev_tools.sh -l $logdir |& tee $logfile_dev_tools
    if [ $? -ne 0 ]; then
        echo_failure_help
    fi
fi

# # Update Aug 24th 2023: Automatic firmware updating during the robot install causes devices to soft brick.
# # This functionality is disabled until the problem can be identified.
# echo ""
# cd $HOME/stretch_install/factory/$factory_osdir
# bash -i ./stretch_install_firmware.sh -l $logdir |& tee $logfile_firmware
# if [ $? -ne 0 ]; then
#     echo_failure_help
# fi

zip -r $logzip $logdir/ > /dev/null

echo ""
echo "#############################################"
echo "DONE! INSTALLATION COMPLETED SUCCESSFULLY."
echo "Perform post install steps: https://docs.hello-robot.com/0.2/stretch-install/docs/robot_install/#post-install-steps"
echo "#############################################"
echo ""