<?php
namespace Concrete\Core\Database;

use Concrete\Core\Application\Application;
use Concrete\Core\Cache\Adapter\DoctrineCacheDriver;
use Concrete\Core\Database\Connection\Connection;
use Concrete\Core\Package\Package;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\Tools\Setup;
use Config;
use Database;
use Events;

class DatabaseManagerORM
{

    /**
     * The application instance.
     *
     * @var Application
     */
    protected $app;

    /**
     * The active entity manager instances.
     * 
     * @var \Doctrine\ORM\EntityManager[]
     */
    protected $entityManagers = array();

    /**
     * Create a new database ORM manager instance.
     *
     * @param Application $app
     */
    public function __construct(Application $app)
    {
        $this->app = $app;
    }

    /**
     * Return all of the available entity managers.
     *
     * @return \Doctrine\ORM\EntityManager[]
     */
    public function getEntityManagers()
    {
        return $this->entityManagers;
    }

    /**
     * Gets a context specific entity manager. Allows easier management of
     * entities where different settings than the core settings are needed
     * for the EntityManager object.
     * 
     * @param  mixed $context
     * @param  string $connectionName
     * @return \Doctrine\ORM\EntityManager
     */
    public function entityManager($context = null, $connectionName = null)
    {
        if ($connectionName === null) {
            $connectionName = Database::getDefaultConnection();
        }
        $name = $connectionName . '_';
        if ($context instanceof Package) {
            $name .= 'pkg_' . $context->getPackageHandle();
        } elseif ($context === 'core') {
            $name .= 'core';
        } else {
            $name .= 'app';
        }
        if (!isset($this->entityManagers[$name])) {
            $this->entityManagers[$name] = static::makeEntityManager(Database::connection($connectionName), $context);
        }
        return $this->entityManagers[$name];
    }

    /**
     * Makes a new entity manager instance for the given context
     * (e.g. a package) or if no context object is given, for the application
     * context. The options for the context are:
     * - A package object, results in a package specific entity manager
     * - The string 'core', results in a core specific entity manager
     * - Null or omitted context, results in an application specific entity
     *   manager
     * 
     * @param  Connection $connection
     * @param  mixed      $context
     * @return \Doctrine\ORM\EntityManager
     */
    public static function makeEntityManager(Connection $connection, $context = null)
    {
        $config = Setup::createConfiguration(
            Config::get('concrete.cache.doctrine_dev_mode'),
            Config::get('database.proxy_classes'),
            new DoctrineCacheDriver('cache/expensive')
        );

        $path = DIR_APPLICATION . '/' . DIRNAME_CLASSES;
        if ($context instanceof Package) {
            $path = $context->getPackageEntitiesPath();
        } elseif ($context === 'core') {
            $path = DIR_BASE_CORE . '/' . DIRNAME_CLASSES;
        } elseif (is_object($context) && method_exists($context, 'getEntitiesPath')) {
            $path = $context->getEntitiesPath();
        }

        $driverImpl = $config->newDefaultAnnotationDriver($path);
        $driverImpl->addExcludePaths(Config::get('database.proxy_exclusions', array()));
        $config->setMetadataDriverImpl($driverImpl);

        $event = new \Symfony\Component\EventDispatcher\GenericEvent();
        $event->setArgument('connection', $connection);
        $event->setArgument('context', $context);
        $event->setArgument('configuration', $config);
        Events::dispatch('on_entity_manager_configure', $event);
        $config = $event->getArgument('configuration');

        return EntityManager::create($connection, $config);
    }

}