EntityAttributeList.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\GraphQl\Model;

use Magento\Eav\Api\AttributeManagementInterface;
use Magento\Eav\Api\AttributeSetRepositoryInterface;
use Magento\Eav\Api\Data\AttributeInterface;
use Magento\Framework\Api\FilterBuilder;
use Magento\Framework\Api\MetadataServiceInterface;
use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Framework\GraphQl\Exception\GraphQlInputException;

class EntityAttributeList
{
    private $attributeManagement;

    private $attributeSetRepository;

    private $searchCriteriaBuilder;

    private $filterBuilder;

    public function __construct(
        AttributeManagementInterface $attributeManagement,
        AttributeSetRepositoryInterface $attributeSetRepository,
        SearchCriteriaBuilder $searchCriteriaBuilder,
        FilterBuilder $filterBuilder
    ) {
        $this->attributeManagement =  $attributeManagement;
        $this->attributeSetRepository = $attributeSetRepository;
        $this->searchCriteriaBuilder = $searchCriteriaBuilder;
        $this->filterBuilder = $filterBuilder;
    }

    public function getDefaultEntityAttributes(
        string $entityCode,
        MetadataServiceInterface $metadataService = null
    ) : array {
        $this->searchCriteriaBuilder->addFilters(
            [
                $this->filterBuilder
                    ->setField('entity_type_code')
                    ->setValue($entityCode)
                    ->setConditionType('eq')
                    ->create(),
            ]
        );
        $attributeSetList = $this->attributeSetRepository->getList($this->searchCriteriaBuilder->create())->getItems();
        $attributes = [];
        foreach ($attributeSetList as $attributeSet) {
            try {
                $attributes = array_merge(
                    $attributes,
                    $this->attributeManagement->getAttributes($entityCode, $attributeSet->getAttributeSetId())
                );
            } catch (NoSuchEntityException $exception) {
                throw new GraphQlInputException(__('Entity code %1 does not exist.', [$entityCode]));
            }
        }
        $attributeCodes = [];
        $metadata = $metadataService ? $metadataService->getCustomAttributesMetadata() : [];
        foreach ($metadata as $customAttribute) {
            if (!array_key_exists($customAttribute->getAttributeCode(), $attributeCodes)) {
                $attributeCodes[$customAttribute->getAttributeCode()] = false;
            }
        }
        foreach ($attributes as $attribute) {
            if (!array_key_exists($attribute->getAttributeCode(), $attributeCodes)) {
                $attributeCodes[$attribute->getAttributeCode()]
                    = ((! $attribute->getIsUserDefined()) && !is_array($attribute));
            }
        }
        return $attributeCodes;
    }
}