Ticket #11847: fix-counts.php

<?php

        function fix_term_count($terms, $taxonomies, $args) { # {{{
                # Currently, when get_terms is used with a custom taxonomy, it returns
                # a count value that includes unpublished content. So wp_tag_cloud will
                # produce unexpected results. The tooltip will show the count including
                # unpublished content, but the page it links to will only show published
                # items. If a term is only attached to unpublished content, then it'll
                # be a 404, which is even worse. So this filter does its own query to
                # get an accurate count and filters things accordingly.

                # This may be unnecessary in a future WordPress version.

                if (!$args['fix_counts'])
                        return $terms;

                global $wpdb;

                $term_ids = array();
                foreach ($terms as $term)
                        $term_ids[] = $term->term_id;

                $posts_where = "AND $wpdb->posts.post_type != 'revision' AND (($wpdb->posts.post_status = 'publish') OR ($wpdb->posts.post_status = 'inherit' AND (p2.post_status = 'publish')))";
                $posts_where = apply_filters('posts_where', $posts_where);

                $query = "
                        SELECT COUNT(1) as count, t.term_id
                                FROM $wpdb->terms AS t
                                INNER JOIN $wpdb->term_taxonomy AS tt ON t.term_id = tt.term_id
                                INNER JOIN $wpdb->term_relationships AS tr ON tt.term_taxonomy_id = tr.term_taxonomy_id
                                INNER JOIN $wpdb->posts ON $wpdb->posts.ID = tr.object_id
                                LEFT JOIN $wpdb->posts AS p2 on $wpdb->posts.post_parent = p2.ID
                                WHERE t.term_id IN (".join(', ', $term_ids).")
                                        $posts_where
                                GROUP BY t.name, t.term_id, tt.parent
                                ";

                $counts = $wpdb->get_results($query);
                $_terms = array();
                $_counts = array();
                foreach ($counts as $c)
                        $_counts[$c->term_id] = $c->count;

                foreach ($terms as $t) {
                        if (!$_counts[$t->term_id]) {
                                if ($args['hide_empty'])
                                        continue;
                                $t->count = 0;
                        } else {
                                $t->count = $_counts[$t->term_id];
                        }
                        $_terms[] = $t;
                }

                # TODO: Worry about hierarchy and padding parent counts or just be
                # satisifed knowing that I didn't set up any of my taxonomies as
                # hierarchical?

                return $_terms;
        } # }}}

# vim:noet:ts=4