1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
#! /usr/bin/env ruby
# frozen_string_literal: true
# this adds ascii colorization to String class
class String
CLS = "\033[0;0f\033\[2J"
@color_codes = {
black: 0, light_black: 60,
red: 1, light_red: 61,
green: 2, light_green: 62,
yellow: 3, light_yellow: 63,
blue: 4, light_blue: 64,
magenta: 5, light_magenta: 65,
cyan: 6, light_cyan: 66,
white: 7, light_white: 67,
default: 9
}
@color_codes.default = 9
@color_modes = {
default: 0, # Turn off all attributes
bold: 1, # Set bold mode
italic: 3, # Set italic mode
underline: 4, # Set underline mode
blink: 5, # Set blink mode
swap: 7, # Exchange foreground and background colors
hide: 8 # Hide text (foreground color would be the same as background)
}
@color_modes.default = 0
@syms = %i[fg bg mode]
class << self
attr_reader :color_codes, :color_modes, :syms
def create_methods
color_codes.each_key do |cc|
next if cc == :default
define_method(cc) { colorize(fg: cc) }
define_method("on_#{cc}") { colorize(bg: cc) }
end
color_modes.each_key do |cc|
next if cc == :default
define_method(cc) { colorize(mode: cc) }
end
end
end
create_methods
def colorize(opts)
# code = compile_code(opts)
code = opts.each_with_object([]) { |(k, v), a| a << resolve(k, v) if self.class.syms.include? k }.join(';')
return self if code.empty?
apply_code(code)
end
private
RESET = "\033[0m"
START_CODE = /^\033\[([0-9;]+)m/.freeze
# negative lookbehind : (?<! ) + ^ => is not at the start of the line
# negative lookahead : (?! ) + $ => is not at the end of the line
MIDDLE_RESET = /(?<!^)\033\[0m(?!$)/.freeze
def apply_code(code)
# prefix with ascii code
# replace all not ending reset with ascii code
if self =~ START_CODE
prev_start_code = ::Regexp.last_match(1)
code = "\033[#{prev_start_code};#{code}m"
s = sub(START_CODE, code).gsub(/(?<!^)\033\[#{prev_start_code}m(?!$)/, code)
else
code = "\033[#{code}m"
s = (code + self).gsub(MIDDLE_RESET, code)
end
# add terminal reset if needed
(s[-4..] == RESET ? s : s + RESET)
end
def resolve(key, var)
return self.class.color_codes[var] + 30 if key == :fg
return self.class.color_codes[var] + 40 if key == :bg
return self.class.color_modes[var] if key == :mode
end
end
if $PROGRAM_NAME == __FILE__
a = 'blue'.colorize(fg: :blue, bg: nil)
b = 'green'.colorize(fg: nil).on_green
puts "RED >> #{a} #{b} << DER".white.on_red.underline
end
|
