Categoryfinder.php

Go to the documentation of this file.

<?php
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
 
    function __construct() {
    }
 
    function seed( $article_ids, $categories, $mode = 'AND' ) {
        $this->articles = $article_ids;
        $this->next = $article_ids;
        $this->mode = $mode;
 
        # Set the list of target categories; convert them to DBKEY form first
        $this->targets = array();
        foreach ( $categories as $c ) {
            $ct = Title::makeTitleSafe( NS_CATEGORY, $c );
            if ( $ct ) {
                $c = $ct->getDBkey();
                $this->targets[$c] = $c;
            }
        }
    }
 
    function run() {
        $this->dbr = wfGetDB( DB_SLAVE );
        while ( count( $this->next ) > 0 ) {
            $this->scan_next_layer();
        }
 
        # Now check if this applies to the individual articles
        $ret = array();
 
        foreach ( $this->articles as $article ) {
            $conds = $this->targets;
            if ( $this->check( $article, $conds ) ) {
                # Matches the conditions
                $ret[] = $article;
            }
        }
        return $ret;
    }
 
    function check( $id, &$conds, $path = array() ) {
        // 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 = array();
                    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;
    }
 
    function scan_next_layer() {
        wfProfileIn( __METHOD__ );
 
        # Find all parents of the article currently in $this->next
        $layer = array();
        $res = $this->dbr->select(
            /* FROM   */ 'categorylinks',
            /* SELECT */ '*',
            /* WHERE  */ array( '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] = array();
            }
            $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 = array();
 
        # 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 ),
                __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;
        }
 
        wfProfileOut( __METHOD__ );
    }
}