Blame view

vendor/ezyang/htmlpurifier/library/HTMLPurifier/Arborize.php 2.48 KB
abf1649b   andryeyev   Чистая установка ...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
  <?php
  
  /**
   * Converts a stream of HTMLPurifier_Token into an HTMLPurifier_Node,
   * and back again.
   *
   * @note This transformation is not an equivalence.  We mutate the input
   * token stream to make it so; see all [MUT] markers in code.
   */
  class HTMLPurifier_Arborize
  {
      public static function arborize($tokens, $config, $context) {
          $definition = $config->getHTMLDefinition();
          $parent = new HTMLPurifier_Token_Start($definition->info_parent);
          $stack = array($parent->toNode());
          foreach ($tokens as $token) {
              $token->skip = null; // [MUT]
              $token->carryover = null; // [MUT]
              if ($token instanceof HTMLPurifier_Token_End) {
                  $token->start = null; // [MUT]
                  $r = array_pop($stack);
                  assert($r->name === $token->name);
                  assert(empty($token->attr));
                  $r->endCol = $token->col;
                  $r->endLine = $token->line;
                  $r->endArmor = $token->armor;
                  continue;
              }
              $node = $token->toNode();
              $stack[count($stack)-1]->children[] = $node;
              if ($token instanceof HTMLPurifier_Token_Start) {
                  $stack[] = $node;
              }
          }
          assert(count($stack) == 1);
          return $stack[0];
      }
  
      public static function flatten($node, $config, $context) {
          $level = 0;
          $nodes = array($level => new HTMLPurifier_Queue(array($node)));
          $closingTokens = array();
          $tokens = array();
          do {
              while (!$nodes[$level]->isEmpty()) {
                  $node = $nodes[$level]->shift(); // FIFO
                  list($start, $end) = $node->toTokenPair();
                  if ($level > 0) {
                      $tokens[] = $start;
                  }
                  if ($end !== NULL) {
                      $closingTokens[$level][] = $end;
                  }
                  if ($node instanceof HTMLPurifier_Node_Element) {
                      $level++;
                      $nodes[$level] = new HTMLPurifier_Queue();
                      foreach ($node->children as $childNode) {
                          $nodes[$level]->push($childNode);
                      }
                  }
              }
              $level--;
              if ($level && isset($closingTokens[$level])) {
                  while ($token = array_pop($closingTokens[$level])) {
                      $tokens[] = $token;
                  }
              }
          } while ($level > 0);
          return $tokens;
      }
  }