<?php
/**
 * This file is part of the Nella Framework (http://nellafw.org).
 *
 * Copyright (c) 2006, 2012 Patrik Votoček (http://patrik.votocek.cz)
 *
 * For the full copyright and license information, please view the file LICENSE.txt that was distributed with this source code.
 */

namespace Nella\Config\Extensions;

/**
 * Nella Framework doctrine extension
 *
 * @author	Patrik Votoček
 */
class DoctrineExtension extends \Nella\NetteAddons\Doctrine\Config\Extension
{
	/** @var array */
	public $emDefaults = array(
		'repositoryClass' => 'Nella\Doctrine\Repository',
	);

	/**
	 * Processes configuration data
	 *
	 * @throws \Nette\InvalidStateException
	 */
	public function loadConfiguration()
	{
		$this->entityManagerDefaults['entityDirs'][] = realpath(__DIR__ . "/../..");
		parent::loadConfiguration();

		$builder = $this->getContainerBuilder();

		// Ignore Testing dir for loading entityes
		if ($builder->hasDefinition($this->prefix('metadataDriver'))) {
			$builder->getDefinition($this->prefix('metadataDriver'))
				->addSetup('addIgnoredDir', array(__DIR__ . "/../../Testing"));
		}
	}

	/**
	 * @param string
	 * @param array
	 */
	protected function processEntityManager($name, array $config)
	{
		$cfg = $config + $this->emDefaults;
		parent::processEntityManager($name, $cfg);
		$builder = $this->getContainerBuilder();

		if ($builder->hasDefinition($this->configurationsPrefix($name.'AnnotationReader'))) {
			$builder->addDefinition($this->configurationsPrefix($name.'DiscriminatorMapDiscovery'))
				->setClass('Nella\Doctrine\Listeners\DiscriminatorMapDiscovery', array(
					$builder->getDefinition($this->configurationsPrefix($name.'AnnotationReader'))
				))
				->addTag('doctrineListener')
				->setAutowired(FALSE);
		}

		if ($builder->hasDefinition($this->configurationsPrefix($name))) {
			$builder->getDefinition($this->configurationsPrefix($name))
				->addSetup('setDefaultRepositoryClassName', array($cfg['repositoryClass']));
		}
	}

	/**
	 * @param \Doctrine\Common\Cache\Cache
	 * @param bool
	 * @return \Doctrine\Common\Annotations\CachedReader
	 */
	public static function createAnnotationReader(\Doctrine\Common\Cache\Cache $cache, $useSimple = FALSE)
	{
		require_once __DIR__ ."/../../Doctrine/Mapping/DiscriminatorEntry.php";
		return parent::createAnnotationReader($cache, $useSimple);
	}
}