module Rainbow
  class StringUtils

    def self.wrap_with_sgr(string, codes)
      return string if codes.empty?

      seq = "\e[" + codes.join(";") + "m"
      match = string.match(/^(\e\[([\d;]+)m)*/)

      if match
        seq_pos = match.end(0)
        string = string[0...seq_pos] + seq + string[seq_pos..-1]
      else
        string = seq + string
      end

      string = string + "\e[0m" unless string =~ /\e\[0m$/

      string
    end

  end
end