<?php

/*

 * This file is part of the Symfony package.

 *

 * (c) Fabien Potencier <fabien@symfony.com>

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

namespace Symfony\Bridge\Doctrine\CacheWarmer;

use Symfony\Bridge\Doctrine\RegistryInterface;

use Symfony\Component\HttpKernel\CacheWarmer\CacheWarmerInterface;

/**

 * The proxy generator cache warmer generates all entity proxies.

 *

 * In the process of generating proxies the cache for all the metadata is primed also,

 * since this information is necessary to build the proxies in the first place.

 *

 * @author Benjamin Eberlei <kontakt@beberlei.de>

 */

class ProxyCacheWarmer implements CacheWarmerInterface

{

    private $registry;

    /**

     * Constructor.

     *

     * @param RegistryInterface $registry A RegistryInterface instance

     */

    public function __construct(RegistryInterface $registry)

    {

        $this->registry = $registry;

    }

    /**

     * This cache warmer is not optional, without proxies fatal error occurs!

     *

     * @return false

     */

    public function isOptional()

    {

        return false;

    }

    public function warmUp($cacheDir)

    {

        foreach ($this->registry->getEntityManagers() as $em) {

            // we need the directory no matter the proxy cache generation strategy

            if (!file_exists($proxyCacheDir = $em->getConfiguration()->getProxyDir())) {

                if (false === @mkdir($proxyCacheDir, 0777, true)) {

                    throw new \RuntimeException(sprintf('Unable to create the Doctrine Proxy directory "%s".', dirname($proxyCacheDir)));

                }

            } elseif (!is_writable($proxyCacheDir)) {

                throw new \RuntimeException(sprintf('The Doctrine Proxy directory "%s" is not writeable for the current system user.', $proxyCacheDir));

            }

            // if proxies are autogenerated we don't need to generate them in the cache warmer

            if ($em->getConfiguration()->getAutoGenerateProxyClasses()) {

                continue;

            }

            $classes = $em->getMetadataFactory()->getAllMetadata();

            $em->getProxyFactory()->generateProxyClasses($classes);

        }

    }

}