symfony/src/Symfony/Bundle/FrameworkBundle/Command/CacheWarmupCommand.php

<?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\Bundle\FrameworkBundle\Command;

use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Symfony\Component\Finder\Finder;

/**
 * Warmup the cache.
 *
 * @author Fabien Potencier <fabien@symfony.com>
 * @author Francis Besset <francis.besset@gmail.com>
 */
class CacheWarmupCommand extends Command
{
    protected $warmupDir;
    protected $kernelTmp;

    /**
     * @see Command
     */
    protected function configure()
    {
        $this
            ->setName('cache:warmup')
            ->setDescription('Warms up an empty cache')
            ->setDefinition(array(
                new InputOption('warmup-dir', '', InputOption::VALUE_REQUIRED, 'Warms up the cache in the specified directory')
            ))
            ->setHelp(<<<EOF
The <info>cache:warmup</info> command warms up the cache.

Before running this command, the cache must be empty.
EOF
            )
        ;
    }

    /**
     * {@inheritdoc}
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $output->writeln('Warming up the cache');

        if ($input->getOption('warmup-dir')) {
            $this->setWarmupDir($input->getOption('warmup-dir'));
        }

        if (!$this->warmupDir) {
            $this->warmUp($this->container);
        } else {
            $class = get_class($this->container->get('kernel'));
            $this->kernelTmp = new $class(
                $this->container->getParameter('kernel.environment'),
                $this->container->getParameter('kernel.debug'),
                $this->warmupDir
            );

            $this->container->get('filesystem')->remove($this->kernelTmp->getCacheDir());

            $this->kernelTmp->boot();
            unlink($this->kernelTmp->getCacheDir().DIRECTORY_SEPARATOR.$this->kernelTmp->getContainerClass().'.php');

            $this->warmUp($this->kernelTmp->getContainer());
        }
    }

    protected function warmUp(ContainerInterface $container)
    {
        $warmer = $container->get('cache_warmer');
        $warmer->enableOptionalWarmers();
        $warmer->warmUp($container->getParameter('kernel.cache_dir'));
    }

    protected function setWarmupDir($dir)
    {
        $this->warmupDir = $dir;
    }
}