Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 1 | <?php |
| 2 | |
| 3 | /* |
| 4 | * This file is part of Twig. |
| 5 | * |
| 6 | * (c) Fabien Potencier |
| 7 | * |
| 8 | * For the full copyright and license information, please view the LICENSE |
| 9 | * file that was distributed with this source code. |
| 10 | */ |
| 11 | |
| 12 | namespace Twig\Profiler\NodeVisitor; |
| 13 | |
| 14 | use Twig\Environment; |
| 15 | use Twig\Node\BlockNode; |
| 16 | use Twig\Node\BodyNode; |
| 17 | use Twig\Node\MacroNode; |
| 18 | use Twig\Node\ModuleNode; |
| 19 | use Twig\Node\Node; |
| 20 | use Twig\NodeVisitor\NodeVisitorInterface; |
| 21 | use Twig\Profiler\Node\EnterProfileNode; |
| 22 | use Twig\Profiler\Node\LeaveProfileNode; |
| 23 | use Twig\Profiler\Profile; |
| 24 | |
| 25 | /** |
| 26 | * @author Fabien Potencier <fabien@symfony.com> |
| 27 | */ |
| 28 | final class ProfilerNodeVisitor implements NodeVisitorInterface |
| 29 | { |
| 30 | private $extensionName; |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 31 | private $varName; |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 32 | |
| 33 | public function __construct(string $extensionName) |
| 34 | { |
| 35 | $this->extensionName = $extensionName; |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 36 | $this->varName = sprintf('__internal_%s', hash(\PHP_VERSION_ID < 80100 ? 'sha256' : 'xxh128', $extensionName)); |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 37 | } |
| 38 | |
| 39 | public function enterNode(Node $node, Environment $env): Node |
| 40 | { |
| 41 | return $node; |
| 42 | } |
| 43 | |
| 44 | public function leaveNode(Node $node, Environment $env): ?Node |
| 45 | { |
| 46 | if ($node instanceof ModuleNode) { |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 47 | $node->setNode('display_start', new Node([new EnterProfileNode($this->extensionName, Profile::TEMPLATE, $node->getTemplateName(), $this->varName), $node->getNode('display_start')])); |
| 48 | $node->setNode('display_end', new Node([new LeaveProfileNode($this->varName), $node->getNode('display_end')])); |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 49 | } elseif ($node instanceof BlockNode) { |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 50 | $node->setNode('body', new BodyNode([ |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 51 | new EnterProfileNode($this->extensionName, Profile::BLOCK, $node->getAttribute('name'), $this->varName), |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 52 | $node->getNode('body'), |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 53 | new LeaveProfileNode($this->varName), |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 54 | ])); |
| 55 | } elseif ($node instanceof MacroNode) { |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 56 | $node->setNode('body', new BodyNode([ |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 57 | new EnterProfileNode($this->extensionName, Profile::MACRO, $node->getAttribute('name'), $this->varName), |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 58 | $node->getNode('body'), |
Matthias Andreas Benkard | 1ba5381 | 2022-12-27 17:32:58 +0100 | [diff] [blame^] | 59 | new LeaveProfileNode($this->varName), |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 60 | ])); |
| 61 | } |
| 62 | |
| 63 | return $node; |
| 64 | } |
| 65 | |
Matthias Andreas Benkard | 12a5735 | 2021-12-28 18:02:04 +0100 | [diff] [blame] | 66 | public function getPriority(): int |
| 67 | { |
| 68 | return 0; |
| 69 | } |
| 70 | } |