Fix colors[i] would burst the index size

sonots · Jan 13, 2013 · 605b31a · 605b31a
1 parent 96b2ac6
commit 605b31a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/capistrano/colorized_stream.rb b/lib/capistrano/colorized_stream.rb
@@ -37,13 +37,16 @@ def colorized(hostname)
             @colorized = {}
             servers = find_servers_for_task(current_task).map(&:to_s)
             len = servers.map(&:length).max
-            servers.each_with_index {|host, i| @colorized[host] = (host.ljust(len) + ' | ').colorize(colors[i]) }
+            servers.each_with_index do |host, i|
+              color = colors[i % colors.size]
+              @colorized[host] = (host.ljust(len) + ' | ').colorize(color)
+            end
           end
           @colorized[hostname]
         end
 
         def colors
-          %w( cyan yellow green magenta red blue light_cyan light_yellow
+          @collors ||= %w( cyan yellow green magenta red blue light_cyan light_yellow
           light_green light_magenta light_red, light_blue ).map(&:to_sym)
         end
       end