KitLoong\MigrationsGenerator\Generators\ForeignKeyGenerator

Complexity

Total Complexity

11

Size/Duplication

Total Lines	82
Duplicated Lines	0 %

Test Coverage

Coverage

11.11%

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	createIndexName()	0	5	1
A	generate()	0	22	5
A	__construct()	0	3	1
A	getName()	0	6	3
A	isDefaultName()	0	3	1

<?php namespace KitLoong\MigrationsGenerator\Generators;

class ForeignKeyGenerator
{
    /**
     * @var string
     */
    protected $table;

    private $decorator;

    public function __construct(Decorator $decorator)
    {
        $this->decorator = $decorator;
    }

    /**
     * Get array of foreign keys
     *
     * @param  string  $table  Table name
     * @param  \Doctrine\DBAL\Schema\AbstractSchemaManager  $schema
     * @param  bool  $ignoreForeignKeyNames
     *
     * @return array
     */
    public function generate(string $table, $schema, bool $ignoreForeignKeyNames): array
    {
        $this->table = $table;
        $fields = [];

        $foreignKeys = $schema->listTableForeignKeys($table);

        if (empty($foreignKeys)) {
            return [];
        }

        foreach ($foreignKeys as $foreignKey) {
            $fields[] = [
                'name' => $this->getName($foreignKey, $ignoreForeignKeyNames),
                'field' => $foreignKey->getLocalColumns()[0],
                'references' => $foreignKey->getForeignColumns()[0],
                'on' => $this->decorator->tableWithoutPrefix($foreignKey->getForeignTableName()),
                'onUpdate' => $foreignKey->hasOption('onUpdate') ? $foreignKey->getOption('onUpdate') : 'RESTRICT',
                'onDelete' => $foreignKey->hasOption('onDelete') ? $foreignKey->getOption('onDelete') : 'RESTRICT',
            ];
        }
        return $fields;
    }

    /**
     * @param  \Doctrine\DBAL\Schema\ForeignKeyConstraint  $foreignKey
     * @param  bool  $ignoreForeignKeyNames
     *
     * @return null|string
     */
    protected function getName($foreignKey, bool $ignoreForeignKeyNames): ?string
    {
        if ($ignoreForeignKeyNames or $this->isDefaultName($foreignKey)) {
            return null;
        }
        return $foreignKey->getName();
    }

    /**
     * @param  \Doctrine\DBAL\Schema\ForeignKeyConstraint  $foreignKey
     *
     * @return bool
     */
    protected function isDefaultName($foreignKey): bool
    {
        return $foreignKey->getName() === $this->createIndexName($foreignKey->getLocalColumns()[0]);
    }

    /**
     * Create a default index name for the table.
     *
     * @param  string  $column
     * @return string
     */
    protected function createIndexName(string $column): string
    {
        $index = strtolower($this->table.'_'.$column.'_foreign');

        return str_replace(['-', '.'], '_', $index);
    }
}