mirror of
https://github.com/Homebrew/brew.git
synced 2025-07-14 16:09:03 +08:00
2cf2c020ba
When stdin is not a tty then the message `stty: stdin isn't a terminal` will be produced. Silence this message and fall back to `tput` when it fails and default to 80 if we get no results at all. Follow-up from #2714.
68 lines
1.2 KiB
Ruby
68 lines
1.2 KiB
Ruby
module Tty
|
|
module_function
|
|
|
|
def strip_ansi(string)
|
|
string.gsub(/\033\[\d+(;\d+)*m/, "")
|
|
end
|
|
|
|
def width
|
|
width = `/bin/stty size 2>/dev/null`.split[1]
|
|
width ||= `/usr/bin/tput cols 2>/dev/null`.split[0]
|
|
width ||= 80
|
|
width.to_i
|
|
end
|
|
|
|
def truncate(string)
|
|
(w = width).zero? ? string.to_s : string.to_s[0, w - 4]
|
|
end
|
|
|
|
COLOR_CODES = {
|
|
red: 31,
|
|
green: 32,
|
|
yellow: 33,
|
|
blue: 34,
|
|
magenta: 35,
|
|
cyan: 36,
|
|
default: 39,
|
|
}.freeze
|
|
|
|
STYLE_CODES = {
|
|
reset: 0,
|
|
bold: 1,
|
|
italic: 3,
|
|
underline: 4,
|
|
strikethrough: 9,
|
|
no_underline: 24,
|
|
}.freeze
|
|
|
|
CODES = COLOR_CODES.merge(STYLE_CODES).freeze
|
|
|
|
def append_to_escape_sequence(code)
|
|
@escape_sequence ||= []
|
|
@escape_sequence << code
|
|
self
|
|
end
|
|
|
|
def current_escape_sequence
|
|
return "" if @escape_sequence.nil?
|
|
"\033[#{@escape_sequence.join(";")}m"
|
|
end
|
|
|
|
def reset_escape_sequence!
|
|
@escape_sequence = nil
|
|
end
|
|
|
|
CODES.each do |name, code|
|
|
define_singleton_method(name) do
|
|
append_to_escape_sequence(code)
|
|
end
|
|
end
|
|
|
|
def to_s
|
|
return "" unless $stdout.tty?
|
|
current_escape_sequence
|
|
ensure
|
|
reset_escape_sequence!
|
|
end
|
|
end
|