<?php

namespace Stemcell\Customer\Ui\Component\DataProvider;

use Magento\Framework\App\ObjectManager;
use Magento\Framework\Session\SessionManagerInterface;
use Magento\Ui\DataProvider\AbstractDataProvider;
use Psr\Log\LoggerInterface;
use Stemcell\Customer\Model\ResourceModel\Lab\CollectionFactory;

class Lab extends AbstractDataProvider
{
    protected SessionManagerInterface $session;

    public function __construct(
        $name,
        $primaryFieldName,
        $requestFieldName,
        SessionManagerInterface $session,
        CollectionFactory $collectionFactory,
        array $meta = [],
        array $data = []
    ) {
        $this->collection = $collectionFactory->create();
        parent::__construct($name, $primaryFieldName, $requestFieldName, $meta, $data);
        $this->session = $session;
    }

    public function getData(): array
    {
        $data = [];
        $salesperson = $this->session->getLab();
        if (!empty($salesperson) && isset($salesperson['lab_id'])) {
            $data[$salesperson['lab_id']]['lab'] = $salesperson;
        }
        $this->session->unsLab();
        return $data;
    }
}