Add database initialization with custom user and credentials

Supports POSTGRES_USER, POSTGRES_PASSWORD, and POSTGRES_DB environment variables during initdb.

Author: adityathebe

cmd/main.go

@@ -154,7 +154,7 @@ Examples:
 	cmd.Flags().StringVar(&authMethod, "auth-method", lo.CoalesceOrEmpty(os.Getenv("PG_AUTH_METHOD"), string(pkg.AuthScramSHA)), "Authentication method for pg_hba.conf (auto-detected if not specified)")
 	cmd.Flags().BoolVar(&opts.Enabled, "pg-tune", true, "Run pg_tune to optimize postgresql.conf before starting")
 	cmd.Flags().Bool("auto-upgrade", true, "Automatically upgrade PostgreSQL if version mismatch detected")
-	cmd.Flags().Bool("auto-reset-password", true, "Reset postgres superuser password on start")
+	cmd.Flags().Bool("auto-reset-password", false, "Reset postgres superuser password on start")
 	cmd.Flags().Bool("auto-init", true, "Automatically initialize database if data directory doesn't exist")
 	cmd.Flags().Int("upgrade-to", 0, "Target PostgreSQL version for upgrade (default: auto-detect latest)")
 

docker-compose.yaml

@@ -0,0 +1,20 @@
+services:
+  postgres:
+    container_name: "pg_test"
+    build:
+      context: .
+      dockerfile: Dockerfile
+      args:
+        PG_VERSION: 17
+    image: flanksource/postgres:17
+    environment:
+      POSTGRES_DB: mc
+      POSTGRES_USER: flanksource
+      POSTGRES_PASSWORD: testpassword
+    ports:
+      - "5435:5432"
+    volumes:
+      - pg-data-test:/var/lib/postgresql/data
+
+volumes:
+  pg-data-test:

docker-entrypoint.sh

@@ -32,12 +32,34 @@ fi
 
 export PGBIN=/usr/lib/postgresql/${PG_VERSION}/bin
 
+# Set defaults for PostgreSQL environment variables
+POSTGRES_USER="${POSTGRES_USER:-postgres}"
+POSTGRES_DB="${POSTGRES_DB:-$POSTGRES_USER}"
 
 if [ ! -f $PGDATA/PG_VERSION ]; then
     echo "Initializing database cluster at $PGDATA ..."
-    # starting and stopping the DB to initialize the directory for tuning
-    $PGBIN/initdb -D $PGDATA
+    echo "Using POSTGRES_USER: $POSTGRES_USER"
+
+    # Initialize database with specified user and password
+    if [ -n "$POSTGRES_PASSWORD" ]; then
+        echo "Initializing with password authentication"
+        $PGBIN/initdb -D $PGDATA -U "$POSTGRES_USER" --pwfile=<(printf "%s\n" "$POSTGRES_PASSWORD")
+    else
+        echo "WARNING: No password set. Initializing without password authentication."
+        $PGBIN/initdb -D $PGDATA -U "$POSTGRES_USER"
+    fi
+
+    # Start PostgreSQL temporarily to create database
     $PGBIN/pg_ctl start -D $PGDATA --wait
+
+    # Create custom database if specified and different from default
+    if [ "$POSTGRES_DB" != "postgres" ]; then
+        echo "Creating database: $POSTGRES_DB"
+        $PGBIN/psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" --dbname postgres <<-EOSQL
+			CREATE DATABASE "$POSTGRES_DB" OWNER "$POSTGRES_USER";
+		EOSQL
+    fi
+
     $PGBIN/pg_ctl stop -D $PGDATA --wait
 else
     echo "Database cluster already initialized at $PGDATA with version $(cat $PGDATA/PG_VERSION)"