<?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.
	 */
	public 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
	 */
	public 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
	 */
	public 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
	 */
	public static function tick( $input = null ) {

		global $wp_filter;

		$filter_name     = current_filter();
		$filter_instance = $wp_filter[ $filter_name ];
		$priority        = $filter_instance instanceof \WP_Hook ? $filter_instance->current_priority() : key( $filter_instance );

		$event = wp_parse_args(
			self::$events[ $filter_name ][ $priority ],
			array(
				'action'   => 'start',
				'category' => null,
			)
		);

		if ( 'stop' === $event['action'] && ! self::$stopwatch->isStarted( $event['event'] ) ) {
			return $input;
		}

		self::$stopwatch->{$event['action']}( $event['event'], $event['category'] );

		return $input;
	}

	/**
	 * Capture SQL queries start times
	 *
	 * @param string $query
	 *
	 * @return string
	 */
	public 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
	 */
	public 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
	 */
	public 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.
	 */
	public 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() );
		}
	}

	public 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 );
	}

	public 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
	 */
	public 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' ),
			)
		);

		if ( self::$stopwatch->isStarted( 'Toolbar' ) ) {
			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;

		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;
				$sql      = trim( $sql );
				$category = 'query-read';

				if ( 0 === stripos( $sql, 'INSERT' ) || 0 === stripos( $sql, 'UPDATE' ) ) {
					$category = 'query-write';
				}

				$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 ),
		) );
	}
}