Twig/extra/cache-extra/Node/CacheNode.php

<?php

/*
 * This file is part of Twig.
 *
 * (c) Fabien Potencier
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Twig\Extra\Cache\Node;

use Twig\Compiler;
use Twig\Node\Expression\AbstractExpression;
use Twig\Node\Node;

class CacheNode extends Node
{
    public function __construct(AbstractExpression $key, ?AbstractExpression $ttl, ?AbstractExpression $tags, Node $body, int $lineno, string $tag)
    {
        $nodes = ['key' => $key, 'body' => $body];
        if (null !== $ttl) {
            $nodes['ttl'] = $ttl;
        }
        if (null !== $tags) {
            $nodes['tags'] = $tags;
        }

        parent::__construct($nodes, [], $lineno, $tag);
    }

    public function compile(Compiler $compiler): void
    {
        $compiler
            ->addDebugInfo($this)
            ->write('$cached = $this->env->getRuntime(\'Twig\Extra\Cache\CacheRuntime\')->getCache()->get(')
            ->subcompile($this->getNode('key'))
            ->raw(", function (\Symfony\Contracts\Cache\ItemInterface \$item) use (\$context) {\n")
            ->indent()
            ->write("try {\n")
            ->indent()
            ->write("\$item->tag('twig');\n")
        ;

        if ($this->hasNode('tags')) {
            $compiler
                ->write("\$item->tag(")
                ->subcompile($this->getNode('tags'))
                ->raw(");\n")
            ;
        }

        $compiler
            ->outdent()
            ->write("} catch (\Psr\Cache\CacheException \$e) {\n")
            ->indent()
            ->write("// cache doesn't support tags\n")
            ->outdent()
            ->write("}\n")
        ;

        if ($this->hasNode('ttl')) {
            $compiler
                ->write('$item->expiresAfter(')
                ->subcompile($this->getNode('ttl'))
                ->raw(");\n")
            ;
        }

        $compiler
            ->write("ob_start(function () { return ''; });\n")
            ->subcompile($this->getNode('body'))
            ->write("\n")
            ->write("return ob_get_clean();\n")
            ->outdent()
            ->write("});\n")
            ->write("echo '' === \$cached ? '' : new Markup(\$cached, \$this->env->getCharset());\n")
        ;
    }
}