WellCommerce\Component\DataSet\Context\FlatTreeContext

Complexity

Total Complexity

5

Size/Duplication

Total Lines	68
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	getResult()	0	6	1
A	makeTree()	0	10	2
A	makeNode()	0	9	1
A	configureOptions()	0	18	1

<?php
/*
 * WellCommerce Open-Source E-Commerce Platform
 *
 * This file is part of the WellCommerce package.
 *
 * (c) Adam Piotrowski <adam@wellcommerce.org>
 *
 * For the full copyright and license information,
 * please view the LICENSE file that was distributed with this source code.
 */

namespace WellCommerce\Component\DataSet\Context;

use Doctrine\ORM\QueryBuilder;
use Symfony\Component\OptionsResolver\OptionsResolver;
use WellCommerce\Component\DataSet\Cache\CacheOptions;
use WellCommerce\Component\DataSet\Column\ColumnCollection;
use WellCommerce\Component\DataSet\Request\DataSetRequestInterface;

/**
 * Class FlatTreeContext
 *
 * @author Adam Piotrowski <adam@wellcommerce.org>
 */
class FlatTreeContext extends ArrayContext
{
    /**
     * {@inheritdoc}
     */
    public function getResult(QueryBuilder $builder, DataSetRequestInterface $request, ColumnCollection $columns, CacheOptions $cache)
    {
        $result = parent::getResult($builder, $request, $columns, $cache);

        return $this->makeTree($result['rows']);
    }

    /**
     * Creates flat tree structure from result rows
     *
     * @param array $result
     *
     * @return array
     */
    private function makeTree(array $result)
    {
        $tree = [];

        foreach ($result as $row) {
            $this->makeNode($row, $tree);
        }

        return $tree;
    }

    /**
     * Converts single row to flat-tree node
     *
     * @param array $row
     * @param array $tree
     */
    private function makeNode(&$row, &$tree)
    {
        $hasChildren = $this->propertyAccessor->getValue($row, "[{$this->options['children_column']}]");
        $weight      = $this->propertyAccessor->getValue($row, "[{$this->options['hierarchy_column']}]");

        $row['hasChildren'] = (bool)($hasChildren > 0);
        $row['weight']      = $weight;
        $tree[$row['id']]   = $row;
    }

    /**
     * {@inheritdoc}
     */
    public function configureOptions(OptionsResolver $resolver)
    {
        parent::configureOptions($resolver);

        $resolver->setRequired([
            'children_column',
            'hierarchy_column',
        ]);

        $resolver->setDefaults([
            'children_column'  => 'children',
            'hierarchy_column' => 'hierarchy',
            'pagination'       => false
        ]);

        $resolver->setAllowedTypes('children_column', 'string');
        $resolver->setAllowedTypes('hierarchy_column', 'string');
    }
}