<?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-cache-warmer
 * @version   1.11.13
 * @copyright Copyright (C) 2026 Mirasvit (https://mirasvit.com/)
 */


namespace Mirasvit\CacheWarmer\Repository;


use Magento\Framework\EntityManager\EntityManager;
use Mirasvit\CacheWarmer\Api\Data\SourceInterface;
use Mirasvit\CacheWarmer\Api\Data\SourceInterfaceFactory;
use Mirasvit\CacheWarmer\Api\Repository\SourceRepositoryInterface;
use Mirasvit\CacheWarmer\Model\ResourceModel\Source\CollectionFactory;

class SourceRepository implements SourceRepositoryInterface
{
    private $entityManager;

    private $collectionFactory;

    private $factory;

    public function __construct(
        EntityManager $entityManager,
        CollectionFactory $collectionFactory,
        SourceInterfaceFactory $factory
    ) {
        $this->entityManager     = $entityManager;
        $this->collectionFactory = $collectionFactory;
        $this->factory           = $factory;
    }

    /**
     * {@inheritdoc}
     */
    public function getCollection($activeOnly = true)
    {
        $collection = $this->collectionFactory->create();
        
        if ($activeOnly) {
            $collection->addFieldToFilter('is_active', 1);
        }
        
        return $collection;
    }

    /**
     * {@inheritdoc}
     */
    public function get($id)
    {
        $model = $this->create();
        $model = $this->entityManager->load($model, $id);

        if (!$model->getId()) {
            return false;
        }

        return $model;
    }

    /**
     * @return SourceInterface|null
     */
    public function getDefaultSource()
    {
        $defaultSource = $this->collectionFactory->create()
            ->addFieldToFilter(SourceInterface::SOURCE_TYPE, SourceInterface::TYPE_VISITOR)
            ->getFirstItem();

        return $defaultSource->getId()
            ? $defaultSource
            : null;
    }

    /**
     * @return SourceInterface|null
     */
    public function getCrawlerSource()
    {
        $crawlerSource = $this->getCollection()
            ->addFieldToFilter(SourceInterface::SOURCE_TYPE, SourceInterface::TYPE_CRAWLER)
            ->getFirstItem();

        return $crawlerSource->getId()
            ? $crawlerSource
            : null;
    }

    /**
     * {@inheritdoc}
     */
    public function create()
    {
        return $this->factory->create();
    }

    /**
     * {@inheritdoc}
     */
    public function delete(SourceInterface $model)
    {
        $this->entityManager->delete($model);
    }

    /**
     * {@inheritdoc}
     */
    public function save(SourceInterface $model)
    {
        return $this->entityManager->save($model);
    }
}