This repository has been archived on 2022-08-01. You can view files and clone it, but cannot push or open issues or pull requests.
Oldphaloskepsis/plugins/jetpack/json-endpoints/class.wpcom-json-api-get-post-counts-v1-1-endpoint.php

108 lines
2.8 KiB
PHP
Raw Normal View History

2017-01-12 23:10:07 +00:00
<?php
class WPCOM_JSON_API_GET_Post_Counts_V1_1_Endpoint extends WPCOM_JSON_API_Endpoint {
private $whitelist = array( 'publish' );
/**
* Build SQL query
*
* @param {String} type - post type
* @param {Number} [author]
* @return {String} SQL query
*/
private function buildCountsQuery( $post_type = 'post', $user_id = null ) {
global $wpdb;
$query = "SELECT post_status as status, count(*) as count ";
$query .= "FROM {$wpdb->posts} ";
$query .= "WHERE post_type = %s ";
if ( isset( $user_id ) ) {
$query .= "AND post_author = %d ";
}
$query .= "GROUP BY status";
return $wpdb->prepare( $query, $post_type, $user_id );
}
/**
* Retrive counts using wp_cache
*
* @param {String} $post_type
* @param {Number} [$id]
*/
private function retrieveCounts( $post_type, $id = null) {
if ( ! isset( $id ) ) {
$counts = array();
foreach( (array) wp_count_posts( $post_type ) as $status => $count ) {
if ( in_array( $status, $this->whitelist ) && $count > 0 ) {
$counts[ $status ] = (int) $count;
}
};
return $counts;
}
global $wpdb;
$key = 'rest-api-' . $id . '-' . _count_posts_cache_key( $post_type );
$counts = wp_cache_get( $key, 'counts' );
if ( false === $counts ) {
$results = $wpdb->get_results( $this->buildCountsQuery( $post_type, $id ) );
$counts = $this->filterStatusesByWhiteslist( $results );
wp_cache_set( $key, $counts, 'counts' );
}
return $counts;
}
private function filterStatusesByWhiteslist( $in ) {
$return = array();
foreach( $in as $result) {
if ( in_array( $result->status, $this->whitelist ) ) {
$return[ $result->status ] = (int) $result->count;
}
};
return $return;
}
public function callback( $path = '', $blog_id = 0, $post_type = 'post' ) {
if ( ! get_current_user_id() ) {
return new WP_Error( 'authorization_required', __( 'An active access token must be used to retrieve post counts.', 'jetpack' ), 403 );
}
$blog_id = $this->api->switch_to_blog_and_validate_user( $this->api->get_blog_id( $blog_id ), false );
if ( is_wp_error( $blog_id ) ) {
return $blog_id;
}
if ( ! post_type_exists( $post_type ) ) {
return new WP_Error( 'unknown_post_type', __( 'Unknown post type requested.', 'jetpack' ), 404 );
}
$args = $this->query_args();
$mine_ID = get_current_user_id();
if ( current_user_can( 'edit_posts' ) ) {
array_push( $this->whitelist, 'draft', 'future', 'pending', 'private', 'trash' );
}
$return = array(
'counts' => (array) array(
'all' => (object) $this->retrieveCounts( $post_type ),
'mine' => (object) $this->retrieveCounts( $post_type, $mine_ID ),
)
);
// AUTHOR
if ( isset( $args['author'] ) ) {
$author_ID = $args['author'];
$return['counts']['author'] = (object) $this->retrieveCounts( $post_type, $author_ID );
}
return (object) $return;
}
}