<?php
/**
 * Mirasvit
 *
 * This source file is subject to the Mirasvit Software License, which is available at https://mirasvit.com/license/.
 * Do not edit or add to this file if you wish to upgrade the to newer versions in the future.
 * If you wish to customize this module for your needs.
 * Please refer to http://www.magentocommerce.com for more information.
 *
 * @category  Mirasvit
 * @package   mirasvit/module-rma
 * @version   2.1.17
 * @copyright Copyright (C) 2020 Mirasvit (https://mirasvit.com/)
 */



namespace Mirasvit\Rma\Model\UI\Status\Column;

use Magento\Framework\Escaper;
use Magento\Framework\View\Element\UiComponent\ContextInterface;
use Magento\Framework\View\Element\UiComponentFactory;
use Magento\Ui\Component\Listing\Columns\Column;
use Mirasvit\Rma\Api\Data\StatusInterface;
use Mirasvit\Rma\Api\Service\Status\StatusManagementInterface;
use Mirasvit\Rma\Model\ResourceModel\Status\Collection;

class ColorColumn extends Column
{
    /**
     * @var Collection
     */
    private $statusCollection;
    /**
     * @var StatusManagementInterface
     */
    private $statusManagement;

    private $escaper;

    /**
     * ColorColumn constructor.
     * @param Escaper $escaper
     * @param Collection $statusCollection
     * @param StatusManagementInterface $statusManagement
     * @param ContextInterface $context
     * @param UiComponentFactory $uiComponentFactory
     * @param array $components
     * @param array $data
     */
    public function __construct(
        Escaper $escaper,
        Collection $statusCollection,
        StatusManagementInterface $statusManagement,
        ContextInterface $context,
        UiComponentFactory $uiComponentFactory,
        array $components = [],
        array $data = []
    ) {
        parent::__construct($context, $uiComponentFactory, $components, $data);

        $this->escaper          = $escaper;
        $this->statusCollection = $statusCollection;
        $this->statusManagement = $statusManagement;
    }

    /**
     * @param array $dataSource
     *
     * @return array
     */
    public function prepareDataSource(array $dataSource)
    {
        if (isset($dataSource['data']['items'])) {
            foreach ($dataSource['data']['items'] as &$item) {

                $css = "mst-rma-badge status-branch-" . $item[$this->getData('name')];

                $item[$this->getData('name')] = "<span class='" . $css . "'>" .
                    $this->escaper->escapeHtml($item[StatusInterface::KEY_NAME]) .
                "</span>";
            }
        }

        return $dataSource;
    }
}
