# coding: utf-8

class StringIrc
  COLOR_CODE = "\x03" # \u0003
  BOLD       = "\x02" # \u0002
  UNDERLINE  = "\x1f" # \u001F
  INVERSE    = "\x16" # \u0016
  CLEAR      = "\x0f" # \u000F

  COLOR_TABLE = {
    0  => %w(white),
    1  => %w(black),
    2  => %w(blue navy),
    3  => %w(green),
    4  => %w(red),
    5  => %w(brown maroon),
    6  => %w(purple),
    7  => %w(orange olive),
    8  => %w(yellow),
    9  => %w(light_green lime),
    10 => %w(teal a_green blue_cyan),
    11 => %w(light_cyan cyan aqua),
    12 => %w(light_blue royal),
    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_TABLE.each do |code, colors|
    colors.each do |color|
      @@color_name_table[color] = code
      define_method(color) do |*args|
        bg_color = args.first || nil
        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}#{"%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
  end

  def underline
    add_code_l UNDERLINE
  end

  def inverse
    add_code_l INVERSE
  end

  def stringfy
    "#{@string}#{CLEAR}"
  end

  alias_method :to_s, :stringfy

  private
  def add_code_l(code)
    @string = "#{code}#{@string}"
    self
  end
end