feat: add commands for exporting current and expected database schema

Signed-off-by: Robin Appelman <robin@icewind.nl>

1 files changed, 115 insertions, 0 deletions

diff --git a/core/Command/Db/SchemaEncoder.php b/core/Command/Db/SchemaEncoder.php
new file mode 100644
index 00000000000..d36a34e8583
--- /dev/null
+++ b/core/Command/Db/SchemaEncoder.php
@@ -0,0 +1,115 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * SPDX-FileCopyrightText: 2024 Robin Appelman <robin@icewind.nl>
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ */
+
+namespace OC\Core\Command\Db;
+
+use Doctrine\DBAL\Platforms\AbstractMySQLPlatform;
+use Doctrine\DBAL\Platforms\AbstractPlatform;
+use Doctrine\DBAL\Platforms\PostgreSQLPlatform;
+use Doctrine\DBAL\Schema\Schema;
+use Doctrine\DBAL\Schema\Table;
+use Doctrine\DBAL\Types\PhpIntegerMappingType;
+use Doctrine\DBAL\Types\Type;
+
+class SchemaEncoder {
+	/**
+	 * Encode a DBAL schema to json, performing some normalization based on the database platform
+	 *
+	 * @param Schema $schema
+	 * @param AbstractPlatform $platform
+	 * @return array
+	 */
+	public function encodeSchema(Schema $schema, AbstractPlatform $platform): array {
+		$encoded = ['tables' => [], 'sequences' => []];
+		foreach ($schema->getTables() as $table) {
+			$encoded[$table->getName()] = $this->encodeTable($table, $platform);
+		}
+		ksort($encoded);
+		return $encoded;
+	}
+
+	/**
+	 * @psalm-type ColumnArrayType =
+	 */
+	private function encodeTable(Table $table, AbstractPlatform $platform): array {
+		$encoded = ['columns' => [], 'indexes' => []];
+		foreach ($table->getColumns() as $column) {
+			/**
+			 * @var array{
+			 *     name: string,
+			 *     default: mixed,
+			 *     notnull: bool,
+			 *     length: ?int,
+			 *     precision: int,
+			 *     scale: int,
+			 *     unsigned: bool,
+			 *     fixed: bool,
+			 *     autoincrement: bool,
+			 *     comment: string,
+			 *     columnDefinition: ?string,
+			 *     collation?: string,
+			 *     charset?: string,
+			 *     jsonb?: bool,
+			 * } $data
+			 **/
+			$data = $column->toArray();
+			$data['type'] = Type::getTypeRegistry()->lookupName($column->getType());
+			$data['default'] = $column->getType()->convertToPHPValue($column->getDefault(), $platform);
+			if ($platform instanceof PostgreSQLPlatform) {
+				$data['unsigned'] = false;
+				if ($column->getType() instanceof PhpIntegerMappingType) {
+					$data['length'] = null;
+				}
+				unset($data['jsonb']);
+			} elseif ($platform instanceof AbstractMySqlPlatform) {
+				if ($column->getType() instanceof PhpIntegerMappingType) {
+					$data['length'] = null;
+				} elseif (in_array($data['type'], ['text', 'blob', 'datetime', 'float', 'json'])) {
+					$data['length'] = 0;
+				}
+				unset($data['collation']);
+				unset($data['charset']);
+			}
+			if ($data['type'] === 'string' && $data['length'] === null) {
+				$data['length'] = 255;
+			}
+			$encoded['columns'][$column->getName()] = $data;
+		}
+		ksort($encoded['columns']);
+		foreach ($table->getIndexes() as $index) {
+			$options = $index->getOptions();
+			if (isset($options['lengths']) && count(array_filter($options['lengths'])) === 0) {
+				unset($options['lengths']);
+			}
+			if ($index->isPrimary()) {
+				if ($platform instanceof PostgreSqlPlatform) {
+					$name = $table->getName() . '_pkey';
+				} elseif ($platform instanceof AbstractMySQLPlatform) {
+					$name = "PRIMARY";
+				} else {
+					$name = $index->getName();
+				}
+			} else {
+				$name = $index->getName();
+			}
+			if ($platform instanceof PostgreSqlPlatform) {
+				$name = strtolower($name);
+			}
+			$encoded['indexes'][$name] = [
+				'name' => $name,
+				'columns' => $index->getColumns(),
+				'unique' => $index->isUnique(),
+				'primary' => $index->isPrimary(),
+				'flags' => $index->getFlags(),
+				'options' => $options,
+			];
+		}
+		ksort($encoded['indexes']);
+		return $encoded;
+	}
+}