wiki.techinc.nl/includes/CategoryFinder.php
James D. Forrester 9acd5508cf Drop the 'CategorizedRecentChanges' feature
It is off by default, it's very rarely used (GitHub finds four config
users setting it true), it has over time had fatals going unnoticed or
fixed for years. Let's just kill it.

Change-Id: Idac2341ed7ecf053ee84069df657e424e50b896c
2018-02-15 10:45:10 -08:00

262 lines
6.7 KiB
PHP

<?php
/**
* 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(
* [ 12345 ],
* [ 'Category 1', 'Category 2' ],
* 'AND'
* );
* $a = $cf->run();
* print implode( ',' , $a );
* @endcode
*
* @deprecated since 1.31
*/
class CategoryFinder {
/** @var int[] The original article IDs passed to the seed function */
protected $articles = [];
/** @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 int Max layer depth **/
protected $maxdepth = -1;
/** @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 array $articleIds Array of article IDs
* @param array $categories FIXME
* @param string $mode FIXME, default 'AND'.
* @param int $maxdepth Maximum layer depth. Where:
* -1 means deep recursion (default);
* 0 means no-parents;
* 1 means one parent layer, etc.
* @todo FIXME: $categories/$mode
*/
public function seed( $articleIds, $categories, $mode = 'AND', $maxdepth = -1 ) {
$this->articles = $articleIds;
$this->next = $articleIds;
$this->mode = $mode;
$this->maxdepth = $maxdepth;
# Set the list of target categories; convert them to DBKEY form first
$this->targets = [];
foreach ( $categories as $c ) {
$ct = Title::makeTitleSafe( NS_CATEGORY, $c );
if ( $ct ) {
$c = $ct->getDBkey();
$this->targets[$c] = $c;
}
}
}
/**
* Iterates through the parent tree starting with the seed values,
* then checks the articles if they match the conditions
* @return array Array of page_ids (those given to seed() that match the conditions)
*/
public function run() {
$this->dbr = wfGetDB( DB_REPLICA );
$i = 0;
$dig = true;
while ( count( $this->next ) && $dig ) {
$this->scanNextLayer();
// Is there any depth limit?
if ( $this->maxdepth !== -1 ) {
$dig = $i < $this->maxdepth;
$i++;
}
}
# Now check if this applies to the individual articles
$ret = [];
foreach ( $this->articles as $article ) {
$conds = $this->targets;
if ( $this->check( $article, $conds ) ) {
# Matches the conditions
$ret[] = $article;
}
}
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 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?
*/
private function check( $id, &$conds, $path = [] ) {
// Check for loops and stop!
if ( in_array( $id, $path ) ) {
return false;
}
$path[] = $id;
# Shortcut (runtime paranoia): No conditions=all matched
if ( count( $conds ) == 0 ) {
return true;
}
if ( !isset( $this->parents[$id] ) ) {
return false;
}
# iterate through the parents
foreach ( $this->parents[$id] as $p ) {
$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 = [];
return true;
} else {
# Assuming "AND" as default
unset( $conds[$pname] );
if ( count( $conds ) == 0 ) {
# All conditions met, done
return true;
}
}
}
# Not done yet, try sub-parents
if ( !isset( $this->name2id[$pname] ) ) {
# No sub-parent
continue;
}
$done = $this->check( $this->name2id[$pname], $conds, $path );
if ( $done || count( $conds ) == 0 ) {
# Subparents have done it!
return true;
}
}
return false;
}
/**
* Scans a "parent layer" of the articles/categories in $this->next
*/
private function scanNextLayer() {
# Find all parents of the article currently in $this->next
$layer = [];
$res = $this->dbr->select(
/* FROM */ 'categorylinks',
/* SELECT */ [ 'cl_to', 'cl_from' ],
/* WHERE */ [ 'cl_from' => $this->next ],
__METHOD__ . '-1'
);
foreach ( $res as $o ) {
$k = $o->cl_to;
# Update parent tree
if ( !isset( $this->parents[$o->cl_from] ) ) {
$this->parents[$o->cl_from] = [];
}
$this->parents[$o->cl_from][$k] = $o;
# Ignore those we already have
if ( in_array( $k, $this->deadend ) ) {
continue;
}
if ( isset( $this->name2id[$k] ) ) {
continue;
}
# Hey, new category!
$layer[$k] = $k;
}
$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 */ [ 'page_id', 'page_title' ],
/* WHERE */ [ 'page_namespace' => NS_CATEGORY, 'page_title' => $layer ],
__METHOD__ . '-2'
);
foreach ( $res as $o ) {
$id = $o->page_id;
$name = $o->page_title;
$this->name2id[$name] = $id;
$this->next[] = $id;
unset( $layer[$name] );
}
}
# Mark dead ends
foreach ( $layer as $v ) {
$this->deadend[$v] = $v;
}
}
}