<?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\CacheWarmer;

use Symfony\Component\HttpKernel\KernelInterface;

use Symfony\Component\Finder\Finder;

use Symfony\Bundle\FrameworkBundle\Templating\TemplateNameParser;

use Symfony\Component\HttpKernel\Bundle\BundleInterface;

/**

 * Finds all the templates.

 *

 * @author Victor Berchet <victor@suumit.com>

 */

class TemplateFinder implements TemplateFinderInterface

{

    private $kernel;

    private $parser;

    private $rootDir;

    private $templates;

    /**

     * Constructor.

     *

     * @param KernelInterface      $kernel  A KernelInterface instance

     * @param TemplateNameParser   $parser  A TemplateNameParser instance

     * @param string               $rootDir The directory where global templates can be stored

     */

    public function __construct(KernelInterface $kernel, TemplateNameParser $parser, $rootDir)

    {

        $this->kernel = $kernel;

        $this->parser = $parser;

        $this->rootDir = $rootDir;

    }

    /**

     * Find all the templates in the bundle and in the kernel Resources folder.

     *

     * @return array An array of templates of type TemplateReferenceInterface

     */

    public function findAllTemplates()

    {

        if (null !== $this->templates) {

            return $this->templates;

        }

        $templates = array();

        foreach ($this->kernel->getBundles() as $name => $bundle) {

            $templates = array_merge($templates, $this->findTemplatesInBundle($bundle));

        }

        $templates = array_merge($templates, $this->findTemplatesInFolder($this->rootDir.'/views'));

        return $this->templates = $templates;

    }

    /**

     * Find templates in the given directory.

     *

     * @param string $dir The folder where to look for templates

     *

     * @return array An array of templates of type TemplateReferenceInterface

     */

    private function findTemplatesInFolder($dir)

    {

        $templates = array();

        if (is_dir($dir)) {

            $finder = new Finder();

            foreach ($finder->files()->followLinks()->in($dir) as $file) {

                $template = $this->parser->parseFromFilename($file->getRelativePathname());

                if (false !== $template) {

                    $templates[] = $template;

                }

            }

        }

        return $templates;

    }

    /**

     * Find templates in the given bundle.

     *

     * @param BundleInterface $bundle The bundle where to look for templates

     *

     * @return array An array of templates of type TemplateReferenceInterface

     */

    private function findTemplatesInBundle(BundleInterface $bundle)

    {

        $templates = $this->findTemplatesInFolder($bundle->getPath().'/Resources/views');

        $name = $bundle->getName();

        foreach ($templates as $i => $template) {

            $templates[$i] = $template->set('bundle', $name);

        }

        return $templates;

    }

}