Thinktomorrow\Vine\Branch\BranchCollection

Complexity

Total Complexity

12

Size/Duplication

Total Lines	60
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

96.43%

Importance

Changes	5
Bugs	0	Features	0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	hasBranches()	0	4	2
A	push()	0	6	1
A	__get()	0	6	2
A	validateBranches()	0	10	3
A	flatten()	0	8	1
A	flattenRecursive()	0	11	2

<?php

namespace Thinktomorrow\Vine\Branch;

use Thinktomorrow\Vine\VineException;

/**
 * Class BranchCollection
 *
 * @property array $branches
 */
class BranchCollection extends ArrayCollection
{
    public function __construct(array $branches = [])
    {
        self::validateBranches($branches);

        $this->branches = $branches;
    }

    public function hasBranches()
    {
        return (!empty($this->branches) && count($this->branches) > 0);
    }

    public function push(Branch $branch)
    {
        $this->branches[] = $branch;

        return $this;
    }

    public function __get($key)
    {
        if(isset($this->$key)) return $this->$key;

        throw VineException::unknownBranchCollectionAttribute($key);
    }

    public static function validateBranches($branches)
    {
        foreach($branches as $branch)
        {
            if(!$branch instanceof Branch)
            {
                throw VineException::requiredAsBranch($branch);
            }
        }
    }

    public function flatten()
    {
        $flatten = array_reduce($this->branches,function($carry,$branch){
            return array_merge($carry,$this->flattenRecursive($branch,0));
        },[]);

        return new static($flatten);
    }

    protected function flattenRecursive(Branch $branch, $level = 0)
    {
        $children = $branch->branches;

        // Remove the children reference
        $output = [new Branch($branch->key,$branch->value,$branch->label)];

        foreach($children as $child) $output = array_merge($output,$this->flattenRecursive($child,$level+1));

        return $output;
    }
}