From d84ec88faec1932f6486eeaecb9ab58ad8bf63ba Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Sun, 27 Nov 2011 16:33:43 -0700
Subject: [PATCH] Add color to the colored UI

---
 lib/vagrant/ui.rb | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/lib/vagrant/ui.rb b/lib/vagrant/ui.rb
index 9adebdcb5..9d2ffafd5 100644
--- a/lib/vagrant/ui.rb
+++ b/lib/vagrant/ui.rb
@@ -88,6 +88,19 @@ module Vagrant
     # of messages. This should only be used with a TTY that supports color,
     # but is up to the user of the class to verify this is the case.
     class Colored < Interface
+      # Terminal colors
+      CLEAR  = "\e[0m"
+      YELLOW = "\e[33m"
+      RED    = "\e[31m"
+      GREEN  = "\e[32m"
+
+      # Mapping between type of message and the color to output
+      COLOR_MAP = {
+        :warn    => YELLOW,
+        :error   => RED,
+        :success => GREEN
+      }
+
       # Use some light meta-programming to create the various methods to
       # output text to the UI. These all delegate the real functionality
       # to `say`.
@@ -137,6 +150,9 @@ module Vagrant
         # Format the message
         message = "[#{env.resource}] #{message}" if opts[:prefix]
 
+        # Colorize the message
+        message = "#{COLOR_MAP[type]}#{message}#{CLEAR}" if COLOR_MAP[type]
+
         # Output!
         # TODO: Color
         channel.send(printer, message)