Permit running requests as frontend vs. backend based on URL (#216)

Co-authored-by: swissspidy <841956+swissspidy@users.noreply.github.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: Pascal Birchler <pascalb@google.com>

Author: 3 people

features/profile-stage.feature

@@ -171,3 +171,32 @@ Feature: Profile the template render stage
       | hook              |
       | init              |
       | wp_loaded:after   |
+
+  @require-wp-4.0
+  Scenario: Admin URL runs as a backend request and skips frontend stages
+    Given a WP install
+
+    When I run `wp profile stage --url=example.com/wp-admin/ --context=admin --fields=stage`
+    Then STDOUT should be a table containing rows:
+      | stage     |
+      | bootstrap |
+    And STDOUT should not contain:
+      """
+      main_query
+      """
+    And STDOUT should not contain:
+      """
+      template
+      """
+    And STDERR should be empty
+
+  @require-wp-4.0
+  Scenario: Admin URL without --context=admin emits an error
+    Given a WP install
+
+    When I try `wp profile stage --url=example.com/wp-admin/ --fields=stage`
+    Then STDERR should contain:
+      """
+      Profiling an admin URL requires --context=admin.
+      """
+    And the return code should be 1

src/Profiler.php

@@ -48,6 +48,8 @@ class Profiler {
 	private $tick_cache_hit_offset  = null;
 	private $tick_cache_miss_offset = null;
 
+	private $is_admin_request = false;
+
 	public function __construct( $type, $focus ) {
 		$this->type  = $type;
 		$this->focus = $focus;
@@ -77,6 +79,23 @@ public function get_loggers() {
 	 * Run the profiler against WordPress
 	 */
 	public function run() {
+		$url  = WP_CLI::get_runner()->config['url'];
+		$path = '';
+		if ( ! empty( $url ) ) {
+			$parsed_url = WP_CLI\Utils\parse_url( $url );
+			if ( false !== $parsed_url && isset( $parsed_url['path'] ) ) {
+				$path = $parsed_url['path'];
+			} else {
+				// Fallback for cases where $url is just a path.
+				$path = $url;
+			}
+		}
+		$this->is_admin_request = ! empty( $path ) && (bool) preg_match( '#/wp-admin(/|$|\?)#i', $path );
+
+		if ( $this->is_admin_request && 'admin' !== WP_CLI::get_runner()->config['context'] ) {
+			WP_CLI::error( 'Profiling an admin URL requires --context=admin.' );
+		}
+
 		WP_CLI::add_wp_hook(
 			'muplugins_loaded',
 			function () {
@@ -437,6 +456,11 @@ private function load_wordpress_with_template() {
 			$this->loggers[] = $logger;
 		}
 
+		// Skip main_query and template stages for admin requests.
+		if ( $this->is_admin_request ) {
+			return;
+		}
+
 		// Set up main_query main WordPress query.
 		if ( 'stage' === $this->type ) {
 			if ( 'main_query' === $this->focus ) {