Fix PostgreSQL setup broken when running as root

Replace $SUDO -u postgres with sudo -u postgres in all psql commands. When SSH user is root, $SUDO is empty, causing -u postgres psql to fail with "bash: -u: command not found". User-switching requires sudo even as root.

Also add TCP connection configuration (listen_addresses, pg_hba.conf) and SSH keepalive options.

Author: devalade

lib/database.sh

@@ -12,7 +12,7 @@ setup_postgresql() {
 
 	info "Setting up PostgreSQL..."
 
-	ssh -p "$SSH_PORT" "$SSH_USER@$SSH_HOST" \
+	ssh -o ServerAliveInterval=30 -o ServerAliveCountMax=5 -p "$SSH_PORT" "$SSH_USER@$SSH_HOST" \
 		DB_NAME="$DB_NAME" \
 		DB_USER="$DB_USER" \
 		DB_PASSWORD="$DB_PASSWORD" \
@@ -38,27 +38,51 @@ setup_postgresql() {
         $SUDO systemctl start postgresql
         $SUDO systemctl enable postgresql
 
+        # Configure PostgreSQL to accept TCP connections on localhost
+        echo "Configuring PostgreSQL for TCP connections..."
+
+        # Find postgresql.conf path
+        PG_VERSION=$(sudo -u postgres psql -t -c "SHOW server_version;" | cut -d. -f1 | tr -d ' ')
+        PG_CONF="/etc/postgresql/$PG_VERSION/main/postgresql.conf"
+        PG_HBA="/etc/postgresql/$PG_VERSION/main/pg_hba.conf"
+
+        # Set listen_addresses if not already configured
+        if ! $SUDO grep -q "^listen_addresses.*localhost" "$PG_CONF"; then
+            echo "Enabling TCP listener on localhost..."
+            $SUDO sed -i "s/^#*listen_addresses.*/listen_addresses = 'localhost'/" "$PG_CONF"
+        fi
+
+        # Add TCP connection rule to pg_hba.conf if not present
+        if ! $SUDO grep -q "host.*all.*all.*127.0.0.1/32.*md5" "$PG_HBA"; then
+            echo "Adding TCP authentication rule..."
+            $SUDO bash -c "echo 'host    all             all             127.0.0.1/32            md5' >> '$PG_HBA'"
+        fi
+
+        # Restart PostgreSQL to apply changes
+        echo "Restarting PostgreSQL..."
+        $SUDO systemctl restart postgresql
+
         # Create database and user
         echo "Creating database '$DB_NAME' and user '$DB_USER'..."
 
         # Create database if it doesn't exist
-        $SUDO -u postgres psql -tc "SELECT 1 FROM pg_database WHERE datname = '$DB_NAME'" | grep -q 1 || \
-            $SUDO -u postgres psql -c "CREATE DATABASE \"$DB_NAME\";"
+        sudo -u postgres psql -tc "SELECT 1 FROM pg_database WHERE datname = '$DB_NAME'" | grep -q 1 || \
+            sudo -u postgres psql -c "CREATE DATABASE \"$DB_NAME\";"
 
         # Create user if it doesn't exist
-        $SUDO -u postgres psql -tc "SELECT 1 FROM pg_user WHERE usename = '$DB_USER'" | grep -q 1 || \
-            $SUDO -u postgres psql -c "CREATE USER \"$DB_USER\" WITH PASSWORD '$DB_PASSWORD';"
+        sudo -u postgres psql -tc "SELECT 1 FROM pg_user WHERE usename = '$DB_USER'" | grep -q 1 || \
+            sudo -u postgres psql -c "CREATE USER \"$DB_USER\" WITH PASSWORD '$DB_PASSWORD';"
 
         # Update password if user already exists
-        $SUDO -u postgres psql -c "ALTER USER \"$DB_USER\" WITH PASSWORD '$DB_PASSWORD';"
+        sudo -u postgres psql -c "ALTER USER \"$DB_USER\" WITH PASSWORD '$DB_PASSWORD';"
 
         # Grant privileges
-        $SUDO -u postgres psql -c "GRANT ALL PRIVILEGES ON DATABASE \"$DB_NAME\" TO \"$DB_USER\";"
+        sudo -u postgres psql -c "GRANT ALL PRIVILEGES ON DATABASE \"$DB_NAME\" TO \"$DB_USER\";"
 
         # For PostgreSQL 15+, grant schema privileges
-        $SUDO -u postgres psql -d "$DB_NAME" -c "GRANT ALL ON SCHEMA public TO \"$DB_USER\";" 2>/dev/null || true
-        $SUDO -u postgres psql -d "$DB_NAME" -c "GRANT ALL PRIVILEGES ON ALL TABLES IN SCHEMA public TO \"$DB_USER\";" 2>/dev/null || true
-        $SUDO -u postgres psql -d "$DB_NAME" -c "GRANT ALL PRIVILEGES ON ALL SEQUENCES IN SCHEMA public TO \"$DB_USER\";" 2>/dev/null || true
+        sudo -u postgres psql -d "$DB_NAME" -c "GRANT ALL ON SCHEMA public TO \"$DB_USER\";" 2>/dev/null || true
+        sudo -u postgres psql -d "$DB_NAME" -c "GRANT ALL PRIVILEGES ON ALL TABLES IN SCHEMA public TO \"$DB_USER\";" 2>/dev/null || true
+        sudo -u postgres psql -d "$DB_NAME" -c "GRANT ALL PRIVILEGES ON ALL SEQUENCES IN SCHEMA public TO \"$DB_USER\";" 2>/dev/null || true
 
         echo "PostgreSQL setup complete. Database '$DB_NAME' is ready."
         echo "Connection string: postgresql://$DB_USER:[REDACTED]@localhost:5432/$DB_NAME"