class Comments

# A target for attaching comments that is based on a specific node's

# location.

attr_reader :node

def initialize(node)

# A target for attaching comments that is based on a location field on a

# node. For example, the `end` token of a ClassNode.

attr_reader :location

def initialize(location)

end

end

attr_reader :parse_result

def initialize(parse_result)

@parse_result = parse_result

end

def attach!

parse_result.comments.each do |comment|

preceding, enclosing, following = nearest_targets(parse_result.value, comment)

# MarkNewlinesVisitor, since that visitor is responsible for marking the

# newlines for JRuby/TruffleRuby.

class Newlines < Visitor

def initialize(newline_marked)

@newline_marked = newline_marked

end

def visit_block_node(node)

old_newline_marked = @newline_marked

@newline_marked = Array.new(old_newline_marked.size, false)

alias_method :visit_lambda_node, :visit_block_node

def visit_if_node(node)

node.set_newline_flag(@newline_marked)

super(node)

alias_method :visit_unless_node, :visit_if_node

def visit_statements_node(node)

node.body.each do |child|

child.set_newline_flag(@newline_marked)