From e0d49b432980365b0c058886ebe92661c0292cff Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Wed, 22 Jan 2025 08:32:02 +0100
Subject: [PATCH] Add vector.rs

---
 postgres/rootfs/etc/cont-init.d/99-run.sh | 30 +++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/postgres/rootfs/etc/cont-init.d/99-run.sh b/postgres/rootfs/etc/cont-init.d/99-run.sh
index 38e9bb7b1..fb5b0f903 100755
--- a/postgres/rootfs/etc/cont-init.d/99-run.sh
+++ b/postgres/rootfs/etc/cont-init.d/99-run.sh
@@ -42,3 +42,33 @@ else
     bashio::log.warning "ARMv7 detected: Starting without vectors.so"
     docker-entrypoint.sh postgres & true
 fi
+
+####################
+# Enable vector.rs #
+####################
+
+bashio::log.info "Waiting for port 5432 to open..."
+
+# Wait for transmission to become available
+( bashio::net.wait_for 5432 localhost 900
+
+bashio::log.info "Enabling vector.rs"
+
+# Set variables for 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
+
+export DB_PORT
+export DB_HOSTNAME
+export DB_USERNAME
+export DB_PASSWORD
+echo "DROP EXTENSION IF EXISTS vectors;
+    CREATE EXTENSION vectors;
+\q" > /setup_postgres.sql
+
+# Enable vectors
+psql "postgres://$DB_USERNAME:$DB_PASSWORD@$DB_HOSTNAME:$DB_PORT" < /setup_postgres.sql >/dev/null || true
+rm /setup_postgres.sql || true
+) & true