From 3ab8b4e15a5414587c099339865b75f5f1bdaf7c Mon Sep 17 00:00:00 2001
From: Alexandre <44178713+alexbelgium@users.noreply.github.com>
Date: Wed, 16 Jul 2025 22:07:57 +0200
Subject: [PATCH] Update 99-run.sh

---
 ente/rootfs/etc/cont-init.d/99-run.sh | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/ente/rootfs/etc/cont-init.d/99-run.sh b/ente/rootfs/etc/cont-init.d/99-run.sh
index 7b8f2af72..beb06d09d 100755
--- a/ente/rootfs/etc/cont-init.d/99-run.sh
+++ b/ente/rootfs/etc/cont-init.d/99-run.sh
@@ -109,20 +109,24 @@ bootstrap_internal_db() {
 	fi
 	bashio::log.info "Creating role/database if needed..."
 
-	# Use psql via local socket (faster, avoids password)
+	# Create role if it doesn't exist
 	psql -v ON_ERROR_STOP=1 -h "$DB_HOST_INTERNAL" -p "$DB_PORT_INTERNAL" -U postgres <<SQL
 DO \$\$
 BEGIN
    IF NOT EXISTS (SELECT 1 FROM pg_roles WHERE rolname = '${DB_USER}') THEN
       EXECUTE 'CREATE ROLE ${DB_USER} LOGIN PASSWORD ''' || replace('${DB_PASS}','''','''''') || '''';
    END IF;
-   IF NOT EXISTS (SELECT 1 FROM pg_database WHERE datname = '${DB_NAME}') THEN
-      EXECUTE 'CREATE DATABASE ${DB_NAME} OWNER ${DB_USER}';
-   END IF;
 END
 \$\$;
 SQL
 
+	# Check and create database if it doesn't exist
+	if ! psql -v ON_ERROR_STOP=1 -h "$DB_HOST_INTERNAL" -p "$DB_PORT_INTERNAL" -U postgres -tAc "SELECT 1 FROM pg_database WHERE datname = '${DB_NAME}'"; then
+		psql -v ON_ERROR_STOP=1 -h "$DB_HOST_INTERNAL" -p "$DB_PORT_INTERNAL" -U postgres <<SQL
+CREATE DATABASE ${DB_NAME} OWNER ${DB_USER};
+SQL
+	fi
+
 	bashio::log.info "Internal Postgres ready."
 }