FileIterator.php

Go to the documentation of this file.

<?php
namespace Magento\Email\Model\Template\Config;

use Magento\Framework\Filesystem\DriverPool;
use Magento\Framework\Filesystem\File\ReadFactory;

class FileIterator extends \Magento\Framework\Config\FileIterator
{
    protected $_moduleDirResolver;

    public function __construct(
        ReadFactory $readFactory,
        array $paths,
        \Magento\Framework\Module\Dir\ReverseResolver $dirResolver
    ) {
        parent::__construct($readFactory, $paths);
        $this->_moduleDirResolver = $dirResolver;
    }

    public function current()
    {
        $path = $this->key();
        $moduleName = $this->_moduleDirResolver->getModuleName($path);
        if (!$moduleName) {
            throw new \UnexpectedValueException(
                sprintf("Unable to determine a module, file '%s' belongs to.", $this->key())
            );
        }

        $fileRead = $this->fileReadFactory->create($this->key(), DriverPool::FILE);
        $contents = $fileRead->readAll();
        return str_replace('<template ', '<template module="' . $moduleName . '" ', $contents);
    }
}