<?php

namespace Autogedal\Retur\Ui\DataProviders;

use Autogedal\Retur\Api\Data\ReturInterface;
use Autogedal\Retur\Model\ResourceModel\Retur\CollectionFactory;
use Magento\Framework\App\Request\DataPersistorInterface;

class ReturDataProvider extends \Magento\Ui\DataProvider\AbstractDataProvider
{
    /**
     * @var DataPersistorInterface
     */
    private $dataPersistor;

    public function __construct(
        $name,
        $primaryFieldName,
        $requestFieldName,
        CollectionFactory $collectionFactory,
        DataPersistorInterface $dataPersistor,
        array $meta = [],
        array $data = []
    ) {
        parent::__construct($name, $primaryFieldName, $requestFieldName, $meta, $data);

        $this->collection = $collectionFactory->create();
        $this->dataPersistor = $dataPersistor;
    }

    public function getData()
    {
        $result = [];

        foreach ($this->collection->getItems() as $item) {
            // prepare data
            $result[$item->getEntityId()] = $item->getData();
        }

        if ($savedData = $this->dataPersistor->get(ReturInterface::FORM_NAMESPACE)) {
            /** @var AreaInterface $model */
            $model = $this->collection->getNewEmptyItem();
            $model->setData($savedData);
            
            $result[$model->getEntityId()] = $model->getData();
        }

        return $result;
    }
}