From 6d7b323412b4ff30d488f102ec6776140e3efde8 Mon Sep 17 00:00:00 2001
From: Alexandre <alexandre.pary@gmail.com>
Date: Sat, 15 Feb 2025 09:54:15 +0100
Subject: [PATCH] Check if vectors is enabled

---
 immich/rootfs/etc/cont-init.d/99-run.sh   | 15 +++++++++++++++
 postgres/rootfs/etc/cont-init.d/99-run.sh |  3 ++-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/immich/rootfs/etc/cont-init.d/99-run.sh b/immich/rootfs/etc/cont-init.d/99-run.sh
index d1c2927b8..585b190f9 100755
--- a/immich/rootfs/etc/cont-init.d/99-run.sh
+++ b/immich/rootfs/etc/cont-init.d/99-run.sh
@@ -150,6 +150,20 @@ EOF
     bashio::log.info "Database setup completed successfully."
 }
 
+# Function to check if vectors extension is enabled
+check_vector_extension() {
+    echo "Checking if 'vectors' extension is enabled..."
+    RESULT=$(psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" -tAc "SELECT extname FROM pg_extension WHERE extname = 'vectors';")
+
+    if [[ "$RESULT" == "vectors" ]]; then
+        echo "✅ 'vectors' extension is enabled."
+        exit 0
+    else
+        bashio::log.warning "❌ 'vectors' extension is NOT enabled."
+        return 1
+    fi
+}
+
 #########################
 # Main script execution #
 #########################
@@ -171,3 +185,4 @@ export_db_env
 
 setup_root_user
 setup_database
+check_vector_extension
diff --git a/postgres/rootfs/etc/cont-init.d/99-run.sh b/postgres/rootfs/etc/cont-init.d/99-run.sh
index e3a707cf5..6f52f24bb 100755
--- a/postgres/rootfs/etc/cont-init.d/99-run.sh
+++ b/postgres/rootfs/etc/cont-init.d/99-run.sh
@@ -59,7 +59,8 @@ bashio::log.info "Enabling vector.rs"
 DB_PORT=5432
 DB_HOSTNAME=localhost
 DB_PASSWORD="$(bashio::config 'POSTGRES_PASSWORD')"
-if bashio::config.has_value "POSTGRES_USER"; then DB_USERNAME="$(bashio::config "POSTGRES_USER")"; else DB_USERNAME=postgres; fi
+DB_USERNAME=postgres
+if bashio::config.has_value "POSTGRES_USER"; then DB_USERNAME="$(bashio::config "POSTGRES_USER")"; fi
 
 export DB_PORT
 export DB_HOSTNAME