TheCodingMachine\TDBM\SchemaVersionControl\SchemaNormalizer

Complexity

Total Complexity

37

Size/Duplication

Total Lines	164
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	normalizeForeignReference()	0	18	4
A	normalizeIndex()	0	19	5
A	normalize()	0	15	3
B	normalizeTable()	0	33	8
A	normalizeForeignKeyConstraint()	0	14	3
F	normalizeColumn()	0	46	14

<?php

namespace TheCodingMachine\TDBM\SchemaVersionControl;

use Doctrine\DBAL\Schema\Column;
use Doctrine\DBAL\Schema\ForeignKeyConstraint;
use Doctrine\DBAL\Schema\Index;
use Doctrine\DBAL\Schema\Schema;
use Doctrine\DBAL\Schema\SchemaConfig;
use Doctrine\DBAL\Schema\Table;

/**
 * Database schema normalizer.
 *
 * Given an instance of Schema, it will construct a deep associative array to describe it. Such an array will then be
 * easy to serialize.
 */
class SchemaNormalizer
{
    protected Schema $schema;
    protected SchemaConfig $schemaConfig;

    /**
     * Normalize a Schema object into an array descriptor
     * @return array
     */
    public function normalize(Schema $schema, SchemaConfig $schemaConfig): array
    {
        $this->schema = $schema;
        $this->schemaConfig = $schemaConfig;
        $schemaDesc = [];
        if (!empty($this->schemaConfig->getDefaultTableOptions())) {
            $schemaDesc['default_table_options'] = $this->schemaConfig->getDefaultTableOptions();
        }
        $schemaDesc['tables'] = [];
        foreach ($schema->getTables() as $table) {
            $schemaDesc['tables'][$table->getName()] = $this->normalizeTable($table);
        }
        ksort($schemaDesc['tables']);

        return $schemaDesc;
    }

    protected function normalizeTable(Table $table)
    {
        $tableDesc = [];

        $primaryKey = $table->getPrimaryKey();
        if ($primaryKey) {
            $pk_columns = $primaryKey->getUnquotedColumns();
        } else {
            $pk_columns = [];
        }

        if ($table->hasOption('comment') && $table->getOption('comment')) {
            $tableDesc['comment'] = $table->getOption('comment');
        }

        // list columns
        foreach ($table->getColumns() as $columnName => $column) {
            $tableDesc['columns'][$column->getName()] = $this->normalizeColumn($column, in_array($column->getName(), $pk_columns));
        }

        // list indexes
        foreach ($table->getIndexes() as $index) {
            if (!$index->isPrimary()) {
                $tableDesc['indexes'][$index->getName()] = $this->normalizeIndex($index);
            }
        }

        // list foreign keys
        foreach ($table->getForeignKeys() as $foreignKey) {
            $tableDesc['foreign_keys'][$foreignKey->getName()] = $this->normalizeForeignKeyConstraint($foreignKey);
        }

        return $tableDesc;
    }

    protected function normalizeColumn(Column $column, bool $isPrimaryKey)
    {
        $columnDesc = [];
        if ($isPrimaryKey) {
            $columnDesc['primary_key'] = $isPrimaryKey;
        }
        $columnDesc['type'] = $column->getType()->getName();

The function Doctrine\DBAL\Types\Type::getName() has been deprecated: this method will be removed in Doctrine DBAL 4.0, use {@see TypeRegistry::lookupName()} instead.

        if ($column->getUnsigned()) {
            $columnDesc['unsigned'] = $column->getUnsigned();
        }
        if ($column->getFixed()) {
            $columnDesc['fixed'] = $column->getFixed();
        }
        if ($column->getLength() !== null) {
            $columnDesc['length'] = $column->getLength();
        }
        if ($column->getPrecision() !== 10) {
            $columnDesc['precision'] = $column->getPrecision();
        }
        if ($column->getScale() !== 0) {
            $columnDesc['scale'] = $column->getScale();
        }
        if ($column->getNotnull()) {
            $columnDesc['not_null'] = $column->getNotnull();
        }
        if ($column->getDefault() !== null) {
            $columnDesc['default'] = $column->getDefault();
        }
        if ($column->getAutoincrement()) {
            $columnDesc['auto_increment'] = $column->getAutoincrement();
        }
        if ($column->getComment() !== null) {
            $columnDesc['comment'] = $column->getComment();
        }
        if (!empty($column->getPlatformOptions())) {
            $custom = array_diff_assoc($column->getPlatformOptions(), $this->schemaConfig->getDefaultTableOptions());
            if (!empty($custom)) {
                $columnDesc['custom'] = $custom;
            }
        }

        if (count($columnDesc) > 1) {
            return $columnDesc;
        }

        return $columnDesc['type'];
    }

    protected function normalizeForeignKeyConstraint(ForeignKeyConstraint $foreignKeyConstraint)
    {
        $constraintDesc = [];
        if (count($foreignKeyConstraint->getLocalColumns()) > 1) {
            $constraintDesc['columns'] = $foreignKeyConstraint->getLocalColumns();
        } else {
            $constraintDesc['column'] = $foreignKeyConstraint->getLocalColumns()[0];
        }

        $constraintDesc['references'] = $this->normalizeForeignReference($foreignKeyConstraint);
        if (!empty($foreignKeyConstraint->getOptions())) {
            $constraintDesc = array_merge($constraintDesc, $foreignKeyConstraint->getOptions());
        }
        return $constraintDesc;
    }

    protected function normalizeForeignReference(ForeignKeyConstraint $foreignKeyConstraint)
    {
        $referenceDesc = [];
        $foreignTableName = $foreignKeyConstraint->getForeignTableName();
        $foreignTable = $this->schema->getTable($foreignTableName);
        $foreignPrimaryKey = $foreignTable->getPrimaryKey();
        if ($foreignPrimaryKey && $foreignPrimaryKey->getColumns() == $foreignKeyConstraint->getForeignColumns()) {
            $referenceDesc = $foreignKeyConstraint->getForeignTableName();
        } else {
            $referenceDesc['table'] = $foreignKeyConstraint->getForeignTableName();
            $fkColumns = $foreignKeyConstraint->getForeignColumns();
            if (count($fkColumns) > 1) {
                $referenceDesc['columns'] = $fkColumns;
            } else {
                $referenceDesc['column'] = $fkColumns[0];
            }
        }
        return $referenceDesc;
    }

    protected function normalizeIndex(Index $index)
    {
        $indexDesc = [];
        $columns = $index->getColumns();
        if (count($columns) > 1) {
            $indexDesc['columns'] = $index->getColumns();
        } else {
            $indexDesc['column'] = $index->getColumns()[0];
        }
        if ($index->isUnique()) {
            $indexDesc['unique'] = $index->isUnique();
        }
        if ($index->isPrimary()) {
            $indexDesc['primary'] = $index->isPrimary();
        }
        if (!empty($index->getOptions())) {
            $indexDesc = array_merge($indexDesc, $index->getOptions());
        }
        return $indexDesc;
    }
}