Contents

require "pp"
require "stringio"

require_relative "format"
require_relative "labeler"
require_relative "location"
require_relative "location_label"
require_relative "source_label"

module CutePrint
  # @api private
  class Formatter

    DEFAULT_WIDTH = 79

    def initialize(opts = {})
      @method = opts.fetch(:method)
      @out = opts.fetch(:out)
      @block = opts.fetch(:block, nil)
      @values = opts.fetch(:values, [])
      @width = opts.fetch(:width, DEFAULT_WIDTH)
      if @block && !@values.empty?
        raise ArgumentError, "arguments and block are mutually exclusive"
      end
      @location_label = nil
    end

    def write
      if values.empty? && !label.empty?
        write_line label.chomp(": ")
      else
        values.each do |value|
          labeler = Labeler.new(@format, @width, label, value)
          write_lines labeler.labeled
        end
      end
    end

    def with_location(format_key)
      location = Location.find
      @location_label = LocationLabel.make(format_key, location)
    end

    def inspect
      @format = Format::Inspect.new
    end

    def pretty_print
      @format = Format::PrettyPrint.new
    end

    private

    def values
      if @block
        [@block.call]
      else
        @values
      end
    end

    def write_lines(lines)
      lines.each do |line|
        write_line line
      end
    end

    def write_line(line)
      line += "\n" unless line =~ /\n\Z/
      @out.print line
    end

    def label
      @label ||= make_label
    end

    def make_label
      [
        (@location_label.to_s if @location_label),
        (SourceLabel.new(@block, @method) if @block),
      ].compact.join
    end

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
cute_print-0.4.0	lib/cute_print/formatter.rb