Ticket #44000: 44000.patch

wp-admin/includes/user.php

@@ -800 +800 @@
                 'plural'   => 'privacy_requests',
                 'singular' => 'privacy_request',
         ) );
+
         $requests_table->process_bulk_action();
         $requests_table->prepare_items();
         ?>
@@ -845 +846 @@
 }
 
 /**
+ * Admin menu bubble for data export or removal requests.
+ *
+ * @since 4.9.6
+ */
+
+add_filter( 'add_menu_classes', '_wp_export_removal_request_bubble' );
+function _wp_export_removal_request_bubble( $menu ){
+        // Start with a count of 0
+        $request_count = 0;
+
+        // Get number of export requests that have been confirmed, and add to $request_count
+        $export_requests = new WP_Privacy_Data_Export_Requests_Table( array(
+                'plural'   => 'privacy_requests',
+                'singular' => 'privacy_request',
+        ) );
+
+        $export_requests->process_bulk_action();
+        $export_requests->prepare_items();
+
+        foreach( $export_requests as $request ){
+                $status = $request[0]->status;
+                if( 'request-confirmed' == $status ){
+                        $request_count++;
+                }
+        }
+
+        // Get number of removal requests that have been confirmed, and add to $request_count
+        $removal_requests = new WP_Privacy_Data_Removal_Requests_Table( array(
+                'plural'   => 'privacy_requests',
+                'singular' => 'privacy_request',
+        ) );
+
+        $removal_requests->process_bulk_action();
+        $removal_requests->prepare_items();
+
+        foreach( $removal_requests as $request ){
+                $status = $request[0]->status;
+                if( 'request-confirmed' == $status ){
+                        $request_count++;
+                }
+        }
+
+        // Loop through menu items
+        foreach( $menu as $menu_key => $menu_data ){
+                // If not the Tools menu, continue
+                if( 'tools.php' != $menu_data[2] )
+                        continue;
+                // On Tools menu add the bubble with the $request_count calculated above.
+                $menu[$menu_key][0] .= " <span class='update-plugins count-$request_count'><span class='plugin-count'>" . number_format_i18n( $request_count ) . '</span></span>';
+        }
+        return $menu;
+}
+
+/**
  * Personal data anonymization.
  *
  * @since 4.9.6