You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
171 lines
5.1 KiB
171 lines
5.1 KiB
<?php |
|
|
|
/* |
|
* This file is part of the Symfony package. |
|
* |
|
* (c) Fabien Potencier <fabien@symfony.com> |
|
* |
|
* For the full copyright and license information, please view the LICENSE |
|
* file that was distributed with this source code. |
|
*/ |
|
|
|
namespace Symfony\Component\CssSelector\XPath\Extension; |
|
|
|
use Symfony\Component\CssSelector\Exception\ExpressionErrorException; |
|
use Symfony\Component\CssSelector\Exception\SyntaxErrorException; |
|
use Symfony\Component\CssSelector\Node\FunctionNode; |
|
use Symfony\Component\CssSelector\Parser\Parser; |
|
use Symfony\Component\CssSelector\XPath\Translator; |
|
use Symfony\Component\CssSelector\XPath\XPathExpr; |
|
|
|
/** |
|
* XPath expression translator function extension. |
|
* |
|
* This component is a port of the Python cssselect library, |
|
* which is copyright Ian Bicking, @see https://github.com/SimonSapin/cssselect. |
|
* |
|
* @author Jean-François Simon <jeanfrancois.simon@sensiolabs.com> |
|
* |
|
* @internal |
|
*/ |
|
class FunctionExtension extends AbstractExtension |
|
{ |
|
/** |
|
* {@inheritdoc} |
|
*/ |
|
public function getFunctionTranslators(): array |
|
{ |
|
return [ |
|
'nth-child' => [$this, 'translateNthChild'], |
|
'nth-last-child' => [$this, 'translateNthLastChild'], |
|
'nth-of-type' => [$this, 'translateNthOfType'], |
|
'nth-last-of-type' => [$this, 'translateNthLastOfType'], |
|
'contains' => [$this, 'translateContains'], |
|
'lang' => [$this, 'translateLang'], |
|
]; |
|
} |
|
|
|
/** |
|
* @throws ExpressionErrorException |
|
*/ |
|
public function translateNthChild(XPathExpr $xpath, FunctionNode $function, bool $last = false, bool $addNameTest = true): XPathExpr |
|
{ |
|
try { |
|
[$a, $b] = Parser::parseSeries($function->getArguments()); |
|
} catch (SyntaxErrorException $e) { |
|
throw new ExpressionErrorException(sprintf('Invalid series: "%s".', implode('", "', $function->getArguments())), 0, $e); |
|
} |
|
|
|
$xpath->addStarPrefix(); |
|
if ($addNameTest) { |
|
$xpath->addNameTest(); |
|
} |
|
|
|
if (0 === $a) { |
|
return $xpath->addCondition('position() = '.($last ? 'last() - '.($b - 1) : $b)); |
|
} |
|
|
|
if ($a < 0) { |
|
if ($b < 1) { |
|
return $xpath->addCondition('false()'); |
|
} |
|
|
|
$sign = '<='; |
|
} else { |
|
$sign = '>='; |
|
} |
|
|
|
$expr = 'position()'; |
|
|
|
if ($last) { |
|
$expr = 'last() - '.$expr; |
|
--$b; |
|
} |
|
|
|
if (0 !== $b) { |
|
$expr .= ' - '.$b; |
|
} |
|
|
|
$conditions = [sprintf('%s %s 0', $expr, $sign)]; |
|
|
|
if (1 !== $a && -1 !== $a) { |
|
$conditions[] = sprintf('(%s) mod %d = 0', $expr, $a); |
|
} |
|
|
|
return $xpath->addCondition(implode(' and ', $conditions)); |
|
|
|
// todo: handle an+b, odd, even |
|
// an+b means every-a, plus b, e.g., 2n+1 means odd |
|
// 0n+b means b |
|
// n+0 means a=1, i.e., all elements |
|
// an means every a elements, i.e., 2n means even |
|
// -n means -1n |
|
// -1n+6 means elements 6 and previous |
|
} |
|
|
|
public function translateNthLastChild(XPathExpr $xpath, FunctionNode $function): XPathExpr |
|
{ |
|
return $this->translateNthChild($xpath, $function, true); |
|
} |
|
|
|
public function translateNthOfType(XPathExpr $xpath, FunctionNode $function): XPathExpr |
|
{ |
|
return $this->translateNthChild($xpath, $function, false, false); |
|
} |
|
|
|
/** |
|
* @throws ExpressionErrorException |
|
*/ |
|
public function translateNthLastOfType(XPathExpr $xpath, FunctionNode $function): XPathExpr |
|
{ |
|
if ('*' === $xpath->getElement()) { |
|
throw new ExpressionErrorException('"*:nth-of-type()" is not implemented.'); |
|
} |
|
|
|
return $this->translateNthChild($xpath, $function, true, false); |
|
} |
|
|
|
/** |
|
* @throws ExpressionErrorException |
|
*/ |
|
public function translateContains(XPathExpr $xpath, FunctionNode $function): XPathExpr |
|
{ |
|
$arguments = $function->getArguments(); |
|
foreach ($arguments as $token) { |
|
if (!($token->isString() || $token->isIdentifier())) { |
|
throw new ExpressionErrorException('Expected a single string or identifier for :contains(), got '.implode(', ', $arguments)); |
|
} |
|
} |
|
|
|
return $xpath->addCondition(sprintf( |
|
'contains(string(.), %s)', |
|
Translator::getXpathLiteral($arguments[0]->getValue()) |
|
)); |
|
} |
|
|
|
/** |
|
* @throws ExpressionErrorException |
|
*/ |
|
public function translateLang(XPathExpr $xpath, FunctionNode $function): XPathExpr |
|
{ |
|
$arguments = $function->getArguments(); |
|
foreach ($arguments as $token) { |
|
if (!($token->isString() || $token->isIdentifier())) { |
|
throw new ExpressionErrorException('Expected a single string or identifier for :lang(), got '.implode(', ', $arguments)); |
|
} |
|
} |
|
|
|
return $xpath->addCondition(sprintf( |
|
'lang(%s)', |
|
Translator::getXpathLiteral($arguments[0]->getValue()) |
|
)); |
|
} |
|
|
|
/** |
|
* {@inheritdoc} |
|
*/ |
|
public function getName(): string |
|
{ |
|
return 'function'; |
|
} |
|
}
|
|
|