142 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			142 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php declare(strict_types = 1);
 | |
| /**
 | |
|  * Admin screen collector.
 | |
|  *
 | |
|  * @package query-monitor
 | |
|  */
 | |
| 
 | |
| if ( ! defined( 'ABSPATH' ) ) {
 | |
| 	exit;
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * @extends QM_DataCollector<QM_Data_Admin>
 | |
|  */
 | |
| class QM_Collector_Admin extends QM_DataCollector {
 | |
| 
 | |
| 	public $id = 'response';
 | |
| 
 | |
| 	public function get_storage(): QM_Data {
 | |
| 		return new QM_Data_Admin();
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * @return array<int, string>
 | |
| 	 */
 | |
| 	public function get_concerned_actions() {
 | |
| 		$actions = array(
 | |
| 			'current_screen',
 | |
| 			'admin_notices',
 | |
| 			'all_admin_notices',
 | |
| 			'network_admin_notices',
 | |
| 			'user_admin_notices',
 | |
| 		);
 | |
| 
 | |
| 		if ( ! empty( $this->data->list_table ) ) {
 | |
| 			$actions[] = $this->data->list_table['column_action'];
 | |
| 		}
 | |
| 
 | |
| 		return $actions;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * @return array<int, string>
 | |
| 	 */
 | |
| 	public function get_concerned_filters() {
 | |
| 		$filters = array();
 | |
| 
 | |
| 		if ( ! empty( $this->data->list_table ) ) {
 | |
| 			$filters[] = $this->data->list_table['columns_filter'];
 | |
| 			$filters[] = $this->data->list_table['sortables_filter'];
 | |
| 		}
 | |
| 
 | |
| 		return $filters;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * @return void
 | |
| 	 */
 | |
| 	public function process() {
 | |
| 		/**
 | |
| 		 * @var string $pagenow
 | |
| 		 * @var ?WP_List_Table $wp_list_table
 | |
| 		 */
 | |
| 		global $pagenow, $wp_list_table;
 | |
| 
 | |
| 		$this->data->pagenow = $pagenow;
 | |
| 		$this->data->typenow = $GLOBALS['typenow'] ?? '';
 | |
| 		$this->data->taxnow = $GLOBALS['taxnow'] ?? '';
 | |
| 		$this->data->hook_suffix = $GLOBALS['hook_suffix'] ?? '';
 | |
| 		$this->data->current_screen = get_current_screen();
 | |
| 
 | |
| 		$screens = array(
 | |
| 			'edit' => true,
 | |
| 			'edit-comments' => true,
 | |
| 			'edit-tags' => true,
 | |
| 			'link-manager' => true,
 | |
| 			'plugins' => true,
 | |
| 			'plugins-network' => true,
 | |
| 			'sites-network' => true,
 | |
| 			'themes-network' => true,
 | |
| 			'upload' => true,
 | |
| 			'users' => true,
 | |
| 			'users-network' => true,
 | |
| 		);
 | |
| 
 | |
| 		if ( empty( $this->data->current_screen ) || ! isset( $screens[ $this->data->current_screen->base ] ) ) {
 | |
| 			return;
 | |
| 		}
 | |
| 
 | |
| 		# And now, WordPress' legendary inconsistency comes into play:
 | |
| 
 | |
| 		$columns = $this->data->current_screen->id;
 | |
| 		$sortables = $this->data->current_screen->id;
 | |
| 		$column = $this->data->current_screen->base;
 | |
| 
 | |
| 		if ( ! empty( $this->data->current_screen->taxonomy ) ) {
 | |
| 			$column = $this->data->current_screen->taxonomy;
 | |
| 		} elseif ( ! empty( $this->data->current_screen->post_type ) ) {
 | |
| 			$column = $this->data->current_screen->post_type . '_posts';
 | |
| 		}
 | |
| 
 | |
| 		if ( ! empty( $this->data->current_screen->post_type ) && empty( $this->data->current_screen->taxonomy ) ) {
 | |
| 			$columns = $this->data->current_screen->post_type . '_posts';
 | |
| 		}
 | |
| 
 | |
| 		if ( 'edit-comments' === $column ) {
 | |
| 			$column = 'comments';
 | |
| 		} elseif ( 'upload' === $column ) {
 | |
| 			$column = 'media';
 | |
| 		} elseif ( 'link-manager' === $column ) {
 | |
| 			$column = 'link';
 | |
| 		}
 | |
| 
 | |
| 		$list_table_data = array(
 | |
| 			'columns_filter' => "manage_{$columns}_columns",
 | |
| 			'sortables_filter' => "manage_{$sortables}_sortable_columns",
 | |
| 			'column_action' => "manage_{$column}_custom_column",
 | |
| 		);
 | |
| 
 | |
| 		if ( ! empty( $wp_list_table ) ) {
 | |
| 			$list_table_data['class_name'] = get_class( $wp_list_table );
 | |
| 		}
 | |
| 
 | |
| 		$this->data->list_table = $list_table_data;
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * @param array<string, QM_Collector> $collectors
 | |
|  * @param QueryMonitor $qm
 | |
|  * @return array<string, QM_Collector>
 | |
|  */
 | |
| function register_qm_collector_admin( array $collectors, QueryMonitor $qm ) {
 | |
| 	$collectors['response'] = new QM_Collector_Admin();
 | |
| 	return $collectors;
 | |
| }
 | |
| 
 | |
| if ( is_admin() ) {
 | |
| 	add_filter( 'qm/collectors', 'register_qm_collector_admin', 10, 2 );
 | |
| }
 |