CustomerDataProvider.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\CustomerGraphQl\Model\Resolver\Customer;

use Magento\Customer\Api\CustomerRepositoryInterface;
use Magento\Framework\Exception\LocalizedException;
use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Framework\Serialize\SerializerInterface;
use Magento\Framework\Webapi\ServiceOutputProcessor;
use Magento\Customer\Api\Data\CustomerInterface;

class CustomerDataProvider
{
    private $customerRepository;

    private $serviceOutputProcessor;

    private $jsonSerializer;

    public function __construct(
        CustomerRepositoryInterface $customerRepository,
        ServiceOutputProcessor $serviceOutputProcessor,
        SerializerInterface $jsonSerializer
    ) {
        $this->customerRepository = $customerRepository;
        $this->serviceOutputProcessor = $serviceOutputProcessor;
        $this->jsonSerializer = $jsonSerializer;
    }

    public function getCustomerById(int $customerId) : array
    {
        try {
            $customerObject = $this->customerRepository->getById($customerId);
        } catch (NoSuchEntityException $e) {
            // No error should be thrown, null result should be returned
            return [];
        }
        return $this->processCustomer($customerObject);
    }

    private function processCustomer(CustomerInterface $customerObject) : array
    {
        $customer = $this->serviceOutputProcessor->process(
            $customerObject,
            CustomerRepositoryInterface::class,
            'get'
        );
        if (isset($customer['extension_attributes'])) {
            $customer = array_merge($customer, $customer['extension_attributes']);
        }
        $customAttributes = [];
        if (isset($customer['custom_attributes'])) {
            foreach ($customer['custom_attributes'] as $attribute) {
                $isArray = false;
                if (is_array($attribute['value'])) {
                    $isArray = true;
                    foreach ($attribute['value'] as $attributeValue) {
                        if (is_array($attributeValue)) {
                            $customAttributes[$attribute['attribute_code']] = $this->jsonSerializer->serialize(
                                $attribute['value']
                            );
                            continue;
                        }
                        $customAttributes[$attribute['attribute_code']] = implode(',', $attribute['value']);
                        continue;
                    }
                }
                if ($isArray) {
                    continue;
                }
                $customAttributes[$attribute['attribute_code']] = $attribute['value'];
            }
        }
        $customer = array_merge($customer, $customAttributes);

        return $customer;
    }
}