class Rainbow::StringUtils

def self.uncolor(string)
  # See http://www.commandlinefu.com/commands/view/3584/remove-color-codes-special-characters-with-sed
  string.gsub(/\e\[[0-9;]*m/, '')
end

def self.wrap_with_sgr(string, codes)
  return string if codes.empty?
  seq = "\e[" + codes.join(";") + "m"
  match = string.match(/^(\e\[([\d;]+)m)*/)
  seq_pos = match.end(0)
  string = string[0...seq_pos] + seq + string[seq_pos..-1]
  string += "\e[0m" unless string =~ /\e\[0m$/
  string
end