--- /dev/null
+<?php
+
+/**
+ * New debugger system that outputs a toolbar on page view
+ *
+ * @todo Clean up HTML generated by the javascript
+ * @todo Profiler support
+ */
+class MWDebug {
+
+ /**
+ * Log lines
+ *
+ * @var array
+ */
+ protected static $log = array();
+
+ /**
+ * Debug messages from wfDebug()
+ *
+ * @var array
+ */
+ protected static $debug = array();
+
+ /**
+ * Queries
+ *
+ * @var array
+ */
+ protected static $query = array();
+
+ /**
+ * Request information
+ *
+ * @var array
+ */
+ protected static $request = array();
+
+ /**
+ * Is the debugger enabled?
+ *
+ * @var bool
+ */
+ protected static $enabled = true;
+
+ /**
+ * Called in Setup.php, initializes the debugger if it is enabled with
+ * $wgDebugToolbar
+ */
+ public static function init() {
+ global $wgDebugToolbar;
+
+ if ( !$wgDebugToolbar ) {
+ self::$enabled = false;
+ return;
+ }
+
+ RequestContext::getMain()->getOutput()->addModules( 'mediawiki.debug' );
+ }
+
+ /**
+ * Adds a line to the log
+ *
+ * This does nothing atm, there's not frontend for it
+ *
+ * @todo Add error and warning log type
+ * @todo Add support for passing objects
+ *
+ * @param $str string
+ */
+ public static function log( $str ) {
+ if ( !self::$enabled ) {
+ return;
+ }
+
+ self::$log[] = $str;
+ }
+
+ /**
+ * This is a method to pass messages from wfDebug to the pretty debugger.
+ * Do NOT use this method, use MWDebug::log or wfDebug()
+ *
+ * @param $str string
+ */
+ public static function debugMsg( $str ) {
+ if ( !self::$enabled ) {
+ return;
+ }
+
+ self::$debug[] = trim( $str );
+ }
+
+ /**
+ * Begins profiling on a database query
+ *
+ * @param $sql string
+ * @param $function string
+ * @param $isMaster bool
+ * @return int ID number of the query to pass to queryTime or -1 if the
+ * debugger is disabled
+ */
+ public static function query( $sql, $function, $isMaster ) {
+ if ( !self::$enabled ) {
+ return -1;
+ }
+
+ self::$query[] = array(
+ 'sql' => $sql,
+ 'function' => $function,
+ 'master' => (bool) $isMaster,
+ 'time' > 0.0,
+ '_start' => microtime( true ),
+ );
+
+ return count( self::$query ) - 1;
+ }
+
+ /**
+ * Calculates how long a query took.
+ *
+ * @param $id int
+ */
+ public static function queryTime( $id ) {
+ if ( $id === -1 || !self::$enabled ) {
+ return;
+ }
+
+ self::$query[$id]['time'] = microtime( true ) - self::$query[$id]['_start'];
+ unset( self::$query[$id]['_start'] );
+ }
+
+ /**
+ * Processes a WebRequest object
+ *
+ * @param $request WebRequest
+ */
+ public static function processRequest( WebRequest $request ) {
+ if ( !self::$enabled ) {
+ return;
+ }
+
+ self::$request = array(
+ 'method' => $_SERVER['REQUEST_METHOD'],
+ 'url' => $request->getRequestURL(),
+ 'headers' => $request->getAllHeaders(),
+ 'params' => $request->getValues(),
+ );
+ }
+
+ /**
+ * Returns a list of files included, along with their size
+ *
+ * @return array
+ */
+ protected static function getFilesIncluded() {
+ $files = get_included_files();
+ $fileList = array();
+ foreach ( $files as $file ) {
+ $size = filesize( $file );
+ $fileList[] = array(
+ 'name' => $file,
+ 'size' => self::formatBytes( $size ),
+ );
+ }
+
+ return $fileList;
+ }
+
+ /**
+ * Returns the HTML to add to the page for the toolbar
+ *
+ * @return string
+ */
+ public static function getDebugHTML() {
+ if ( !self::$enabled ) {
+ return '';
+ }
+
+ global $wgVersion, $wgRequestTime;
+ $debugInfo = array(
+ 'mwVersion' => $wgVersion,
+ 'phpVersion' => PHP_VERSION,
+ 'time' => microtime( true ) - $wgRequestTime,
+ 'log' => self::$log,
+ 'debugLog' => self::$debug,
+ 'queries' => self::$query,
+ 'request' => self::$request,
+ 'memory' => self::formatBytes( memory_get_usage() ),
+ 'memoryPeak' => self::formatBytes( memory_get_peak_usage() ),
+ 'includes' => self::getFilesIncluded(),
+ );
+ // TODO: Clean this up
+ $html = Html::openElement( 'script' );
+ $html .= 'var debugInfo = ' . Xml::encodeJsVar( $debugInfo ) . ';';
+ $html .= " $(function() { mw.loader.using( 'mediawiki.debug', function() { mw.Debug.init( debugInfo ) } ); }); ";
+ $html .= Html::closeElement( 'script' );
+
+ return $html;
+ }
+
+ /**
+ * Formats raw bytes integer into a human readable format
+ *
+ * @author John Himmelman - http://stackoverflow.com/a/2510540/343911
+ * @param $size int
+ * @param $precision int
+ * @return string
+ */
+ protected static function formatBytes( $size, $precision = 2 ) {
+ $base = log( $size ) / log( 1024 );
+ // If we ever use 1TB of RAM we're fucked
+ $suffixes = array( '', 'kb', 'MB', 'GB', 'TB' );
+
+ return round( pow( 1024, $base - floor( $base ) ), $precision ) . $suffixes[floor( $base )];
+ }
+}
\ No newline at end of file