WP-LIT-MCP/tests/Unit/Integrations/Divi/Compose/ComposeFixture.php

<?php

declare(strict_types=1);

namespace LIT\MCP\Tests\Unit\Integrations\Divi\Compose;

use LIT\MCP\Integrations\Divi\Compose\AttrWriter;
use LIT\MCP\Integrations\Divi\Compose\MediaResolver;
use LIT\MCP\Integrations\Divi\Compose\Placeholders;
use LIT\MCP\Integrations\Divi\Compose\ValueSanitizer;
use LIT\MCP\Tests\Unit\Integrations\Divi\Design\DesignSystemTest;
use LIT\MCP\Tests\Unit\Integrations\Divi\Schema\ModuleCatalogTest;

/**
 * Compose classes wired with the real Divi 5.2.1 sample catalog, the sample design system and
 * predictable fakes for sanitizing and media (no WordPress). $unfiltered switches the fake
 * sanitizer between a user with and without unfiltered_html (ruling S7).
 */
final class ComposeFixture
{
    public static function sanitizer(bool $unfiltered = false): ValueSanitizer
    {
        return new class ($unfiltered) implements ValueSanitizer {
            public function __construct(private bool $unfiltered)
            {
            }

            public function text(string $value): string
            {
                return trim(strip_tags($value));
            }

            /** Like WpValueSanitizer (ruling S4): raw for users with unfiltered_html. */
            public function html(string $value): string
            {
                return $this->unfiltered ? $value : (string) preg_replace('#<script\b[^>]*>.*?</script>#is', '', $value);
            }

            public function code(string $value): string
            {
                return $value;
            }

            public function url(string $value): string
            {
                return str_starts_with(strtolower(trim($value)), 'javascript:') ? '' : trim($value);
            }

            public function allowsUnfilteredHtml(): bool
            {
                return $this->unfiltered;
            }
        };
    }

    public static function media(): MediaResolver
    {
        return new class implements MediaResolver {
            public function image(int $id): ?array
            {
                return match ($id) {
                    33 => ['url' => 'https://kunde.test/wp-content/uploads/team.jpg', 'alt' => 'Unser Team'],
                    12 => ['url' => 'https://kunde.test/wp-content/uploads/halle.jpg', 'alt' => ''],
                    default => null,
                };
            }
        };
    }

    public static function placeholders(): Placeholders
    {
        return new Placeholders(DesignSystemTest::sample());
    }

    public static function attrWriter(bool $unfiltered = false): AttrWriter
    {
        return new AttrWriter(ModuleCatalogTest::sampleCatalog(), self::placeholders(), self::sanitizer($unfiltered));
    }
}