Move rewRevisionFromEditComplete call down till after page rows set
[lhc/web/wiklou.git] / includes / ParserOutput.php
index 0fd98b2..f98d564 100644 (file)
@@ -1,6 +1,7 @@
 <?php
 /**
  * @todo document
+ * @ingroup Parser
  */
 class ParserOutput
 {
@@ -13,12 +14,21 @@ class ParserOutput
                $mTitleText,        # title text of the chosen language variant
                $mLinks,            # 2-D map of NS/DBK to ID for the links in the document. ID=zero for broken.
                $mTemplates,        # 2-D map of NS/DBK to ID for the template references. ID=zero for broken.
+               $mTemplateIds,      # 2-D map of NS/DBK to rev ID for the template references. ID=zero for broken.
                $mImages,           # DB keys of the images used, in the array key only
                $mExternalLinks,    # External link URLs, in the key only
-               $mHTMLtitle,            # Display HTML title
-               $mSubtitle,                     # Additional subtitle
-               $mNewSection,           # Show a new section link?
-               $mNoGallery;            # No gallery on category page? (__NOGALLERY__)
+               $mNewSection,       # Show a new section link?
+               $mNoGallery,        # No gallery on category page? (__NOGALLERY__)
+               $mHeadItems,        # Items to put in the <head> section
+               $mOutputHooks,      # Hook tags as per $wgParserOutputHooks
+               $mWarnings,         # Warning text to be returned to the user. Wikitext formatted, in the key only
+               $mSections,         # Table of contents
+               $mProperties;       # Name/value pairs to be cached in the DB
+
+       /**
+        * Overridden title for display
+        */
+       private $displayTitle = false;
 
        function ParserOutput( $text = '', $languageLinks = array(), $categoryLinks = array(),
                $containsOldMagic = false, $titletext = '' )
@@ -28,30 +38,37 @@ class ParserOutput
                $this->mCategories = $categoryLinks;
                $this->mContainsOldMagic = $containsOldMagic;
                $this->mCacheTime = '';
-               $this->mVersion = MW_PARSER_VERSION;
+               $this->mVersion = Parser::VERSION;
                $this->mTitleText = $titletext;
+               $this->mSections = array();
                $this->mLinks = array();
                $this->mTemplates = array();
                $this->mImages = array();
                $this->mExternalLinks = array();
-               $this->mHTMLtitle = "" ;
-               $this->mSubtitle = "" ;
                $this->mNewSection = false;
                $this->mNoGallery = false;
+               $this->mHeadItems = array();
+               $this->mTemplateIds = array();
+               $this->mOutputHooks = array();
+               $this->mWarnings = array();
+               $this->mProperties = array();
        }
 
        function getText()                   { return $this->mText; }
-       function &getLanguageLinks()          { return $this->mLanguageLinks; }
+       function &getLanguageLinks()         { return $this->mLanguageLinks; }
        function getCategoryLinks()          { return array_keys( $this->mCategories ); }
        function &getCategories()            { return $this->mCategories; }
        function getCacheTime()              { return $this->mCacheTime; }
        function getTitleText()              { return $this->mTitleText; }
+       function getSections()               { return $this->mSections; }
        function &getLinks()                 { return $this->mLinks; }
        function &getTemplates()             { return $this->mTemplates; }
        function &getImages()                { return $this->mImages; }
        function &getExternalLinks()         { return $this->mExternalLinks; }
        function getNoGallery()              { return $this->mNoGallery; }
        function getSubtitle()               { return $this->mSubtitle; }
+       function getOutputHooks()            { return (array)$this->mOutputHooks; }
+       function getWarnings()               { return array_keys( $this->mWarnings ); }
 
        function containsOldMagic()          { return $this->mContainsOldMagic; }
        function setText( $text )            { return wfSetVar( $this->mText, $text ); }
@@ -59,13 +76,17 @@ class ParserOutput
        function setCategoryLinks( $cl )     { return wfSetVar( $this->mCategories, $cl ); }
        function setContainsOldMagic( $com ) { return wfSetVar( $this->mContainsOldMagic, $com ); }
        function setCacheTime( $t )          { return wfSetVar( $this->mCacheTime, $t ); }
