MediaWikiTitleCodec.php

Go to the documentation of this file.

<?php
class MediaWikiTitleCodec implements TitleFormatter, TitleParser {

    protected $language;

    protected $genderCache;

    protected $localInterwikis;

    public function __construct( Language $language, GenderCache $genderCache, $localInterwikis = array() ) {
        $this->language = $language;
        $this->genderCache = $genderCache;
        $this->localInterwikis = (array)$localInterwikis;
    }

    public function getNamespaceName( $namespace, $text ) {
        if ( $this->language->needsGenderDistinction() &&
            MWNamespace::hasGenderDistinction( $namespace ) ) {

            //NOTE: we are assuming here that the title text is a user name!
            $gender = $this->genderCache->getGenderOf( $text, __METHOD__ );
            $name = $this->language->getGenderNsText( $namespace, $gender );
        } else {
            $name = $this->language->getNsText( $namespace );
        }

        if ( $name === false ) {
            throw new InvalidArgumentException( 'Unknown namespace ID: ' . $namespace );
        }

        return $name;
    }

    public function formatTitle( $namespace, $text, $fragment = '' ) {
        if ( $namespace !== false ) {
            $namespace = $this->getNamespaceName( $namespace, $text );

            if ( $namespace !== '' ) {
                $text = $namespace . ':' . $text;
            }
        }

        if ( $fragment !== '' ) {
            $text = $text . '#' . $fragment;
        }

        $text = str_replace( '_', ' ', $text );

        return $text;
    }

    public function parseTitle( $text, $defaultNamespace ) {
        // NOTE: this is an ugly cludge that allows this class to share the
        // code for parsing with the old Title class. The parser code should
        // be refactored to avoid this.
        $parts = $this->splitTitleString( $text, $defaultNamespace );

        // Interwiki links are not supported by TitleValue
        if ( $parts['interwiki'] !== '' ) {
            throw new MalformedTitleException( 'Title must not contain an interwiki prefix: ' . $text );
        }

        // Relative fragment links are not supported by TitleValue
        if ( $parts['dbkey'] === '' ) {
            throw new MalformedTitleException( 'Title must not be empty: ' . $text );
        }

        return new TitleValue( $parts['namespace'], $parts['dbkey'], $parts['fragment'] );
    }

    public function getText( TitleValue $title ) {
        return $this->formatTitle( false, $title->getText(), '' );
    }

    public function getPrefixedText( TitleValue $title ) {
        return $this->formatTitle( $title->getNamespace(), $title->getText(), '' );
    }

    public function getFullText( TitleValue $title ) {
        return $this->formatTitle( $title->getNamespace(), $title->getText(), $title->getFragment() );
    }

