table.rb in herodot-0.1.6

- old
+ new

@@ -1,9 +1,13 @@
 require 'terminal-table'
 
 class Herodot::Table
   HEADERS = %w(Project Branch Time).freeze
+  EMPTY_WORKLOG_MESSAGE = Rainbow('Not enough entries in the worklog.').red +
+                          ' On a tracked repository `git checkouts`'\
+                          ' and `git commits` will add entries.'.freeze
+  COLORS = %i(green yellow blue magenta cyan aqua silver aliceblue indianred).freeze
 
   class << self
     def format_time(time_is_seconds)
       total_seconds = time_is_seconds.to_i
       seconds = total_seconds % 60
@@ -11,10 +15,11 @@
       hours = total_seconds / (60 * 60)
       "#{hours}:#{minutes.to_s.rjust(2, '0')}:#{seconds.to_s.rjust(2, '0')}"
     end
 
     def print(worklogs_totals_per_day)
+      abort EMPTY_WORKLOG_MESSAGE if worklogs_totals_per_day.empty?
       Terminal::Table.new(headings: HEADERS) do |table|
         worklogs_totals_per_day.each do |date, times|
           table.add_separator
           table << [date]
           table.add_separator
@@ -24,13 +29,17 @@
       end
     end
 
     private
 
+    def colorize(project)
+      Rainbow(project).color(COLORS[project.chars.map(&:ord).reduce(:+) % COLORS.size])
+    end
+
     def print_day(times)
       times.values.group_by { |log| log[:project] }.flat_map do |_, logs|
         logs.map do |log|
-          [log[:project], log[:branch], format_time(log[:time])]
+          [colorize(log[:project]), log[:branch], format_time(log[:time])]
         end
       end
     end
   end
 end