RemexCompatFormatter.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki\Tidy;
 
use MediaWiki\Parser\Sanitizer;
use Wikimedia\RemexHtml\HTMLData;
use Wikimedia\RemexHtml\Serializer\HtmlFormatter;
use Wikimedia\RemexHtml\Serializer\SerializerNode;
 
class RemexCompatFormatter extends HtmlFormatter {
    private const MARKED_EMPTY_ELEMENTS = [
        'li' => true,
        'p' => true,
        'tr' => true,
    ];
 
    private $textProcessor;
 
    public function __construct( array $options = [] ) {
        parent::__construct( $options );
        // Escape non-breaking space
        $this->attributeEscapes["\u{00A0}"] = '&#160;';
        $this->textEscapes["\u{00A0}"] = '&#160;';
        // Escape U+0338 (T387130)
        $this->textEscapes["\u{0338}"] = '&#x338;';
        // Disable escaping of '&', because we expect to see entities, due to 'ignoreCharRefs'
        unset( $this->attributeEscapes["&"] );
        unset( $this->textEscapes["&"] );
        $this->textProcessor = $options['textProcessor'] ?? null;
    }
 
    public function startDocument( $fragmentNamespace, $fragmentName ) {
        return '';
    }
 
    public function characters( SerializerNode $parent, $text, $start, $length ) {
        $text = parent::characters( $parent, $text, $start, $length );
 
        if ( $parent->namespace !== HTMLData::NS_HTML
            || !isset( $this->rawTextElements[$parent->name] )
        ) {
            if ( $this->textProcessor !== null ) {
                $text = ( $this->textProcessor )( $text );
            }
 
            // Ensure a consistent representation for all entities
            $text = Sanitizer::normalizeCharReferences( $text );
        }
 
        return $text;
    }
 
    public function element( SerializerNode $parent, SerializerNode $node, $contents ) {
        $data = $node->snData;
        if ( $data && $data->isPWrapper ) {
            if ( $data->nonblankNodeCount ) {
                return "<p>$contents</p>";
            } else {
                return $contents;
            }
        }
 
        $name = $node->name;
        $attrs = $node->attrs;
        if ( isset( self::MARKED_EMPTY_ELEMENTS[$name] ) && $attrs->count() === 0
            && strspn( $contents, "\t\n\f\r " ) === strlen( $contents )
        ) {
            return "<{$name} class=\"mw-empty-elt\">$contents</{$name}>";
        }
 
        $s = "<$name";
        foreach ( $attrs->getValues() as $attrName => $attrValue ) {
            $encValue = strtr( $attrValue, $this->attributeEscapes );
            $encValue = Sanitizer::normalizeCharReferences( $encValue );
            $s .= " $attrName=\"$encValue\"";
        }
        if ( $node->namespace === HTMLData::NS_HTML && isset( $this->voidElements[$name] ) ) {
            $s .= ' />';
            return $s;
        }
 
        $s .= '>';
        if ( $node->namespace === HTMLData::NS_HTML
            && isset( $contents[0] ) && $contents[0] === "\n"
            && isset( $this->prefixLfElements[$name] )
        ) {
            $s .= "\n$contents</$name>";
        } else {
            $s .= "$contents</$name>";
        }
        return $s;
    }
}