mediawiki-extensions-TemplateStylesExtender/includes/Hooks/StylesheetSanitizerHook.php

<?php
/**
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * @file
 */

declare( strict_types=1 );

namespace MediaWiki\Extension\TemplateStylesExtender\Hooks;

use MediaWiki\Extension\TemplateStylesExtender\MatcherFactoryExtender;
use MediaWiki\Extension\TemplateStylesExtender\StylePropertySanitizerExtender;
use MediaWiki\Extension\TemplateStylesExtender\TemplateStylesExtender;
use TemplateStylesMatcherFactory;
use Wikimedia\CSS\Sanitizer\MediaAtRuleSanitizer;
use Wikimedia\CSS\Sanitizer\StylePropertySanitizer;
use Wikimedia\CSS\Sanitizer\StylesheetSanitizer;

class StylesheetSanitizerHook {

	/**
	 * This adds new matchers
	 *
	 * @param StylesheetSanitizer $sanitizer
	 * @param StylePropertySanitizer $propertySanitizer
	 * @param TemplateStylesMatcherFactory $matcherFactory
	 */
	public static function onSanitize( $sanitizer, $propertySanitizer, $matcherFactory ): void {
		if ( TemplateStylesExtender::getConfigValue(
			'TemplateStylesExtenderEnablePrefersColorScheme',
			false ) === true ) {
			$factory = new MatcherFactoryExtender();

			$newRules = $sanitizer->getRuleSanitizers();
			$newRules['@media'] = new MediaAtRuleSanitizer( $factory->cssMediaQueryList() );
			$newRules['@media']->setRuleSanitizers( $newRules );

			$sanitizer->setRuleSanitizers( $newRules );

		}

		$extended = new TemplateStylesExtender();

		$extender = new StylePropertySanitizerExtender( $matcherFactory );

		$extended->addVarSelector( $propertySanitizer );

		$extended->addImageRendering( $extender );
		$extended->addRuby( $extender );
		$extended->addScrollMarginProperties( $extender, $matcherFactory );

		$propertySanitizer->setKnownProperties( $extender->getKnownProperties() );
	}
}