Merge "Whitelist the <wbr> element."

[mediawiki.git] / includes / Message.php

<?php
/**
 * Fetching and processing of interface messages.
 *
 * 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.
 * http://www.gnu.org/copyleft/gpl.html
 *
 * @file
 * @author Niklas Laxström
 */

/**
 * The Message class provides methods which fulfil two basic services:
 *  - fetching interface messages
 *  - processing messages into a variety of formats
 *
 * First implemented with MediaWiki 1.17, the Message class is intended to
 * replace the old wfMsg* functions that over time grew unusable.
 * @see https://www.mediawiki.org/wiki/Manual:Messages_API for equivalences
 * between old and new functions.
 *
 * You should use the wfMessage() global function which acts as a wrapper for
 * the Message class. The wrapper let you pass parameters as arguments.
 *
 * The most basic usage cases would be:
 *
 * @code
 *     // Initialize a Message object using the 'some_key' message key
 *     $message = wfMessage( 'some_key' );
 *
 *     // Using two parameters those values are strings 'value1' and 'value2':
 *     $message = wfMessage( 'some_key',
 *          'value1', 'value2'
 *     );
 * @endcode
 *
 * @section message_global_fn Global function wrapper:
 *
 * Since wfMessage() returns a Message instance, you can chain its call with
 * a method. Some of them return a Message instance too so you can chain them.
 * You will find below several examples of wfMessage() usage.
 *
 * Fetching a message text for interface message:
 *
 * @code
 *    $button = Xml::button(
 *         wfMessage( 'submit' )->text()
 *    );
 * @endcode
 *
 * A Message instance can be passed parameters after it has been constructed,
 * use the params() method to do so:
 *
 * @code
 *     wfMessage( 'welcome-to' )
 *         ->params( $wgSitename )
 *         ->text();
 * @endcode
 *
 * {{GRAMMAR}} and friends work correctly:
 *
 * @code
 *    wfMessage( 'are-friends',
 *        $user, $friend
 *    );
 *    wfMessage( 'bad-message' )
 *         ->rawParams( '<script>...</script>' )
 *         ->escaped();
 * @endcode
 *
 * @section message_language Changing language:
 *
 * Messages can be requested in a different language or in whatever current
 * content language is being used. The methods are:
 *     - Message->inContentLanguage()
 *     - Message->inLanguage()
 *
 * Sometimes the message text ends up in the database, so content language is
 * needed:
 *
 * @code
 *    wfMessage( 'file-log',
 *        $user, $filename
 *    )->inContentLanguage()->text();
 * @endcode
 *
 * Checking whether a message exists:
 *
 * @code
 *    wfMessage( 'mysterious-message' )->exists()
 *    // returns a boolean whether the 'mysterious-message' key exist.
 * @endcode
 *
 * If you want to use a different language:
 *
 * @code
 *    $userLanguage = $user->getOption( 'language' );
 *    wfMessage( 'email-header' )
 *         ->inLanguage( $userLanguage )
 *         ->plain();
 * @endcode
 *
 * @note You can parse the text only in the content or interface languages
 *
 * @section message_compare_old Comparison with old wfMsg* functions:
 *
 * Use full parsing:
 *
 * @code
 *     // old style:
 *     wfMsgExt( 'key', array( 'parseinline' ), 'apple' );
 *     // new style:
 *     wfMessage( 'key', 'apple' )->parse();
 * @endcode
 *
 * Parseinline is used because it is more useful when pre-building HTML.
 * In normal use it is better to use OutputPage::(add|wrap)WikiMsg.
 *
 * Places where HTML cannot be used. {{-transformation is done.
 * @code
 *     // old style:
 *     wfMsgExt( 'key', array( 'parsemag' ), 'apple', 'pear' );
 *     // new style:
 *     wfMessage( 'key', 'apple', 'pear' )->text();
 * @endcode
 *
 * Shortcut for escaping the message too, similar to wfMsgHTML(), but
 * parameters are not replaced after escaping by default.
 * @code
 *     $escaped = wfMessage( 'key' )
 *          ->rawParams( 'apple' )
 *          ->escaped();
 * @endcode
 *
 * @section message_appendix Appendix:
 *
 * @todo
 * - test, can we have tests?
 * - this documentation needs to be extended
 *
 * @see https://www.mediawiki.org/wiki/WfMessage()
 * @see https://www.mediawiki.org/wiki/New_messages_API
 * @see https://www.mediawiki.org/wiki/Localisation
 *
 * @since 1.17
 */
