Use an app_env to get color intensity

1 files changed, 17 insertions, 7 deletions

diff --git a/src/rebar_log.erl b/src/rebar_log.erl
index eb37869..9eb3da6 100644
--- a/src/rebar_log.erl
+++ b/src/rebar_log.erl
@@ -30,6 +30,7 @@
          set_level/1,
          error_level/0,
          default_level/0,
+         intensity/0,
          log/3,
          is_verbose/1]).
 
@@ -43,14 +44,23 @@
 %% Public API
 %% ===================================================================
 
+%% @doc Returns the color intensity, we first check the application envorinment
+%% if that is not set we check the environment variable REBAR_COLOR.
 intensity() ->
-    case os:getenv("REBAR_COLOR") of
-        "high" ->
-            high;
-        "low" ->
-            low;
-        _ ->
-            ?DFLT_INTENSITY
+    case application:get_env(rebar, color_intensity) of
+        undefined ->
+            R = case os:getenv("REBAR_COLOR") of
+                    "high" ->
+                        high;
+                    "low" ->
+                        low;
+                    _ ->
+                        ?DFLT_INTENSITY
+                end,
+            application:set_env(rebar, color_intensity, R),
+            R;
+        {ok, Mode} ->
+            Mode
     end.
 
 init(Caller, Verbosity) ->