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 );
 | 
						|
}
 |