Contents

require 'reek/source/ast_node_class_map'

module Reek
  module Source
    #
    # Adorns an abstract syntax tree with mix-in modules to make accessing
    # the tree more understandable and less implementation-dependent.
    #
    class TreeDresser
      def initialize(klass_map = AstNodeClassMap.new)
        @klass_map = klass_map
      end

      def dress(sexp, comment_map)
        return sexp unless sexp.is_a? AST::Node
        type = sexp.type
        children = sexp.children.map { |child| dress(child, comment_map) }
        comments = comment_map[sexp]
        @klass_map.klass_for(type).new(type, children,
                                       location: sexp.loc, comments: comments)
      end
    end
  end
end

Version data entries

14 entries across 14 versions & 1 rubygems

Version	Path
reek-2.0.4	lib/reek/source/tree_dresser.rb
reek-2.0.3	lib/reek/source/tree_dresser.rb
reek-2.0.2	lib/reek/source/tree_dresser.rb
reek-2.0.1	lib/reek/source/tree_dresser.rb
reek-2.0.0	lib/reek/source/tree_dresser.rb
reek-1.6.6	lib/reek/source/tree_dresser.rb
reek-1.6.5	lib/reek/source/tree_dresser.rb
reek-1.6.4	lib/reek/source/tree_dresser.rb
reek-1.6.3	lib/reek/source/tree_dresser.rb
reek-1.6.2	lib/reek/source/tree_dresser.rb
reek-1.6.1	lib/reek/source/tree_dresser.rb
reek-1.6.0	lib/reek/source/tree_dresser.rb
reek-1.5.1	lib/reek/source/tree_dresser.rb
reek-1.5.0	lib/reek/source/tree_dresser.rb