File: string_utils.rb

package info (click to toggle)
ruby-rainbow 3.1.1-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 132 kB
  • sloc: ruby: 559; makefile: 6
file content (22 lines) | stat: -rw-r--r-- 525 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# frozen_string_literal: true

module Rainbow
  class StringUtils
    def self.wrap_with_sgr(string, codes)
      return string if codes.empty?

      seq = "\e[" + codes.join(";") + "m"

      string = string.sub(/^(\e\[([\d;]+)m)*/) { |m| m + seq }

      return string if string.end_with? "\e[0m"

      string + "\e[0m"
    end

    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
  end
end