#!/usr/bin/env bash
#
# Update Immich on TrueNAS safely: release notes, snapshots, one script
# https://www.zadran.net/guides/update-immich-on-truenas-safely
# Generated from the guide, last updated 2026-10-18.
#
# NO WARRANTY. This script is provided as is. You are responsible for
# understanding what it does before running it, and for the consequences.
# Take a backup first. See https://www.zadran.net/disclaimer
#
# Destructive steps are left commented out on purpose -- uncomment them
# only once you have understood what they do.
#
# Where the guide writes a file with <<'EOF' and your values are in it,
# this script uses <<EOF so that they expand. Every other $, ` and \
# gets a backslash, so the rest comes out exactly as the guide shows it.

set -euo pipefail

# ---- Set these for your environment ----
# From: sudo grep '^IMMICH_VERSION' .env
CURRENT_VERSION='v3.2.2'
# Tag from the GitHub release page, with the v
NEW_VERSION='v3.3.1'
# Folder with docker-compose.yaml and .env
COMPOSE_DIR='/mnt/nvme/apps/immich'
# Where the script is saved
SCRIPT_PATH='/mnt/nvme/apps/immich-update.sh'
# Dataset of the compose folder and model cache
APPS_DATASET='nvme/apps/immich'
# Dataset behind DB_DATA_LOCATION
DB_DATASET='nvme/databases/immich'
# Dataset behind UPLOAD_LOCATION
LIBRARY_DATASET='tank/immich'

cd ${COMPOSE_DIR}
ls -la
sudo grep -n '^IMMICH_VERSION' .env
sudo docker compose config --images
sudo docker ps --format '{{.Names}}  {{.Image}}' | grep immich

sudo cp docker-compose.yaml docker-compose.yaml.bak-${CURRENT_VERSION}

sudo tee ${SCRIPT_PATH} >/dev/null <<EOF
#!/bin/bash
# Usage: sudo bash ${SCRIPT_PATH} v1.2.3
set -euo pipefail
NEW="\${1:?Usage: sudo bash \$0 v1.2.3}"
DIR=${COMPOSE_DIR}
TAG="pre-\${NEW}"

cd "\$DIR"
OLD=\$(grep '^IMMICH_VERSION=' .env | cut -d= -f2)
echo ">> Current: \$OLD   New: \$NEW"
[ "\$OLD" = "\$NEW" ] && { echo ">> Already on \$NEW, nothing to do."; exit 0; }

snap() { zfs list -t snapshot "\$2@\$TAG" >/dev/null 2>&1 && echo ">> Snapshot \$2@\$TAG exists" || { zfs snapshot \$1 "\$2@\$TAG"; echo ">> Snapshot \$2@\$TAG created"; }; }
snap -r ${APPS_DATASET}
snap ""  ${DB_DATASET}
snap ""  ${LIBRARY_DATASET}

cp .env ".env.bak-\$OLD"
sed -i "s/^IMMICH_VERSION=.*/IMMICH_VERSION=\$NEW/" .env
echo ">> .env now: \$(grep '^IMMICH_VERSION=' .env)"

docker compose pull
docker compose up -d
echo ">> Waiting 30 s for startup..."
sleep 30
docker ps --format '{{.Names}}  {{.Image}}  {{.Status}}' | grep immich
echo
echo ">> Rollback if needed:"
echo "   cd \$DIR && sed -i 's/^IMMICH_VERSION=.*/IMMICH_VERSION=\$OLD/' .env && docker compose down"
echo "   zfs rollback ${DB_DATASET}@\$TAG && docker compose up -d"
echo ">> After a few good days: zfs destroy -r ${APPS_DATASET}@\$TAG; zfs destroy ${DB_DATASET}@\$TAG; zfs destroy ${LIBRARY_DATASET}@\$TAG"
EOF
sudo chmod 700 ${SCRIPT_PATH}
ls -l ${SCRIPT_PATH}

sudo bash ${SCRIPT_PATH} ${NEW_VERSION}

sudo docker ps --format '{{.Names}}  {{.Status}}' | grep immich
sudo docker logs --tail 15 immich_server

sudo zfs destroy -r ${APPS_DATASET}@pre-${NEW_VERSION}
sudo zfs destroy ${DB_DATASET}@pre-${NEW_VERSION}
sudo zfs destroy ${LIBRARY_DATASET}@pre-${NEW_VERSION}
sudo docker image prune -f

cd ${COMPOSE_DIR}
sudo sed -i 's/^IMMICH_VERSION=.*/IMMICH_VERSION='"${CURRENT_VERSION}"'/' .env
sudo docker compose down
sudo zfs rollback ${DB_DATASET}@pre-${NEW_VERSION}
sudo docker compose up -d