lib/dtc_rake/ui.rb in dtc_rake-0.2.0 vs lib/dtc_rake/ui.rb in dtc_rake-1.0.0.pre.beta.1

@@ -1,39 +1,47 @@
 module DtcRake
   module UI
     def error(msg)
-      msg = Rainbow(msg).red if colorize?
+      if DtcRake::UI.colorize?
+        msg = Rainbow(msg).red
+      end
       abort msg
     end
     module_function :error
 
     def announce(msg)
-      msg = Rainbow(msg).aqua if colorize?
+      if DtcRake::UI.colorize?
+        msg = Rainbow(msg).aqua
+      end
       puts msg
     end
     module_function :announce
 
     def success(msg)
-      msg = Rainbow(msg).green if colorize?
+      if DtcRake::UI.colorize?
+        msg = Rainbow(msg).green
+      end
       puts msg
     end
     module_function :success
 
     def warning(msg)
-      msg = Rainbow(msg).yellow if colorize?
+      if DtcRake::UI.colorize?
+        msg = Rainbow(msg).yellow
+      end
       puts msg
     end
     module_function :warning
 
     def info(msg)
-      msg = Rainbow(msg).white if colorize?
+      if DtcRake::UI.colorize?
+        msg = Rainbow(msg).white
+      end
       puts msg
     end
     module_function :info
 
-    private
-
     def colorize?
       return @colorize if defined? @colorize
 
       colorize = Config.instance.colorize
       if colorize
@@ -43,7 +51,8 @@
           colorize = false
         end
       end
       @colorize = colorize
     end
+    module_function :colorize?
   end
 end