PhutilPygmentizeParser.php

<?php

/**
 * Parser that converts `pygmetize` output or similar HTML blocks from "class"
 * attributes to "style" attributes.
 */
final class PhutilPygmentizeParser extends Phobject {

  private $map = array();

  public function setMap(array $map) {
    $this->map = $map;
    return $this;
  }

  public function getMap() {
    return $this->map;
  }

  public function parse($block) {
    $class_look = 'class="';
    $class_len = strlen($class_look);

    $class_start = null;

    $map = $this->map;

    $len = strlen($block);
    $out = '';
    $mode = 'text';
    for ($ii = 0; $ii < $len; $ii++) {
      $c = $block[$ii];
      switch ($mode) {
        case 'text':
          // We're in general text between tags, and just passing characers
          // through unmodified.
          if ($c == '<') {
            $mode = 'tag';
          }
          $out .= $c;
          break;
        case 'tag':
          // We're inside a tag, and looking for `class="` so we can rewrite
          // it.
          if ($c == '>') {
            $mode = 'text';
          }
          if ($c == 'c') {
            if (!substr_compare($block, $class_look, $ii, $class_len)) {
              $mode = 'class';
              $ii += $class_len;
              $class_start = $ii;
            }
          }

          if ($mode != 'class') {
            $out .= $c;
          }
          break;
        case 'class':
          // We're inside a `class="..."` tag, and looking for the ending quote
          // so we can replace it.
          if ($c == '"') {
            $class = substr($block, $class_start, $ii - $class_start);

            // If this class is present in the map, rewrite it into an inline
            // style attribute.
            if (isset($map[$class])) {
              $out .= 'style="'.phutil_escape_html($map[$class]).'"';
            } else {
              $out .= 'class="'.$class.'"';
            }

            $mode = 'tag';
          }
          break;
      }
    }

    return $out;
  }

}