lib/string-irc.rb in string-irc-0.2.3 vs lib/string-irc.rb in string-irc-0.3.0

@@ -1,9 +1,8 @@
-# -*- coding: utf-8 -*-
+# coding: utf-8
 
 class StringIrc
-
   COLOR_CODE = "\x03" # \u0003
   BOLD       = "\x02" # \u0002
   UNDERLINE  = "\x1f" # \u001F
   INVERSE    = "\x16" # \u0016
   CLEAR      = "\x0f" # \u000F
@@ -25,27 +24,50 @@
     13 => %w(pink light_purple fuchsia),
     14 => %w(grey),
     15 => %w(light_grey silver),
   }
 
+  ORIGINAL_COLOR_SCHEME = {
+    "rainbow"      => %w(red orange yellow green blue light_blue purple), # order is referred: http://ja.wikipedia.org/wiki/%E8%99%B9
+    "seven_eleven" => %w(orange green red)
+  }
+
   def initialize(string = nil)
     @string = string
   end
 
-  color_name_table = {}
+  @@color_name_table = {}
 
-  COLOR_TABLE.each do | code, colors |
+  COLOR_TABLE.each do |code, colors|
     colors.each do |color|
-      color_name_table[color] = code
-      define_method(color) do | *args |
+      @@color_name_table[color] = code
+      define_method(color) do |*args|
         bg_color = args.first || nil
-        if (!!bg_color and color_name_table.include?(bg_color))
-          color_code = "#{COLOR_CODE}#{sprintf("%02d,%02d", code, color_name_table[bg_color])}"
+        raise ArgumentError, "`#{bg_color}' is not allowed to use as background color" if ORIGINAL_COLOR_SCHEME.include?(bg_color)
+        if bg_color && @@color_name_table.include?(bg_color)
+          color_code = "#{COLOR_CODE}#{"%02d,%02d" % [code, @@color_name_table[bg_color]]}"
         else
-          color_code = "#{COLOR_CODE}#{sprintf("%02d", code)}"
+          color_code = "#{COLOR_CODE}#{"%02d" % code}"
         end
         add_code_l(color_code)
       end
+    end
+  end
+
+  ORIGINAL_COLOR_SCHEME.each do |key, order|
+    define_method(key) do |*args|
+      index    = 0
+      bg_color = args.first || nil
+      @string = @string.each_char.map { |char|
+        order = index % ORIGINAL_COLOR_SCHEME[key].size
+        index += 1
+        if bg_color && @@color_name_table.include?(bg_color)
+          "#{COLOR_CODE}#{"%02d,%02d" % [@@color_name_table[ORIGINAL_COLOR_SCHEME[key][order]], @@color_name_table[bg_color]]}#{char}"
+        else
+          "#{COLOR_CODE}#{"%02d" % @@color_name_table[ORIGINAL_COLOR_SCHEME[key][order]]}#{char}"
+        end
+      }.join
+      self
     end
   end
 
   def bold
     add_code_l BOLD