class Message {
        /**
         * In which language to get this message. True, which is the default,
         * means the current interface language, false content language.
         */
        protected $interface = true;

        /**
         * In which language to get this message. Overrides the $interface
         * variable.
         *
         * @var Language
         */
        protected $language = null;

        /**
         * The message key.
         */
        protected $key;

        /**
         * List of parameters which will be substituted into the message.
         */
        protected $parameters = array();

        /**
         * Format for the message.
         * Supported formats are:
         * * text (transform)
         * * escaped (transform+htmlspecialchars)
         * * block-parse
         * * parse (default)
         * * plain
         */
        protected $format = 'parse';

        /**
         * Whether database can be used.
         */
        protected $useDatabase = true;

        /**
         * Title object to use as context
         */
        protected $title = null;

        /**
         * Content object representing the message
         */
        protected $content = null;

        /**
         * @var string
         */
        protected $message;

        /**
         * Constructor.
         * @since 1.17
         * @param $key: message key, or array of message keys to try and use the first non-empty message for
         * @param array $params message parameters
         * @return Message: $this
         */
        public function __construct( $key, $params = array() ) {
                global $wgLang;
                $this->key = $key;
                $this->parameters = array_values( $params );
                $this->language = $wgLang;
        }

        /**
         * Returns the message key
         *
         * @since 1.21
         *
         * @return string
         */
        public function getKey() {
                if ( is_array( $this->key ) ) {
                        // May happen if some kind of fallback is applied.
                        // For now, just use the first key. We really need a better solution.
                        return $this->key[0];
                } else {
                        return $this->key;
                }
        }

        /**
         * Returns the message parameters
         *
         * @since 1.21
         *
         * @return string[]
         */
        public function getParams() {
                return $this->parameters;
        }

        /**
         * Returns the message format
         *
         * @since 1.21
         *
         * @return string
         */
        public function getFormat() {
                return $this->format;
        }

        /**
         * Factory function that is just wrapper for the real constructor. It is
         * intended to be used instead of the real constructor, because it allows
         * chaining method calls, while new objects don't.
         * @since 1.17
         * @param string $key message key
         * @param Varargs: parameters as Strings
         * @return Message: $this
         */
        public static function newFromKey( $key /*...*/ ) {
                $params = func_get_args();
                array_shift( $params );
                return new self( $key, $params );
        }

        /**
         * Factory function accepting multiple message keys and returning a message instance
         * for the first message which is non-empty. If all messages are empty then an
         * instance of the first message key is returned.
         * @since 1.18
         * @param Varargs: message keys (or first arg as an array of all the message keys)
         * @return Message: $this
         */
        public static function newFallbackSequence( /*...*/ ) {
                $keys = func_get_args();
                if ( func_num_args() == 1 ) {
                        if ( is_array( $keys[0] ) ) {
                                // Allow an array to be passed as the first argument instead
                                $keys = array_values( $keys[0] );
                        } else {
                                // Optimize a single string to not need special fallback handling
                                $keys = $keys[0];
                        }
                }
                return new self( $keys );
        }

        /**
         * Adds parameters to the parameter list of this message.
         * @since 1.17
         * @param Varargs: parameters as Strings, or a single argument that is an array of Strings
         * @return Message: $this
         */
        public function params( /*...*/ ) {
                $args = func_get_args();
                if ( isset( $args[0] ) && is_array( $args[0] ) ) {
                        $args = $args[0];
                }
                $args_values = array_values( $args );
                $this->parameters = array_merge( $this->parameters, $args_values );
                return $this;
        }

