root/trunk/wp-includes/query.php

<?php

/*
 * The Big Query.
 */

function get_query_var($var) {
    global $wp_query;

    return $wp_query->get($var);
}

function set_query_var($var, $value) {
    global $wp_query;

    return $wp_query->set($var, $value);
}

function &query_posts($query) {
    unset($GLOBALS['wp_query']);
    $GLOBALS['wp_query'] =& new WP_Query();
    return $GLOBALS['wp_query']->query($query);
}

function wp_reset_query() {
    unset($GLOBALS['wp_query']);
    $GLOBALS['wp_query'] =& $GLOBALS['wp_the_query'];
    global $wp_query;
    if ( !empty($wp_query->post) ) {
        $GLOBALS['post'] = $wp_query->post;
        setup_postdata($wp_query->post);
    }
}

/*
 * Query type checks.
 */

function is_admin () {
    if ( defined('WP_ADMIN') )
        return WP_ADMIN;
    return false;
}

function is_archive () {
    global $wp_query;

    return $wp_query->is_archive;
}

function is_attachment () {
    global $wp_query;

    return $wp_query->is_attachment;
}

function is_author ($author = '') {
    global $wp_query;

    if ( !$wp_query->is_author )
        return false;

    if ( empty($author) )
        return true;

    $author_obj = $wp_query->get_queried_object();

    $author = (array) $author;

    if ( in_array( $author_obj->ID, $author ) )
        return true;
    elseif ( in_array( $author_obj->nickname, $author ) )
        return true;
    elseif ( in_array( $author_obj->user_nicename, $author ) )
        return true;

    return false;
}

function is_category ($category = '') {
    global $wp_query;

    if ( !$wp_query->is_category )
        return false;

    if ( empty($category) )
        return true;

    $cat_obj = $wp_query->get_queried_object();

    $category = (array) $category;

    if ( in_array( $cat_obj->term_id, $category ) )
        return true;
    elseif ( in_array( $cat_obj->name, $category ) )
        return true;
    elseif ( in_array( $cat_obj->slug, $category ) )
        return true;

    return false;
}

function is_tag( $slug = '' ) {
    global $wp_query;

    if ( !$wp_query->is_tag )
        return false;

    if ( empty( $slug ) )
        return true;

    $tag_obj = $wp_query->get_queried_object();

    $slug = (array) $slug;

    if ( in_array( $tag_obj->slug, $slug ) )
        return true;

    return false;
}

function is_tax( $slug = '' ) {
    global $wp_query;
    
    if ( !$wp_query->is_tax )
        return false;

    if ( empty($slug) )
        return true;

    $term = $wp_query->get_queried_object();

    $slug = (array) $slug;

    if ( in_array( $term->slug, $slug ) )
        return true;

    return false;
}

function is_comments_popup () {
    global $wp_query;

    return $wp_query->is_comments_popup;
}

function is_date () {
    global $wp_query;

    return $wp_query->is_date;
}

function is_day () {
    global $wp_query;

    return $wp_query->is_day;
}

function is_feed () {
    global $wp_query;

    return $wp_query->is_feed;
}

/**
 * is_front_page() - Is it the front of the site, whether blog view or a WP Page?
 *
 * @since 2.5
 * @uses is_home
 * @uses get_option
 *
 * @return bool True if front of site
 */
function is_front_page () {
    // most likely case
    if ( 'posts' == get_option('show_on_front') && is_home() )
        return true;
    elseif ( 'page' == get_option('show_on_front') && get_option('page_on_front') && is_page(get_option('page_on_front')) )
        return true;
    else
        return false;
}

/**
 * is_home() - Is it the blog view homepage?
 *
 * @since 2.1
 * @global object $wp_query
 *
 * @return bool True if blog view homepage
 */
function is_home () {
    global $wp_query;

    return $wp_query->is_home;
}

function is_month () {
    global $wp_query;

    return $wp_query->is_month;
}

function is_page ($page = '') {
    global $wp_query;

    if ( !$wp_query->is_page )
        return false;

    if ( empty($page) )
        return true;

    $page_obj = $wp_query->get_queried_object();

    $page = (array) $page;

    if ( in_array( $page_obj->ID, $page ) )
        return true;
    elseif ( in_array( $page_obj->post_title, $page ) )
        return true;
    else if ( in_array( $page_obj->post_name, $page ) )
        return true;

    return false;
}

