Added new hook 'MessageCacheReplace' into MessageCache.php. For instance
[lhc/web/wiklou.git] / includes / MessageCache.php
index 39e6410..185b617 100644 (file)
@@ -25,7 +25,7 @@ class MessageCache {
        var $mKeys, $mParserOptions, $mParser;
        var $mExtensionMessages = array();
        var $mInitialised = false;
-       var $mAllMessagesLoaded; // Extension messages
+       var $mAllMessagesLoaded = array(); // Extension messages
 
        // Variable for tracking which variables are loaded
        var $mLoadedLanguages = array();
@@ -44,7 +44,6 @@ class MessageCache {
 
        /**
         * ParserOptions is lazy initialised.
-        * Access should probably be protected.
         */
        function getParserOptions() {
                if ( !$this->mParserOptions ) {
@@ -110,7 +109,7 @@ class MessageCache {
                global $wgLocalMessageCache;
 
                $filename = "$wgLocalMessageCache/messages-" . wfWikiID() . "-$code";
-               wfMkdirParents( $wgLocalMessageCache, 0777 ); // might fail
+               wfMkdirParents( $wgLocalMessageCache ); // might fail
 
                wfSuppressWarnings();
                $file = fopen( $filename, 'w' );
@@ -131,7 +130,7 @@ class MessageCache {
 
                $filename = "$wgLocalMessageCache/messages-" . wfWikiID() . "-$code";
                $tempFilename = $filename . '.tmp';
-       wfMkdirParents( $wgLocalMessageCache, 0777 ); // might fail
+               wfMkdirParents( $wgLocalMessageCache ); // might fail
 
                wfSuppressWarnings();
                $file = fopen( $tempFilename, 'w');
@@ -261,12 +260,23 @@ class MessageCache {
 
                        $this->lock($cacheKey);
 
-                       $cache = $this->loadFromDB( $code );
-                       $success = $this->setCache( $cache, $code );
+                       # Limit the concurrency of loadFromDB to a single process 
+                       # This prevents the site from going down when the cache expires
+                       $statusKey = wfMemcKey( 'messages', $code, 'status' );
+                       $success = $this->mMemc->add( $statusKey, 'loading', MSG_LOAD_TIMEOUT );
                        if ( $success ) {
-                               $this->saveToCaches( $cache, true, $code );
+                               $cache = $this->loadFromDB( $code );
+                               $success = $this->setCache( $cache, $code );
+                       }
+                       if ( $success ) {
+                               $success = $this->saveToCaches( $cache, true, $code );
+                               if ( $success ) {
+                                       $this->mMemc->delete( $statusKey );
+                               } else {
+                                       $this->mMemc->set( $statusKey, 'error', 60*5 );
+                                       wfDebug( "MemCached set error in MessageCache: restart memcached server!\n" );
+                               }
                        }
-
                        $this->unlock($cacheKey);
                }
 
@@ -397,6 +407,8 @@ class MessageCache {
                $sidebarKey = wfMemcKey( 'sidebar', $code );
                $parserMemc->delete( $sidebarKey );
 
+               wfRunHooks( "MessageCacheReplace", array( $title, $text ) );
+
                wfProfileOut( __METHOD__ );
        }
 
@@ -414,10 +426,6 @@ class MessageCache {
                global $wgLocalMessageCache, $wgLocalMessageCacheSerialized;
 
                $cacheKey = wfMemcKey( 'messages', $code );
-               $statusKey = wfMemcKey( 'messages', $code, 'status' );
-
-               $success = $this->mMemc->add( $statusKey, 'loading', MSG_LOAD_TIMEOUT );
-               if ( !$success ) return true; # Other process should be updating them now
 
                $i = 0;
                if ( $memc ) {
@@ -444,11 +452,8 @@ class MessageCache {
                }
 
                if ( $i == 20 ) {
-                       $this->mMemc->set( $statusKey, 'error', 60*5 );
-                       wfDebug( "MemCached set error in MessageCache: restart memcached server!\n" );
                        $success = false;
                } else {
-                       $this->mMemc->delete( $statusKey );
                        $success = true;
                }
                wfProfileOut( __METHOD__ );
@@ -644,23 +649,30 @@ class MessageCache {
                return $message;
        }
 
-       function transform( $message, $interface = false ) {
+       function transform( $message, $interface = false, $language = null ) {
                // Avoid creating parser if nothing to transfrom
                if( strpos( $message, '{{' ) === false ) {
                        return $message;
                }
 
-               global $wgParser;
+               global $wgParser, $wgParserConf;
                if ( !$this->mParser && isset( $wgParser ) ) {
                        # Do some initialisation so that we don't have to do it twice
                        $wgParser->firstCallInit();
                        # Clone it and store it
-                       $this->mParser = clone $wgParser;
+                       $class = $wgParserConf['class'];
+                       if ( $class == 'Parser_DiffTest' ) {
+                               # Uncloneable
+                               $this->mParser = new $class( $wgParserConf );
+                       } else {
+                               $this->mParser = clone $wgParser;
+                       }
                        #wfDebug( __METHOD__ . ": following contents triggered transform: $message\n" );
                }
                if ( $this->mParser ) {
                        $popts = $this->getParserOptions();
                        $popts->setInterfaceMessage( $interface );
+                       $popts->setTargetLanguage( $language );
                        $message = $this->mParser->transformMsg( $message, $popts );
                }
                return $message;
@@ -692,7 +704,10 @@ class MessageCache {
         * @param string $lang The messages language, English by default
         */
        function addMessage( $key, $value, $lang = 'en' ) {
-               $this->mExtensionMessages[$lang][$key] = $value;
+               global $wgContLang;
+               # Normalise title-case input
+               $lckey = str_replace( ' ', '_', $wgContLang->lcfirst( $key ) );
+               $this->mExtensionMessages[$lang][$lckey] = $value;
        }
 
        /**
@@ -761,12 +776,13 @@ class MessageCache {
                }
        }
 
-       function loadAllMessages() {
+       function loadAllMessages( $lang = false ) {
                global $wgExtensionMessagesFiles;
-               if ( $this->mAllMessagesLoaded ) {
+               $key = $lang === false ? '*' : $lang;
+               if ( isset( $this->mAllMessagesLoaded[$key] ) ) {
                        return;
                }
-               $this->mAllMessagesLoaded = true;
+               $this->mAllMessagesLoaded[$key] = true;
 
                # Some extensions will load their messages when you load their class file
                wfLoadAllExtensions();
@@ -774,7 +790,7 @@ class MessageCache {
                wfRunHooks( 'LoadAllMessages' );
                # Some register their messages in $wgExtensionMessagesFiles
                foreach ( $wgExtensionMessagesFiles as $name => $file ) {
-                       wfLoadExtensionMessages( $name );
+                       wfLoadExtensionMessages( $name, $lang );
                }
                # Still others will respond to neither, they are EVIL. We sometimes need to know!
        }
@@ -789,6 +805,7 @@ class MessageCache {
         */
        function loadMessagesFile( $filename, $langcode = false ) {
                global $wgLang, $wgContLang;
+               wfProfileIn( __METHOD__ );
                $messages = $magicWords = false;
                require( $filename );
 
@@ -811,6 +828,7 @@ class MessageCache {
                        global $wgContLang;
                        $wgContLang->addMagicWordsByLang( $magicWords );
                }
+               wfProfileOut( __METHOD__ );
        }
 
        /**
@@ -820,6 +838,7 @@ class MessageCache {
         * @param string $langcode Language code to process.
         */
        function processMessagesArray( $messages, $langcode ) {
+               wfProfileIn( __METHOD__ );
                $fallbackCode = $langcode;
                $mergedMessages = array();
                do {
@@ -831,17 +850,22 @@ class MessageCache {
                
                if ( !empty($mergedMessages) )
                        $this->addMessages( $mergedMessages, $langcode );
+               wfProfileOut( __METHOD__ );
        }
 
        public function figureMessage( $key ) {
                global $wgContLanguageCode;
-               $pieces = explode('/', $key, 2);
+               $pieces = explode( '/', $key );
+               if( count( $pieces ) < 2 )
+                       return array( $key, $wgContLanguageCode );
 
-               $key = $pieces[0];
+               $lang = array_pop( $pieces );
+               $validCodes = Language::getLanguageNames();
+               if( !array_key_exists( $lang, $validCodes ) )
+                       return array( $key, $wgContLanguageCode );
 
-               # Language the user is translating to
-               $langCode = isset($pieces[1]) ? $pieces[1] : $wgContLanguageCode;
-               return array( $key, $langCode );
+               $message = implode( '/', $pieces );
+               return array( $message, $lang );
        }
 
 }