102 lines
2.3 KiB
Ruby
102 lines
2.3 KiB
Ruby
require "utils/tty"
|
|
|
|
module Formatter
|
|
module_function
|
|
|
|
def arrow(string, color: nil)
|
|
prefix("==>", string, color)
|
|
end
|
|
|
|
def headline(string, color: nil)
|
|
arrow("#{Tty.bold}#{string}#{Tty.reset}", color: color)
|
|
end
|
|
|
|
def identifier(string)
|
|
"#{Tty.green}#{string}#{Tty.default}"
|
|
end
|
|
|
|
def option(string)
|
|
"#{Tty.bold}#{string}#{Tty.reset}"
|
|
end
|
|
|
|
def success(string, label: nil)
|
|
label(label, string, :green)
|
|
end
|
|
|
|
def warning(string, label: nil)
|
|
label(label, string, :yellow)
|
|
end
|
|
|
|
def error(string, label: nil)
|
|
label(label, string, :red)
|
|
end
|
|
|
|
def wrap(s, width = 172)
|
|
s.gsub(/(.{1,#{width}})(\s+|\Z)/, "\\1\n")
|
|
end
|
|
|
|
def url(string)
|
|
"#{Tty.underline}#{string}#{Tty.no_underline}"
|
|
end
|
|
|
|
def label(label, string, color)
|
|
label = "#{label}:" unless label.nil?
|
|
prefix(label, string, color)
|
|
end
|
|
private_class_method :label
|
|
|
|
def prefix(prefix, string, color)
|
|
if prefix.nil? && color.nil?
|
|
string
|
|
elsif prefix.nil?
|
|
"#{Tty.send(color)}#{string}#{Tty.reset}"
|
|
elsif color.nil?
|
|
"#{prefix} #{string}"
|
|
else
|
|
"#{Tty.send(color)}#{prefix}#{Tty.reset} #{string}"
|
|
end
|
|
end
|
|
private_class_method :prefix
|
|
|
|
def columns(*objects, gap_size: 2)
|
|
objects = objects.flatten.map(&:to_s)
|
|
|
|
fallback = proc do
|
|
return objects.join("\n").concat("\n")
|
|
end
|
|
|
|
fallback.call if objects.empty?
|
|
fallback.call if respond_to?(:tty?) ? !tty? : !$stdout.tty?
|
|
|
|
console_width = Tty.width
|
|
object_lengths = objects.map { |obj| Tty.strip_ansi(obj).length }
|
|
cols = (console_width + gap_size) / (object_lengths.max + gap_size)
|
|
|
|
fallback.call if cols < 2
|
|
|
|
rows = (objects.count + cols - 1) / cols
|
|
cols = (objects.count + rows - 1) / rows # avoid empty trailing columns
|
|
|
|
col_width = (console_width + gap_size) / cols - gap_size
|
|
|
|
gap_string = "".rjust(gap_size)
|
|
|
|
output = ""
|
|
|
|
rows.times do |row_index|
|
|
item_indices_for_row = row_index.step(objects.size - 1, rows).to_a
|
|
|
|
first_n = item_indices_for_row[0...-1].map do |index|
|
|
objects[index] + "".rjust(col_width - object_lengths[index])
|
|
end
|
|
|
|
# don't add trailing whitespace to last column
|
|
last = objects.values_at(item_indices_for_row.last)
|
|
|
|
output.concat((first_n + last).join(gap_string)).concat("\n")
|
|
end
|
|
|
|
output
|
|
end
|
|
end
|