HtmlHelper.php

Go to the documentation of this file.

<?php
 
namespace MediaWiki;
 
use Wikimedia\Assert\Assert;
use Wikimedia\RemexHtml\HTMLData;
use Wikimedia\RemexHtml\Serializer\HtmlFormatter;
use Wikimedia\RemexHtml\Serializer\Serializer;
use Wikimedia\RemexHtml\Serializer\SerializerNode;
use Wikimedia\RemexHtml\Tokenizer\Tokenizer;
use Wikimedia\RemexHtml\TreeBuilder\Dispatcher;
use Wikimedia\RemexHtml\TreeBuilder\TreeBuilder;
 
class HtmlHelper {
 
    public static function modifyElements(
        string $htmlFragment,
        callable $shouldModifyCallback,
        callable $modifyCallback
    ) {
        $formatter = new class( $options = [], $shouldModifyCallback, $modifyCallback ) extends HtmlFormatter {
            private $shouldModifyCallback;
 
            private $modifyCallback;
 
            public function __construct( $options, $shouldModifyCallback, $modifyCallback ) {
                parent::__construct( $options );
                $this->shouldModifyCallback = $shouldModifyCallback;
                $this->modifyCallback = $modifyCallback;
            }
 
            public function element( SerializerNode $parent, SerializerNode $node, $contents ) {
                if ( ( $this->shouldModifyCallback )( $node ) ) {
                    $node = clone $node;
                    $node->attrs = clone $node->attrs;
                    $newNode = ( $this->modifyCallback )( $node );
                    Assert::parameterType( SerializerNode::class, $newNode, 'return value' );
                    return parent::element( $parent, $newNode, $contents );
                } else {
                    return parent::element( $parent, $node, $contents );
                }
            }
 
            public function startDocument( $fragmentNamespace, $fragmentName ) {
                return '';
            }
        };
        $serializer = new Serializer( $formatter );
        $treeBuilder = new TreeBuilder( $serializer );
        $dispatcher = new Dispatcher( $treeBuilder );
        $tokenizer = new Tokenizer( $dispatcher, $htmlFragment );
 
        $tokenizer->execute( [
            'fragmentNamespace' => HTMLData::NS_HTML,
            'fragmentName' => 'body',
        ] );
 
        return $serializer->getResult();
    }
 
}