init.sh

#!/bin/bash
echo "Starting initialization script for run control."

# =================================================
# Populate dependencies folder
# =================================================
# copies config from .gitmodules to .git/config
git submodule sync --recursive
# clones submodules if not already cloned
git submodule update --init --recursive
# updates submodules to latest commit
git submodule update --remote --recursive

# =================================================
# Set up conda environment
# =================================================
ENV_NAME="runcontrol"
ENV_FILE="./dependencies/conda_rc.yml"
echo "Setting up conda environment..."
# update conda to latest version
conda update -y -n base conda
if conda info --envs | grep -q "$ENV_NAME"; then
  # if runcontrol environment already exists, update it
  echo "Conda environment \"$ENV_NAME\" already exists. Updating...";
  conda env update --file "$ENV_FILE" --prune;
else
  # if runcontrol environment doesn't exist, create it
  echo "Conda environment \"$ENV_NAME\" doesn't exist. Creating environment...";
  conda env create --file "$ENV_FILE";
fi
conda activate "$ENV_NAME";

# =================================================
# Generate all ui and resources files
# =================================================
echo "Generating Qt UI and resources files."
# generate all .py files from .ui files
for f in ./ui/*.ui; do
  name="${f%%.ui}";
  pyside6-uic $f -o "$name.py";
done
# generate all .py files from .qrc files
pyside6-rcc ./resources/resources.qrc -o ./resources_rc.py

# =================================================
# Add dependencies folder to PATH
# =================================================
export PATH=$PATH:$PWD/dependencies

# =================================================
# Install arduino packages
# =================================================
# user need to be in dialout group on linux
echo "Setting up Arduino cli and libraries..."
curl -fsSL https://raw.githubusercontent.com/arduino/arduino-cli/master/install.sh | BINDIR=./dependencies sh
arduino-cli config init
arduino-cli update
arduino-cli upgrade
arduino-cli core install arduino:avr
arduino-cli lib install incbin ArduinoJson Ethernet

# =================================================
# Check and prepare SMB token
# =================================================
# Crete config directory if it doesn't exist
mkdir -p "$HOME/.config/runcontrol"
FILE="$HOME/.config/runcontrol/smb_token"

# Check if file already exists
if [ -e "$FILE" ]; then
    echo "File '$FILE' already exists. Skipping password save."
else
    # Prompt for password without echo
    read -s -p "Enter SMB user name: " USERNAME
    echo
    read -s -p "Enter SMB password: " PASSWORD
    echo
    echo -e "username=$USERNAME\npassword=$PASSWORD" > "$FILE"
    echo "Password saved to '$FILE'."
fi

FILE_PERM=$(stat -c "%a" "$FILE")
if [ "$FILE_PERM" != "600" ]; then
    chmod 600 "$FILE"
    echo "File permissions for '$FILE' have been corrected to 600."
fi

# =================================================
# Save and prepare SQL token
# =================================================
FILE="$HOME/.config/runcontrol/sql_token"
LINE="export SQL_DAQ_TOKEN=\$(<\"$FILE\" tr -d '\\n')"
BASHRC="$HOME/.bashrc"

# Check if file already exists
if [ -e "$FILE" ]; then
    echo "File '$FILE' already exists. Skipping password save."
else
    # Prompt for password without echo
    read -s -p "Enter SQL password: " PASSWORD
    echo
    echo "$PASSWORD" > "$FILE"
    echo "Password saved to '$FILE'."
fi

FILE_PERM=$(stat -c "%a" "$FILE")
if [ "$FILE_PERM" != "600" ]; then
    chmod 600 "$FILE"
    echo "File permissions for '$FILE' have been corrected to 600."
fi

# =================================================
# Install root init script
# =================================================
ROOT_SCRIPT_DEST="/usr/local/bin/rc-init"
ROOT_SCRIPT_SOURCE="./dependencies/root-init.sh"
SUDOERS_FILE="/etc/sudoers.d/runcontrol"
CURRENT_USER=$(whoami)
SCRIPT_UPDATED=false
SUDOER_SET=false

# Check if root script already installed
if [ -f "$ROOT_SCRIPT_DEST" ]; then
    echo "Root components already installed."
    SOURCE_HASH=$(sha256sum "$ROOT_SCRIPT_SOURCE" | awk '{print $1}')
    DESTINATION_HASH=$(sha256sum "$ROOT_SCRIPT_DEST" | awk '{print $1}')
    # Check if the script is up to date
    if [ "$SOURCE_HASH" = "$DESTINATION_HASH" ]; then
        echo "Root script is up to date."
        SCRIPT_UPDATED=true
    else
        echo "Root script is not up to date."
        echo "Source hash: $SOURCE_HASH, Destination hash: $DESTINATION_HASH"
        echo "Updating..."
        SCRIPT_UPDATED=false
    fi
else
    echo "Root script not installed yet."
    SCRIPT_UPDATED=false
fi

if [ "$SCRIPT_UPDATED" = "false" ]; then
    echo "Installing root script."
    sudo install -o root -g root -m 755 "$ROOT_SCRIPT_SOURCE" "$ROOT_SCRIPT_DEST" || {
        echo "Failed to install root script"
        exit 1
    }
    echo "Root script successfully installed."
fi

# Check if sudoer file has been set
if [ -f "$SUDOERS_FILE" ]; then
    echo "Sudoer file already exists."
    SUDOER_SET=true
fi

# Set up sudoer file if not set
if [ "$SUDOER_SET" = "false" ]; then
    echo "Setting sudoer file"
    sudo bash -c "echo '$CURRENT_USER ALL=(ALL) NOPASSWD: $ROOT_SCRIPT_DEST' > $SUDOERS_FILE" || {
        echo "Failed to create sudoers entry"
        sudo rm -f "$ROOT_SCRIPT_DEST"
        exit 1
    }

    # Set correct permission of sudoer
    sudo chmod 440 "$SUDOERS_FILE"

    # Check if sudoer file is valid
    if ! sudo visudo -cf "$SUDOERS_FILE"; then
        echo "Invalid sudo configuration. Rolling back..."
        sudo rm -f "$SUDOERS_FILE" "$ROOT_SCRIPT_DEST"
        exit 1
    fi
fi

# =================================================
# Handle group permissions for current user
# =================================================
PERMISSION_CHANGED=false
if ! groups "$CURRENT_USER" | grep -q '\busbusers\b'; then
    sudo usermod -aG usbusers "$CURRENT_USER"
    echo "Added $CURRENT_USER to usbusers group."
    PERMISSION_CHANGED=true
fi
if ! groups "$CURRENT_USER" | grep -q '\bdialout\b'; then
    sudo usermod -aG dialout "$CURRENT_USER"
    echo "Added $CURRENT_USER to dialout group."
    PERMISSION_CHANGED=true
fi
if [ "$PERMISSION_CHANGED" = "true" ]; then
    echo "User $CURRENT_USER has been added to the required groups. Please log out and log back in for the changes to take effect."
else
    echo "User $CURRENT_USER is already in the required groups."
fi

# =================================================
# Execute root init script
# =================================================
echo "Executing root script"
sudo "$ROOT_SCRIPT_DEST"
# Compile SBC-Piezo-Base-Code
make -C ./dependencies/gati-linux-driver/Sdk/
echo "Run Control environment successfully initialized."