Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of the Symfony package.
Chris@0:  *
Chris@0:  * (c) Fabien Potencier <fabien@symfony.com>
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Symfony\Component\CssSelector\XPath\Extension;
Chris@0: 
Chris@0: use Symfony\Component\CssSelector\XPath\Translator;
Chris@0: use Symfony\Component\CssSelector\XPath\XPathExpr;
Chris@0: 
Chris@0: /**
Chris@0:  * XPath expression translator attribute extension.
Chris@0:  *
Chris@0:  * This component is a port of the Python cssselect library,
Chris@0:  * which is copyright Ian Bicking, @see https://github.com/SimonSapin/cssselect.
Chris@0:  *
Chris@0:  * @author Jean-François Simon <jeanfrancois.simon@sensiolabs.com>
Chris@0:  *
Chris@0:  * @internal
Chris@0:  */
Chris@0: class AttributeMatchingExtension extends AbstractExtension
Chris@0: {
Chris@0:     /**
Chris@0:      * {@inheritdoc}
Chris@0:      */
Chris@0:     public function getAttributeMatchingTranslators()
Chris@0:     {
Chris@0:         return array(
Chris@0:             'exists' => array($this, 'translateExists'),
Chris@0:             '=' => array($this, 'translateEquals'),
Chris@0:             '~=' => array($this, 'translateIncludes'),
Chris@0:             '|=' => array($this, 'translateDashMatch'),
Chris@0:             '^=' => array($this, 'translatePrefixMatch'),
Chris@0:             '$=' => array($this, 'translateSuffixMatch'),
Chris@0:             '*=' => array($this, 'translateSubstringMatch'),
Chris@0:             '!=' => array($this, 'translateDifferent'),
Chris@0:         );
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateExists(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition($attribute);
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateEquals(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition(sprintf('%s = %s', $attribute, Translator::getXpathLiteral($value)));
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateIncludes(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition($value ? sprintf(
Chris@0:             '%1$s and contains(concat(\' \', normalize-space(%1$s), \' \'), %2$s)',
Chris@0:             $attribute,
Chris@0:             Translator::getXpathLiteral(' '.$value.' ')
Chris@0:         ) : '0');
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateDashMatch(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition(sprintf(
Chris@0:             '%1$s and (%1$s = %2$s or starts-with(%1$s, %3$s))',
Chris@0:             $attribute,
Chris@0:             Translator::getXpathLiteral($value),
Chris@0:             Translator::getXpathLiteral($value.'-')
Chris@0:         ));
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translatePrefixMatch(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition($value ? sprintf(
Chris@0:             '%1$s and starts-with(%1$s, %2$s)',
Chris@0:             $attribute,
Chris@0:             Translator::getXpathLiteral($value)
Chris@0:         ) : '0');
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateSuffixMatch(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition($value ? sprintf(
Chris@0:             '%1$s and substring(%1$s, string-length(%1$s)-%2$s) = %3$s',
Chris@0:             $attribute,
Chris@0:             strlen($value) - 1,
Chris@0:             Translator::getXpathLiteral($value)
Chris@0:         ) : '0');
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateSubstringMatch(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition($value ? sprintf(
Chris@0:             '%1$s and contains(%1$s, %2$s)',
Chris@0:             $attribute,
Chris@0:             Translator::getXpathLiteral($value)
Chris@0:         ) : '0');
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @param XPathExpr $xpath
Chris@0:      * @param string    $attribute
Chris@0:      * @param string    $value
Chris@0:      *
Chris@0:      * @return XPathExpr
Chris@0:      */
Chris@0:     public function translateDifferent(XPathExpr $xpath, $attribute, $value)
Chris@0:     {
Chris@0:         return $xpath->addCondition(sprintf(
Chris@0:             $value ? 'not(%1$s) or %1$s != %2$s' : '%s != %s',
Chris@0:             $attribute,
Chris@0:             Translator::getXpathLiteral($value)
Chris@0:         ));
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * {@inheritdoc}
Chris@0:      */
Chris@0:     public function getName()
Chris@0:     {
Chris@0:         return 'attribute-matching';
Chris@0:     }
Chris@0: }