#!/bin/bash
# MeshBot Update Script
# Usage: bash update.sh or ./update.sh after making it executable with chmod +x update.sh

# Check if the mesh_bot.service or pong_bot.service 
service_stopped=false
if systemctl is-active --quiet mesh_bot.service; then
    echo "Stopping mesh_bot.service..."
    systemctl stop mesh_bot.service
    service_stopped=true
fi
if systemctl is-active --quiet pong_bot.service; then
    echo "Stopping pong_bot.service..."
    systemctl stop pong_bot.service
    service_stopped=true
fi
if systemctl is-active --quiet mesh_bot_reporting.service; then
    echo "Stopping mesh_bot_reporting.service..."
    systemctl stop mesh_bot_reporting.service
    service_stopped=true
fi
if systemctl is-active --quiet mesh_bot_w3.service; then
    echo "Stopping mesh_bot_w3.service..."
    systemctl stop mesh_bot_w3.service
    service_stopped=true
fi

# Fetch latest changes from GitHub
echo "Fetching latest changes from GitHub..."
if ! git fetch origin; then
    echo "Error: Failed to fetch from GitHub, check your network connection."
    exit 1
fi

# git pull with rebase to avoid unnecessary merge commits
echo "Pulling latest changes from GitHub..."
if ! git pull origin main --rebase; then
    read -p "Git pull resulted in conflicts. Do you want to reset hard to origin/main? This will discard local changes. (y/n): " choice
    if [[ "$choice" == "y" || "$choice" == "Y" ]]; then
        git fetch --all
        git reset --hard origin/main
        echo "Local repository updated."
    else
        echo "Update aborted due to git conflicts."
    fi
fi

# copy modules/custom_scheduler.py template if it does not exist
if [[ ! -f modules/custom_scheduler.py ]]; then
    cp -n etc/custom_scheduler.py modules/
    printf "\nCustom scheduler template copied to modules/custom_scheduler.py\n"
elif ! cmp -s modules/custom_scheduler.template etc/custom_scheduler.py; then
    echo "custom_scheduler.py is set. To check changes run: diff etc/custom_scheduler.py modules/custom_scheduler.py"
fi

# Backup the data/ directory
echo "Backing up data/ directory..."
#backup_file="backup_$(date +%Y%m%d_%H%M%S).tar.gz"
backup_file="data_backup.tar.gz"
path2backup="data/"
#copy custom_scheduler.py if it exists
if [[ -f "modules/custom_scheduler.py" ]]; then
    echo "Including custom_scheduler.py in backup..."
    cp modules/custom_scheduler.py data/
fi
# Check config.ini ownership and permissions
if [[ -f "config.ini" ]]; then
    owner=$(stat -f "%Su" config.ini)
    perms=$(stat -f "%A" config.ini)
    echo "config.ini is owned by: $owner"
    echo "config.ini permissions: $perms"
    if [[ "$owner" == "root" ]]; then
        echo "Warning: config.ini is owned by root check out the etc/set-permissions.sh script"
    fi
    if [[ $(stat -f "%Lp" config.ini) =~ .*[7,6,2]$ ]]; then
        echo "Warning: config.ini is world-writable or world-readable! check out the etc/set-permissions.sh script"
    fi

    echo "Including config.ini in backup..."

    cp config.ini data/config.backup
fi
#create the tar.gz backup
tar -czf "$backup_file" "$path2backup"
if [ $? -ne 0 ]; then
    echo "Error: Backup failed."
else
    echo "Backup of ${path2backup} completed: ${backup_file}"
fi

# Build a config_new.ini file merging user config with new defaults
echo "Merging configuration files..."
python3 script/configMerge.py > ini_merge_log.txt 2>&1
if [[ -f ini_merge_log.txt ]]; then
    if grep -q "Error during configuration merge" ini_merge_log.txt; then
        echo "Configuration merge encountered errors. Please check ini_merge_log.txt for details."
    else
        echo "Configuration merge completed. Please review config_new.ini and ini_merge_log.txt."
    fi
else
    echo "Configuration merge log (ini_merge_log.txt) not found. check out the script/configMerge.py tool!"
fi

if [[ "$service_stopped" = true ]]; then
    echo "Restarting services..."
    for svc in mesh_bot.service pong_bot.service mesh_bot_reporting.service mesh_bot_w3.service; do
        if systemctl list-unit-files | grep -q "^$svc"; then
            systemctl start "$svc"
            echo "$svc restarted."
        fi
    done
fi

# Print completion message
echo "Update completed successfully?"
exit 0
# End of script