        /**
         * Add parameters that are substituted after parsing or escaping.
         * In other words the parsing process cannot access the contents
         * of this type of parameter, and you need to make sure it is
         * sanitized beforehand.  The parser will see "$n", instead.
         * @since 1.17
         * @param Varargs: raw parameters as Strings (or single argument that is an array of raw parameters)
         * @return Message: $this
         */
        public function rawParams( /*...*/ ) {
                $params = func_get_args();
                if ( isset( $params[0] ) && is_array( $params[0] ) ) {
                        $params = $params[0];
                }
                foreach ( $params as $param ) {
                        $this->parameters[] = self::rawParam( $param );
                }
                return $this;
        }

        /**
         * Add parameters that are numeric and will be passed through
         * Language::formatNum before substitution
         * @since 1.18
         * @param Varargs: numeric parameters (or single argument that is array of numeric parameters)
         * @return Message: $this
         */
        public function numParams( /*...*/ ) {
                $params = func_get_args();
                if ( isset( $params[0] ) && is_array( $params[0] ) ) {
                        $params = $params[0];
                }
                foreach ( $params as $param ) {
                        $this->parameters[] = self::numParam( $param );
                }
                return $this;
        }

        /**
         * Set the language and the title from a context object
         * @since 1.19
         * @param $context IContextSource
         * @return Message: $this
         */
        public function setContext( IContextSource $context ) {
                $this->inLanguage( $context->getLanguage() );
                $this->title( $context->getTitle() );
                $this->interface = true;

                return $this;
        }

        /**
         * Request the message in any language that is supported.
         * As a side effect interface message status is unconditionally
         * turned off.
         * @since 1.17
         * @param $lang Mixed: language code or Language object.
         * @throws MWException
         * @return Message: $this
         */
        public function inLanguage( $lang ) {
                if ( $lang instanceof Language || $lang instanceof StubUserLang ) {
                        $this->language = $lang;
                } elseif ( is_string( $lang ) ) {
                        if ( $this->language->getCode() != $lang ) {
                                $this->language = Language::factory( $lang );
                        }
                } else {
                        $type = gettype( $lang );
                        throw new MWException( __METHOD__ . " must be "
                                . "passed a String or Language object; $type given"
                        );
                }
                $this->interface = false;
                return $this;
        }

        /**
         * Request the message in the wiki's content language,
         * unless it is disabled for this message.
         * @since 1.17
         * @see $wgForceUIMsgAsContentMsg
         * @return Message: $this
         */
        public function inContentLanguage() {
                global $wgForceUIMsgAsContentMsg;
                if ( in_array( $this->key, (array)$wgForceUIMsgAsContentMsg ) ) {
                        return $this;
                }

                global $wgContLang;
                $this->interface = false;
                $this->language = $wgContLang;
                return $this;
        }

        /**
         * Allows manipulating the interface message flag directly.
         * Can be used to restore the flag after setting a language.
         * @param $value bool
         * @return Message: $this
         * @since 1.20
         */
        public function setInterfaceMessageFlag( $value ) {
                $this->interface = (bool) $value;
                return $this;
        }

        /**
         * Enable or disable database use.
         * @since 1.17
         * @param $value Boolean
         * @return Message: $this
         */
        public function useDatabase( $value ) {
                $this->useDatabase = (bool) $value;
                return $this;
        }

        /**
         * Set the Title object to use as context when transforming the message
         * @since 1.18
         * @param $title Title object
         * @return Message: $this
         */
        public function title( $title ) {
                $this->title = $title;
                return $this;
        }

        /**
         * Returns the message as a Content object.
         * @return Content
         */
        public function content() {
                if ( !$this->content ) {
                        $this->content = new MessageContent( $this );
                }

                return $this->content;
        }

