Add initial support for color output in the build system

lib/naclports/util.py

 # Copyright 2014 The Native Client Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import fcntl
 import hashlib
 import os
 import shutil
 import subprocess
 import sys
+import termcolor
 
 from naclports import error, paths
 
 GS_URL = 'http://storage.googleapis.com/'
 GS_BUCKET = 'naclports'
 GS_MIRROR_URL = '%s%s/mirror' % (GS_URL, GS_BUCKET)
 
 arch_to_pkgarch = {
   'x86_64': 'x86-64',
   'i686': 'i686',
   'arm': 'arm',
   'pnacl': 'pnacl',
 }
 
 # Inverse of arch_to_pkgarch
 pkgarch_to_arch = {v:k for k, v in arch_to_pkgarch.items()}
 
 verbose = False
+color_mode = 'auto'
+
+def Color(message, color):
+  if Color.enabled:
+    return termcolor.colored(message, color)
+  else:
+    return message
+
+
+def CheckStdoutForColorSupport():
+  if color_mode == 'auto':
+    Color.enabled = sys.stdout.isatty()
 
 
 def Memoize(f):
   """Memoization decorator for functions taking one or more arguments."""
   class Memo(dict):
     def __init__(self, f):
       super(Memo, self).__init__()
       self.f = f
 
     def __call__(self, *args):
@@ skipping 10 matching lines @@
   global verbose
   verbose = verbosity
 
 
 def Log(message):
   """Log a message to the console (stdout)."""
   sys.stdout.write(str(message) + '\n')
   sys.stdout.flush()
 
 
+def LogHeading(message, suffix=''):
+  """Log a colored/highlighted message with optional suffix."""
+  if Color.enabled:
+    Log(Color(message, 'green') + suffix)
+  else:
+    if verbose:
+      # When running in verbose mode make sure heading standout
+      Log('###################################################################')
+      Log(message + suffix)
+      Log('###################################################################')
+    else:
+      Log(message + suffix)
+
+
 def Warn(message):
   Log('warning: ' + message)
 
 
 def Trace(message):
   """Log a message to the console if running in verbose mode (-v)."""
   if verbose:
     Log(message)
 
 
@@ skipping 168 matching lines @@
   This file is written at install time.
   """
   root = GetInstallStampRoot(config)
   return os.path.join(root, package_name + '.list')
 
 
 def IsInstalled(package_name, config, stamp_content=None):
   """Returns True if the given package is installed."""
   stamp = GetInstallStamp(package_name, config)
   result = CheckStamp(stamp, stamp_content)
-  Trace("IsInstalled: %s -> %s" % (package_name, result))
   return result
 
 
 def CheckSDKRoot():
   """Check validity of NACL_SDK_ROOT."""
   root = GetSDKRoot()
   if not root:
     raise error.Error('$NACL_SDK_ROOT not set')
 
   if not os.path.isdir(root):
@@ skipping 86 matching lines @@
   """Lock used when building a package (essentially a lock on OUT_DIR)"""
   def __init__(self):
     super(BuildLock, self).__init__(paths.OUT_DIR)
 
 
 class InstallLock(Lock):
   """Lock used when installing/uninstalling package"""
   def __init__(self, config):
     root = GetInstallRoot(config)
     super(InstallLock, self).__init__(root)
+
+
+CheckStdoutForColorSupport()