(bug 5167) Add {{SUBPAGENAME}} variable
[lhc/web/wiklou.git] / includes / LinksUpdate.php
index be61ebd..e997724 100644 (file)
@@ -15,13 +15,14 @@ class LinksUpdate {
         */
        var $mId,            # Page ID of the article linked from
                $mTitle,         # Title object of the article linked from
-               $mParserOutput,  # Parser output containing the links to be inserted into the database
                $mLinks,         # Map of title strings to IDs for the links in the document
                $mImages,        # DB keys of the images used, in the array key only
                $mTemplates,     # Map of title strings to IDs for the template references, including broken ones
+               $mExternals,     # URLs of external links, array key only
                $mCategories,    # Map of category names to sort keys
                $mDb,            # Database connection reference
-               $mOptions;       # SELECT options to be used (array)
+               $mOptions,       # SELECT options to be used (array)
+               $mRecursive;     # Whether to queue jobs for recursive updates
        /**#@-*/
 
        /**
@@ -30,7 +31,7 @@ class LinksUpdate {
         * @param integer $id
         * @param string $title
         */
-       function LinksUpdate( $title, $parserOutput ) {
+       function LinksUpdate( $title, $parserOutput, $recursive = true ) {
                global $wgAntiLockFlags;
 
                if ( $wgAntiLockFlags & ALF_NO_LINK_LOCK ) {
@@ -46,13 +47,13 @@ class LinksUpdate {
                }
                $this->mTitle = $title;
                $this->mId = $title->getArticleID();
-               $this->mParserOutput = $parserOutput;
 
-               // Shortcut aliases
-               $this->mLinks =& $this->mParserOutput->getLinks();
-               $this->mImages =& $this->mParserOutput->getImages();
-               $this->mTemplates =& $this->mParserOutput->getTemplates();
-               $this->mCategories =& $this->mParserOutput->getCategories();
+               $this->mLinks = $parserOutput->getLinks();
+               $this->mImages = $parserOutput->getImages();
+               $this->mTemplates = $parserOutput->getTemplates();
+               $this->mExternals = $parserOutput->getExternalLinks();
+               $this->mCategories = $parserOutput->getCategories();
+               $this->mRecursive = $recursive;
 
        }
 
@@ -77,24 +78,40 @@ class LinksUpdate {
                $this->incrTableUpdate( 'pagelinks', 'pl', $this->getLinkDeletions( $existing ),
                        $this->getLinkInsertions( $existing ) );
 
+               # Image links
+               $existing = $this->getExistingImages();
+               $this->incrTableUpdate( 'imagelinks', 'il', $this->getImageDeletions( $existing ),
+                       $this->getImageInsertions( $existing ) );
+
+               # Invalidate all image description pages which had links added or removed
+               $imageUpdates = array_diff_key( $existing, $this->mImages ) + array_diff_key( $this->mImages, $existing );
+               $this->invalidateImageDescriptions( $imageUpdates );
+
+               # External links
+               $existing = $this->getExistingExternals();
+               $this->incrTableUpdate( 'externallinks', 'el', $this->getExternalDeletions( $existing ),
+                       $this->getExternalInsertions( $existing ) );
+
                # Template links
                $existing = $this->getExistingTemplates();
                $this->incrTableUpdate( 'templatelinks', 'tl', $this->getTemplateDeletions( $existing ),
                        $this->getTemplateInsertions( $existing ) );
 
-               # Image links
-               $existing = $this->getExistingImages();
-               $this->incrTableUpdate( 'imagelinks', 'il', $this->getImageDeletions( $existing ),
-                       $this->getImageInsertions( $existing ) );
+               # Refresh links of all pages including this page
+               if ( $this->mRecursive ) {
+                       $tlto = $this->mTitle->getTemplateLinksTo();
+                       if ( count( $tlto ) ) {
+                               require_once( 'JobQueue.php' );
+                               Job::queueLinksJobs( $tlto );
+                       }
+               }
 
                # Category links
                $existing = $this->getExistingCategories();
                $this->incrTableUpdate( 'categorylinks', 'cl', $this->getCategoryDeletions( $existing ),
                        $this->getCategoryInsertions( $existing ) );
 
-               # I think this works out to a set XOR operation, the idea is to invalidate all
-               # categories which were added, deleted or changed
-               # FIXME: surely there's a more appropriate place to put this update?
+               # Invalidate all categories which were added, deleted or changed (set symmetric difference)
                $categoryUpdates = array_diff_assoc( $existing, $this->mCategories ) + array_diff_assoc( $this->mCategories, $existing );
                $this->invalidateCategories( $categoryUpdates );
 
@@ -110,16 +127,30 @@ class LinksUpdate {
                $fname = 'LinksUpdate::doDumbUpdate';
                wfProfileIn( $fname );
 
+               # Refresh category pages and image description pages
                $existing = $this->getExistingCategories();
                $categoryUpdates = array_diff_assoc( $existing, $this->mCategories ) + array_diff_assoc( $this->mCategories, $existing );
+               $existing = $this->getExistingImages();
+               $imageUpdates = array_diff_key( $existing, $this->mImages ) + array_diff_key( $this->mImages, $existing );
+
+               # Refresh links of all pages including this page
+               if ( $this->mRecursive ) {
+                       $tlto = $this->mTitle->getTemplateLinksTo();
+                       if ( count( $tlto ) ) {
+                               require_once( 'JobQueue.php' );
+                               Job::queueLinksJobs( $tlto );
+                       }
+               }
 
                $this->dumbTableUpdate( 'pagelinks',     $this->getLinkInsertions(),     'pl_from' );
                $this->dumbTableUpdate( 'imagelinks',    $this->getImageInsertions(),    'il_from' );
                $this->dumbTableUpdate( 'categorylinks', $this->getCategoryInsertions(), 'cl_from' );
                $this->dumbTableUpdate( 'templatelinks', $this->getTemplateInsertions(), 'tl_from' );
+               $this->dumbTableUpdate( 'externallinks', $this->getExternalInsertions(), 'el_from' );
 
-               # Update the cache of all the category pages
+               # Update the cache of all the category pages and image description pages which were changed
                $this->invalidateCategories( $categoryUpdates );
+               $this->invalidateImageDescriptions( $imageUpdates );
 
                wfProfileOut( $fname );
        }
@@ -136,13 +167,25 @@ class LinksUpdate {
                }
        }
 
+       function invalidateImageDescriptions( $images ) {
+               $fname = 'LinksUpdate::invalidateImageDescriptions';
+               if ( count( $images ) ) {
+                       $this->mDb->update( 'page', array( 'page_touched' => $this->mDb->timestamp() ),
+                               array(
+                                       'page_namespace' => NS_IMAGE,
+                                       'page_title IN (' . $this->mDb->makeList( array_keys( $images ) ) . ')'
+                               ), $fname
+                       );
+               }
+       }
+
        function dumbTableUpdate( $table, $insertions, $fromField ) {
                $fname = 'LinksUpdate::dumbTableUpdate';
                $this->mDb->delete( $table, array( $fromField => $this->mId ), $fname );
                if ( count( $insertions ) ) {
                        # The link array was constructed without FOR UPDATE, so there may be collisions
-                       # Ignoring for now, I'm not sure if that causes problems or not, but I'm fairly
-                       # sure it's better than without IGNORE
+                       # This may cause minor link table inconsistencies, which is better than 
+                       # crippling the site with lock contention.
                        $this->mDb->insert( $table, $insertions, $fname, array( 'IGNORE' ) );
                }
        }
@@ -238,7 +281,7 @@ class LinksUpdate {
        function getImageInsertions( $existing = array() ) {
                $arr = array();
                $diffs = array_diff_key( $this->mImages, $existing );
-               foreach( $diffs as $iname => $val ) {
+               foreach( $diffs as $iname => $dummy ) {
                        $arr[] = array(
                                'il_from' => $this->mId,
                                'il_to'   => $iname
@@ -247,6 +290,23 @@ class LinksUpdate {
                return $arr;
        }
 
+       /**
+        * Get an array of externallinks insertions. Skips the names specified in $existing
+        * @access private
+        */
+       function getExternalInsertions( $existing = array() ) {
+               $arr = array();
+               $diffs = array_diff_key( $this->mExternals, $existing );
+               foreach( $diffs as $url => $dummy ) {
+                       $arr[] = array(
+                               'el_from'   => $this->mId,
+                               'el_to'     => $url,
+                               'el_index'  => wfMakeUrlIndex( $url ),
+                       );
+               }
+               return $arr;
+       }
+
        /**
         * Get an array of category insertions
         * @param array $existing Array mapping existing category names to sort keys. If both
@@ -260,7 +320,8 @@ class LinksUpdate {
                        $arr[] = array(
                                'cl_from'    => $this->mId,
                                'cl_to'      => $name,
-                               'cl_sortkey' => $sortkey
+                               'cl_sortkey' => $sortkey,
+                               'cl_timestamp' => $this->mDb->timestamp()
                        );
                }
                return $arr;
@@ -309,6 +370,15 @@ class LinksUpdate {
                return array_diff_key( $existing, $this->mImages );
        }
 
+       /** 
+        * Given an array of existing external links, returns those links which are not 
+        * in $this and thus should be deleted.
+        * @access private
+        */
+       function getExternalDeletions( $existing ) {
+               return array_diff_key( $existing, $this->mExternals );
+       }
+
        /**
         * Given an array of existing categories, returns those categories which are not in $this
         * and thus should be deleted.
@@ -333,6 +403,7 @@ class LinksUpdate {
                        }
                        $arr[$row->pl_namespace][$row->pl_title] = 1;
                }
+               $this->mDb->freeResult( $res );
                return $arr;
        }
 
@@ -351,6 +422,7 @@ class LinksUpdate {
                        }
                        $arr[$row->tl_namespace][$row->tl_title] = 1;
                }
+               $this->mDb->freeResult( $res );
                return $arr;
        }
 
@@ -366,6 +438,23 @@ class LinksUpdate {
                while ( $row = $this->mDb->fetchObject( $res ) ) {
                        $arr[$row->il_to] = 1;
                }
+               $this->mDb->freeResult( $res );
+               return $arr;
+       }
+
+       /**
+        * Get an array of existing external links, URLs in the keys
+        * @access private
+        */
+       function getExistingExternals() {
+               $fname = 'LinksUpdate::getExistingExternals';
+               $res = $this->mDb->select( 'externallinks', array( 'el_to' ),
+                       array( 'el_from' => $this->mId ), $fname, $this->mOptions );
+               $arr = array();
+               while ( $row = $this->mDb->fetchObject( $res ) ) {
+                       $arr[$row->el_to] = 1;
+               }
+               $this->mDb->freeResult( $res );
                return $arr;
        }
 
@@ -381,6 +470,7 @@ class LinksUpdate {
                while ( $row = $this->mDb->fetchObject( $res ) ) {
                        $arr[$row->cl_to] = $row->cl_sortkey;
                }
+               $this->mDb->freeResult( $res );
                return $arr;
        }
 }