Sha256: 73539288323bbab4a2ec849b3440a2a0f7438e5e281c4c6e6f5e1655ab003ae3
Contents?: true
Size: 1.2 KB
Versions: 12
Compression:
Stored size: 1.2 KB
Contents
# frozen_string_literal: true
# InsertAction to add code to the node.
class NodeMutation::InsertAction < NodeMutation::Action
# Initialize an InsertAction.
#
# @param ndoe [Node]
# @param code [String] to be inserted
# @param at [String] position to insert, beginning or end
# @param to [<nil|String>] name of child node
# @param and_comma [Boolean] insert extra comma.
def initialize(node, code, at: 'end', to: nil, and_comma: false)
super(node, code)
@at = at
@to = to
@and_comma = and_comma
end
# The rewritten source code.
#
# @return [String] rewritten code.
def new_code
if @and_comma
@at == 'end' ? ", #{rewritten_source}" : "#{rewritten_source}, "
else
rewritten_source
end
end
private
# Calculate the begin and end positions.
def calculate_position
@start =
if @at == 'end'
if @to
NodeMutation.adapter.child_node_range(@node, @to).end
else
NodeMutation.adapter.get_end(@node)
end
else
if @to
NodeMutation.adapter.child_node_range(@node, @to).start
else
NodeMutation.adapter.get_start(@node)
end
end
@end = @start
end
end
Version data entries
12 entries across 12 versions & 1 rubygems