51 lines
1.3 KiB
PHP
51 lines
1.3 KiB
PHP
<?php
|
|
|
|
/**
|
|
* This file is part of the rybakit/twig-deferred-extension package.
|
|
*
|
|
* (c) Eugene Leonovich <gen.work@gmail.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Twig\DeferredExtension;
|
|
|
|
use Twig\Environment;
|
|
use Twig\Node\ModuleNode;
|
|
use Twig\Node\Node;
|
|
use Twig\NodeVisitor\NodeVisitorInterface;
|
|
|
|
final class DeferredNodeVisitor implements NodeVisitorInterface
|
|
{
|
|
private $hasDeferred = false;
|
|
|
|
public function enterNode(Node $node, Environment $env) : Node
|
|
{
|
|
if (!$this->hasDeferred && $node instanceof DeferredBlockNode) {
|
|
$this->hasDeferred = true;
|
|
}
|
|
|
|
return $node;
|
|
}
|
|
|
|
public function leaveNode(Node $node, Environment $env) : ?Node
|
|
{
|
|
if ($this->hasDeferred && $node instanceof ModuleNode) {
|
|
$node->getNode('constructor_end')->setNode('deferred_initialize', new DeferredInitializeNode());
|
|
$node->getNode('display_end')->setNode('deferred_resolve', new DeferredResolveNode());
|
|
$node->getNode('class_end')->setNode('deferred_declare', new DeferredDeclareNode());
|
|
$this->hasDeferred = false;
|
|
}
|
|
|
|
return $node;
|
|
}
|
|
|
|
public function getPriority() : int
|
|
{
|
|
return 0;
|
|
}
|
|
}
|