module HighLine::Wrapper

def self.actual_length(string_with_escapes)
  string_with_escapes.to_s.gsub(/\e\[\d{1,2}m/, "").length
end

def self.wrap(text, wrap_at)
  return text unless wrap_at
  wrap_at = Integer(wrap_at)
  wrapped = []
  text.each_line do |line|
    # take into account color escape sequences when wrapping
    wrap_at += (line.length - actual_length(line))
    while line =~ /([^\n]{#{wrap_at + 1},})/
      search  = Regexp.last_match(1).dup
      replace = Regexp.last_match(1).dup
      index = replace.rindex(" ", wrap_at)
      if index
        replace[index, 1] = "\n"
        replace.sub!(/\n[ \t]+/, "\n")
        line.sub!(search, replace)
      else
        line[$LAST_MATCH_INFO.begin(1) + wrap_at, 0] = "\n"
      end
    end
    wrapped << line
  end
  wrapped.join
end