RegexInsertablesSuggester.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\TranslatorInterface\Insertable;
 
use InvalidArgumentException;
 
class RegexInsertablesSuggester implements InsertablesSuggester {
    protected $regex = null;
    protected $display = null;
    protected $pre = null;
    protected $post = null;
 
    public function __construct( $params ) {
        if ( is_string( $params ) ) {
            $this->regex = $params;
        } elseif ( is_array( $params ) ) {
            // Validate if the array is in a proper format.
            $this->regex = $params['regex'] ?? null;
            $this->display = $params['display'] ?? null;
            $this->pre = $params['pre'] ?? null;
            $this->post = $params['post'] ?? null;
        }
 
        if ( $this->regex === null ) {
            throw new InvalidArgumentException(
                'Invalid configuration for the RegexInsertablesSuggester. Did not find a regex specified'
            );
        }
 
        if ( $this->display !== null && $this->pre === null ) {
            // if display value is set, and pre value is not set, set the display to pre.
            // makes the configuration easier.
            $this->pre = $this->display;
        }
    }
 
    public function getInsertables( string $text ): array {
        $matches = [];
        preg_match_all( $this->regex, $text, $matches, PREG_SET_ORDER );
 
        return array_map( [ $this, 'mapInsertables' ], $matches );
    }
 
    protected function mapInsertables( array $match ) {
        if ( $this->display === null ) {
            return new Insertable( $match[0], $match[0] );
        }
 
        $replacements = [];
        // add '$' to the other keys for replacement.
        foreach ( $match as $key => $value ) {
            if ( !is_int( $key ) ) {
                $tmpKey = '$' . $key;
                $replacements[ $tmpKey ] = $value;
            }
        }
 
        $displayVal = strtr( $this->display, $replacements );
        $preVal = strtr( $this->pre, $replacements );
        $postVal = '';
        if ( $this->post !== null ) {
            $postVal = strtr( $this->post, $replacements );
        }
 
        return new Insertable( $displayVal, $preVal, $postVal );
    }
}