]> scripts.mit.edu Git - autoinstallsdev/mediawiki.git/blobdiff - includes/Categoryfinder.php
MediaWiki 1.30.2
[autoinstallsdev/mediawiki.git] / includes / Categoryfinder.php
index 1f08b7f8378a9327ec362a230df4344a1f1d18a9..89bf5c7327058dd3bfd643d5b8ed5f82642aa7d7 100644 (file)
@@ -1,55 +1,87 @@
 <?php
 /**
- * The "Categoryfinder" class takes a list of articles, creates an internal
+ * Recent changes filtering by category.
+ *
+ * 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
+ */
+
+use Wikimedia\Rdbms\IDatabase;
+
+/**
+ * The "CategoryFinder" class takes a list of articles, creates an internal
  * representation of all their parent categories (as well as parents of
  * parents etc.). From this representation, it determines which of these
  * articles are in one or all of a given subset of categories.
  *
  * Example use :
- * <code>
- *     # Determines whether the article with the page_id 12345 is in both
- *     # "Category 1" and "Category 2" or their subcategories, respectively
- *
- *     $cf = new Categoryfinder;
- *     $cf->seed(
- *             array( 12345 ),
- *             array( 'Category 1', 'Category 2' ),
- *             'AND'
- *     );
- *     $a = $cf->run();
- *     print implode( ',' , $a );
- * </code>
+ * @code
+ *     # Determines whether the article with the page_id 12345 is in both
+ *     # "Category 1" and "Category 2" or their subcategories, respectively
  *
+ *     $cf = new CategoryFinder;
+ *     $cf->seed(
+ *         [ 12345 ],
+ *         [ 'Category 1', 'Category 2' ],
+ *         'AND'
+ *     );
+ *     $a = $cf->run();
+ *     print implode( ',' , $a );
+ * @endcode
  */