        /**
         * Returns the message parsed from wikitext to HTML.
         * @since 1.17
         * @return String: HTML
         */
        public function toString() {
                $string = $this->fetchMessage();

                if ( $string === false ) {
                        $key = htmlspecialchars( is_array( $this->key ) ? $this->key[0] : $this->key );
                        if ( $this->format === 'plain' ) {
                                return '<' . $key . '>';
                        }
                        return '&lt;' . $key . '&gt;';
                }

                # Replace $* with a list of parameters for &uselang=qqx.
                if ( strpos( $string, '$*' ) !== false ) {
                        $paramlist = '';
                        if ( $this->parameters !== array() ) {
                                $paramlist = ': $' . implode( ', $', range( 1, count( $this->parameters ) ) );
                        }
                        $string = str_replace( '$*', $paramlist, $string );
                }

                # Replace parameters before text parsing
                $string = $this->replaceParameters( $string, 'before' );

                # Maybe transform using the full parser
                if ( $this->format === 'parse' ) {
                        $string = $this->parseText( $string );
                        $m = array();
                        if ( preg_match( '/^<p>(.*)\n?<\/p>\n?$/sU', $string, $m ) ) {
                                $string = $m[1];
                        }
                } elseif ( $this->format === 'block-parse' ) {
                        $string = $this->parseText( $string );
                } elseif ( $this->format === 'text' ) {
                        $string = $this->transformText( $string );
                } elseif ( $this->format === 'escaped' ) {
                        $string = $this->transformText( $string );
                        $string = htmlspecialchars( $string, ENT_QUOTES, 'UTF-8', false );
                }

                # Raw parameter replacement
                $string = $this->replaceParameters( $string, 'after' );

                return $string;
        }

        /**
         * Magic method implementation of the above (for PHP >= 5.2.0), so we can do, eg:
         *     $foo = Message::get( $key );
         *     $string = "<abbr>$foo</abbr>";
         * @since 1.18
         * @return String
         */
        public function __toString() {
                // PHP doesn't allow __toString to throw exceptions and will
                // trigger a fatal error if it does. So, catch any exceptions.

                try {
                        return $this->toString();
                } catch ( Exception $ex ) {
                        try {
                                trigger_error( "Exception caught in " . __METHOD__ . " (message " . $this->key . "): "
                                        . $ex, E_USER_WARNING );
                        } catch ( Exception $ex ) {
                                // Doh! Cause a fatal error after all?
                        }

                        if ( $this->format === 'plain' ) {
                                return '<' . $this->key . '>';
                        }
                        return '&lt;' . $this->key . '&gt;';
                }
        }

        /**
         * Fully parse the text from wikitext to HTML
         * @since 1.17
         * @return String parsed HTML
         */
        public function parse() {
                $this->format = 'parse';
                return $this->toString();
        }

        /**
         * Returns the message text. {{-transformation is done.
         * @since 1.17
         * @return String: Unescaped message text.
         */
        public function text() {
                $this->format = 'text';
                return $this->toString();
        }

        /**
         * Returns the message text as-is, only parameters are substituted.
         * @since 1.17
         * @return String: Unescaped untransformed message text.
         */
        public function plain() {
                $this->format = 'plain';
                return $this->toString();
        }

        /**
         * Returns the parsed message text which is always surrounded by a block element.
         * @since 1.17
         * @return String: HTML
         */
        public function parseAsBlock() {
                $this->format = 'block-parse';
                return $this->toString();
        }

        /**
         * Returns the message text. {{-transformation is done and the result
         * is escaped excluding any raw parameters.
         * @since 1.17
         * @return String: Escaped message text.
         */
        public function escaped() {
                $this->format = 'escaped';
                return $this->toString();
        }

        /**
         * Check whether a message key has been defined currently.
         * @since 1.17
         * @return Bool: true if it is and false if not.
         */
        public function exists() {
                return $this->fetchMessage() !== false;
        }

        /**
         * Check whether a message does not exist, or is an empty string
         * @since 1.18
         * @return Bool: true if is is and false if not
         * @todo FIXME: Merge with isDisabled()?
         */
        public function isBlank() {
                $message = $this->fetchMessage();
                return $message === false || $message === '';
        }

        /**
         * Check whether a message does not exist, is an empty string, or is "-"
         * @since 1.18
         * @return Bool: true if it is and false if not
         */
        public function isDisabled() {
                $message = $this->fetchMessage();
                return $message === false || $message === '' || $message === '-';
        }

        /**
         * @since 1.17
         * @param $value
         * @return array
         */
        public static function rawParam( $value ) {
                return array( 'raw' => $value );
        }

