301 lines
7.8 KiB
PHP
301 lines
7.8 KiB
PHP
<?php
|
|
|
|
namespace Rarst\Laps;
|
|
|
|
use Symfony\Component\Stopwatch\Stopwatch;
|
|
|
|
/**
|
|
* Main plugin's class.
|
|
*/
|
|
class Laps {
|
|
|
|
/** @var Stopwatch $stopwatch */
|
|
public static $stopwatch;
|
|
public static $events = array();
|
|
public static $query_starts = array();
|
|
|
|
/**
|
|
* Start Stopwatch and timing plugin load immediately, then set up core events and needed hooks.
|
|
*/
|
|
static function on_load() {
|
|
|
|
if ( defined( 'DOING_AJAX' ) && DOING_AJAX ) {
|
|
return;
|
|
}
|
|
|
|
self::$stopwatch = new Stopwatch();
|
|
self::$stopwatch->start( 'Plugins Load', 'plugin' );
|
|
$events = new Core_Events();
|
|
self::add_events( $events->get() );
|
|
|
|
add_action( 'pre_update_option_active_plugins', array( __CLASS__, 'pre_update_option_active_plugins' ) );
|
|
add_action( 'pre_update_site_option_active_sitewide_plugins', array( __CLASS__, 'pre_update_option_active_plugins' ) );
|
|
add_action( 'pre_http_request', array( __CLASS__, 'pre_http_request' ), 10, 3 );
|
|
add_action( 'http_api_debug', array( __CLASS__, 'http_api_debug' ), 10, 5 );
|
|
add_action( 'after_setup_theme', array( __CLASS__, 'after_setup_theme' ), 15 );
|
|
add_action( 'init', array( __CLASS__, 'init' ) );
|
|
|
|
if ( defined( 'SAVEQUERIES' ) && SAVEQUERIES ) {
|
|
add_filter( 'query', array( __CLASS__, 'query' ), 20 );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Hook events by name and priority from array.
|
|
*
|
|
* @param array $stops
|
|
*/
|
|
static function add_events( $stops ) {
|
|
|
|
self::$events = array_merge( self::$events, $stops );
|
|
|
|
foreach ( $stops as $hook_name => $data ) {
|
|
|
|
foreach ( array_keys( $data ) as $priority ) {
|
|
|
|
add_action( $hook_name, array( __CLASS__, 'tick' ), $priority );
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Reorder active plugins so Laps is first and starts timing load early.
|
|
*
|
|
* @param array $plugins
|
|
*
|
|
* @return array
|
|
*/
|
|
static function pre_update_option_active_plugins( $plugins ) {
|
|
|
|
$plugin = plugin_basename( dirname( __DIR__ ) . '/laps.php' );
|
|
$key = array_search( $plugin, $plugins );
|
|
|
|
if ( false !== $key && $key > 0 ) {
|
|
|
|
unset( $plugins[ $key ] );
|
|
array_unshift( $plugins, $plugin );
|
|
$plugins = array_values( $plugins );
|
|
}
|
|
|
|
return $plugins;
|
|
}
|
|
|
|
/**
|
|
* Mark action for the event on Stopwatch.
|
|
*
|
|
* @param mixed $input pass through if added to filter
|
|
*
|
|
* @return mixed
|
|
*/
|
|
static function tick( $input = null ) {
|
|
|
|
global $wp_filter;
|
|
|
|
$filter = current_filter();
|
|
$priority = key( $wp_filter[ $filter ] );
|
|
|
|
$event = wp_parse_args(
|
|
self::$events[ $filter ][ $priority ],
|
|
array(
|
|
'action' => 'start',
|
|
'category' => null,
|
|
)
|
|
);
|
|
|
|
self::$stopwatch->$event['action']( $event['event'], $event['category'] );
|
|
|
|
return $input;
|
|
}
|
|
|
|
/**
|
|
* Capture SQL queries start times
|
|
*
|
|
* @param string $query
|
|
*
|
|
* @return string
|
|
*/
|
|
static function query( $query ) {
|
|
|
|
global $wpdb;
|
|
|
|
if ( empty( self::$query_starts ) && ! empty( $wpdb->queries ) ) {
|
|
self::$query_starts[ count( $wpdb->queries ) ] = microtime( true ) * 1000;
|
|
} else {
|
|
self::$query_starts[] = microtime( true ) * 1000;
|
|
}
|
|
|
|
return $query;
|
|
}
|
|
|
|
/**
|
|
* Capture start time of HTTP request
|
|
*
|
|
* @param boolean $false
|
|
* @param array $args
|
|
* @param string $url
|
|
*
|
|
* @return boolean
|
|
*/
|
|
static function pre_http_request( $false, $args, $url ) {
|
|
|
|
self::$stopwatch->start( $url, 'http' );
|
|
|
|
return $false;
|
|
}
|
|
|
|
/**
|
|
* Capture end time of HTTP request
|
|
*
|
|
* @param array|\WP_Error $response
|
|
* @param string $type
|
|
* @param object $class
|
|
* @param array $args
|
|
* @param string $url
|
|
*
|
|
* @return mixed
|
|
*/
|
|
static function http_api_debug( $response, $type, $class, $args, $url ) {
|
|
|
|
self::$stopwatch->stop( $url );
|
|
|
|
return $response;
|
|
}
|
|
|
|
/**
|
|
* When theme is done possibly add vendor-specific events.
|
|
*/
|
|
static function after_setup_theme() {
|
|
|
|
foreach ( array( 'THA', 'Hybrid', 'Genesis', 'Thematic', 'Yoast' ) as $vendor ) {
|
|
|
|
$class = "Rarst\\Laps\\{$vendor}_Events";
|
|
/** @var Laps_Events $events */
|
|
$events = new $class;
|
|
self::add_events( $events->get() );
|
|
}
|
|
}
|
|
|
|
static function init() {
|
|
|
|
add_action( 'wp_enqueue_scripts', array( __CLASS__, 'enqueue_scripts' ) );
|
|
add_action( 'admin_enqueue_scripts', array( __CLASS__, 'enqueue_scripts' ) );
|
|
add_action( 'admin_bar_menu', array( __CLASS__, 'admin_bar_menu' ), 100 );
|
|
}
|
|
|
|
static function enqueue_scripts() {
|
|
|
|
$suffix = ( defined( 'SCRIPT_DEBUG' ) && SCRIPT_DEBUG ) ? '' : '.min';
|
|
wp_register_script( 'laps', plugins_url( "js/laps{$suffix}.js", __DIR__ ), array( 'jquery' ), '3.3.1', true );
|
|
wp_register_style( 'laps', plugins_url( "css/laps{$suffix}.css", __DIR__ ) );
|
|
|
|
if ( is_admin_bar_showing() ) {
|
|
wp_enqueue_script( 'laps' );
|
|
wp_enqueue_style( 'laps' );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Render interface and add to the toolbar.
|
|
*
|
|
* @param \WP_Admin_Bar $wp_admin_bar
|
|
*/
|
|
static function admin_bar_menu( $wp_admin_bar ) {
|
|
|
|
if ( ! apply_filters( 'laps_can_see', current_user_can( 'manage_options' ) ) ) {
|
|
return;
|
|
}
|
|
|
|
global $timestart, $wpdb;
|
|
|
|
$mustache = new \Mustache_Engine(
|
|
array(
|
|
'loader' => new \Mustache_Loader_FilesystemLoader( dirname( __DIR__ ) . '/views' ),
|
|
'cache' => new Mustache_Cache_FrozenCache( dirname( __DIR__ ) . '/views/cache' ),
|
|
)
|
|
);
|
|
|
|
self::$stopwatch->stop( 'Toolbar' );
|
|
|
|
$events = self::$stopwatch->getSectionEvents( '__root__' );
|
|
$start = $timestart * 1000;
|
|
$end = microtime( true ) * 1000;
|
|
$total = $end - $start;
|
|
$event_data = array();
|
|
$http_data = array();
|
|
|
|
foreach ( $events as $name => $event ) {
|
|
|
|
$offset = round( ( $event->getOrigin() - $start ) / $total * 100, 2 );
|
|
$duration = $event->getDuration();
|
|
$width = round( $duration / $total * 100, 2 );
|
|
$category = $event->getCategory();
|
|
|
|
if ( 'http' === $category ) {
|
|
$http_data[] = compact( 'name', 'offset', 'duration', 'width', 'category' );
|
|
continue;
|
|
}
|
|
|
|
$memory = $event->getMemory() / 1024 / 1024;
|
|
|
|
$event_data[] = compact( 'name', 'offset', 'duration', 'width', 'category', 'memory' );
|
|
}
|
|
|
|
$query_data = array();
|
|
$last_query_end = 0;
|
|
$last_offset = 0;
|
|
$last_duration = 0;
|
|
$category = 'query';
|
|
|
|
if ( defined( 'SAVEQUERIES' ) && SAVEQUERIES ) {
|
|
|
|
foreach ( $wpdb->queries as $key => $query ) {
|
|
$query_start = isset( self::$query_starts[ $key ] ) ? self::$query_starts[ $key ] : $last_query_end;
|
|
list( $sql, $duration, $trace ) = $query;
|
|
$duration *= 1000;
|
|
$last_query_end = $query_start + $duration;
|
|
$offset = round( ( $query_start - $start ) / $total * 100, 2 );
|
|
|
|
// if query is indistinguishably close to previous then stack it
|
|
if ( $offset === $last_offset ) {
|
|
$key = count( $query_data ) - 1;
|
|
$query_data[ $key ]['sql'] .= '<br />' . $sql;
|
|
|
|
$last_duration += $duration;
|
|
$width = round( $last_duration / $total * 100, 2 );
|
|
$query_data[ $key ]['width'] = $width;
|
|
|
|
continue;
|
|
}
|
|
|
|
$width = round( $duration / $total * 100, 2 );
|
|
$last_offset = $offset;
|
|
$last_duration = $duration;
|
|
|
|
$query_data[] = compact( 'sql', 'duration', 'offset', 'width', 'category' );
|
|
}
|
|
}
|
|
|
|
$html = $mustache->render(
|
|
'laps',
|
|
array(
|
|
'events' => $event_data,
|
|
'queries' => $query_data,
|
|
'savequeries' => defined( 'SAVEQUERIES' ) && SAVEQUERIES,
|
|
'http' => $http_data,
|
|
'savehttp' => ! empty( $http_data ),
|
|
)
|
|
);
|
|
|
|
$wp_admin_bar->add_node( array(
|
|
'id' => 'laps',
|
|
'title' => sprintf( 'Lap: %ss', round( $total / 1000, 3 ) ),
|
|
) );
|
|
|
|
$wp_admin_bar->add_node( array(
|
|
'id' => 'laps_output',
|
|
'parent' => 'laps',
|
|
'meta' => array( 'html' => $html ),
|
|
) );
|
|
}
|
|
}
|