CategoryTree.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\CatalogGraphQl\Model\Resolver;

use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
use Magento\Framework\GraphQl\Config\Element\Field;
use Magento\Framework\GraphQl\Exception\GraphQlInputException;
use Magento\Framework\GraphQl\Query\ResolverInterface;

class CategoryTree implements ResolverInterface
{
    const CATEGORY_INTERFACE = 'CategoryInterface';

    private $categoryTree;

    public function __construct(
        \Magento\CatalogGraphQl\Model\Resolver\Products\DataProvider\CategoryTree $categoryTree
    ) {
        $this->categoryTree = $categoryTree;
    }

    private function getCategoryId(array $args) : int
    {
        if (!isset($args['id'])) {
            throw new GraphQlInputException(__('"id for category should be specified'));
        }

        return (int)$args['id'];
    }

    public function resolve(Field $field, $context, ResolveInfo $info, array $value = null, array $args = null)
    {
        if (isset($value[$field->getName()])) {
            return $value[$field->getName()];
        }

        $rootCategoryId = $this->getCategoryId($args);
        $categoriesTree = $this->categoryTree->getTree($info, $rootCategoryId);
        if (!empty($categoriesTree)) {
            return current($categoriesTree);
        } else {
            return null;
        }
    }
}