Initial commit

src/class-laps.php

@@ -0,0 +1,175 @@
+<?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();
+
+	/**
+	 * 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( 'after_setup_theme', array( __CLASS__, 'after_setup_theme' ), 15 );
+		add_action( 'init', array( __CLASS__, 'init' ) );
+	}
+
+	/**
+	 * 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;
+	}
+
+	/**
+	 * When theme is done possibly add theme-specific events.
+	 */
+	static function after_setup_theme() {
+
+		foreach ( array( 'THA', 'Hybrid', 'Genesis', 'Thematic' ) as $theme ) {
+
+			$class = "Rarst\\Laps\\{$theme}_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() {
+
+		wp_register_style( 'laps-hint', plugins_url( 'css/hint.css', __DIR__ ), array(), '1.2.1' );
+		wp_register_style( 'laps', plugins_url( 'css/laps.css', __DIR__ ), array( 'laps-hint' ) );
+
+		if ( is_admin_bar_showing() )
+			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;
+
+		$mustache = new \Mustache_Engine( array(
+			'loader' => new \Mustache_Loader_FilesystemLoader( dirname( __DIR__ ) . '/views' ),
+		) );
+
+		$events     = self::$stopwatch->getSectionEvents( '__root__' );
+		$start      = $timestart * 1000;
+		$end        = microtime( true ) * 1000;
+		$total      = $end - $start;
+		$event_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();
+			$memory   = $event->getMemory() / 1024 / 1024;
+
+			$event_data[] = compact( 'name', 'offset', 'duration', 'width', 'category', 'memory' );
+		}
+
+		$html = $mustache->render( 'laps', array( 'events' => $event_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 ),
+		) );
+	}
+}