121 lines
3.0 KiB
Ruby
121 lines
3.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class Pry
|
|
module Helpers
|
|
def self.tablify_or_one_line(heading, things, config = Pry.config)
|
|
plain_heading = Pry::Helpers::Text.strip_color(heading)
|
|
attempt = Table.new(things, column_count: things.size)
|
|
if attempt.fits_on_line?(Terminal.width! - plain_heading.size - 2)
|
|
"#{heading}: #{attempt}\n"
|
|
else
|
|
"#{heading}: \n#{tablify_to_screen_width(things, { indent: ' ' }, config)}\n"
|
|
end
|
|
end
|
|
|
|
def self.tablify_to_screen_width(things, options, config = Pry.config)
|
|
options ||= {}
|
|
things = things.compact
|
|
if (indent = options[:indent])
|
|
usable_width = Terminal.width! - indent.size
|
|
tablify(things, usable_width, config).to_s.gsub(/^/, indent)
|
|
else
|
|
tablify(things, Terminal.width!, config).to_s
|
|
end
|
|
end
|
|
|
|
def self.tablify(things, line_length, config = Pry.config)
|
|
table = Table.new(things, { column_count: things.size }, config)
|
|
until (table.column_count == 1) || table.fits_on_line?(line_length)
|
|
table.column_count -= 1
|
|
end
|
|
table
|
|
end
|
|
|
|
class Table
|
|
attr_reader :items, :column_count
|
|
def initialize(items, args, config = Pry.config)
|
|
@column_count = args[:column_count]
|
|
@config = config
|
|
self.items = items
|
|
end
|
|
|
|
def to_s
|
|
rows_to_s.join("\n")
|
|
end
|
|
|
|
def rows_to_s(style = :color_on)
|
|
widths = columns.map { |e| _max_width(e) }
|
|
@rows_without_colors.map do |r|
|
|
padded = []
|
|
r.each_with_index do |e, i|
|
|
next unless e
|
|
|
|
item = e.ljust(widths[i])
|
|
item.sub! e, _recall_color_for(e) if style == :color_on
|
|
padded << item
|
|
end
|
|
padded.join(@config.ls.separator)
|
|
end
|
|
end
|
|
|
|
def items=(items)
|
|
@items = items
|
|
_rebuild_colorless_cache
|
|
_recolumn
|
|
end
|
|
|
|
def column_count=(count)
|
|
@column_count = count
|
|
_recolumn
|
|
end
|
|
|
|
def fits_on_line?(line_length)
|
|
_max_width(rows_to_s(:no_color)) <= line_length
|
|
end
|
|
|
|
def columns
|
|
@rows_without_colors.transpose
|
|
end
|
|
|
|
def ==(other)
|
|
items == other.to_a
|
|
end
|
|
|
|
def to_a
|
|
items.to_a
|
|
end
|
|
|
|
private
|
|
|
|
def _max_width(things)
|
|
things.compact.map(&:size).max || 0
|
|
end
|
|
|
|
def _rebuild_colorless_cache
|
|
@colorless_cache = {}
|
|
@plain_items = []
|
|
items.map do |e|
|
|
plain = Pry::Helpers::Text.strip_color(e)
|
|
@colorless_cache[plain] = e
|
|
@plain_items << plain
|
|
end
|
|
end
|
|
|
|
def _recolumn
|
|
@rows_without_colors = []
|
|
return if items.size.zero?
|
|
|
|
row_count = (items.size.to_f / column_count).ceil
|
|
row_count.times do |i|
|
|
row_indices = (0...column_count).map { |e| row_count * e + i }
|
|
@rows_without_colors << row_indices.map { |e| @plain_items[e] }
|
|
end
|
|
end
|
|
|
|
def _recall_color_for(thing)
|
|
@colorless_cache[thing]
|
|
end
|
|
end
|
|
end
|
|
end
|