function is_paged () {
    global $wp_query;

    return $wp_query->is_paged;
}

function is_plugin_page() {
    global $plugin_page;

    if ( isset($plugin_page) )
        return true;

    return false;
}

function is_preview() {
    global $wp_query;

    return $wp_query->is_preview;
}

function is_robots() {
    global $wp_query;

    return $wp_query->is_robots;
}

function is_search () {
    global $wp_query;

    return $wp_query->is_search;
}

function is_single ($post = '') {
    global $wp_query;

    if ( !$wp_query->is_single )
        return false;

    if ( empty( $post) )
        return true;

    $post_obj = $wp_query->get_queried_object();

    $post = (array) $post;

    if ( in_array( $post_obj->ID, $post ) )
        return true;
    elseif ( in_array( $post_obj->post_title, $post ) )
        return true;
    elseif ( in_array( $post_obj->post_name, $post ) )
        return true;

    return false;
}

function is_singular() {
    global $wp_query;

    return $wp_query->is_singular;
}

function is_time () {
    global $wp_query;

    return $wp_query->is_time;
}

function is_trackback () {
    global $wp_query;

    return $wp_query->is_trackback;
}

function is_year () {
    global $wp_query;

    return $wp_query->is_year;
}

function is_404 () {
    global $wp_query;

    return $wp_query->is_404;
}

/*
 * The Loop.  Post loop control.
 */

function have_posts() {
    global $wp_query;

    return $wp_query->have_posts();
}

function in_the_loop() {
    global $wp_query;

    return $wp_query->in_the_loop;
}

function rewind_posts() {
    global $wp_query;

    return $wp_query->rewind_posts();
}

function the_post() {
    global $wp_query;

    $wp_query->the_post();
}

/*
 * Comments loop.
 */

function have_comments() {
    global $wp_query;
    return $wp_query->have_comments();
}

function the_comment() {
    global $wp_query;
    return $wp_query->the_comment();
}

/*
 * WP_Query
 */

class WP_Query {
    var $query;
    var $query_vars = array();
    var $queried_object;
    var $queried_object_id;
    var $request;

    var $posts;
    var $post_count = 0;
    var $current_post = -1;
    var $in_the_loop = false;
    var $post;

    var $comments;
    var $comment_count = 0;
    var $current_comment = -1;
    var $comment;

    var $found_posts = 0;
    var $max_num_pages = 0;

    var $is_single = false;
    var $is_preview = false;
    var $is_page = false;
    var $is_archive = false;
    var $is_date = false;
    var $is_year = false;
    var $is_month = false;
    var $is_day = false;
    var $is_time = false;
    var $is_author = false;
    var $is_category = false;
    var $is_tag = false;
    var $is_tax = false;
    var $is_search = false;
    var $is_feed = false;
    var $is_comment_feed = false;
    var $is_trackback = false;
    var $is_home = false;
    var $is_404 = false;
    var $is_comments_popup = false;
    var $is_admin = false;
    var $is_attachment = false;
    var $is_singular = false;
    var $is_robots = false;
    var $is_posts_page = false;

    function init_query_flags() {
        $this->is_single = false;
        $this->is_page = false;
        $this->is_archive = false;
        $this->is_date = false;
        $this->is_year = false;
        $this->is_month = false;
        $this->is_day = false;
        $this->is_time = false;
        $this->is_author = false;
        $this->is_category = false;
        $this->is_tag = false;
        $this->is_tax = false;
        $this->is_search = false;
        $this->is_feed = false;
        $this->is_comment_feed = false;
        $this->is_trackback = false;
        $this->is_home = false;
        $this->is_404 = false;
        $this->is_paged = false;
        $this->is_admin = false;
        $this->is_attachment = false;
        $this->is_singular = false;
        $this->is_robots = false;
        $this->is_posts_page = false;
    }

    function init () {
        unset($this->posts);
        unset($this->query);
        $this->query_vars = array();
        unset($this->queried_object);
        unset($this->queried_object_id);
        $this->post_count = 0;
        $this->current_post = -1;
        $this->in_the_loop = false;

        $this->init_query_flags();
    }

    // Reparse the query vars.
    function parse_query_vars() {
        $this->parse_query('');
    }

