* (bug 24563) Entries on Special:WhatLinksHere now have a link to their history
[lhc/web/wiklou.git] / includes / Category.php
index f386bee..44f266e 100644 (file)
@@ -1,6 +1,8 @@
 <?php
 /**
- * Category objects are immutable, strictly speaking.  If you call methods that change the database, like to refresh link counts, the objects will be appropriately reinitialized.  Member variables are lazy-initialized.
+ * Category objects are immutable, strictly speaking. If you call methods that change the database,
+ * like to refresh link counts, the objects will be appropriately reinitialized.
+ * Member variables are lazy-initialized.
  *
  * TODO: Move some stuff from CategoryPage.php to here, and use that.
  *
@@ -11,39 +13,52 @@ class Category {
        /** Name of the category, normalized to DB-key form */
        private $mName = null;
        private $mID = null;
+       /** Category page title */
+       private $mTitle = null;
        /** Counts of membership (cat_pages, cat_subcats, cat_files) */
        private $mPages = null, $mSubcats = null, $mFiles = null;
 
-       private function __construct() {}
+       private function __construct() { }
 
        /**
         * Set up all member variables using a database query.
         * @return bool True on success, false on failure.
         */
        protected function initialize() {
-               if( $this->mName === null && $this->mID === null ) {
-                       throw new MWException( __METHOD__.' has both names and IDs null' );
-               }
-               $dbr = wfGetDB( DB_SLAVE );
-               if( $this->mID === null ) {
+               if ( $this->mName === null && $this->mID === null ) {
+                       throw new MWException( __METHOD__ . ' has both names and IDs null' );
+               } elseif ( $this->mID === null ) {
                        $where = array( 'cat_title' => $this->mName );
-               } elseif( $this->mName === null ) {
+               } elseif ( $this->mName === null ) {
                        $where = array( 'cat_id' => $this->mID );
                } else {
                        # Already initialized
                        return true;
                }
+               $dbr = wfGetDB( DB_SLAVE );
                $row = $dbr->selectRow(
                        'category',
-                       array( 'cat_id', 'cat_title', 'cat_pages', 'cat_subcats',
-                               'cat_files' ),
+                       array( 'cat_id', 'cat_title', 'cat_pages', 'cat_subcats', 'cat_files' ),
                        $where,
                        __METHOD__
                );
-               if( !$row ) {
+
+               if ( !$row ) {
                        # Okay, there were no contents.  Nothing to initialize.
-                       return false;
+                       if ( $this->mTitle ) {
+                               # If there is a title object but no record in the category table, treat this as an empty category
+                               $this->mID      = false;
+                               $this->mName    = $this->mTitle->getDBkey();
+                               $this->mPages   = 0;
+                               $this->mSubcats = 0;
+                               $this->mFiles   = 0;
+
+                               return true;
+                       } else {
+                               return false; # Fail
+                       }
                }
+
                $this->mID      = $row->cat_id;
                $this->mName    = $row->cat_title;
                $this->mPages   = $row->cat_pages;
@@ -53,8 +68,7 @@ class Category {
                # (bug 13683) If the count is negative, then 1) it's obviously wrong
                # and should not be kept, and 2) we *probably* don't have to scan many
                # rows to obtain the correct figure, so let's risk a one-time recount.
-               if( $this->mPages < 0 || $this->mSubcats < 0 ||
-               $this->mFiles < 0 ) {
+               if ( $this->mPages < 0 || $this->mSubcats < 0 || $this->mFiles < 0 ) {
                        $this->refreshCounts();
                }
 
@@ -64,17 +78,35 @@ class Category {
        /**
         * Factory function.
         *
-        * @param array $name A category name (no "Category:" prefix).  It need
+        * @param $name Array: A category name (no "Category:" prefix).  It need
         *   not be normalized, with spaces replaced by underscores.
         * @return mixed Category, or false on a totally invalid name
         */
        public static function newFromName( $name ) {
                $cat = new self();
-               $title = Title::newFromText( "Category:$name" );
-               if( !is_object( $title ) ) {
+               $title = Title::makeTitleSafe( NS_CATEGORY, $name );
+
+               if ( !is_object( $title ) ) {
                        return false;
                }
-               $cat->mName = $title->getDBKey();
+
+               $cat->mTitle = $title;
+               $cat->mName = $title->getDBkey();
+
+               return $cat;
+       }
+
+       /**
+        * Factory function.
+        *
+        * @param $title Title for the category page
+        * @return Mixed: category, or false on a totally invalid name
+        */
+       public static function newFromTitle( $title ) {
+               $cat = new self();
+
+               $cat->mTitle = $title;
+               $cat->mName = $title->getDBkey();
 
                return $cat;
        }
@@ -82,7 +114,7 @@ class Category {
        /**
         * Factory function.
         *
-        * @param array $id A category id
+        * @param $id Integer: a category id
         * @return Category
         */
        public static function newFromID( $id ) {
@@ -91,14 +123,61 @@ class Category {
                return $cat;
        }
 
+       /**
+        * Factory function, for constructing a Category object from a result set
+        *
+        * @param $row result set row, must contain the cat_xxx fields. If the fields are null,
+        *        the resulting Category object will represent an empty category if a title object
+        *        was given. If the fields are null and no title was given, this method fails and returns false.
+        * @param $title optional title object for the category represented by the given row.
+        *        May be provided if it is already known, to avoid having to re-create a title object later.
+        * @return Category
+        */
+       public static function newFromRow( $row, $title = null ) {
+               $cat = new self();
+               $cat->mTitle = $title;
+
+               # NOTE: the row often results from a LEFT JOIN on categorylinks. This may result in
+               #       all the cat_xxx fields being null, if the category page exists, but nothing
+               #       was ever added to the category. This case should be treated linke an empty
+               #       category, if possible.
+
+               if ( $row->cat_title === null ) {
+                       if ( $title === null ) {
+                               # the name is probably somewhere in the row, for example as page_title,
+                               # but we can't know that here...
+                               return false;
+                       } else {
+                               $cat->mName = $title->getDBkey(); # if we have a title object, fetch the category name from there
+                       }
+
+                       $cat->mID =   false;
+                       $cat->mSubcats = 0;
+                       $cat->mPages   = 0;
+                       $cat->mFiles   = 0;
+               } else {
+                       $cat->mName    = $row->cat_title;
+                       $cat->mID      = $row->cat_id;
+                       $cat->mSubcats = $row->cat_subcats;
+                       $cat->mPages   = $row->cat_pages;
+                       $cat->mFiles   = $row->cat_files;
+               }
+
+               return $cat;
+       }
+
        /** @return mixed DB key name, or false on failure */
        public function getName() { return $this->getX( 'mName' ); }
+
        /** @return mixed Category ID, or false on failure */
        public function getID() { return $this->getX( 'mID' ); }
+
        /** @return mixed Total number of member pages, or false on failure */
        public function getPageCount() { return $this->getX( 'mPages' ); }
+
        /** @return mixed Number of subcategories, or false on failure */
        public function getSubcatCount() { return $this->getX( 'mSubcats' ); }
+
        /** @return mixed Number of member files, or false on failure */
        public function getFileCount() { return $this->getX( 'mFiles' ); }
 
@@ -106,18 +185,55 @@ class Category {
         * @return mixed The Title for this category, or false on failure.
         */
        public function getTitle() {
-               if( !$this->initialize() ) {
+               if ( $this->mTitle ) return $this->mTitle;
+
+               if ( !$this->initialize() ) {
                        return false;
                }
-               return Title::makeTitleSafe( NS_CATEGORY, $this->mName );
+
+               $this->mTitle = Title::makeTitleSafe( NS_CATEGORY, $this->mName );
+               return $this->mTitle;
+       }
+
+       /**
+        * Fetch a TitleArray of up to $limit category members, beginning after the
+        * category sort key $offset.
+        * @param $limit integer
+        * @param $offset string
+        * @return TitleArray object for category members.
+        */
+       public function getMembers( $limit = false, $offset = '' ) {
+               $dbr = wfGetDB( DB_SLAVE );
+
+               $conds = array( 'cl_to' => $this->getName(), 'cl_from = page_id' );
+               $options = array( 'ORDER BY' => 'cl_sortkey' );
+
+               if ( $limit ) {
+                       $options[ 'LIMIT' ] = $limit;
+               }
+
+               if ( $offset !== '' ) {
+                       $conds[] = 'cl_sortkey > ' . $dbr->addQuotes( $offset );
+               }
+
+               return TitleArray::newFromResult(
+                       $dbr->select(
+                               array( 'page', 'categorylinks' ),
+                               array( 'page_id', 'page_namespace', 'page_title', 'page_len',
+                                       'page_is_redirect', 'page_latest' ),
+                               $conds,
+                               __METHOD__,
+                               $options
+                       )
+               );
        }
 
        /** Generic accessor */
        private function getX( $key ) {
-               if( !$this->initialize() ) {
+               if ( !$this->initialize() ) {
                        return false;
                }
-               return $this->{$key};
+               return $this-> { $key } ;
        }
 
        /**
@@ -126,29 +242,33 @@ class Category {
         * @return bool True on success, false on failure
         */
        public function refreshCounts() {
-               if( wfReadOnly() ) {
+               if ( wfReadOnly() ) {
                        return false;
                }
                $dbw = wfGetDB( DB_MASTER );
                $dbw->begin();
                # Note, we must use names for this, since categorylinks does.
-               if( $this->mName === null ) {
-                       if( !$this->initialize() ) {
+               if ( $this->mName === null ) {
+                       if ( !$this->initialize() ) {
                                return false;
                        }
                } else {
                        # Let's be sure that the row exists in the table.  We don't need to
                        # do this if we got the row from the table in initialization!
+                       $seqVal = $dbw->nextSequenceValue( 'category_cat_id_seq' );
                        $dbw->insert(
                                'category',
-                               array( 'cat_title' => $this->mName ),
+                               array(
+                                       'cat_id' => $seqVal,
+                                       'cat_title' => $this->mName
+                               ),
                                __METHOD__,
                                'IGNORE'
                        );
                }
 
-               $cond1 = $dbw->conditional( 'page_namespace='.NS_CATEGORY, 1, 'NULL' );
-               $cond2 = $dbw->conditional( 'page_namespace='.NS_IMAGE, 1, 'NULL' );
+               $cond1 = $dbw->conditional( 'page_namespace=' . NS_CATEGORY, 1, 'NULL' );
+               $cond2 = $dbw->conditional( 'page_namespace=' . NS_FILE, 1, 'NULL' );
                $result = $dbw->selectRow(
                        array( 'categorylinks', 'page' ),
                        array( 'COUNT(*) AS pages',