Zenstruck\Foundry\PersistenceManager

Complexity

Total Complexity

14

Size/Duplication

Total Lines	71
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	register()	0	3	1
A	managerRegistry()	0	7	2
A	persist()	0	7	2
A	objectManagerFor()	0	9	3
A	repositoryFor()	0	13	4
A	proxy()	0	7	2

<?php

namespace Zenstruck\Foundry;

use Doctrine\Persistence\ManagerRegistry;
use Doctrine\Persistence\ObjectManager;
use Doctrine\Persistence\ObjectRepository;

/**
 * @author Kevin Bond <kevinbond@gmail.com>
 */
final class PersistenceManager
{
    private static ?ManagerRegistry $managerRegistry = null;

    /**
     * @param object|string $objectOrClass
     *
     * @return RepositoryProxy|ObjectRepository
     */
    public static function repositoryFor($objectOrClass, bool $proxy = true): ObjectRepository
    {
        if ($objectOrClass instanceof Proxy) {
            $objectOrClass = $objectOrClass->object();
        }

        if (!\is_string($objectOrClass)) {
            $objectOrClass = \get_class($objectOrClass);
        }

        $repository = self::managerRegistry()->getRepository($objectOrClass);

        return $proxy ? new RepositoryProxy($repository) : $repository;
    }

    /**
     * @return Proxy|object
     */
    public static function persist(object $object, bool $proxy = true): object
    {
        $objectManager = self::objectManagerFor($object);
        $objectManager->persist($object);
        $objectManager->flush();

        return $proxy ? self::proxy($object) : $object;
    }

    public static function proxy(object $object): Proxy
    {
        if ($object instanceof Proxy) {
            return $object;
        }

        return new Proxy($object);
    }

    public static function register(ManagerRegistry $managerRegistry): void
    {
        self::$managerRegistry = $managerRegistry;
    }

    /**
     * @param object|string $objectOrClass
     */
    public static function objectManagerFor($objectOrClass): ObjectManager
    {
        $class = \is_string($objectOrClass) ? $objectOrClass : \get_class($objectOrClass);

        if (!$objectManager = self::managerRegistry()->getManagerForClass($class)) {
            throw new \RuntimeException(\sprintf('No object manager registered for "%s".', $class));
        }

        return $objectManager;
    }

    private static function managerRegistry(): ManagerRegistry
    {
        if (null === self::$managerRegistry) {
            throw new \RuntimeException('ManagerRegistry not registered...'); // todo improve
        }

        return self::$managerRegistry;
    }
}