fix: robust postgres startup, wait loop, and permission fixes

start.sh

@@ -1,32 +1,70 @@
 #!/bin/bash
+set -e # Exit immediately if any command fails
 
-# --- 1. Fix PostgreSQL Path ---
-# Add common PostgreSQL bin paths to $PATH so 'initdb' and 'pg_ctl' are found
-export PATH=/usr/lib/postgresql/17/bin:/usr/lib/postgresql/16/bin:/usr/lib/postgresql/15/bin:/usr/lib/postgresql/14/bin:$PATH
+# --- 1. Dynamic PostgreSQL Path Detection ---
+# Find the directory containing pg_ctl (works for any postgres version)
+PG_BIN_DIR=$(find /usr/lib/postgresql -name pg_ctl | head -n 1 | xargs dirname)
 
-# --- 2. Set Critical Environment Variables (Fixes Pydantic Error) ---
+if [ -z "$PG_BIN_DIR" ]; then
+    echo "❌ Error: Could not find PostgreSQL binaries."
+    exit 1
+fi
+
+echo "✅ Found PostgreSQL binaries at $PG_BIN_DIR"
+export PATH="$PG_BIN_DIR:$PATH"
+
+# --- 2. Set Critical Environment Variables ---
 export DATABASE_URL="postgresql+asyncpg://prepuser:password@127.0.0.1:5432/studentdb"
+export PGDATA=/home/user/postgres_data
+export HOME=/home/user
+# ChromaDB settings
 export chroma_host="127.0.0.1"
 export chroma_port="8080"
 export chroma_collection="prepai_collection"
-export PGDATA=/home/user/postgres_data
 
-# --- 3. Initialize & Start PostgreSQL ---
+# --- 3. Database Initialization ---
+# If PGDATA exists but is missing the version file, it's corrupt. Wipe it.
+if [ -d "$PGDATA" ] && [ ! -f "$PGDATA/PG_VERSION" ]; then
+    echo "⚠️  $PGDATA exists but is not a valid cluster. Wiping to start fresh..."
+    rm -rf "$PGDATA"
+fi
+
 if [ ! -d "$PGDATA" ]; then
-    echo "Initializing database..."
-    initdb -D "$PGDATA"
+    echo "⚙️  Initializing database..."
+    # Initialize with trust auth for local connections to avoid password issues during setup
+    initdb -D "$PGDATA" --auth-local=trust --no-locale --encoding=UTF8
 fi
 
-echo "Starting PostgreSQL..."
-# Start DB in background
+# --- 4. Start PostgreSQL ---
+echo "🚀 Starting PostgreSQL..."
 pg_ctl -D "$PGDATA" -l /home/user/postgres.log start
-sleep 3
 
-# Create User and Database if they don't exist
-psql -h 127.0.0.1 -d postgres -c "CREATE USER prepuser WITH PASSWORD 'password';" || true
-createdb -h 127.0.0.1 -O prepuser studentdb || true
+# CRITICAL: Wait for Postgres to be ready before continuing
+echo "⏳ Waiting for PostgreSQL to accept connections..."
+until pg_isready -h 127.0.0.1 -p 5432; do
+    echo "   ...waiting for DB..."
+    sleep 2
+done
+echo "✅ PostgreSQL is up!"
 
-# --- 4. Start Nginx (Non-Root Mode) ---
+# --- 5. User & DB Setup ---
+echo "🛠️  Configuring Database..."
+# Idempotent creation: Only create if they don't exist
+psql -h 127.0.0.1 -d postgres -tAc "SELECT 1 FROM pg_roles WHERE rolname='prepuser'" | grep -q 1 || \
+psql -h 127.0.0.1 -d postgres -c "CREATE USER prepuser WITH PASSWORD 'password';"
+
+psql -h 127.0.0.1 -d postgres -tAc "SELECT 1 FROM pg_database WHERE datname='studentdb'" | grep -q 1 || \
+createdb -h 127.0.0.1 -O prepuser studentdb
+
+# --- 6. ChromaDB Setup ---
+echo "🎨 Setting up ChromaDB..."
+# Ensure directory exists and we own it
+mkdir -p ./chroma_store
+# Start Chroma in the background
+chroma run --host 0.0.0.0 --port 8080 --path ./chroma_store &
+
+# --- 7. Nginx Setup (Non-root Mode) ---
+echo "🌐 Starting Nginx..."
 # Create temp directories for Nginx to write to (avoids permission errors)
 mkdir -p /tmp/nginx/body \
          /tmp/nginx/proxy \
@@ -34,7 +72,7 @@ mkdir -p /tmp/nginx/body \
          /tmp/nginx/uwsgi \
          /tmp/nginx/scgi
 
-# Generate a non-root compatible nginx config on the fly
+# Generate a temporary nginx.conf that listens on port 7860
 cat <<EOF > /tmp/nginx.conf
 worker_processes 1;
 daemon off;
@@ -70,15 +108,10 @@ http {
 }
 EOF
 
-# Start Nginx in the background using the custom config
-echo "Starting Nginx..."
+# Start Nginx using the temp config
 nginx -c /tmp/nginx.conf &
 
-# --- 5. Start ChromaDB ---
-echo "Starting ChromaDB..."
-chroma run --host 0.0.0.0 --port 8080 --path ./chroma_store &
-
-# --- 6. Start Backend ---
-echo "Starting Backend..."
+# --- 8. Start Backend ---
+echo "🐍 Starting Backend..."
 cd Backend
 python run.py