<?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.
 */

/**
 * Twig_SupTwgSgg_NodeVisitor_Sandbox implements sandboxing.
 *
 * @final
 *
 * @author Fabien Potencier <fabien@symfony.com>
 */
class Twig_SupTwgSgg_NodeVisitor_Sandbox extends Twig_SupTwgSgg_BaseNodeVisitor
{
  protected $inAModule = false;
  protected $tags;
  protected $filters;
  protected $functions;

  protected function doEnterNode(Twig_SupTwgSgg_Node $node, Twig_SupTwgSgg_Environment $env)
  {
    if ($node instanceof Twig_SupTwgSgg_Node_Module) {
      $this->inAModule = true;
      $this->tags = [];
      $this->filters = [];
      $this->functions = [];

      return $node;
    } elseif ($this->inAModule) {
      // look for tags
      if ($node->getNodeTag() && !isset($this->tags[$node->getNodeTag()])) {
        $this->tags[$node->getNodeTag()] = $node;
      }

      // look for filters
      if ($node instanceof Twig_SupTwgSgg_Node_Expression_Filter && !isset($this->filters[$node->getNode('filter')->getAttribute('value')])) {
        $this->filters[$node->getNode('filter')->getAttribute('value')] = $node;
      }

      // look for functions
      if ($node instanceof Twig_SupTwgSgg_Node_Expression_Function && !isset($this->functions[$node->getAttribute('name')])) {
        $this->functions[$node->getAttribute('name')] = $node;
      }

      // wrap print to check __toString() calls
      if ($node instanceof Twig_SupTwgSgg_Node_Print) {
        return new Twig_SupTwgSgg_Node_SandboxedPrint($node->getNode('expr'), $node->getTemplateLine(), $node->getNodeTag());
      }
    }

    return $node;
  }

  protected function doLeaveNode(Twig_SupTwgSgg_Node $node, Twig_SupTwgSgg_Environment $env)
  {
    if ($node instanceof Twig_SupTwgSgg_Node_Module) {
      $this->inAModule = false;

      $node->setNode('display_start', new Twig_SupTwgSgg_Node([new Twig_SupTwgSgg_Node_CheckSecurity($this->filters, $this->tags, $this->functions), $node->getNode('display_start')]));
    }

    return $node;
  }

  public function getPriority()
  {
    return 0;
  }
}
