Encima\Albero\Extensions\Eloquent\Collection

Complexity

Total Complexity

8

Size/Duplication

Total Lines	37
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	toHierarchy()	0	10	2
A	hierarchical()	0	23	6

<?php

namespace Encima\Albero\Extensions\Eloquent;

use Illuminate\Database\Eloquent\Collection as BaseCollection;

class Collection extends BaseCollection
{
    public function toHierarchy(): BaseCollection
    {
        $dict = $this->getDictionary();

        // Enforce sorting by $orderColumn setting in Baum\Node instance
        uasort($dict, function ($a, $b) {
            return ($a->getOrder() >= $b->getOrder()) ? 1 : -1;
        });

        return new BaseCollection($this->hierarchical($dict));
    }

    protected function hierarchical(array $result): array
    {
        foreach ($result as $key => $node) {
            $node->setRelation('children', new BaseCollection());
        }

        $nestedKeys = [];

        foreach ($result as $key => $node) {
            $parentKey = $node->getParentId();

            if (!is_null($parentKey) && array_key_exists($parentKey, $result)) {
                $result[$parentKey]->children[] = $node;

                $nestedKeys[] = $node->getKey();
            }
        }

        foreach ($nestedKeys as $key) {
            unset($result[$key]);
        }

        return $result;
    }
}