Arp\LaminasEntity\Service\EntityRepositoryFactory

Complexity

Total Complexity

3

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	getRepository()	0	12	2

<?php

declare(strict_types=1);

namespace Arp\LaminasEntity\Service;

use Arp\DoctrineEntityRepository\EntityRepositoryInterface;
use Arp\DoctrineEntityRepository\EntityRepositoryProviderInterface;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Repository\RepositoryFactory;
use Doctrine\Persistence\ObjectRepository;

/**
 * @author  Alex Patterson <alex.patterson.webdev@gmail.com>
 * @package Arp\LaminasEntity\Service
 */
class EntityRepositoryFactory implements RepositoryFactory
{
    /**
     * @var EntityRepositoryProviderInterface
     */
    private EntityRepositoryProviderInterface $repositoryProvider;

    /**
     * The default (fallback) repository factory.
     *
     * @var RepositoryFactory
     */
    private RepositoryFactory $repositoryFactory;

    /**
     * @param EntityRepositoryProviderInterface $repositoryProvider
     * @param RepositoryFactory                 $repositoryFactory
     */
    public function __construct(
        EntityRepositoryProviderInterface $repositoryProvider,
        RepositoryFactory $repositoryFactory
    ) {
        $this->repositoryProvider = $repositoryProvider;
        $this->repositoryFactory = $repositoryFactory;
    }

    /**
     * @param EntityManagerInterface $entityManager
     * @param string                 $entityName
     *
     * @return EntityRepositoryInterface|ObjectRepository
     *
     * @throws \Throwable
     */
    public function getRepository(EntityManagerInterface $entityManager, $entityName): ObjectRepository
    {
        if ($this->repositoryProvider->hasRepository($entityName)) {
            $options = [
                'entity_name'    => $entityName,
                'entity_manager' => $entityManager,
            ];

            return $this->repositoryProvider->getRepository($entityName, $options);
        }

        return $this->repositoryFactory->getRepository($entityManager, $entityName);
    }
}