GravityMedia\Commander\Provider\EntityManagerProvider

Complexity

Total Complexity

7

Size/Duplication

Total Lines	110
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Test Coverage

Coverage

0%

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	6	1
A	getEntityManagerConfig()	0	23	3
B	getEntityManager()	0	24	3

<?php
/**
 * This file is part of the Commander project.
 *
 * @author Daniel Schröder <daniel.schroeder@gravitymedia.de>
 */

namespace GravityMedia\Commander\Provider;

use Doctrine\Common\Cache\Cache;
use Doctrine\Common\EventManager;
use Doctrine\Common\Persistence\Mapping\Driver\MappingDriver;
use Doctrine\ORM\Configuration as EntityManagerConfig;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Mapping\Driver\AnnotationDriver;
use Gedmo\Timestampable\TimestampableListener;
use GravityMedia\Commander\Commander;
use GravityMedia\Commander\Config;

/**
 * Entity manager provider class.
 *
 * @package GravityMedia\Commander\Provider
 */
class EntityManagerProvider
{
    /**
     * The config.
     *
     * @var Config
     */
    protected $config;

    /**
     * The cache.
     *
     * @var Cache
     */
    protected $cache;

    /**
     * The mapping driver.
     *
     * @var MappingDriver
     */
    protected $mappingDriver;

    /**
     * The entity manager config.
     *
     * @var EntityManagerConfig
     */
    protected $entityManagerConfig;

    /**
     * The entity manager.
     *
     * @var EntityManagerInterface
     */
    protected $entityManager;

    /**
     * Create entity manager provider object.
     *
     * @param Config        $config
     * @param Cache         $cache
     * @param MappingDriver $mappingDriver
     */
    public function __construct(Config $config, Cache $cache, MappingDriver $mappingDriver)
    {
        $this->config = $config;
        $this->cache = $cache;
        $this->mappingDriver = $mappingDriver;
    }

    /**
     * Get entity manager config.
     *
     * @return EntityManagerConfig
     */
    public function getEntityManagerConfig()
    {
        if (null === $this->entityManagerConfig) {
            $cacheDirectory = $this->config->getCacheDirectory();
            if (null === $cacheDirectory) {
                $cacheDirectory = sys_get_temp_dir();
            }

            $config = new EntityManagerConfig();
            $config->setAutoGenerateProxyClasses(true);
            $config->setProxyDir(rtrim($cacheDirectory, '/') . '/proxy');
            $config->setProxyNamespace(Commander::ENTITY_NAMESPACE);
            $config->setMetadataDriverImpl($this->mappingDriver);
            $config->setMetadataCacheImpl($this->cache);
            $config->setQueryCacheImpl($this->cache);
            $config->setResultCacheImpl($this->cache);
            $config->setHydrationCacheImpl($this->cache);

            $this->entityManagerConfig = $config;
        }

        return $this->entityManagerConfig;
    }

    /**
     * Get entity manager.
     *
     * @return EntityManagerInterface
     */
    public function getEntityManager()
    {
        if (null === $this->entityManager) {
            $connection = [
                'driver' => 'pdo_sqlite',
                'path' => $this->config->getDatabaseFilePath()
            ];

            $config = $this->getEntityManagerConfig();
            $metadataDriver = $config->getMetadataDriverImpl();

            $timestampableListener = new TimestampableListener();
            if ($metadataDriver instanceof AnnotationDriver) {
                $timestampableListener->setAnnotationReader($metadataDriver->getReader());
            }

            $eventManager = new EventManager();
            $eventManager->addEventSubscriber($timestampableListener);

            $this->entityManager = EntityManager::create($connection, $config, $eventManager);
        }

        return $this->entityManager;
    }
}