    public function splitTitleString( $text, $defaultNamespace = NS_MAIN ) {
        $dbkey = str_replace( ' ', '_', $text );

        # Initialisation
        $parts = array(
            'interwiki' => '',
            'fragment' => '',
            'namespace' => $defaultNamespace,
            'dbkey' => $dbkey,
            'user_case_dbkey' => $dbkey,
        );

        # Strip Unicode bidi override characters.
        # Sometimes they slip into cut-n-pasted page titles, where the
        # override chars get included in list displays.
        $dbkey = preg_replace( '/\xE2\x80[\x8E\x8F\xAA-\xAE]/S', '', $dbkey );

        # Clean up whitespace
        # Note: use of the /u option on preg_replace here will cause
        # input with invalid UTF-8 sequences to be nullified out in PHP 5.2.x,
        # conveniently disabling them.
        $dbkey = preg_replace( '/[ _\xA0\x{1680}\x{180E}\x{2000}-\x{200A}\x{2028}\x{2029}\x{202F}\x{205F}\x{3000}]+/u', '_', $dbkey );
        $dbkey = trim( $dbkey, '_' );

        if ( strpos( $dbkey, UTF8_REPLACEMENT ) !== false ) {
            # Contained illegal UTF-8 sequences or forbidden Unicode chars.
            throw new MalformedTitleException( 'Bad UTF-8 sequences found in title: ' . $text );
        }

        $parts['dbkey'] = $dbkey;

        # Initial colon indicates main namespace rather than specified default
        # but should not create invalid {ns,title} pairs such as {0,Project:Foo}
        if ( $dbkey !== '' && ':' == $dbkey[0] ) {
            $parts['namespace'] = NS_MAIN;
            $dbkey = substr( $dbkey, 1 ); # remove the colon but continue processing
            $dbkey = trim( $dbkey, '_' ); # remove any subsequent whitespace
        }

        if ( $dbkey == '' ) {
            throw new MalformedTitleException( 'Empty title: ' . $text );
        }

        # Namespace or interwiki prefix
        $firstPass = true;
        $prefixRegexp = "/^(.+?)_*:_*(.*)$/S";
        do {
            $m = array();
            if ( preg_match( $prefixRegexp, $dbkey, $m ) ) {
                $p = $m[1];
                if ( ( $ns = $this->language->getNsIndex( $p ) ) !== false ) {
                    # Ordinary namespace
                    $dbkey = $m[2];
                    $parts['namespace'] = $ns;
                    # For Talk:X pages, check if X has a "namespace" prefix
                    if ( $ns == NS_TALK && preg_match( $prefixRegexp, $dbkey, $x ) ) {
                        if ( $this->language->getNsIndex( $x[1] ) ) {
                            # Disallow Talk:File:x type titles...
                            throw new MalformedTitleException( 'Bad namespace prefix: ' . $text );
                        } elseif ( Interwiki::isValidInterwiki( $x[1] ) ) {
                            //TODO: get rid of global state!
                            # Disallow Talk:Interwiki:x type titles...
                            throw new MalformedTitleException( 'Interwiki prefix found in title: ' . $text );
                        }
                    }
                } elseif ( Interwiki::isValidInterwiki( $p ) ) {
                    if ( !$firstPass ) {
                        //TODO: get rid of global state!
                        # Can't make a local interwiki link to an interwiki link.
                        # That's just crazy!
                        throw new MalformedTitleException( 'Interwiki prefix found in title: ' . $text );
                    }

                    # Interwiki link
                    $dbkey = $m[2];
                    $parts['interwiki'] = $this->language->lc( $p );

                    # Redundant interwiki prefix to the local wiki
                    foreach ( $this->localInterwikis as $localIW ) {
                        if ( 0 == strcasecmp( $parts['interwiki'], $localIW ) ) {
                            if ( $dbkey == '' ) {
                                # Can't have an empty self-link
                                throw new MalformedTitleException( 'Local interwiki with empty title: ' . $text );
                            }
                            $parts['interwiki'] = '';
                            $firstPass = false;

                            # Do another namespace split...
                            continue 2;
                        }
                    }

                    # If there's an initial colon after the interwiki, that also
                    # resets the default namespace
                    if ( $dbkey !== '' && $dbkey[0] == ':' ) {
                        $parts['namespace'] = NS_MAIN;
                        $dbkey = substr( $dbkey, 1 );
                    }
                }
                # If there's no recognized interwiki or namespace,
                # then let the colon expression be part of the title.
            }
            break;
        } while ( true );

        $fragment = strstr( $dbkey, '#' );
        if ( false !== $fragment ) {
            $parts['fragment'] = str_replace( '_', ' ', substr( $fragment, 1 ) );
            $dbkey = substr( $dbkey, 0, strlen( $dbkey ) - strlen( $fragment ) );
            # remove whitespace again: prevents "Foo_bar_#"
            # becoming "Foo_bar_"
            $dbkey = preg_replace( '/_*$/', '', $dbkey );
        }

        # Reject illegal characters.
        $rxTc = Title::getTitleInvalidRegex();
        if ( preg_match( $rxTc, $dbkey ) ) {
            throw new MalformedTitleException( 'Illegal characters found in title: ' . $text );
        }

        # Pages with "/./" or "/../" appearing in the URLs will often be un-
        # reachable due to the way web browsers deal with 'relative' URLs.
        # Also, they conflict with subpage syntax.  Forbid them explicitly.
        if (
            strpos( $dbkey, '.' ) !== false &&
            (
                $dbkey === '.' || $dbkey === '..' ||
                strpos( $dbkey, './' ) === 0 ||
                strpos( $dbkey, '../' ) === 0 ||
                strpos( $dbkey, '/./' ) !== false ||
                strpos( $dbkey, '/../' ) !== false ||
                substr( $dbkey, -2 ) == '/.' ||
                substr( $dbkey, -3 ) == '/..'
            )
        ) {
            throw new MalformedTitleException( 'Bad title: ' . $text );
        }

        # Magic tilde sequences? Nu-uh!
        if ( strpos( $dbkey, '~~~' ) !== false ) {
            throw new MalformedTitleException( 'Bad title: ' . $text );
        }

        # Limit the size of titles to 255 bytes. This is typically the size of the
        # underlying database field. We make an exception for special pages, which
        # don't need to be stored in the database, and may edge over 255 bytes due
        # to subpage syntax for long titles, e.g. [[Special:Block/Long name]]
        if (
            ( $parts['namespace'] != NS_SPECIAL && strlen( $dbkey ) > 255 )
            || strlen( $dbkey ) > 512
        ) {
            throw new MalformedTitleException( 'Title too long: ' . substr( $dbkey, 0, 255 ) . '...' );
        }

        # Normally, all wiki links are forced to have an initial capital letter so [[foo]]
        # and [[Foo]] point to the same place.  Don't force it for interwikis, since the
        # other site might be case-sensitive.
        $parts['user_case_dbkey'] = $dbkey;
        if ( $parts['interwiki'] === '' ) {
            $dbkey = Title::capitalize( $dbkey, $parts['namespace'] );
        }

        # Can't make a link to a namespace alone... "empty" local links can only be
        # self-links with a fragment identifier.
        if ( $dbkey == '' && $parts['interwiki'] === '' ) {
            if ( $parts['namespace'] != NS_MAIN ) {
                throw new MalformedTitleException( 'Empty title: ' . $text );
            }
        }

        // Allow IPv6 usernames to start with '::' by canonicalizing IPv6 titles.
        // IP names are not allowed for accounts, and can only be referring to
        // edits from the IP. Given '::' abbreviations and caps/lowercaps,
        // there are numerous ways to present the same IP. Having sp:contribs scan
        // them all is silly and having some show the edits and others not is
        // inconsistent. Same for talk/userpages. Keep them normalized instead.
        if ( $parts['namespace'] == NS_USER || $parts['namespace'] == NS_USER_TALK ) {
            $dbkey = IP::sanitizeIP( $dbkey );
        }

        // Any remaining initial :s are illegal.
        if ( $dbkey !== '' && ':' == $dbkey[0] ) {
            throw new MalformedTitleException( 'Title must not start with a colon: ' . $text );
        }

        # Fill fields
        $parts['dbkey'] = $dbkey;
        return $parts;
    }

}