first

wp-content/plugins/query-monitor/dispatchers/REST_Envelope.php

@@ -0,0 +1,80 @@
+<?php declare(strict_types = 1);
+/**
+ * REST API enveloped request dispatcher.
+ *
+ * @package query-monitor
+ */
+
+class QM_Dispatcher_REST_Envelope extends QM_Dispatcher {
+
+	public $id = 'rest_envelope';
+
+	public function __construct( QM_Plugin $qm ) {
+		parent::__construct( $qm );
+
+		add_filter( 'rest_envelope_response', array( $this, 'filter_rest_envelope_response' ), 999, 2 );
+	}
+
+	/**
+	 * Filters the enveloped form of a REST API response to add QM's data.
+	 *
+	 * @param array<string, mixed> $envelope Envelope data.
+	 * @param WP_REST_Response     $response Original response data.
+	 * @return array<string, mixed> Envelope data.
+	 */
+	public function filter_rest_envelope_response( array $envelope, WP_REST_Response $response ) {
+		if ( ! $this->should_dispatch() ) {
+			return $envelope;
+		}
+
+		$data = array();
+
+		$this->before_output();
+
+		/** @var array<string, QM_Output_Raw> $outputters */
+		$outputters = $this->get_outputters( 'raw' );
+
+		foreach ( $outputters as $id => $output ) {
+			$data[ $id ] = $output->get_output();
+		}
+
+		$this->after_output();
+
+		$envelope['qm'] = $data;
+
+		return $envelope;
+	}
+
+	/**
+	 * @return void
+	 */
+	protected function before_output() {
+		foreach ( (array) glob( $this->qm->plugin_path( 'output/raw/*.php' ) ) as $file ) {
+			include_once $file;
+		}
+	}
+
+	/**
+	 * @return bool
+	 */
+	public function is_active() {
+		if ( ! self::user_can_view() ) {
+			return false;
+		}
+
+		return true;
+	}
+
+}
+
+/**
+ * @param array<string, QM_Dispatcher> $dispatchers
+ * @param QM_Plugin $qm
+ * @return array<string, QM_Dispatcher>
+ */
+function register_qm_dispatcher_rest_envelope( array $dispatchers, QM_Plugin $qm ) {
+	$dispatchers['rest_envelope'] = new QM_Dispatcher_REST_Envelope( $qm );
+	return $dispatchers;
+}
+
+add_filter( 'qm/dispatchers', 'register_qm_dispatcher_rest_envelope', 10, 2 );