[Summary view] [Print] [Text view]

   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  }