-class Categoryfinder {
-       var $articles = array(); # The original article IDs passed to the seed function
-       var $deadend = array(); # Array of DBKEY category names for categories that don't have a page
-       var $parents = array(); # Array of [ID => array()]
-       var $next = array(); # Array of article/category IDs
-       var $targets = array(); # Array of DBKEY category names
-       var $name2id = array();
-       var $mode; # "AND" or "OR"
-       var $dbr; # Read-DB slave
+class CategoryFinder {
+       /** @var int[] The original article IDs passed to the seed function */
+       protected $articles = [];
 
-       /**
-        * Constructor (currently empty).
-        */
-       function __construct() {
-       }
+       /** @var array Array of DBKEY category names for categories that don't have a page */
+       protected $deadend = [];
+
+       /** @var array Array of [ ID => [] ] */
+       protected $parents = [];
+
+       /** @var array Array of article/category IDs */
+       protected $next = [];
+
+       /** @var array Array of DBKEY category names */
+       protected $targets = [];
+
+       /** @var array */
+       protected $name2id = [];
+
+       /** @var string "AND" or "OR" */
+       protected $mode;
+
+       /** @var IDatabase Read-DB replica DB */
+       protected $dbr;
 
        /**
         * Initializes the instance. Do this prior to calling run().
-        * @param $article_ids Array of article IDs
-        * @param $categories FIXME
-        * @param $mode String: FIXME, default 'AND'.
+        * @param array $articleIds Array of article IDs
+        * @param array $categories FIXME
+        * @param string $mode FIXME, default 'AND'.
+        * @todo FIXME: $categories/$mode
         */
-       function seed( $article_ids, $categories, $mode = 'AND' ) {
-               $this->articles = $article_ids;
-               $this->next = $article_ids;
+       public function seed( $articleIds, $categories, $mode = 'AND' ) {
+               $this->articles = $articleIds;
+               $this->next = $articleIds;
                $this->mode = $mode;
 
                # Set the list of target categories; convert them to DBKEY form first
-               $this->targets = array();
+               $this->targets = [];
                foreach ( $categories as $c ) {
                        $ct = Title::makeTitleSafe( NS_CATEGORY, $c );
                        if ( $ct ) {
@@ -62,16 +94,16 @@ class Categoryfinder {
        /**
         * Iterates through the parent tree starting with the seed values,
         * then checks the articles if they match the conditions
-        * @return array of page_ids (those given to seed() that match the conditions)
+        * @return array Array of page_ids (those given to seed() that match the conditions)
         */
-       function run() {
-               $this->dbr = wfGetDB( DB_SLAVE );
+       public function run() {
+               $this->dbr = wfGetDB( DB_REPLICA );
                while ( count( $this->next ) > 0 ) {
-                       $this->scan_next_layer();
+                       $this->scanNextLayer();
                }
 
                # Now check if this applies to the individual articles
-               $ret = array();
+               $ret = [];
 
                foreach ( $this->articles as $article ) {
                        $conds = $this->targets;
@@ -83,14 +115,22 @@ class Categoryfinder {
                return $ret;
        }
 
+       /**
+        * Get the parents. Only really useful if run() has been called already
+        * @return array
+        */
+       public function getParents() {
+               return $this->parents;
+       }
+
        /**
         * This functions recurses through the parent representation, trying to match the conditions
-        * @param $id The article/category to check
-        * @param $conds The array of categories to match
-        * @param $path used to check for recursion loops
+        * @param int $id The article/category to check
+        * @param array $conds The array of categories to match
+        * @param array $path Used to check for recursion loops
         * @return bool Does this match the conditions?
         */
-       function check( $id, &$conds, $path = array() ) {
+       private function check( $id, &$conds, $path = [] ) {
                // Check for loops and stop!
                if ( in_array( $id, $path ) ) {
                        return false;
@@ -98,7 +138,7 @@ class Categoryfinder {
 
                $path[] = $id;
 
-               # Shortcut (runtime paranoia): No contitions=all matched
+               # Shortcut (runtime paranoia): No conditions=all matched
                if ( count( $conds ) == 0 ) {
                        return true;
                }
@@ -109,14 +149,14 @@ class Categoryfinder {
 
                # iterate through the parents
                foreach ( $this->parents[$id] as $p ) {
-                       $pname = $p->cl_to ;
+                       $pname = $p->cl_to;
 
                        # Is this a condition?
                        if ( isset( $conds[$pname] ) ) {
                                # This key is in the category list!
                                if ( $this->mode == 'OR' ) {
                                        # One found, that's enough!
-                                       $conds = array();
+                                       $conds = [];
                                        return true;
                                } else {
                                        # Assuming "AND" as default
@@ -145,13 +185,13 @@ class Categoryfinder {
        /**
         * Scans a "parent layer" of the articles/categories in $this->next
         */
-       function scan_next_layer() {
+       private function scanNextLayer() {
                # Find all parents of the article currently in $this->next
-               $layer = array();
+               $layer = [];
                $res = $this->dbr->select(
                        /* FROM   */ 'categorylinks',
                        /* SELECT */ '*',
-                       /* WHERE  */ array( 'cl_from' => $this->next ),
+                       /* WHERE  */ [ 'cl_from' => $this->next ],
                        __METHOD__ . '-1'
                );
                foreach ( $res as $o ) {
@@ -159,7 +199,7 @@ class Categoryfinder {
 
                        # Update parent tree
                        if ( !isset( $this->parents[$o->cl_from] ) ) {
-                               $this->parents[$o->cl_from] = array();
+                               $this->parents[$o->cl_from] = [];
                        }
                        $this->parents[$o->cl_from][$k] = $o;
 
@@ -176,14 +216,14 @@ class Categoryfinder {
                        $layer[$k] = $k;
                }
 
-               $this->next = array();
+               $this->next = [];
 
                # Find the IDs of all category pages in $layer, if they exist
                if ( count( $layer ) > 0 ) {
                        $res = $this->dbr->select(
                                /* FROM   */ 'page',
-                               /* SELECT */ array( 'page_id', 'page_title' ),
-                               /* WHERE  */ array( 'page_namespace' => NS_CATEGORY , 'page_title' => $layer ),
+                               /* SELECT */ [ 'page_id', 'page_title' ],
+                               /* WHERE  */ [ 'page_namespace' => NS_CATEGORY, 'page_title' => $layer ],
                                __METHOD__ . '-2'
                        );
                        foreach ( $res as $o ) {
@@ -200,5 +240,4 @@ class Categoryfinder {
                        $this->deadend[$v] = $v;
                }
        }
-
 }