From 75ec695a62b0f4c7312f8e386d1005a75ae76a6b Mon Sep 17 00:00:00 2001
From: Vitalii Elenhaupt <velenhaupt@gmail.com>
Date: Wed, 1 Nov 2017 22:37:21 +0200
Subject: [PATCH] Colorize dot formatter

closes #1
---
 src/ameba/formatter.cr | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/ameba/formatter.cr b/src/ameba/formatter.cr
index cd666b1f..42b0111e 100644
--- a/src/ameba/formatter.cr
+++ b/src/ameba/formatter.cr
@@ -13,7 +13,7 @@ module Ameba
 
     def start(sources)
       puts formatter.before sources
-      puts "\n\n"
+      puts "\n"
     end
 
     def report(source)
@@ -36,7 +36,7 @@ module Ameba
     end
 
     def format(source : Source)
-      source.errors.size == 0 ? "." : "F"
+      source.valid? ? ".".colorize(:green) : "F".colorize(:red)
     end
 
     def after(sources)
@@ -44,15 +44,17 @@ module Ameba
         failures = sources.select { |s| s.errors.any? }
         l = failures.map { |f| f.errors.size }.sum
 
-        mes << "#{sources.size} inspected, #{l} failure#{"s" if l != 1}.\n\n"
-
         failures.each do |failure|
           failure.errors.each do |error|
-            mes << "#{failure.path}:#{error.pos}"
-            mes << "\n"
-            mes << "#{error.rule.name}: #{error.message}\n\n"
+            mes << "#{failure.path}:#{error.pos}\n".colorize(:cyan)
+            mes << "#{error.rule.name}: #{error.message}".colorize(:red)
+            mes << "\n\n"
           end
         end
+
+        color = l == 0 ? :green : :red
+        mes << "#{sources.size} inspected, #{l} failure#{"s" if l != 1}."
+          .colorize(color)
       end
     end
   end