class Alf::Renderer class Text < Alf::Renderer module Utils def looks_a_relation?(value) value.is_a?(Alf::Iterator) or (value.is_a?(Array) && !value.empty? && value.all?{|v| v.is_a?(Hash)}) end def max(x, y) return y if x.nil? return x if y.nil? x > y ? x : y end end include Utils class Cell include Utils def initialize(value) @value = value end def min_width @min_width ||= rendering_lines.inject(0) do |maxl,line| max(maxl,line.size) end end def rendering_lines(size = nil) if size.nil? text_rendering.split(/\n/) elsif @value.is_a?(Numeric) rendering_lines(nil).collect{|l| "%#{size}s" % l} else rendering_lines(nil).collect{|l| "%-#{size}s" % l} end end def text_rendering @text_rendering ||= case (value = @value) when NilClass "[nil]" when Symbol value.inspect when Float "%.7f" % value when Hash value.inspect when Alf::Iterator Text.render(value, "") when Array array_rendering(value) else value.to_s end end def array_rendering(value) if looks_a_relation?(value) Text.render(value, "") elsif value.empty? "[]" else values = value.collect{|x| Cell.new(x).text_rendering} if values.inject(0){|memo,s| memo + s.size} < 20 "[" + values.join(", ") + "]" else "[" + values.join(",\n ") + "]" end end end end # class Cell class Row include Utils def initialize(values) @cells = values.collect{|v| Cell.new(v)} end def min_widths @cells.collect{|cell| cell.min_width} end def rendering_lines(sizes = min_widths) nb_lines = 0 by_cell = @cells.zip(sizes).collect do |cell,size| lines = cell.rendering_lines(size) nb_lines = max(nb_lines, lines.size) lines end grid = (0...nb_lines).collect do |line_i| "| " + by_cell.zip(sizes).collect{|cell_lines, size| cell_lines[line_i] || " "*size }.join(" | ") + " |" end grid.empty? ? ["| |"] : grid end end # class Row class Table include Utils def initialize(records, attributes) @header = Row.new(attributes) @rows = records.collect{|r| Row.new(r)} end def render(buffer = "") sizes = @rows.inject(@header.min_widths) do |memo,row| memo.zip(row.min_widths).collect{|x,y| max(x,y)} end sep = '+-' << sizes.collect{|s| '-' * s}.join('-+-') << '-+' buffer << sep << "\n" buffer << @header.rendering_lines(sizes).first << "\n" buffer << sep << "\n" @rows.each do |row| row.rendering_lines(sizes).each do |line| buffer << line << "\n" end end buffer << sep << "\n" buffer end end # class Table protected def render(input, output) relation = input.to_a attrs = relation.inject([]){|memo,t| memo | t.keys } records = relation.collect{|t| attrs.collect{|a| t[a]} } Table.new(records, attrs).render(output) end def self.render(input, output) new(input).execute(output) end Alf::Renderer.register(:text, "as a text table", self) end # class Text end # class Alf