Shoot\Shoot\Twig\PatchingCompiler

Complexity

Total Complexity

4

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Test Coverage

Coverage

90%

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	raw()	0	3	1
A	write()	0	5	1
A	patch()	0	10	2

<?php
declare(strict_types=1);

namespace Shoot\Shoot\Twig;

use Twig\Compiler;

/**
 * This compiler patches a few crucial lines in some core Twig classes. It allows Shoot to be used with extend, embed,
 * and blocks. Out of all hacks to make that work, this one seemed to be the most straightforward.
 *
 * @internal
 */
final class PatchingCompiler extends Compiler
{
    /**
     * @param mixed $string
     *
     * @return Compiler
     */
    public function raw($string)
    {
        return parent::raw($this->patch($string));
    }

    /**
     * @param mixed[] ...$strings
     *
     * @return Compiler
     */
    public function write(...$strings)
    {
        $strings = array_map([$this, 'patch'], $strings);

        return parent::write(...$strings);
    }

    /**
     * @param mixed $string
     *
     * @return mixed
     */
    private function patch($string)
    {
        static $patterns = [
            '/(->display\()\$context(, array_merge\(\$this->blocks, \$blocks\)\);\n)/',
            '/(\$this->displayBlock\(\'[^\']+\', )\$context(, \$blocks\);\n)/',
        ];

        return !is_string($string)
            ? $string
            : preg_replace($patterns, '$1array_merge($context, $originalContext ?? [])$2', $string);
    }
}