-       function setTitleText( $t )          { return wfSetVar($this->mTitleText, $t); }
-       function setSubtitle( $st )          { return wfSetVar( $this->mSubtitle, $st ); }
+       function setTitleText( $t )          { return wfSetVar( $this->mTitleText, $t ); }
+       function setSections( $toc )         { return wfSetVar( $this->mSections, $toc ); }
 
        function addCategory( $c, $sort )    { $this->mCategories[$c] = $sort; }
-       function addImage( $name )           { $this->mImages[$name] = 1; }
        function addLanguageLink( $t )       { $this->mLanguageLinks[] = $t; }
        function addExternalLink( $url )     { $this->mExternalLinks[$url] = 1; }
+       function addWarning( $s )            { $this->mWarnings[$s] = 1; }
+
+       function addOutputHook( $hook, $data = false ) {
+               $this->mOutputHooks[] = array( $hook, $data );
+       }
 
        function setNewSection( $value ) {
                $this->mNewSection = (bool)$value;
@@ -86,13 +107,21 @@ class ParserOutput
                $this->mLinks[$ns][$dbk] = $id;
        }
 
-       function addTemplate( $title, $id ) {
+       function addImage( $name ) {
+               $this->mImages[$name] = 1;
+       }
+
+       function addTemplate( $title, $page_id, $rev_id ) {
                $ns = $title->getNamespace();
                $dbk = $title->getDBkey();
                if ( !isset( $this->mTemplates[$ns] ) ) {
                        $this->mTemplates[$ns] = array();
                }
-               $this->mTemplates[$ns][$dbk] = $id;
+               $this->mTemplates[$ns][$dbk] = $page_id;
+               if ( !isset( $this->mTemplateIds[$ns] ) ) {
+                       $this->mTemplateIds[$ns] = array();
+               }
+               $this->mTemplateIds[$ns][$dbk] = $rev_id; // For versioning
        }
 
        /**
@@ -110,8 +139,68 @@ class ParserOutput
                       $this->getCacheTime() < $touched ||
                       $this->getCacheTime() <= $wgCacheEpoch ||
                       !isset( $this->mVersion ) ||
-                      version_compare( $this->mVersion, MW_PARSER_VERSION, "lt" );
+                      version_compare( $this->mVersion, Parser::VERSION, "lt" );
+       }
+
+       /**
+        * Add some text to the <head>.
+        * If $tag is set, the section with that tag will only be included once
+        * in a given page.
+        */
+       function addHeadItem( $section, $tag = false ) {
+               if ( $tag !== false ) {
+                       $this->mHeadItems[$tag] = $section;
+               } else {
+                       $this->mHeadItems[] = $section;
+               }
+       }
+
+       /**
+        * Override the title to be used for display
+        * -- this is assumed to have been validated
+        * (check equal normalisation, etc.)
+        *
+        * @param string $text Desired title text
+        */
+       public function setDisplayTitle( $text ) {
+               $this->displayTitle = $text;
        }
-}
 
-?>
+       /**
+        * Get the title to be used for display
+        *
+        * @return string
+        */
+       public function getDisplayTitle() {
+               return $this->displayTitle;
+       }
+
+       /**
+        * Fairly generic flag setter thingy.
+        */
+       public function setFlag( $flag ) {
+               $this->mFlags[$flag] = true;
+       }
+
+       public function getFlag( $flag ) {
+               return isset( $this->mFlags[$flag] );
+       }
+
+       /**
+        * Set a property to be cached in the DB
+        */
+       public function setProperty( $name, $value ) {
+               $this->mProperties[$name] = $value;
+       }
+
+       public function getProperty( $name ){
+               return isset( $this->mProperties[$name] ) ? $this->mProperties[$name] : false;
+       }
+
+       public function getProperties() {
+               if ( !isset( $this->mProperties ) ) {
+                       $this->mProperties = array();
+               }
+               return $this->mProperties;
+       }
+}