Project

General

Profile

1
<?php
2

    
3
/*
4
 * This file is part of Twig.
5
 *
6
 * (c) 2009 Fabien Potencier
7
 * (c) 2009 Armin Ronacher
8
 *
9
 * For the full copyright and license information, please view the LICENSE
10
 * file that was distributed with this source code.
11
 */
12

    
13
/**
14
 * Represents a module node.
15
 *
16
 * @package    twig
17
 * @author     Fabien Potencier <fabien@symfony.com>
18
 */
19
class Twig_Node_SandboxedModule extends Twig_Node_Module
20
{
21
    protected $usedFilters;
22
    protected $usedTags;
23
    protected $usedFunctions;
24

    
25
    public function __construct(Twig_Node_Module $node, array $usedFilters, array $usedTags, array $usedFunctions)
26
    {
27
        parent::__construct($node->getNode('body'), $node->getNode('parent'), $node->getNode('blocks'), $node->getNode('macros'), $node->getNode('traits'), $node->getAttribute('filename'), $node->getLine(), $node->getNodeTag());
28

    
29
        $this->usedFilters = $usedFilters;
30
        $this->usedTags = $usedTags;
31
        $this->usedFunctions = $usedFunctions;
32
    }
33

    
34
    protected function compileDisplayBody(Twig_Compiler $compiler)
35
    {
36
        $compiler->write("\$this->checkSecurity();\n");
37

    
38
        parent::compileDisplayBody($compiler);
39
    }
40

    
41
    protected function compileDisplayFooter(Twig_Compiler $compiler)
42
    {
43
        parent::compileDisplayFooter($compiler);
44

    
45
        $compiler
46
            ->write("protected function checkSecurity() {\n")
47
            ->indent()
48
            ->write("\$this->env->getExtension('sandbox')->checkSecurity(\n")
49
            ->indent()
50
            ->write(!$this->usedTags ? "array(),\n" : "array('".implode('\', \'', $this->usedTags)."'),\n")
51
            ->write(!$this->usedFilters ? "array(),\n" : "array('".implode('\', \'', $this->usedFilters)."'),\n")
52
            ->write(!$this->usedFunctions ? "array()\n" : "array('".implode('\', \'', $this->usedFunctions)."')\n")
53
            ->outdent()
54
            ->write(");\n")
55
            ->outdent()
56
            ->write("}\n\n")
57
        ;
58
    }
59
}
(17-17/22)