1
0
Fork 0
mirror of https://github.com/pry/pry.git synced 2022-11-09 12:35:05 -05:00
pry--pry/lib/pry/helpers/text.rb

109 lines
2.9 KiB
Ruby
Raw Normal View History

2011-05-06 11:29:41 -04:00
class Pry
module Helpers
# The methods defined on {Text} are available to custom commands via {Pry::Command#text}.
module Text
COLORS =
2011-05-06 11:29:41 -04:00
{
"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|
Pry.color ? "\033[0;#{30+value}m#{text}\033[0m" : text.to_s
end
2011-05-06 11:29:41 -04:00
define_method "bright_#{color}" do |text|
Pry.color ? "\033[1;#{30+value}m#{text}\033[0m" : text.to_s
end
2011-05-06 11:29:41 -04:00
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
2011-05-06 11:29:41 -04:00
# Returns _text_ as bold text for use on a terminal.
# _Pry.color_ must be true for this method to perform any transformations.
#
# @param [String, #to_s] text
# @return [String] _text_
def bold(text)
Pry.color ? "\e[1m#{text}\e[0m" : text.to_s
end
2011-05-06 11:29:41 -04:00
2012-06-27 01:30:00 -04:00
# Returns `text` in the default foreground colour.
2011-10-10 00:41:33 -04:00
# Use this instead of "black" or "white" when you mean absence of colour.
#
2012-06-27 01:30:00 -04:00
# @param [String, #to_s] text
# @return [String]
2011-10-10 00:41:33 -04:00
def default(text)
text.to_s
end
alias_method :bright_default, :bold
2012-06-27 01:30:00 -04:00
# Executes the block with `Pry.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
2012-06-27 01:30:00 -04:00
# 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
2011-05-06 11:29:41 -04:00
# 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
2011-11-06 02:59:49 -05:00
# 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
2011-05-06 11:29:41 -04:00
end
end
end