AllowedAttributes.php
3 months ago
AllowedTags.php
3 months ago
AttributeInterface.php
3 months ago
TagInterface.php
3 months ago
XPath.php
3 months ago
XPath.php
65 lines
| 1 | <?php |
| 2 | namespace enshrined\svgSanitize\data; |
| 3 | |
| 4 | class XPath extends \DOMXPath |
| 5 | { |
| 6 | const DEFAULT_NAMESPACE_PREFIX = 'svg'; |
| 7 | |
| 8 | /** |
| 9 | * @var string |
| 10 | */ |
| 11 | protected $defaultNamespaceURI; |
| 12 | |
| 13 | public function __construct(\DOMDocument $doc) |
| 14 | { |
| 15 | parent::__construct($doc); |
| 16 | $this->handleDefaultNamespace(); |
| 17 | } |
| 18 | |
| 19 | /** |
| 20 | * @param string $nodeName |
| 21 | * @return string |
| 22 | */ |
| 23 | public function createNodeName($nodeName) |
| 24 | { |
| 25 | if (empty($this->defaultNamespaceURI)) { |
| 26 | return $nodeName; |
| 27 | } |
| 28 | return self::DEFAULT_NAMESPACE_PREFIX . ':' . $nodeName; |
| 29 | } |
| 30 | |
| 31 | protected function handleDefaultNamespace() |
| 32 | { |
| 33 | $rootElements = $this->getRootElements(); |
| 34 | |
| 35 | if (count($rootElements) !== 1) { |
| 36 | throw new \LogicException( |
| 37 | sprintf('Got %d svg elements, expected exactly one', count($rootElements)), |
| 38 | 1570870568 |
| 39 | ); |
| 40 | } |
| 41 | $this->defaultNamespaceURI = (string)$rootElements[0]->namespaceURI; |
| 42 | |
| 43 | if ($this->defaultNamespaceURI !== '') { |
| 44 | $this->registerNamespace(self::DEFAULT_NAMESPACE_PREFIX, $this->defaultNamespaceURI); |
| 45 | } |
| 46 | } |
| 47 | |
| 48 | /** |
| 49 | * @return \DOMElement[] |
| 50 | */ |
| 51 | protected function getRootElements() |
| 52 | { |
| 53 | $rootElements = []; |
| 54 | $elements = $this->document->getElementsByTagName('svg'); |
| 55 | /** @var \DOMElement $element */ |
| 56 | foreach ($elements as $element) { |
| 57 | if ($element->parentNode !== $this->document) { |
| 58 | continue; |
| 59 | } |
| 60 | $rootElements[] = $element; |
| 61 | } |
| 62 | return $rootElements; |
| 63 | } |
| 64 | } |
| 65 |