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
12namespace Twig\TokenParser;
13
14use Twig\Error\SyntaxError;
15use Twig\Node\SetNode;
16use Twig\Token;
17
18/**
19 * Defines a variable.
20 *
21 *  {% set foo = 'foo' %}
22 *  {% set foo = [1, 2] %}
23 *  {% set foo = {'foo': 'bar'} %}
24 *  {% set foo = 'foo' ~ 'bar' %}
25 *  {% set foo, bar = 'foo', 'bar' %}
26 *  {% set foo %}Some content{% endset %}
27 */
28final class SetTokenParser extends AbstractTokenParser
29{
30    public function parse(Token $token)
31    {
32        $lineno = $token->getLine();
33        $stream = $this->parser->getStream();
34        $names = $this->parser->getExpressionParser()->parseAssignmentExpression();
35
36        $capture = false;
37        if ($stream->nextIf(/* Token::OPERATOR_TYPE */ 8, '=')) {
38            $values = $this->parser->getExpressionParser()->parseMultitargetExpression();
39
40            $stream->expect(/* Token::BLOCK_END_TYPE */ 3);
41
42            if (\count($names) !== \count($values)) {
43                throw new SyntaxError('When using set, you must have the same number of variables and assignments.', $stream->getCurrent()->getLine(), $stream->getSourceContext());
44            }
45        } else {
46            $capture = true;
47
48            if (\count($names) > 1) {
49                throw new SyntaxError('When using set with a block, you cannot have a multi-target.', $stream->getCurrent()->getLine(), $stream->getSourceContext());
50            }
51
52            $stream->expect(/* Token::BLOCK_END_TYPE */ 3);
53
54            $values = $this->parser->subparse([$this, 'decideBlockEnd'], true);
55            $stream->expect(/* Token::BLOCK_END_TYPE */ 3);
56        }
57
58        return new SetNode($capture, $names, $values, $lineno, $this->getTag());
59    }
60
61    public function decideBlockEnd(Token $token)
62    {
63        return $token->test('endset');
64    }
65
66    public function getTag()
67    {
68        return 'set';
69    }
70}
71
72class_alias('Twig\TokenParser\SetTokenParser', 'Twig_TokenParser_Set');
73