    function fill_query_vars($array) {
        $keys = array(
            'error'
            , 'm'
            , 'p'
            , 'post_parent'
            , 'subpost'
            , 'subpost_id'
            , 'attachment'
            , 'attachment_id'
            , 'name'
            , 'hour'
            , 'static'
            , 'pagename'
            , 'page_id'
            , 'second'
            , 'minute'
            , 'hour'
            , 'day'
            , 'monthnum'
            , 'year'
            , 'w'
            , 'category_name'
            , 'tag'
            , 'tag_id'
            , 'author_name'
            , 'feed'
            , 'tb'
            , 'paged'
            , 'comments_popup'
            , 'meta_key'
            , 'meta_value'
            , 'preview'
        );

        foreach ($keys as $key) {
            if ( !isset($array[$key]))
                $array[$key] = '';
        }

        $array_keys = array('category__in', 'category__not_in', 'category__and', 'post__in', 'post__not_in',
            'tag__in', 'tag__not_in', 'tag__and', 'tag_slug__in', 'tag_slug__and');

        foreach ( $array_keys as $key ) {
            if ( !isset($array[$key]))
                $array[$key] = array();
        }
        return $array;
    }

    // Parse a query string and set query type booleans.
    function parse_query ($query) {
        if ( !empty($query) || !isset($this->query) ) {
            $this->init();
            if ( is_array($query) )
                $this->query_vars = $query;
            else
                parse_str($query, $this->query_vars);
            $this->query = $query;
        }

        $this->query_vars = $this->fill_query_vars($this->query_vars);
        $qv = &$this->query_vars;

        if ( ! empty($qv['robots']) )
            $this->is_robots = true;

        $qv['p'] =  absint($qv['p']);
        $qv['page_id'] =  absint($qv['page_id']);
        $qv['year'] = absint($qv['year']);
        $qv['monthnum'] = absint($qv['monthnum']);
        $qv['day'] = absint($qv['day']);
        $qv['w'] = absint($qv['w']);
        $qv['m'] = absint($qv['m']);
        $qv['cat'] = preg_replace( '|[^0-9,-]|', '', $qv['cat'] ); // comma separated list of positive or negative integers
        if ( '' !== $qv['hour'] ) $qv['hour'] = absint($qv['hour']);
        if ( '' !== $qv['minute'] ) $qv['minute'] = absint($qv['minute']);
        if ( '' !== $qv['second'] ) $qv['second'] = absint($qv['second']);

        // Compat.  Map subpost to attachment.
        if ( '' != $qv['subpost'] )
            $qv['attachment'] = $qv['subpost'];
        if ( '' != $qv['subpost_id'] )
            $qv['attachment_id'] = $qv['subpost_id'];

        $qv['attachment_id'] = absint($qv['attachment_id']);

        if ( ('' != $qv['attachment']) || !empty($qv['attachment_id']) ) {
            $this->is_single = true;
            $this->is_attachment = true;
        } elseif ( '' != $qv['name'] ) {
            $this->is_single = true;
        } elseif ( $qv['p'] ) {
            $this->is_single = true;
        } elseif ( ('' !== $qv['hour']) && ('' !== $qv['minute']) &&('' !== $qv['second']) && ('' != $qv['year']) && ('' != $qv['monthnum']) && ('' != $qv['day']) ) {
            // If year, month, day, hour, minute, and second are set, a single
            // post is being queried.
            $this->is_single = true;
        } elseif ( '' != $qv['static'] || '' != $qv['pagename'] || !empty($qv['page_id']) ) {
            $this->is_page = true;
            $this->is_single = false;
        } elseif ( !empty($qv['s']) ) {
            $this->is_search = true;
        } else {
        // Look for archive queries.  Dates, categories, authors.

            if ( '' !== $qv['second'] ) {
                $this->is_time = true;
                $this->is_date = true;
            }

            if ( '' !== $qv['minute'] ) {
                $this->is_time = true;
                $this->is_date = true;
            }

            if ( '' !== $qv['hour'] ) {
                $this->is_time = true;
                $this->is_date = true;
            }

            if ( $qv['day'] ) {
                if (! $this->is_date) {
                    $this->is_day = true;
                    $this->is_date = true;
                }
            }

            if ( $qv['monthnum'] ) {
                if (!