Cycle\Schema\Relation\Traits\FieldTrait

Complexity

Total Complexity

7

Size/Duplication

Total Lines	60
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	hasField()	0	3	1
A	getField()	0	3	1
A	ensureField()	0	27	5

<?php
declare(strict_types=1);
/**
 * Spiral Framework.
 *
 * @license   MIT
 * @author    Anton Titov (Wolfy-J)
 */

namespace Cycle\Schema\Relation\Traits;

use Cycle\ORM\Relation;
use Cycle\Schema\Definition\Entity;
use Cycle\Schema\Definition\Field;
use Cycle\Schema\Generator\Table\ColumnSchema;
use Cycle\Schema\Relation\Util\OptionSchema;

trait FieldTrait
{
    /**
     * @param Entity $entity
     * @param int    $field
     * @return bool
     */
    protected function hasField(Entity $entity, int $field): bool
    {
        return $entity->getFields()->has($this->getOptions()->get($field));
    }

    /**
     * @param Entity $entity
     * @param int    $field
     * @return Field
     */
    protected function getField(Entity $entity, int $field): Field
    {
        return $entity->getFields()->get($this->getOptions()->get($field));
    }

    /**
     * @param Entity $target
     * @param Field  $source
     * @param string $name
     */
    protected function ensureField(Entity $target, Field $source, string $name)
    {
        if ($target->getFields()->has($name)) {
            // field already exists and defined by the user
            return;
        }

        $field = new Field();
        $field->setColumn($name);
        $field->setTypecast($source->getTypecast());

        switch ($source->getType()) {
            case 'primary':
                $field->setType('int');
                break;
            case 'bigPrimary':
                $field->setType('bigint');
                break;
            default:
                $field->setType($source->getType());
        }

        if ($this->getOptions()->get(Relation::NULLABLE)) {
            $field->getOptions()->set(ColumnSchema::OPT_NULLABLE, true);
        }

        $target->getFields()->set($name, $field);
    }

    /**
     * @return OptionSchema
     */
    abstract protected function getOptions(): OptionSchema;
}