Sha256: 45d9ca89aa9ab3fc546180c610a78cefaa6f6920731dc1d651cf837f340b3601

Contents?: true

Size: 1.23 KB

Versions: 8

Compression:

Stored size: 1.23 KB

Contents

# Base class for formatted textual dump of a "model"
#
class Puppet::Pops::Model::TreeDumper
  attr_accessor :indent_count
  def initialize initial_indentation = 0
    @@dump_visitor ||= Puppet::Pops::Visitor.new(nil,"dump",0,0)
    @indent_count = initial_indentation
  end

  def dump(o)
    format(do_dump(o))
  end

  def do_dump(o)
    @@dump_visitor.visit_this(self, o)
  end

  def indent
    "  " * indent_count
  end

  def format(x)
    result = ""
    parts = format_r(x)
    parts.each_index do |i|
      if i > 0
        # separate with space unless previous ends with whitepsace or (
        result << ' ' if parts[i] != ")" && parts[i-1] !~ /.*(?:\s+|\()$/ && parts[i] !~ /\s+/
      end
      result << parts[i]
    end
    result
  end

  def format_r(x)
    result = []
    case x
    when :break
      result << "\n" + indent
    when :indent
      @indent_count += 1
    when :dedent
      @indent_count -= 1
    when Array
      result << '('
      result += x.collect {|a| format_r(a) }.flatten
      result << ')'
    when Symbol
      result << x.to_s # Allows Symbols in arrays e.g. ["text", =>, "text"]
    else
      result << x
    end
    result
  end

  def is_nop? o
    o.nil? || o.is_a?(Puppet::Pops::Model::Nop)
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
puppet-3.2.4 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.3 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.3.rc1 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.2 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.1 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.1.rc1 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.0.rc2 lib/puppet/pops/model/tree_dumper.rb
puppet-3.2.0.rc1 lib/puppet/pops/model/tree_dumper.rb