Sirius\Orm\Entity\GenericEntityHydrator

Complexity

Total Complexity

11

Size/Duplication

Total Lines	65
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

5 Methods

Rating	Name	Duplication	Size	Complexity
A	mapColumnsToAttributes()	0	15	4
A	mapAttributesToColumns()	0	15	4
A	extract()	0	5	1
A	hydrate()	0	6	1
A	__construct()	0	4	1

<?php
declare(strict_types=1);

namespace Sirius\Orm\Entity;

use Sirius\Orm\Helpers\Arr;
use Sirius\Orm\Mapper;
use Sirius\Orm\Orm;

class GenericEntityHydrator implements HydratorInterface
{
    /**
     * @var Orm
     */
    protected $orm;

    /**
     * @var Mapper
     */
    protected $mapper;

    public function __construct(Orm $orm, Mapper $mapper)
    {
        $this->orm    = $orm;
        $this->mapper = $mapper;
    }

    public function hydrate($attributes = [])
    {
        $attributes = $this->mapColumnsToAttributes($attributes);
        $class      = $this->mapper->getEntityClass() ?? GenericEntity::class;

        return new $class($attributes, $this->orm->getCastingManager());
    }

    protected function mapColumnsToAttributes($arr)
    {
        $map = $this->mapper->getColumnAttributeMap();
        if (empty($map)) {
            return $arr;
        }

        foreach ($map as $column => $attribute) {
            if (isset($arr[$column])) {
                $arr[$attribute] = $arr[$column];
                unset($arr[$column]);
            }
        }

        return $arr;
    }

    public function extract(EntityInterface $entity)
    {
        $data = $this->mapAttributesToColumns($entity->getArrayCopy());

        return Arr::only($data, $this->mapper->getColumns());
    }

    public function mapAttributesToColumns($arr)
    {
        $map = $this->mapper->getColumnAttributeMap();
        if (empty($map)) {
            return $arr;
        }

        foreach ($map as $column => $attribute) {
            if (isset($arr[$attribute])) {
                $arr[$column] = $arr[$attribute];
                unset($arr[$attribute]);
            }
        }

        return $arr;
    }
}