Cater for Windows bash. Use function_exists. Check TERM. Test arg.

Author: gitlost

lib/cli/Shell.php

@@ -21,27 +21,43 @@ class Shell {
 	/**
 	 * Returns the number of columns the current shell has for display.
 	 *
+	 * @param mixed $test For testing only.
+	 *
 	 * @return int  The number of columns.
 	 * @todo Test on more systems.
 	 */
-	static public function columns() {
+	static public function columns( $test = null ) {
 		static $columns;
 
+		if ( null !== $test ) {
+			$columns = null;
+		}
 		if ( null === $columns ) {
-			if (self::is_windows() ) {
-				$output = array();
-				exec('mode CON', $output);
-				foreach ($output as $line) {
-					if (preg_match('/Columns:( )*([0-9]+)/', $line, $matches)) {
-						$columns = (int)$matches[2];
-						break;
+			if ( function_exists( 'exec' ) ) {
+				if ( self::is_windows( 'WIN' === $test ) ) {
+					// Cater for shells such as Cygwin and Git bash where `mode CON` returns an incorrect value for columns.
+					if ( ( $shell = getenv( 'SHELL' ) ) && preg_match( '/(?:bash|zsh)(?:.exe)?$/', $shell ) && getenv( 'TERM' ) ) {
+						$columns = (int) exec( 'tput cols' );
+					}
+					if ( ! $columns ) {
+						$return_var = -1;
+						$output = array();
+						exec( 'mode CON', $output, $return_var );
+						if ( 0 === $return_var && $output ) {
+							// Look for second line ending in ": <number>" (searching for "Columns:" will fail on non-English locales).
+							if ( preg_match( '/:\s*[0-9]+\n[^:]+:\s*([0-9]+)\n/', implode( "\n", $output ), $matches ) ) {
+								$columns = (int) $matches[1];
+							}
+						}
+					}
+				} else {
+					if ( getenv( 'TERM' ) ) {
+						$columns = (int) exec( '/usr/bin/env tput cols' );
 					}
 				}
-			} else if (!preg_match('/(^|,)(\s*)?exec(\s*)?(,|$)/', ini_get('disable_functions'))) {
-				$columns = (int) exec('/usr/bin/env tput cols');
 			}
 
-			if ( !$columns ) {
+			if ( ! $columns ) {
 				$columns = 80; // default width of cmd window on Windows OS
 			}
 		}

tests/test-shell.php

@@ -0,0 +1,39 @@
+<?php
+
+use cli\Shell;
+
+/**
+ * Class TestShell
+ */
+class TestShell extends PHPUnit_Framework_TestCase {
+
+    /**
+     * Test getting TERM columns.
+     */
+    function testColumns() {
+		// Save.
+		$env_term = getenv( 'TERM' );
+		$env_columns = getenv( 'COLUMNS' );
+
+		// No TERM should result in default 80.
+
+		putenv( 'TERM' );
+		$columns = cli\Shell::columns( true /*test*/ );
+		$this->assertSame( 80, $columns );
+		$columns = cli\Shell::columns( 'WIN' /*test*/ );
+		$this->assertSame( 80, $columns );
+
+		// TERM and COLUMNS should result in whatever COLUMNS is.
+
+		putenv( 'TERM=vt100' );
+		putenv( 'COLUMNS=100' );
+		$columns = cli\Shell::columns( true /*test*/ );
+		$this->assertSame( 100, $columns );
+		$columns = cli\Shell::columns( 'WIN' /*test*/ );
+		$this->assertSame( 100, $columns );
+
+		// Restore.
+		putenv( false === $env_term ? 'TERM' : "TERM=$env_term" );
+		putenv( false === $env_columns ? 'COLUMNS' : "COLUMNS=$env_columns" );
+	}
+}