DBFaker\Generators\CompoundColumnGenerator

Complexity

Total Complexity

8

Size/Duplication

Total Lines	67
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	22	5
A	__invoke()	0	11	3

<?php

namespace DBFaker\Generators;

use DBFaker\DBFaker;
use DBFaker\Exceptions\SchemaLogicException;
use DBFaker\Helpers\DBFakerSchemaManager;
use DBFaker\Helpers\SchemaHelper;
use Doctrine\DBAL\Schema\AbstractSchemaManager;
use Doctrine\DBAL\Schema\Index;
use Doctrine\DBAL\Schema\Table;

class CompoundColumnGenerator implements FakeDataGeneratorInterface
{

    /**
     * @var mixed[]
     */
    private $possibleValues = [];

    /**
     * @var mixed[]
     */
    private $generatedValues = [];


    /**
     * CompoundColumnGenerator constructor.
     * @param Table $table
     * @param Index $index
     * @param SchemaHelper $schemaHelper
     * @param DBFaker $dbFaker
     * @param AbstractSchemaManager $schemaManager
     * @param int $valuesCount
     * @throws \DBFaker\Exceptions\SchemaLogicException
     * @throws \DBFaker\Exceptions\UnsupportedDataTypeException
     * @throws \Doctrine\DBAL\Schema\SchemaException
     * @throws \Doctrine\DBAL\DBALException
     */
    public function __construct(Table $table, Index $index, SchemaHelper $schemaHelper, DBFaker $dbFaker, AbstractSchemaManager $schemaManager, DBFakerSchemaManager $fakerManagerHelper, int $valuesCount)
    {
        foreach ($index->getColumns() as $columnName) {
            //FK or normal column ?
            $column = $table->getColumn($columnName);
            if ($schemaHelper->isColumnPartOfForeignKeyConstraint($table, $column)) {
                $fkConstraint = $schemaHelper->getForeignKeyConstraintByLocal($table, $column);
                if ($fkConstraint === null) {
                    throw new SchemaLogicException($column->getName() . ' was detected as foreign key but could not get it');
                }
                $foreignTableName = $fkConstraint->getForeignTableName();
                $foreignColumn = $fakerManagerHelper->getForeignColumn($table, $column);
                $foreignTable = $schemaManager->listTableDetails($foreignTableName);
                $pkRegistry = $dbFaker->getPkRegistry($foreignTable);
                $values = $pkRegistry->loadValuesFromTable()->getAllValues();
                $this->possibleValues[$columnName] = array_map(function ($value) use ($foreignColumn) {
                    return $value[$foreignColumn->getName()];
                }, $values);
            } else {
                $generator = $dbFaker->getSimpleColumnGenerator($table, $column);
                for ($i = 0; $i < $valuesCount; $i++) {
                    $this->possibleValues[$columnName][] = $generator();
                }
            }
        }
    }
    /**
     * @return mixed[]
     */
    public function __invoke() : array
    {
        $returnVal = [];
        foreach ($this->possibleValues as $columnName => $values) {
            $returnVal[$columnName] = $values[array_rand($values)];
        }
        if (!\in_array($returnVal, $this->generatedValues, true)) {
            $this->generatedValues[] = $returnVal;
            return $returnVal;
        }
        return $this();
    }
}