Edit - Extension_NewRelic

Edit: Extension_NewRelic_Plugin.php

<?php
/**
 * File: Extension_NewRelic_Plugin.php
 *
 * @package W3TC
 */

namespace W3TC;

/**
 * Class Extension_NewRelic_Plugin
 *
 * phpcs:disable PSR2.Classes.PropertyDeclaration.Underscore
 * phpcs:disable PSR2.Methods.MethodDeclaration.Underscore
 */
class Extension_NewRelic_Plugin {
	/**
	 * New Relic reject reason
	 *
	 * @var string
	 */
	private $newrelic_reject_reason = '';

/**
	 * Config
	 *
	 * @var Config
	 */
	private $_config = null;

/**
	 * Initializes the NewRelic extension.
	 *
	 * @return void
	 */
	public function __construct() {
		$this->_config = Dispatcher::config();
	}

/**
	 * Runs the NewRelic extension and completes necessary setup.
	 *
	 * @return void
	 */
	public function run() {
		add_filter( 'w3tc_config_default_values', array( $this, 'w3tc_config_default_values' ) );

$config = Dispatcher::config();
		// remainder only when extension is frontend-active.
		if ( ! $config->is_extension_active_frontend( 'newrelic' ) ) {
			return;
		}

if ( 'browser' === $this->_config->get_string( array( 'newrelic', 'monitoring_type' ) ) ) {
			Util_Bus::add_ob_callback( 'newrelic', array( $this, 'ob_callback_browser' ) );
		} else {
			require_once W3TC_LIB_NEWRELIC_DIR . '/NewRelicWrapper.php';

$this->set_appname();

if ( defined( 'DOING_CRON' ) && DOING_CRON ) {
				$this->background_task();
			}

Util_Bus::add_ob_callback( 'newrelic', array( $this, 'ob_callback_apm' ) );
		}

add_filter( 'w3tc_footer_comment', array( $this, 'w3tc_footer_comment' ) );
	}

/**
	 * Sets default configuration values for the NewRelic extension.
	 *
	 * @param array $default_values Default configuration values.
	 *
	 * @return array Modified default configuration values.
	 */
	public function w3tc_config_default_values( $default_values ) {
		$default_values['newrelic'] = array(
			'monitoring_type'     => 'apm',
			'accept.logged_roles' => false,
			'accept.roles'        => array(),
			'use_php_function'    => true,
			'cache_time'          => 5,
			'include_rum'         => true,
		);

return $default_values;
	}

/**
	 * Handles the output buffer for browser-based monitoring.
	 *
	 * @param string $buffer The output buffer content.
	 *
	 * @return string The modified output buffer.
	 */
	public function ob_callback_browser( $buffer ) {
		$core = Dispatcher::component( 'Extension_NewRelic_Core' );
		$app  = $core->get_effective_browser_application();
		if ( isset( $app['loader_script'] ) && $this->_can_add_tracker_script( $buffer ) ) {
			$buffer = preg_replace( '~<head(\s+[^>]*)*>~Ui', '\\0' . $app['loader_script'], $buffer, 1 );
		}

$buffer = str_replace(
			'{w3tc_newrelic_reject_reason}',
			( '' !== $this->newrelic_reject_reason ? sprintf( ' (%s)', $this->newrelic_reject_reason ) : '' ),
			$buffer
		);

return $buffer;
	}

/**
	 * Handles the output buffer for APM-based monitoring.
	 *
	 * @param string $buffer The output buffer content.
	 *
	 * @return string The modified output buffer.
	 */
	public function ob_callback_apm( $buffer ) {
		if ( ! $this->_can_add_tracker_script( $buffer ) ) {
			$this->disable_auto_rum();
		} elseif ( $this->_config->get_boolean( array( 'newrelic', 'include_rum' ) ) ) {
			$buffer = preg_replace( '~<head(\s+[^>]*)*>~Ui', '\\0' . \NewRelicWrapper::get_browser_timing_header(), $buffer, 1 );
			$buffer = preg_replace( '~<\\/body>~', \NewRelicWrapper::get_browser_timing_footer() . '\\0', $buffer, 1 );
		}

$buffer = str_replace(
			'{w3tc_newrelic_reject_reason}',
			( '' !== $this->newrelic_reject_reason ? sprintf( ' (%s)', $this->newrelic_reject_reason ) : '' ),
			$buffer
		);

return $buffer;
	}

/**
	 * Marks the current job as a background task.
	 *
	 * @return void
	 */
	public function background_task() {
		\NewRelicWrapper::mark_as_background_job();
	}

/**
	 * Disables automatic RUM (Real User Monitoring) for the current request.
	 *
	 * @return void
	 */
	public function disable_auto_rum() {
		\NewRelicWrapper::disable_auto_rum();
	}

/**
	 * Checks whether a tracker script can be added to the output buffer.
	 *
	 * @param string $buffer The output buffer content.
	 *
	 * @return bool Whether the tracker script can be added.
	 */
	public function _can_add_tracker_script( $buffer ) {
		$v = '';
		if ( preg_match( '~^\s*<\?xml[^>]*>\s*<xsl:stylesheet~', $buffer, $v ) ) {
			$this->newrelic_reject_reason = __( 'XSL not tracked', 'w3-total-cache' );
			return false;
		}

$reject_reason = apply_filters( 'w3tc_newrelic_should_disable_auto_rum', null );
		if ( ! empty( $reject_reason ) ) {
			$this->newrelic_reject_reason = __( 'rejected by filter: ', 'w3-total-cache' ) . $reject_reason;
			return false;
		}

/**
		 * Disable for AJAX so its not messed up
		 */
		if ( defined( 'DOING_AJAX' ) && DOING_AJAX ) {
			$this->newrelic_reject_reason = __( 'DOING_AJAX constant is defined', 'w3-total-cache' );

return false;
		}

/**
		 * Check for DONOTAUTORUM constant
		 */
		if ( defined( 'DONOTAUTORUM' ) && DONOTAUTORUM ) {
			$this->newrelic_reject_reason = __( 'DONOTAUTORUM constant is defined', 'w3-total-cache' );

return false;
		}

/**
		 * Check logged users roles
		 */
		if ( $this->_config->get_boolean( array( 'newrelic', 'accept.logged_roles' ) ) &&
			$this->_check_logged_in_role_not_allowed() ) {
			$this->newrelic_reject_reason = __( 'logged in role is rejected', 'w3-total-cache' );

return false;
		}

return true;
	}

/**
	 * Checks whether the logged-in user's role is allowed for NewRelic monitoring.
	 *
	 * @return bool Whether the user's role is allowed for monitoring.
	 */
	private function _check_logged_in_role_not_allowed() {
		$current_user = wp_get_current_user();

if ( ! is_user_logged_in() ) {
			return false;
		}

$roles = $this->_config->get_array( array( 'newrelic', 'accept.roles' ) );

if ( empty( $roles ) || empty( $current_user->roles ) || ! is_array( $current_user->roles ) ) {
			return true;
		}

foreach ( $current_user->roles as $role ) {
			if ( in_array( $role, $roles, true ) ) {
				return false;
			}
		}

return true;
	}

/**
	 * Sets the application name for NewRelic if not already set.
	 *
	 * @return void
	 */
	public function set_appname() {
		static $appname_set;
		if ( ! $appname_set && ( $this->_config->get_boolean( array( 'newrelic', 'use_php_function' ) ) || Util_Environment::is_wpmu() ) ) {
			$appname_set = true;
			$service     = Dispatcher::component( 'Extension_NewRelic_Service' );
			$appname     = $service->get_effective_appname();

$enable_xmit = $this->_config->get_boolean( array( 'newrelic', 'enable_xmit' ) );
			\NewRelicWrapper::set_appname( $appname, '', $enable_xmit );
		}
	}

/**
	 * Adds a footer comment with NewRelic application monitoring information.
	 *
	 * @param array $strings The current footer strings.
	 *
	 * @return array The modified footer strings.
	 */
	public function w3tc_footer_comment( $strings ) {
		$strings[] = sprintf(
			// Translators: 1 reject reason.
			__(
				'Application Monitoring using New Relic%1$s',
				'w3-total-cache'
			),
			'{w3tc_newrelic_reject_reason}'
		);

return $strings;
	}
}

$p = new Extension_NewRelic_Plugin();
$p->run();

if ( is_admin() ) {
	$p = new Extension_NewRelic_Plugin_Admin();
	$p->run();
}

Cancel