setup.sh

#!/bin/bash

# Check if the script is run as root
if [[ $(/usr/bin/id -u) -ne 0 ]]; then
    echo "Not running as root"
    exit
fi

# Change directory to the parent of the script directory
cd "$(dirname "$0")/.."

# Create a virtual environment if it doesn't exist
if [ ! -d ".venv" ]; then
    echo "Creating virtual environment"
    python3 -m venv .venv
fi

# Activate the virtual environment
source .venv/bin/activate

# Install the required packages
echo "Installing required packages"
pip3 install -r requirements.txt --use-pep517

# Remove old systemd service files
for service in /etc/systemd/system/balloon-*.service; do
    if [ "$service" == "/etc/systemd/system/balloon-*.service" ]; then
        echo "No systemd service files found"
        break
    fi
    echo "Removing old service $(basename "$service")"
    systemctl stop "$(basename "$service")"
    systemctl disable "$(basename "$service")"
    rm "$service"
done

# Copy systemd service files (exclude the template service file) from all directorys to /etc/systemd/system/
for service in $(find . -name "balloon-*.service" ! -name "balloon-template.service"); do
    echo "Copying systemd service file $service to /etc/systemd/system/"
    cp "$service" /etc/systemd/system/
done

# Reload systemd
echo "Reloading systemd"
systemctl daemon-reload

# Enable the services
for service in /etc/systemd/system/balloon-*.service; do
    if [ "$service" == "/etc/systemd/system/balloon-*.service" ]; then
        echo "No systemd service files found"
        break
    fi
    echo "Enabling service $(basename "$service")"
    systemctl enable "$(basename "$service")"
done

# Create configuration file
if [ ! -f "config.yml" ]; then
    echo "Creating configuration file"
    cp resources/templates/config.yml.example config.yml
else
    echo "Configuration file already exists"
    read -p "Do you want to override it? (y/n) " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        echo "Making a copy of existing configuration"
        cp config.yml config.yml.old
        echo "Overriding configuration file"
        cp resources/templates/config.yml.example config.yml
    fi
fi

# Futher instructions
echo "Basic setup completed"
echo "Please edit the configuration file config.yml"
echo "Run reset.sh to initialize data storage (WARNING: this will delete all existing data)"
echo "Run start.sh to start the services"
echo "(Uninstall everything with uninstall.sh)"