shootphp / shoot

src/Twig/Node/DisplayEndNode.php

<?php
declare(strict_types=1);

namespace Shoot\Shoot\Twig\Node;

use Shoot\Shoot\Extension;
use Shoot\Shoot\SuppressedException;
use Shoot\Shoot\View;
use Twig\Compiler;
use Twig\Node\ModuleNode;
use Twig\Node\Node;
use Twig\Source;

/**
 * This node is added to the bottom of the display method of a Twig template and is used by Shoot to wrap its contents
 * in a callback.
 *
 * @internal
 */
final class DisplayEndNode extends Node
{
    /** @var ModuleNode */
    private $module;

    /**
     * @param ModuleNode $module
     */
    public function __construct(ModuleNode $module)
    {
        parent::__construct();

        $this->module = $module;

        $this->setSourceContext($module->getSourceContext());

It seems like $module->getSourceContext() can also be of type null; however, parameter $source of Twig\Node\Node::setSourceContext() does only seem to accept Twig\Source, maybe add an additional type check?

    }

    /**
     * @param Compiler $compiler
     *
     * @return void
     */
    public function compile(Compiler $compiler): void
    {
        if ($this->module->hasAttribute('is_embedded')) {
            return;
        }

        /** @var Source $source */
        $source = $this->getSourceContext();
        $templateName = $source->getName();

        $extensionClass = Extension::class;
        $suppressedExceptionClass = SuppressedException::class;
        $viewClass = View::class;

        $compiler
            ->raw("\n")
            ->write("if (\$suppressedException instanceof {$suppressedExceptionClass}) {\n")
            ->indent()
            ->write("throw \$suppressedException;\n")
            ->outdent()
            ->write("}\n")
            ->outdent()
            ->write("};\n\n")
            ->write("\$this->env\n")
            ->indent()
            ->write("->getExtension({$extensionClass}::class)\n")
            ->write("->process(new {$viewClass}('{$templateName}', \$presentationModel, \$callback));\n")
            ->outdent();
    }
}