#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
STACKS_DIR="$(dirname "$SCRIPT_DIR")"
LOG_FILE="$SCRIPT_DIR/update.log"

set -a; source "$SCRIPT_DIR/.env"; set +a

log()    { echo "[$(date '+%Y-%m-%d %H:%M:%S')] $*" | tee -a "$LOG_FILE"; }
notify() {
  local color="$1" text="$2"
  curl -sf -o /dev/null -X POST "$WEBHOOK_URL" \
    -H 'Content-Type: application/json' \
    -d "{\"message\":\"$text\",\"color\":\"$color\"}"
}

: > "$LOG_FILE"
log "Starting image update check"

updated=(); failed=()

for stack_dir in "$STACKS_DIR"/*/; do
  stack=$(basename "$stack_dir")
  [[ "$stack" == _* ]] && continue
  [[ ! -f "$stack_dir/compose.yml" ]] && continue

  log "Checking $stack"
  cd "$stack_dir"
  pull_output=$(docker compose pull 2>&1 | tee -a "$LOG_FILE")

  if echo "$pull_output" | grep -q " Pulled"; then
    log "  Updates found — recreating containers"
    if docker compose up -d 2>&1 | tee -a "$LOG_FILE"; then
      updated+=("$stack")
    else
      log "  FAILED to recreate $stack"
      failed+=("$stack")
    fi
  else
    log "  Up to date"
  fi
done

log "Pruning dangling images"
docker image prune -f 2>&1 | tee -a "$LOG_FILE"

if [ ${#updated[@]} -eq 0 ] && [ ${#failed[@]} -eq 0 ]; then
  notify "#36a64f" "✅ **Update check complete** — all images up to date"
elif [ ${#failed[@]} -gt 0 ]; then
  notify "#cc0000" "❌ **Update failed**\nUpdated: ${updated[*]:-none}\nFailed: ${failed[*]}"
else
  notify "#36a64f" "✅ **Updates applied**\nStacks: ${updated[*]}"
fi

log "Update check complete"