GeneratedClassesAutoloader.php 1.98 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
<?php
/**
 *  Copyright © Magento, Inc. All rights reserved.
 *  See COPYING.txt for license details.
 */

namespace Magento\Framework\TestFramework\Unit\Autoloader;

use Magento\Framework\Code\Generator\Io;

/**
 * Autoloader that initiates auto-generation of requested classes
 */
class GeneratedClassesAutoloader
{
    /**
     * @var Io
     */
    private $generatorIo;

    /**
     * @var GeneratorInterface[]
     */
    private $generators;

    /**
     * @param GeneratorInterface[] $generators
     * @param Io $generatorIo
     */
    public function __construct(array $generators, Io $generatorIo)
    {
        foreach ($generators as $generator) {
            if (!($generator instanceof GeneratorInterface)) {
                throw new \InvalidArgumentException(
                    sprintf(
                        "Instance of '%s' is expected, instance of '%s' is received",
                        \Magento\Framework\TestFramework\Unit\Autoloader\GeneratorInterface::class,
                        get_class($generator)
                    )
                );
            }
        }
        $this->generators = $generators;
        $this->generatorIo = $generatorIo;
    }

    /**
     * Load class
     *
     * @param string $className
     * @return bool
     */
    public function load($className)
    {
        $classSourceFile = $this->generatorIo->generateResultFileName($className);
        if ($this->generatorIo->fileExists($classSourceFile)) {
            include $classSourceFile;
            return true;
        } else {
            foreach ($this->generators as $generator) {
                $content = $generator->generate($className);
                if ($content) {
                    $this->generatorIo->makeResultFileDirectory($className);
                    $this->generatorIo->writeResultFile($classSourceFile, $content);
                    include $classSourceFile;
                    return true;
                }
            }
        }

        return false;
    }
}