From 5365c1a32f396e1e699dea597d7656138f649573 Mon Sep 17 00:00:00 2001
From: Robin Appelman <icewind@owncloud.com>
Date: Thu, 21 Jul 2016 12:44:02 +0200
Subject: [PATCH] handle postgres setup when we cant connect as admin

---
 lib/private/Setup/AbstractDatabase.php |  1 +
 lib/private/Setup/PostgreSQL.php       | 96 ++++++++++++++------------
 2 files changed, 54 insertions(+), 43 deletions(-)

diff --git a/lib/private/Setup/AbstractDatabase.php b/lib/private/Setup/AbstractDatabase.php
index 2b9d7afb36..dbf46888ff 100644
--- a/lib/private/Setup/AbstractDatabase.php
+++ b/lib/private/Setup/AbstractDatabase.php
@@ -114,6 +114,7 @@ abstract class AbstractDatabase {
 			'user' => $this->dbUser,
 			'password' => $this->dbPassword,
 			'tablePrefix' => $this->tablePrefix,
+			'dbname' => $this->dbName
 		);
 
 		// adding port support through installer
diff --git a/lib/private/Setup/PostgreSQL.php b/lib/private/Setup/PostgreSQL.php
index 85755ea421..14325a18ef 100644
--- a/lib/private/Setup/PostgreSQL.php
+++ b/lib/private/Setup/PostgreSQL.php
@@ -35,50 +35,60 @@ class PostgreSQL extends AbstractDatabase {
 	public $dbprettyname = 'PostgreSQL';
 
 	public function setupDatabase($username) {
-		$connection = $this->connect([
-			'dbname' => 'postgres'
-		]);
-		//check for roles creation rights in postgresql
-		$builder = $connection->getQueryBuilder();
-		$builder->automaticTablePrefix(false);
-		$query = $builder
-			->select('rolname')
-			->from('pg_roles')
-			->where($builder->expr()->eq('rolcreaterole', new Literal('TRUE')))
-			->andWhere($builder->expr()->eq('rolname', $builder->createNamedParameter($this->dbUser)));
-
-		try {
-			$result = $query->execute();
-			$canCreateRoles = $result->rowCount() > 0;
-		} catch (DatabaseException $e) {
-			$canCreateRoles = false;
-		}
-
-		if($canCreateRoles) {
-			//use the admin login data for the new database user
-
-			//add prefix to the postgresql user name to prevent collisions
-			$this->dbUser='oc_'.$username;
-			//create a new password so we don't need to store the admin config in the config file
-			$this->dbPassword = \OC::$server->getSecureRandom()->generate(30, \OCP\Security\ISecureRandom::CHAR_LOWER.\OCP\Security\ISecureRandom::CHAR_DIGITS);
-
-			$this->createDBUser($connection);
-		}
-
 		$systemConfig = $this->config->getSystemConfig();
-		$systemConfig->setValues([
-			'dbuser'		=> $this->dbUser,
-			'dbpassword'	=> $this->dbPassword,
-		]);
+		try {
+			$connection = $this->connect([
+				'dbname' => 'postgres'
+			]);
+			//check for roles creation rights in postgresql
+			$builder = $connection->getQueryBuilder();
+			$builder->automaticTablePrefix(false);
+			$query = $builder
+				->select('rolname')
+				->from('pg_roles')
+				->where($builder->expr()->eq('rolcreaterole', new Literal('TRUE')))
+				->andWhere($builder->expr()->eq('rolname', $builder->createNamedParameter($this->dbUser)));
 
-		//create the database
-		$this->createDatabase($connection);
-		$query = $connection->prepare("select count(*) FROM pg_class WHERE relname=? limit 1");
-		$query->execute([$this->tablePrefix . "users"]);
-		$tablesSetup = $query->fetchColumn() > 0;
+			try {
+				$result = $query->execute();
+				$canCreateRoles = $result->rowCount() > 0;
+			} catch (DatabaseException $e) {
+				$canCreateRoles = false;
+			}
 
-		// the connection to dbname=postgres is not needed anymore
-		$connection->close();
+			if ($canCreateRoles) {
+				//use the admin login data for the new database user
+
+				//add prefix to the postgresql user name to prevent collisions
+				$this->dbUser = 'oc_' . $username;
+				//create a new password so we don't need to store the admin config in the config file
+				$this->dbPassword = \OC::$server->getSecureRandom()->generate(30, \OCP\Security\ISecureRandom::CHAR_LOWER . \OCP\Security\ISecureRandom::CHAR_DIGITS);
+
+				$this->createDBUser($connection);
+			}
+
+			$systemConfig->setValues([
+				'dbuser' => $this->dbUser,
+				'dbpassword' => $this->dbPassword,
+			]);
+
+			//create the database
+			$this->createDatabase($connection);
+			$query = $connection->prepare("select count(*) FROM pg_class WHERE relname=? limit 1");
+			$query->execute([$this->tablePrefix . "users"]);
+			$tablesSetup = $query->fetchColumn() > 0;
+
+			// the connection to dbname=postgres is not needed anymore
+			$connection->close();
+		} catch (\Exception $e) {
+			$this->logger->logException($e);
+			$this->logger->warning('Error trying to connect as "postgres", assuming database is setup and tables need to be created');
+			$tablesSetup = false;
+			$systemConfig->setValues([
+				'dbuser' => $this->dbUser,
+				'dbpassword' => $this->dbPassword,
+			]);
+		}
 
 		// connect to the ownCloud database (dbname=$this->dbname) and check if it needs to be filled
 		$this->dbUser = $systemConfig->getValue('dbuser');
@@ -93,13 +103,13 @@ class PostgreSQL extends AbstractDatabase {
 		}
 
 
-		if(!$tablesSetup) {
+		if (!$tablesSetup) {
 			\OC_DB::createDbFromStructure($this->dbDefinitionFile);
 		}
 	}
 
 	private function createDatabase(IDBConnection $connection) {
-		if(!$this->databaseExists($connection)) {
+		if (!$this->databaseExists($connection)) {
 			//The database does not exists... let's create it
 			$query = $connection->prepare("CREATE DATABASE " . addslashes($this->dbName) . " OWNER " . addslashes($this->dbUser));
 			try {