MagicWordArray.php

Go to the documentation of this file.

<?php
 
use MediaWiki\Logger\LoggerFactory;
use MediaWiki\MediaWikiServices;
 
class MagicWordArray {
    public $names = [];
 
    private $factory;
 
    private $hash;
 
    private $baseRegex;
 
    private $regex;
 
    public function __construct( $names = [], MagicWordFactory $factory = null ) {
        $this->names = $names;
        $this->factory = $factory ?: MediaWikiServices::getInstance()->getMagicWordFactory();
    }
 
    public function add( $name ) {
        $this->names[] = $name;
        $this->hash = $this->baseRegex = $this->regex = null;
    }
 
    public function addArray( $names ) {
        $this->names = array_merge( $this->names, array_values( $names ) );
        $this->hash = $this->baseRegex = $this->regex = null;
    }
 
    public function getHash() {
        if ( $this->hash === null ) {
            $this->hash = [ 0 => [], 1 => [] ];
            foreach ( $this->names as $name ) {
                $magic = $this->factory->get( $name );
                $case = intval( $magic->isCaseSensitive() );
                foreach ( $magic->getSynonyms() as $syn ) {
                    if ( !$case ) {
                        $syn = $this->factory->getContentLanguage()->lc( $syn );
                    }
                    $this->hash[$case][$syn] = $name;
                }
            }
        }
        return $this->hash;
    }
 
    public function getBaseRegex( bool $capture = true, string $delimiter = '/' ): array {
        if ( $capture && $delimiter === '/' && $this->baseRegex !== null ) {
            return $this->baseRegex;
        }
        $regex = [ 0 => [], 1 => [] ];
        $allGroups = [];
        foreach ( $this->names as $name ) {
            $magic = $this->factory->get( $name );
            $case = $magic->isCaseSensitive() ? 1 : 0;
            foreach ( $magic->getSynonyms() as $i => $syn ) {
                if ( $capture ) {
                    // Group name must start with a non-digit in PCRE 8.34+
                    $it = strtr( $i, '0123456789', 'abcdefghij' );
                    $groupName = $it . '_' . $name;
                    $group = '(?P<' . $groupName . '>' . preg_quote( $syn, $delimiter ) . ')';
                    // look for same group names to avoid same named subpatterns in the regex
                    if ( isset( $allGroups[$groupName] ) ) {
                        throw new MWException(
                            __METHOD__ . ': duplicate internal name in magic word array: ' . $name
                        );
                    }
                    $allGroups[$groupName] = true;
                    $regex[$case][] = $group;
                } else {
                    $regex[$case][] = preg_quote( $syn, $delimiter );
                }
            }
        }
        '@phan-var array<int,string[]> $regex';
        foreach ( $regex as $case => &$re ) {
            $re = count( $re ) ? implode( '|', $re ) : '(?!)';
            if ( !$case ) {
                $re = "(?i:{$re})";
            }
        }
        '@phan-var array<int,string> $regex';
 
        if ( $capture && $delimiter === '/' ) {
            $this->baseRegex = $regex;
        }
        return $regex;
    }
 
    public function getRegex() {
        if ( $this->regex === null ) {
            $this->regex = [];
            $base = $this->getBaseRegex( true, '/' );
            foreach ( $base as $case => $re ) {
                $this->regex[$case] = "/{$re}/S";
            }
            // As a performance optimization, turn on unicode mode only for
            // case-insensitive matching.
            $this->regex[0] .= 'u';
        }
        return $this->regex;
    }
 
    public function getVariableRegex() {
        return str_replace( "\\$1", "(.*?)", $this->getRegex() );
    }
 
    public function getRegexStart() {
        $newRegex = [];
        $base = $this->getBaseRegex( true, '/' );
        foreach ( $base as $case => $re ) {
            $newRegex[$case] = "/^(?:{$re})/S";
        }
        // As a performance optimization, turn on unicode mode only for
        // case-insensitive matching.
        $newRegex[0] .= 'u';
        return $newRegex;
    }
 
