]>
Commit | Line | Data |
---|---|---|
a4565e88 NL |
1 | <?php |
2 | ||
3 | /* | |
4 | * This file is part of Twig. | |
5 | * | |
6 | * (c) 2010 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 | /** | |
13 | * Represents a trans node. | |
14 | * | |
15 | * @package twig | |
16 | * @author Fabien Potencier <fabien.potencier@symfony-project.com> | |
17 | */ | |
18 | class Twig_Extensions_Node_Trans extends Twig_Node | |
19 | { | |
20 | public function __construct(Twig_NodeInterface $body, Twig_NodeInterface $plural = null, Twig_Node_Expression $count = null, $lineno, $tag = null) | |
21 | { | |
22 | parent::__construct(array('count' => $count, 'body' => $body, 'plural' => $plural), array(), $lineno, $tag); | |
23 | } | |
24 | ||
25 | /** | |
26 | * Compiles the node to PHP. | |
27 | * | |
28 | * @param Twig_Compiler A Twig_Compiler instance | |
29 | */ | |
30 | public function compile(Twig_Compiler $compiler) | |
31 | { | |
32 | $compiler->addDebugInfo($this); | |
33 | ||
34 | list($msg, $vars) = $this->compileString($this->getNode('body')); | |
35 | ||
36 | if (null !== $this->getNode('plural')) { | |
37 | list($msg1, $vars1) = $this->compileString($this->getNode('plural')); | |
38 | ||
39 | $vars = array_merge($vars, $vars1); | |
40 | } | |
41 | ||
42 | $function = null === $this->getNode('plural') ? 'gettext' : 'ngettext'; | |
43 | ||
44 | if ($vars) { | |
45 | $compiler | |
46 | ->write('echo strtr('.$function.'(') | |
47 | ->subcompile($msg) | |
48 | ; | |
49 | ||
50 | if (null !== $this->getNode('plural')) { | |
51 | $compiler | |
52 | ->raw(', ') | |
53 | ->subcompile($msg1) | |
54 | ->raw(', abs(') | |
55 | ->subcompile($this->getNode('count')) | |
56 | ->raw(')') | |
57 | ; | |
58 | } | |
59 | ||
60 | $compiler->raw('), array('); | |
61 | ||
62 | foreach ($vars as $var) { | |
63 | if ('count' === $var->getAttribute('name')) { | |
64 | $compiler | |
65 | ->string('%count%') | |
66 | ->raw(' => abs(') | |
67 | ->subcompile($this->getNode('count')) | |
68 | ->raw('), ') | |
69 | ; | |
70 | } else { | |
71 | $compiler | |
72 | ->string('%'.$var->getAttribute('name').'%') | |
73 | ->raw(' => ') | |
74 | ->subcompile($var) | |
75 | ->raw(', ') | |
76 | ; | |
77 | } | |
78 | } | |
79 | ||
80 | $compiler->raw("));\n"); | |
81 | } else { | |
82 | $compiler | |
83 | ->write('echo '.$function.'(') | |
84 | ->subcompile($msg) | |
85 | ; | |
86 | ||
87 | if (null !== $this->getNode('plural')) { | |
88 | $compiler | |
89 | ->raw(', ') | |
90 | ->subcompile($msg1) | |
91 | ->raw(', abs(') | |
92 | ->subcompile($this->getNode('count')) | |
93 | ->raw(')') | |
94 | ; | |
95 | } | |
96 | ||
97 | $compiler->raw(");\n"); | |
98 | } | |
99 | } | |
100 | ||
101 | protected function compileString(Twig_NodeInterface $body) | |
102 | { | |
103 | if ($body instanceof Twig_Node_Expression_Name || $body instanceof Twig_Node_Expression_Constant || $body instanceof Twig_Node_Expression_TempName) { | |
104 | return array($body, array()); | |
105 | } | |
106 | ||
107 | $vars = array(); | |
108 | if (count($body)) { | |
109 | $msg = ''; | |
110 | ||
111 | foreach ($body as $node) { | |
112 | if (get_class($node) === 'Twig_Node' && $node->getNode(0) instanceof Twig_Node_SetTemp) { | |
113 | $node = $node->getNode(1); | |
114 | } | |
115 | ||
116 | if ($node instanceof Twig_Node_Print) { | |
117 | $n = $node->getNode('expr'); | |
118 | while ($n instanceof Twig_Node_Expression_Filter) { | |
119 | $n = $n->getNode('node'); | |
120 | } | |
121 | $msg .= sprintf('%%%s%%', $n->getAttribute('name')); | |
122 | $vars[] = new Twig_Node_Expression_Name($n->getAttribute('name'), $n->getLine()); | |
123 | } else { | |
124 | $msg .= $node->getAttribute('data'); | |
125 | } | |
126 | } | |
127 | } else { | |
128 | $msg = $body->getAttribute('data'); | |
129 | } | |
130 | ||
131 | return array(new Twig_Node(array(new Twig_Node_Expression_Constant(trim($msg), $body->getLine()))), $vars); | |
132 | } | |
133 | } |