module Style
  @map = {
    :colour => {
      :red => 31,
      :green => 32,
      :yellow => 33,
      :blue => 34,
      :magenta => 35,
      :cyan => 36,
      :white => 37,
      :null => 0
    },
    :style => {
      :reset => 0,
      :bold => 1,
      :underline => 4,
      :normal => ""
    }
  }

  def self.format(message, colour = nil, style = nil)
    c = @map[:colour][colour.to_sym] if !colour.nil?

    if style.nil?
      t = 0
    else
      t = @map[:style][style.to_sym]
    end
    
    "\e[#{t};#{c}m#{message}\e[0m"
  end
end