    public function getVariableStartToEndRegex() {
        $newRegex = [];
        $base = $this->getBaseRegex( true, '/' );
        foreach ( $base as $case => $re ) {
            $newRegex[$case] = str_replace( "\\$1", "(.*?)", "/^(?:{$re})$/S" );
        }
        // As a performance optimization, turn on unicode mode only for
        // case-insensitive matching.
        $newRegex[0] .= 'u';
        return $newRegex;
    }
 
    public function getNames() {
        return $this->names;
    }
 
    public function parseMatch( $m ) {
        reset( $m );
        while ( ( $key = key( $m ) ) !== null ) {
            $value = current( $m );
            next( $m );
            if ( $key === 0 || $value === '' ) {
                continue;
            }
            $parts = explode( '_', $key, 2 );
            if ( count( $parts ) != 2 ) {
                // This shouldn't happen
                // continue;
                throw new MWException( __METHOD__ . ': bad parameter name' );
            }
            list( /* $synIndex */, $magicName ) = $parts;
            $paramValue = next( $m );
            return [ $magicName, $paramValue ];
        }
        // This shouldn't happen either
        throw new MWException( __METHOD__ . ': parameter not found' );
    }
 
    public function matchVariableStartToEnd( $text ) {
        $regexes = $this->getVariableStartToEndRegex();
        foreach ( $regexes as $regex ) {
            $m = [];
            if ( preg_match( $regex, $text, $m ) ) {
                return $this->parseMatch( $m );
            }
        }
        return [ false, false ];
    }
 
    public function matchStartToEnd( $text ) {
        $hash = $this->getHash();
        if ( isset( $hash[1][$text] ) ) {
            return $hash[1][$text];
        }
        $lc = $this->factory->getContentLanguage()->lc( $text );
        return $hash[0][$lc] ?? false;
    }
 
    public function matchAndRemove( &$text ) {
        $found = [];
        $regexes = $this->getRegex();
        foreach ( $regexes as $regex ) {
            $matches = [];
            $res = preg_match_all( $regex, $text, $matches, PREG_SET_ORDER );
            if ( $res === false ) {
                $error = preg_last_error();
                // TODO: Remove function_exists when we require PHP8
                $errorText = function_exists( 'preg_last_error_msg' ) ? preg_last_error_msg() : '';
                LoggerFactory::getInstance( 'parser' )->warning( 'preg_match_all error: {code} {errorText}', [
                    'code' => $error,
                    'regex' => $regex,
                    'text' => $text,
                    'errorText' => $errorText
                ] );
                throw new Exception( "preg_match_all error $error: $errorText" );
            } elseif ( $res ) {
                foreach ( $matches as $m ) {
                    list( $name, $param ) = $this->parseMatch( $m );
                    $found[$name] = $param;
                }
            }
            $res = preg_replace( $regex, '', $text );
            if ( $res === null ) {
                $error = preg_last_error();
                // TODO: Remove function_exists when we require PHP8
                $errorText = function_exists( 'preg_last_error_msg' ) ? preg_last_error_msg() : '';
                LoggerFactory::getInstance( 'parser' )->warning( 'preg_replace error: {code} {errorText}', [
                    'code' => $error,
                    'regex' => $regex,
                    'text' => $text,
                    'errorText' => $errorText
                ] );
                throw new Exception( "preg_replace error $error: $errorText" );
            }
            $text = $res;
        }
        return $found;
    }
 
    public function matchStartAndRemove( &$text ) {
        $regexes = $this->getRegexStart();
        foreach ( $regexes as $regex ) {
            if ( preg_match( $regex, $text, $m ) ) {
                list( $id, ) = $this->parseMatch( $m );
                if ( strlen( $m[0] ) >= strlen( $text ) ) {
                    $text = '';
                } else {
                    $text = substr( $text, strlen( $m[0] ) );
                }
                return $id;
            }
        }
        return false;
    }
}