From 433b60786bf54377f714534ec652d34006204911 Mon Sep 17 00:00:00 2001
From: Brion Vibber <brion@users.mediawiki.org>
Date: Thu, 23 Sep 2004 05:40:10 +0000
Subject: [PATCH] Add color to the output; assumes a more or less ANSI-flavored
 terminal. Won't display color on Windows by default; use --color or
 --color=no to override.

---
 maintenance/parserTests.php | 38 +++++++++++++++++++++++++++++++++----
 1 file changed, 34 insertions(+), 4 deletions(-)

diff --git a/maintenance/parserTests.php b/maintenance/parserTests.php
index e88d0ded28..012bfea4e1 100644
--- a/maintenance/parserTests.php
+++ b/maintenance/parserTests.php
@@ -31,6 +31,20 @@ include_once( 'InitialiseMessages.inc' );
 $wgTitle = Title::newFromText( 'Parser test script' );
 
 class ParserTest {
+	function ParserTest() {
+		if( isset( $_SERVER['argv'] ) && in_array( '--color', $_SERVER['argv'] ) ) {
+			$this->color = true;
+		} elseif( isset( $_SERVER['argv'] ) && in_array( '--color=yes', $_SERVER['argv'] ) ) {
+			$this->color = true;
+		} elseif( isset( $_SERVER['argv'] ) && in_array( '--color=no', $_SERVER['argv'] ) ) {
+			$this->color = false;
+		} elseif( wfIsWindows() ) {
+			$this->color = false;
+		} else {
+			$this->color = true;
+		}
+	}
+	
 	function runTestsFromFile( $filename ) {
 		$infile = fopen( $filename, 'rt' );
 		if( !$infile ) {
@@ -95,7 +109,7 @@ class ParserTest {
 	 * @return bool
 	 */
 	function runTest( $desc, $input, $result ) {
-		print "Running test $desc...";
+		print "Running test $desc... ";
 
 		$this->setupGlobals();
 		
@@ -152,12 +166,12 @@ class ParserTest {
 	}
 	
 	function showSuccess( $desc ) {
-		print "ok\n";
+		print $this->termColor( '1;32' ) . 'PASSED' . $this->termReset() . "\n";
 		return true;
 	}
 	
 	function showFailure( $desc, $result, $html ) {
-		print "FAILED\n";
+		print $this->termColor( '1;31' ) . 'FAILED!' . $this->termReset() . "\n";
 		#print "!! Expected:\n$result\n";
 		#print "!! Received:\n$html\n!!\n";
 		print $this->quickDiff( $result, $html );
@@ -177,7 +191,7 @@ class ParserTest {
 		unlink( $infile );
 		unlink( $outfile );
 		
-		return $diff;
+		return $this->colorDiff( $diff );
 	}
 	
 	function dumpToFile( $data, $filename ) {
@@ -185,6 +199,22 @@ class ParserTest {
 		fwrite( $file, rtrim( $data ) . "\n" );
 		fclose( $file );
 	}
+	
+	function termColor( $color ) {
+		return $this->color ? "\x1b[{$color}m" : '';
+	}
+	
+	function termReset() {
+		return $this->color ? "\x1b[0m" : '';
+	}
+	
+	function colorDiff( $text ) {
+		return preg_replace(
+			array( '/^(-.*)$/m', '/^(\+.*)$/m' ),
+			array( $this->termColor( 34 ) . '$1' . $this->termReset(),
+			       $this->termColor( 31 ) . '$1' . $this->termReset() ),
+			$text );
+	}
 }
 
 $tester =& new ParserTest();
-- 
2.20.1