asserts control

<?php
namespace WOW\Service;

use Nette\Object;
use Nette\Application\UI\Presenter;
use Nette\DI\Container;
use Nette\Utils\Finder;
use Nette\Caching\Cache;
use Nette\Reflection\ClassType;

class ComponentRegister extends Object
{
    /** @var Container */
    private $context;

    private $types = array(
            'jpg' => '%wwwDir%/images',
            'png' => '%wwwDir%/images',
            'gif' => '%wwwDir%/images',
            'css' => '%wwwDir%/css',
            'js' => '%wwwDir%/js',
        );



    /**
     * @param Container $container
     */
    public function __construct(Container $context)
    {
        $this->context = $context;
        $this->context->addService('cache', new Cache($this->context->cacheStorage, 'WOW.ComponentInstaller'));
    }



    /**
     * @param Presenter $presenter
     */
    public function installPresenterComponents(Presenter $presenter)
    {
        if ($this->isInstalled($presenter)) {
            return;
        }

        $files = array();
        foreach ($this->getPresenterComponents($presenter) as $component) {
            $files = array_merge($files, $this->prepareAsserts($component));
        }

        $this->setInstalled($presenter, $files);
    }



    /**
     * @param Presenter $presenters
     * @return array of files
     */
    public function getPresenterAsserts(Presenter $presenter)
    {
        if (!$this->isInstalled($presenter)) {
            throw new Nette\InvalidArgumentException("Presenter " . get_class($presenter) . " is not installed");
        }

        return $this->context->cache->load($presenter->getAction(TRUE));
    }



    /**
     * @param Presenter $presenters
     * @return boolean
     */
    private function isInstalled(Presenter $presenter)
    {
        return $this->context->cache->load($presenter->getAction(TRUE)) !== NULL;
    }



    /**
     * @param Presenter $presenters
     * @param array $files
     */
    private function setInstalled(Presenter $presenter, array $files)
    {
        $this->context->cache->save(
                $presenter->getAction(TRUE), // vrací "nette" cestu k presenteru
                $files, // list of files to include, Control will resolve...
                array(
                        Cache::FILES => $files + array('presenter' => $presenter->getReflection()->getFileName())
                    )
            );
    }



    /**
     * @param object $component
     * @return array of sources
     */
    private function prepareAsserts($component)
    {
        $files = array();
        foreach ($this->types as $type => $publicPath) {
            foreach (Finder::findFiles('*.' . $type)->from($component->path) as $file) {
                $files[] = $source = $file->getRealPath();
                @copy($source, $this->context->expand($publicPath . '/' . basename($source)));
            }
        }

        return $files;
    }



    /**
     * @param Presenter $presenter
     * @return array components
     */
    private function getPresenterComponents(Presenter $presenter)
    {
        $components = array();
        foreach (ClassType::from($presenter)->getMethods() as $method) {
            if (substr($method->name, 0, 15) !== 'createComponent') {
                continue;
            }

            if ($method->hasAnnotation('component')) {
                if (!$method->getAnnotation('return')) {
                    throw new Nette\UnexpectedValueException("Set annotation @return for " . $method->getName() . " in " . get_class($presenter));
                }

                $name = lcFirst(substr($method->getName(), 15));
                $components[] = (object)array(
                        'name' => $name,
                        'factory' => $method,
                        'class' => $reflection->getAnnotation('return'),
                        'path' => ClassType::from($reflection->getAnnotation('return'))->getFileName();
                    );
            }
        }

        return $components;
    }

}



abstract class BasePresenter extends Nette\Application\UI\Presenter
{

    protected function startup()
    {
        parent::startup();
        $this->context->componentRegister->installPresenterComponents($this);
    }



    /**
     * @return AssertsControl
     */
    protected function createComponentAsserts()
    {
        return new AssertsControl($this->context->componentRegister);
    }

}