#   - This script is used to update the Home Assistant Docker containers
#   - It will pull the latest images, check if the image ID has changed, and restart the container if needed

# Original Repo: https://github.com/CCOSTAN/Home-AssistantConfig
# Follow me on https://www.vcloudinfo.com/click-here

#!/bin/bash

set -euo pipefail

# Update system packages
sudo apt-get update && sudo apt-get upgrade -y

DC=(docker compose)

# Pull the latest images
"${DC[@]}" pull

# Get list of services from docker-compose.yml
EXISTING_SERVICES=$("${DC[@]}" config --services)

# Get list of running service containers
RUNNING_CONTAINERS=$("${DC[@]}" ps --services)

# Loop through each running service and check if its image has changed
for service in $RUNNING_CONTAINERS; do
    if echo "$EXISTING_SERVICES" | grep -qw "$service"; then
        # Get the current running image ID (remove sha256: prefix)
        CURRENT_IMAGE_ID=$(docker inspect --format='{{.Image}}' "$service" 2>/dev/null | sed 's/^sha256://')

        # Get the latest image ID from docker compose
        LATEST_IMAGE_ID=$("${DC[@]}" images -q "$service" 2>/dev/null)

        # If the image ID is different, restart the container
        if [ "$CURRENT_IMAGE_ID" != "$LATEST_IMAGE_ID" ] && [ -n "$LATEST_IMAGE_ID" ]; then
            echo "Updating container: $service"
            "${DC[@]}" stop "$service"
            "${DC[@]}" rm -f "$service"
            "${DC[@]}" up -d "$service"
        else
            echo "No update needed for: $service"
        fi
    else
        echo "Skipping non-existent service: $service"
    fi
done