Document.php

Go to the documentation of this file.

<?php
namespace Magento\Sales\Ui\Component\DataProvider;

use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Sales\Model\Order\Invoice;
use Magento\Customer\Api\GroupRepositoryInterface;
use Magento\Framework\Api\AttributeValueFactory;

class Document extends \Magento\Framework\View\Element\UiComponent\DataProvider\Document
{
    private static $stateAttributeCode = 'state';

    private static $customerGroupAttributeCode = 'customer_group_id';

    private $groupRepository;

    public function __construct(
        AttributeValueFactory $attributeValueFactory,
        GroupRepositoryInterface $groupRepository
    ) {
        parent::__construct($attributeValueFactory);
        $this->groupRepository = $groupRepository;
    }

    public function getCustomAttribute($attributeCode)
    {
        switch ($attributeCode) {
            case self::$stateAttributeCode:
                $this->setStateValue();
                break;
            case self::$customerGroupAttributeCode:
                $this->setCustomerGroupValue();
                break;
        }
        return parent::getCustomAttribute($attributeCode);
    }

    private function setStateValue()
    {
        $value = $this->getData(self::$stateAttributeCode);
        $state = Invoice::getStates()[$value];

        $this->setCustomAttribute(self::$stateAttributeCode, $state->getText());
    }

    private function setCustomerGroupValue()
    {
        $value = $this->getData(self::$customerGroupAttributeCode);
        try {
            $group = $this->groupRepository->getById($value);
            $this->setCustomAttribute(self::$customerGroupAttributeCode, $group->getCode());
        } catch (NoSuchEntityException $e) {
            $this->setCustomAttribute(self::$customerGroupAttributeCode, 'N/A');
        }
    }
}