silverstripe-postgresql/code/PostgreSQLDatabaseConfigurationHelper.php

<?php
/**
 * This is a helper class for the SS installer.
 * 
 * It does all the specific checking for PostgreSQLDatabase
 * to ensure that the configuration is setup correctly.
 * 
 * @package mssql
 */
class PostgreSQLDatabaseConfigurationHelper implements DatabaseConfigurationHelper {

	/**
	 * Ensure that the database function pg_connect
	 * is available. If it is, we assume the PHP module for this
	 * database has been setup correctly.
	 * 
	 * @param array $databaseConfig Associative array of database configuration, e.g. "server", "username" etc
	 * @return boolean
	 */
	public function requireDatabaseFunctions($databaseConfig) {
		return (function_exists('pg_connect')) ? true : false;
	}

	/**
	 * Ensure that the database server exists.
	 * @param array $databaseConfig Associative array of db configuration, e.g. "server", "username" etc
	 * @return array Result - e.g. array('okay' => true, 'error' => 'details of error')
	 */
	public function requireDatabaseServer($databaseConfig) {
		$okay = false;
		$error = '';
		$username = $databaseConfig['username'] ? $databaseConfig['username'] : '';
		$password = $databaseConfig['password'] ? $databaseConfig['password'] : '';
		$server = $databaseConfig['server'];
		$userPart = $username ? " user=$username" : '';
		$passwordPart = $password ? " password=$password" : '';
		$connstring = "host=$server port=5432 {$userPart}{$passwordPart}";

		$conn = @pg_connect($connstring);
		if($conn) {
			$okay = true;
		} else {
			$okay = false;
			$error = 'PostgreSQL requires a valid username and password to determine if the server exists.';
		}
		
		return array(
			'okay' => $okay,
			'error' => $error
		);
	}

	/**
	 * Ensure a database connection is possible using credentials provided.
	 * The established connection resource is returned with the results as well.
	 * 
	 * @param array $databaseConfig Associative array of db configuration, e.g. "server", "username" etc
	 * @return array Result - e.g. array('okay' => true, 'connection' => mysql link, 'error' => 'details of error')
	 */
	public function requireDatabaseConnection($databaseConfig) {
		$okay = false;
		$error = '';
		$username = $databaseConfig['username'] ? $databaseConfig['username'] : '';
		$password = $databaseConfig['password'] ? $databaseConfig['password'] : '';
		$server = $databaseConfig['server'];
		$userPart = $username ? " user=$username" : '';
		$passwordPart = $password ? " password=$password" : '';
		$connstring = "host=$server port=5432 {$userPart}{$passwordPart}";
		
		$conn = @pg_connect($connstring);
		if($conn) {
			$okay = true;
		} else {
			$okay = false;
			$error = '';
		}
		
		return array(
			'okay' => $okay,
			'connection' => $conn,
			'error' => $error
		);
	}

	/**
	 * Ensure that the database connection is able to use an existing database,
	 * or be able to create one if it doesn't exist.
	 * 
	 * @param array $databaseConfig Associative array of db configuration, e.g. "server", "username" etc
	 * @return array Result - e.g. array('okay' => true, 'existsAlready' => 'true')
	 */
	public function requireDatabaseOrCreatePermissions($databaseConfig) {
		$okay = false;
		$existsAlready = false;
		
		$check = $this->requireDatabaseConnection($databaseConfig);
		$conn = $check['connection'];
		
		$result = pg_query($conn, "SELECT datname FROM pg_database WHERE datname = '$databaseConfig[database]'");
		if(pg_fetch_array($result)) {
			$okay = true;
			$existsAlready = true;
		} else {
			if(@pg_query($conn, "CREATE DATABASE testing123")) {
				pg_query($conn, "DROP DATABASE testing123");
				$okay = true;
				$existsAlready = false;
			}
		}
		
		return array(
			'okay' => $okay,
			'existsAlready' => $existsAlready
		);
	}

}