mirror of
https://github.com/pry/pry.git
synced 2022-11-09 12:35:05 -05:00
5d7c887129
All credit to johnny5- for this idea:
952568b933
Pros: No longer need to check _pry_.config.color everywhere.
All color is stripped (not just that added by pry)
Cons: Inefficient to add colors and then remove them again.
107 lines
2.8 KiB
Ruby
107 lines
2.8 KiB
Ruby
class Pry
|
|
module Helpers
|
|
|
|
# The methods defined on {Text} are available to custom commands via {Pry::Command#text}.
|
|
module Text
|
|
|
|
COLORS =
|
|
{
|
|
"black" => 0,
|
|
"red" => 1,
|
|
"green" => 2,
|
|
"yellow" => 3,
|
|
"blue" => 4,
|
|
"purple" => 5,
|
|
"magenta" => 5,
|
|
"cyan" => 6,
|
|
"white" => 7
|
|
}
|
|
|
|
class << self
|
|
|
|
COLORS.each_pair do |color, value|
|
|
define_method color do |text|
|
|
"\033[0;#{30+value}m#{text}\033[0m"
|
|
end
|
|
|
|
define_method "bright_#{color}" do |text|
|
|
"\033[1;#{30+value}m#{text}\033[0m"
|
|
end
|
|
end
|
|
|
|
# Remove any color codes from _text_.
|
|
#
|
|
# @param [String, #to_s] text
|
|
# @return [String] _text_ stripped of any color codes.
|
|
def strip_color(text)
|
|
text.to_s.gsub(/\e\[.*?(\d)+m/ , '')
|
|
end
|
|
|
|
# Returns _text_ as bold text for use on a terminal.
|
|
#
|
|
# @param [String, #to_s] text
|
|
# @return [String] _text_
|
|
def bold(text)
|
|
"\e[1m#{text}\e[0m"
|
|
end
|
|
|
|
# Returns `text` in the default foreground colour.
|
|
# Use this instead of "black" or "white" when you mean absence of colour.
|
|
#
|
|
# @param [String, #to_s] text
|
|
# @return [String]
|
|
def default(text)
|
|
text.to_s
|
|
end
|
|
alias_method :bright_default, :bold
|
|
|
|
# Executes the block with `Pry.config.color` set to false.
|
|
# @yield
|
|
# @return [void]
|
|
def no_color(&block)
|
|
boolean = Pry.config.color
|
|
Pry.config.color = false
|
|
yield
|
|
ensure
|
|
Pry.config.color = boolean
|
|
end
|
|
|
|
# Executes the block with `Pry.config.pager` set to false.
|
|
# @yield
|
|
# @return [void]
|
|
def no_pager(&block)
|
|
boolean = Pry.config.pager
|
|
Pry.config.pager = false
|
|
yield
|
|
ensure
|
|
Pry.config.pager = boolean
|
|
end
|
|
|
|
# Returns _text_ in a numbered list, beginning at _offset_.
|
|
#
|
|
# @param [#each_line] text
|
|
# @param [Fixnum] offset
|
|
# @return [String]
|
|
def with_line_numbers(text, offset, color=:blue)
|
|
lines = text.each_line.to_a
|
|
max_width = (offset + lines.count).to_s.length
|
|
lines.each_with_index.map do |line, index|
|
|
adjusted_index = (index + offset).to_s.rjust(max_width)
|
|
"#{self.send(color, adjusted_index)}: #{line}"
|
|
end.join
|
|
end
|
|
|
|
# Returns _text_ indented by _chars_ spaces.
|
|
#
|
|
# @param [String] text
|
|
# @param [Fixnum] chars
|
|
def indent(text, chars)
|
|
text.lines.map { |l| "#{' ' * chars}#{l}" }.join
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
|