74 lines
2.0 KiB
PHP
74 lines
2.0 KiB
PHP
|
<?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.
|
||
|
*/
|
||
|
|
||
|
namespace Twig\TokenParser;
|
||
|
|
||
|
use Twig\Error\SyntaxError;
|
||
|
use Twig\Node\Node;
|
||
|
use Twig\Node\SetNode;
|
||
|
use Twig\Token;
|
||
|
|
||
|
/**
|
||
|
* Defines a variable.
|
||
|
*
|
||
|
* {% set foo = 'foo' %}
|
||
|
* {% set foo = [1, 2] %}
|
||
|
* {% set foo = {'foo': 'bar'} %}
|
||
|
* {% set foo = 'foo' ~ 'bar' %}
|
||
|
* {% set foo, bar = 'foo', 'bar' %}
|
||
|
* {% set foo %}Some content{% endset %}
|
||
|
*
|
||
|
* @internal
|
||
|
*/
|
||
|
final class SetTokenParser extends AbstractTokenParser
|
||
|
{
|
||
|
public function parse(Token $token): Node
|
||
|
{
|
||
|
$lineno = $token->getLine();
|
||
|
$stream = $this->parser->getStream();
|
||
|
$names = $this->parser->getExpressionParser()->parseAssignmentExpression();
|
||
|
|
||
|
$capture = false;
|
||
|
if ($stream->nextIf(/* Token::OPERATOR_TYPE */ 8, '=')) {
|
||
|
$values = $this->parser->getExpressionParser()->parseMultitargetExpression();
|
||
|
|
||
|
$stream->expect(/* Token::BLOCK_END_TYPE */ 3);
|
||
|
|
||
|
if (\count($names) !== \count($values)) {
|
||
|
throw new SyntaxError('When using set, you must have the same number of variables and assignments.', $stream->getCurrent()->getLine(), $stream->getSourceContext());
|
||
|
}
|
||
|
} else {
|
||
|
$capture = true;
|
||
|
|
||
|
if (\count($names) > 1) {
|
||
|
throw new SyntaxError('When using set with a block, you cannot have a multi-target.', $stream->getCurrent()->getLine(), $stream->getSourceContext());
|
||
|
}
|
||
|
|
||
|
$stream->expect(/* Token::BLOCK_END_TYPE */ 3);
|
||
|
|
||
|
$values = $this->parser->subparse([$this, 'decideBlockEnd'], true);
|
||
|
$stream->expect(/* Token::BLOCK_END_TYPE */ 3);
|
||
|
}
|
||
|
|
||
|
return new SetNode($capture, $names, $values, $lineno, $this->getTag());
|
||
|
}
|
||
|
|
||
|
public function decideBlockEnd(Token $token): bool
|
||
|
{
|
||
|
return $token->test('endset');
|
||
|
}
|
||
|
|
||
|
public function getTag(): string
|
||
|
{
|
||
|
return 'set';
|
||
|
}
|
||
|
}
|