        /**
         * @since 1.18
         * @param $value
         * @return array
         */
        public static function numParam( $value ) {
                return array( 'num' => $value );
        }

        /**
         * Substitutes any parameters into the message text.
         * @since 1.17
         * @param string $message the message text
         * @param string $type either before or after
         * @return String
         */
        protected function replaceParameters( $message, $type = 'before' ) {
                $replacementKeys = array();
                foreach ( $this->parameters as $n => $param ) {
                        list( $paramType, $value ) = $this->extractParam( $param );
                        if ( $type === $paramType ) {
                                $replacementKeys['$' . ( $n + 1 )] = $value;
                        }
                }
                $message = strtr( $message, $replacementKeys );
                return $message;
        }

        /**
         * Extracts the parameter type and preprocessed the value if needed.
         * @since 1.18
         * @param string|array $param Parameter as defined in this class.
         * @return Tuple(type, value)
         */
        protected function extractParam( $param ) {
                if ( is_array( $param ) && isset( $param['raw'] ) ) {
                        return array( 'after', $param['raw'] );
                } elseif ( is_array( $param ) && isset( $param['num'] ) ) {
                        // Replace number params always in before step for now.
                        // No support for combined raw and num params
                        return array( 'before', $this->language->formatNum( $param['num'] ) );
                } elseif ( !is_array( $param ) ) {
                        return array( 'before', $param );
                } else {
                        trigger_error(
                                "Invalid message parameter: " . htmlspecialchars( serialize( $param ) ),
                                E_USER_WARNING
                        );
                        return array( 'before', '[INVALID]' );
                }
        }

        /**
         * Wrapper for what ever method we use to parse wikitext.
         * @since 1.17
         * @param string $string Wikitext message contents
         * @return string Wikitext parsed into HTML
         */
        protected function parseText( $string ) {
                $out = MessageCache::singleton()->parse( $string, $this->title, /*linestart*/true, $this->interface, $this->language );
                return is_object( $out ) ? $out->getText() : $out;
        }

        /**
         * Wrapper for what ever method we use to {{-transform wikitext.
         * @since 1.17
         * @param string $string Wikitext message contents
         * @return string Wikitext with {{-constructs replaced with their values.
         */
        protected function transformText( $string ) {
                return MessageCache::singleton()->transform( $string, $this->interface, $this->language, $this->title );
        }

        /**
         * Wrapper for what ever method we use to get message contents
         * @since 1.17
         * @throws MWException
         * @return string
         */
        protected function fetchMessage() {
                if ( !isset( $this->message ) ) {
                        $cache = MessageCache::singleton();
                        if ( is_array( $this->key ) ) {
                                if ( !count( $this->key ) ) {
                                        throw new MWException( "Given empty message key array." );
                                }
                                foreach ( $this->key as $key ) {
                                        $message = $cache->get( $key, $this->useDatabase, $this->language );
                                        if ( $message !== false && $message !== '' ) {
                                                break;
                                        }
                                }
                                $this->message = $message;
                        } else {
                                $this->message = $cache->get( $this->key, $this->useDatabase, $this->language );
                        }
                }
                return $this->message;
        }

}

/**
 * Variant of the Message class.
 *
 * Rather than treating the message key as a lookup
 * value (which is passed to the MessageCache and
 * translated as necessary), a RawMessage key is
 * treated as the actual message.
 *
 * All other functionality (parsing, escaping, etc.)
 * is preserved.
 *
 * @since 1.21
 */
class RawMessage extends Message {
        /**
         * Call the parent constructor, then store the key as
         * the message.
         *
         * @param string $key Message to use
         * @param array $params Parameters for the message
         * @see Message::__construct
         */
        public function __construct( $key, $params = array() ) {
                parent::__construct( $key, $params );
                // The key is the message.
                $this->message = $key;
        }

        /**
         * Fetch the message (in this case, the key).
         *
         * @return string
         */
        public function fetchMessage() {
                // Just in case the message is unset somewhere.
                if ( !isset( $this->message ) ) {
                        $this->message = $this->key;
                }
                return $this->message;
        }
}