1 files changed, 134 insertions, 0 deletions
diff --git a/lib/private/DB/SchemaWrapper.php b/lib/private/DB/SchemaWrapper.php
new file mode 100644
index 00000000000..0d5b2040513
--- /dev/null
+++ b/lib/private/DB/SchemaWrapper.php
@@ -0,0 +1,134 @@
+<?php
+
+/**
+ * SPDX-FileCopyrightText: 2017 Nextcloud GmbH and Nextcloud contributors
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+namespace OC\DB;
+
+use Doctrine\DBAL\Exception;
+use Doctrine\DBAL\Platforms\AbstractPlatform;
+use Doctrine\DBAL\Schema\Schema;
+use OCP\DB\ISchemaWrapper;
+
+class SchemaWrapper implements ISchemaWrapper {
+	/** @var Connection */
+	protected $connection;
+
+	/** @var Schema */
+	protected $schema;
+
+	/** @var array */
+	protected $tablesToDelete = [];
+
+	public function __construct(Connection $connection, ?Schema $schema = null) {
+		$this->connection = $connection;
+		if ($schema) {
+			$this->schema = $schema;
+		} else {
+			$this->schema = $this->connection->createSchema();
+		}
+	}
+
+	public function getWrappedSchema() {
+		return $this->schema;
+	}
+
+	public function performDropTableCalls() {
+		foreach ($this->tablesToDelete as $tableName => $true) {
+			$this->connection->dropTable($tableName);
+			foreach ($this->connection->getShardConnections() as $shardConnection) {
+				$shardConnection->dropTable($tableName);
+			}
+			unset($this->tablesToDelete[$tableName]);
+		}
+	}
+
+	/**
+	 * Gets all table names
+	 *
+	 * @return array
+	 */
+	public function getTableNamesWithoutPrefix() {
+		$tableNames = $this->schema->getTableNames();
+		return array_map(function ($tableName) {
+			if (str_starts_with($tableName, $this->connection->getPrefix())) {
+				return substr($tableName, strlen($this->connection->getPrefix()));
+			}
+
+			return $tableName;
+		}, $tableNames);
+	}
+
+	// Overwritten methods
+
+	/**
+	 * @return array
+	 */
+	public function getTableNames() {
+		return $this->schema->getTableNames();
+	}
+
+	/**
+	 * @param string $tableName
+	 *
+	 * @return \Doctrine\DBAL\Schema\Table
+	 * @throws \Doctrine\DBAL\Schema\SchemaException
+	 */
+	public function getTable($tableName) {
+		return $this->schema->getTable($this->connection->getPrefix() . $tableName);
+	}
+
+	/**
+	 * Does this schema have a table with the given name?
+	 *
+	 * @param string $tableName
+	 *
+	 * @return boolean
+	 */
+	public function hasTable($tableName) {
+		return $this->schema->hasTable($this->connection->getPrefix() . $tableName);
+	}
+
+	/**
+	 * Creates a new table.
+	 *
+	 * @param string $tableName
+	 * @return \Doctrine\DBAL\Schema\Table
+	 */
+	public function createTable($tableName) {
+		unset($this->tablesToDelete[$tableName]);
+		return $this->schema->createTable($this->connection->getPrefix() . $tableName);
+	}
+
+	/**
+	 * Drops a table from the schema.
+	 *
+	 * @param string $tableName
+	 * @return \Doctrine\DBAL\Schema\Schema
+	 */
+	public function dropTable($tableName) {
+		$this->tablesToDelete[$tableName] = true;
+		return $this->schema->dropTable($this->connection->getPrefix() . $tableName);
+	}
+
+	/**
+	 * Gets all tables of this schema.
+	 *
+	 * @return \Doctrine\DBAL\Schema\Table[]
+	 */
+	public function getTables() {
+		return $this->schema->getTables();
+	}
+
+	/**
+	 * Gets the DatabasePlatform for the database.
+	 *
+	 * @return AbstractPlatform
+	 *
+	 * @throws Exception
+	 */
+	public function getDatabasePlatform() {
+		return $this->connection->getDatabasePlatform();
+	}
+}