Ticket #44034: 44034.diff

src/wp-admin/includes/user.php

@@ -995 +995 @@
         add_submenu_page( 'tools.php', __( 'Erase Personal Data' ), __( 'Erase Personal Data' ), 'erase_others_personal_data', 'remove_personal_data', '_wp_personal_data_removal_page' );
 }
 
+/**
+ * Count user requests of a given type.
+ *
+ * @since 4.9.7
+ *
+ * @param  string $type   Request type. Supported types are 
+ *                        'remove_personal_data' and 'export_personal_data'.
+ * @return object $counts Number of request types for each status.
+ */
+function wp_count_user_requests( $type = '' ) {
+        global $wpdb;
+
+        $types = array( 'remove_personal_data', 'export_personal_data' );
+
+        if ( ! in_array( $type, $types, true ) ) {
+                return new stdClass;
+        }
+                
+        $cache_key = 'user-request-' . $type;
+        $counts    = wp_cache_get( $cache_key, 'counts' );
+
+        if ( false !== $counts ) {
+                return $counts;
+        }
+
+        $query = "
+                SELECT post_status, COUNT( * ) AS num_posts
+                FROM {$wpdb->posts}
+                WHERE post_type = %s
+                AND post_name = %s
+                GROUP BY post_status";
+        
+        $results = (array) $wpdb->get_results( $wpdb->prepare( $query, 'user_request', $type ), ARRAY_A );
+        $counts  = array_fill_keys( get_post_stati(), 0 );
+
+        foreach ( $results as $row ) {
+                $counts[ $row['post_status'] ] = $row['num_posts'];
+        }
+
+        $counts = (object) $counts;
+        wp_cache_set( $cache_key, $counts, 'counts' );
+
+        /**
+         * Modify returned user request counts, of a given type, by status.
+         *
+         * @since 4.9.7
+         *
+         * @param object $counts Number of request types for each status.
+         * @param string $type   Request type.
+         */
+        return apply_filters( 'wp_count_user_requests', $counts, $type );
+}
+
 // TODO: move the following classes in new files.
 if ( ! class_exists( 'WP_List_Table' ) ) {
         require_once( ABSPATH . 'wp-admin/includes/class-wp-list-table.php' );