Prism::Translation::Parser::Compiler

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 39
def initialize(parser, offset_cache, forwarding: [], in_destructure: false, in_pattern: false)
  @parser = parser
  @builder = parser.builder
  @source_buffer = parser.source_buffer
  @offset_cache = offset_cache

  @forwarding = forwarding
  @in_destructure = in_destructure
  @in_pattern = in_pattern
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 2045
def chomped_bytesize(line)
  chomped = line.chomp
  chomped.bytesize + (chomped == line ? 0 : 1)
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1916
def copy_compiler(forwarding: self.forwarding, in_destructure: self.in_destructure, in_pattern: self.in_pattern)
  Compiler.new(parser, offset_cache, forwarding: forwarding, in_destructure: in_destructure, in_pattern: in_pattern)
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1923
def find_forwarding(node)
  return [] if node.nil?

  forwarding = []
  forwarding << :* if node.rest.is_a?(RestParameterNode) && node.rest.name.nil?
  forwarding << :** if node.keyword_rest.is_a?(KeywordRestParameterNode) && node.keyword_rest.name.nil?
  forwarding << :& if !node.block.nil? && node.block.name.nil?
  forwarding |= [:&, :"..."] if node.keyword_rest.is_a?(ForwardingParameterNode)

  forwarding
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1936
def multi_target_elements(node)
  elements = [*node.lefts]
  elements << node.rest if !node.rest.nil? && !node.rest.is_a?(ImplicitRestNode)
  elements.concat(node.rights)
  elements
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1948
def numeric_negate(message_loc, receiver)
  case receiver.type
  when :integer_node, :float_node
    receiver.copy(value: -receiver.value, location: message_loc.join(receiver.location))
  when :rational_node
    receiver.copy(numerator: -receiver.numerator, location: message_loc.join(receiver.location))
  when :imaginary_node
    receiver.copy(numeric: numeric_negate(message_loc, receiver.numeric), location: message_loc.join(receiver.location))
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1962
def procarg0?(parameters)
  parameters &&
    parameters.requireds.length == 1 &&
    parameters.optionals.empty? &&
    parameters.rest.nil? &&
    parameters.posts.empty? &&
    parameters.keywords.empty? &&
    parameters.keyword_rest.nil? &&
    parameters.block.nil?
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1979
def srange(location)
  Range.new(source_buffer, offset_cache[location.start_offset], offset_cache[location.end_offset]) if location
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1994
def srange_find(start_offset, end_offset, tokens)
  if (match = source_buffer.source.byteslice(start_offset...end_offset).match(/\A(\s*)(#{tokens.join("|")})/))
    _, whitespace, token = *match
    token_offset = start_offset + whitespace.bytesize

    [token, Range.new(source_buffer, offset_cache[token_offset], offset_cache[token_offset + token.bytesize])]
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 1984
def srange_offsets(start_offset, end_offset)
  Range.new(source_buffer, offset_cache[start_offset], offset_cache[end_offset])
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 2004
def token(location)
  [location.slice, Range.new(source_buffer, offset_cache[location.start_offset], offset_cache[location.end_offset])] if location
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 58
def visit_alias_global_variable_node(node)
  builder.alias(token(node.keyword_loc), visit(node.new_name), visit(node.old_name))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 52
def visit_alias_method_node(node)
  builder.alias(token(node.keyword_loc), visit(node.new_name), visit(node.old_name))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 64
def visit_alternation_pattern_node(node)
  builder.match_alt(visit(node.left), token(node.operator_loc), visit(node.right))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 70
def visit_and_node(node)
  builder.logical_op(:and, visit(node.left), token(node.operator_loc), visit(node.right))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 105
def visit_arguments_node(node)
  visit_all(node.arguments)
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 76
def visit_array_node(node)
  builder.array(token(node.opening_loc), visit_all(node.elements), token(node.closing_loc))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 82
def visit_array_pattern_node(node)
  elements = [*node.requireds]
  elements << node.rest if !node.rest.nil? && !node.rest.is_a?(ImplicitRestNode)
  elements.concat(node.posts)
  visited = visit_all(elements)

  if node.rest.is_a?(ImplicitRestNode)
    visited[-1] = builder.match_with_trailing_comma(visited[-1], token(node.rest.location))
  end

  if node.constant
    if visited.empty?
      builder.const_pattern(visit(node.constant), token(node.opening_loc), builder.array_pattern(token(node.opening_loc), visited, token(node.closing_loc)), token(node.closing_loc))
    else
      builder.const_pattern(visit(node.constant), token(node.opening_loc), builder.array_pattern(nil, visited, nil), token(node.closing_loc))
    end
  else
    builder.array_pattern(token(node.opening_loc), visited, token(node.closing_loc))
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 111
def visit_assoc_node(node)
  key = node.key

  if in_pattern
    if node.value.is_a?(ImplicitNode)
      if key.is_a?(SymbolNode)
        if key.opening.nil?
          builder.match_hash_var([key.unescaped, srange(key.location)])
        else
          builder.match_hash_var_from_str(token(key.opening_loc), [builder.string_internal([key.unescaped, srange(key.value_loc)])], token(key.closing_loc))
        end
      else
        builder.match_hash_var_from_str(token(key.opening_loc), visit_all(key.parts), token(key.closing_loc))
      end
    elsif key.opening.nil?
      builder.pair_keyword([key.unescaped, srange(key.location)], visit(node.value))
    else
      builder.pair_quoted(token(key.opening_loc), [builder.string_internal([key.unescaped, srange(key.value_loc)])], token(key.closing_loc), visit(node.value))
    end
  elsif node.value.is_a?(ImplicitNode)
    if (value = node.value.value).is_a?(LocalVariableReadNode)
      builder.pair_keyword(
        [key.unescaped, srange(key)],
        builder.ident([value.name, srange(key.value_loc)]).updated(:lvar)
      )
    else
      builder.pair_label([key.unescaped, srange(key.location)])
    end
  elsif node.operator_loc
    builder.pair(visit(key), token(node.operator_loc), visit(node.value))
  elsif key.is_a?(SymbolNode) && key.opening_loc.nil?
    builder.pair_keyword([key.unescaped, srange(key.location)], visit(node.value))
  else
    parts =
      if key.is_a?(SymbolNode)
        [builder.string_internal([key.unescaped, srange(key.value_loc)])]
      else
        visit_all(key.parts)
      end

    builder.pair_quoted(token(key.opening_loc), parts, token(key.closing_loc), visit(node.value))
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 160
def visit_assoc_splat_node(node)
  if in_pattern
    builder.match_rest(token(node.operator_loc), token(node.value&.location))
  elsif node.value.nil? && forwarding.include?(:**)
    builder.forwarded_kwrestarg(token(node.operator_loc))
  else
    builder.kwsplat(token(node.operator_loc), visit(node.value))
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 172
def visit_back_reference_read_node(node)
  builder.back_ref(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 178
def visit_begin_node(node)
  rescue_bodies = []

  if (rescue_clause = node.rescue_clause)
    begin
      find_start_offset = (rescue_clause.reference&.location || rescue_clause.exceptions.last&.location || rescue_clause.keyword_loc).end_offset
      find_end_offset = (rescue_clause.statements&.location&.start_offset || rescue_clause.consequent&.location&.start_offset || (find_start_offset + 1))

      rescue_bodies << builder.rescue_body(
        token(rescue_clause.keyword_loc),
        rescue_clause.exceptions.any? ? builder.array(nil, visit_all(rescue_clause.exceptions), nil) : nil,
        token(rescue_clause.operator_loc),
        visit(rescue_clause.reference),
        srange_find(find_start_offset, find_end_offset, [";"]),
        visit(rescue_clause.statements)
      )
    end until (rescue_clause = rescue_clause.consequent).nil?
  end

  begin_body =
    builder.begin_body(
      visit(node.statements),
      rescue_bodies,
      token(node.else_clause&.else_keyword_loc),
      visit(node.else_clause),
      token(node.ensure_clause&.ensure_keyword_loc),
      visit(node.ensure_clause&.statements)
    )

  if node.begin_keyword_loc
    builder.begin_keyword(token(node.begin_keyword_loc), begin_body, token(node.end_keyword_loc))
  else
    begin_body
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 2009
def visit_block(call, block)
  if block
    parameters = block.parameters
    implicit_parameters = parameters.is_a?(NumberedParametersNode) || parameters.is_a?(ItParametersNode)

    builder.block(
      call,
      token(block.opening_loc),
      if parameters.nil?
        builder.args(nil, [], nil, false)
      elsif implicit_parameters
        visit(parameters)
      else
        builder.args(
          token(parameters.opening_loc),
          if procarg0?(parameters.parameters)
            parameter = parameters.parameters.requireds.first
            visited = parameter.is_a?(RequiredParameterNode) ? visit(parameter) : parameter.accept(copy_compiler(in_destructure: true))
            [builder.procarg0(visited)].concat(visit_all(parameters.locals))
          else
            visit(parameters)
          end,
          token(parameters.closing_loc),
          false
        )
      end,
      block.body&.accept(copy_compiler(forwarding: implicit_parameters ? [] : find_forwarding(parameters&.parameters))),
      token(block.closing_loc)
    )
  else
    call
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 216
def visit_block_argument_node(node)
  builder.block_pass(token(node.operator_loc), visit(node.expression))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 222
def visit_block_local_variable_node(node)
  builder.shadowarg(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 227
def visit_block_node(node)
  raise CompilationError, "Cannot directly compile block nodes"
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 233
def visit_block_parameter_node(node)
  builder.blockarg(token(node.operator_loc), token(node.name_loc))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 238
def visit_block_parameters_node(node)
  [*visit(node.parameters)].concat(visit_all(node.locals))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 247
def visit_break_node(node)
  builder.keyword_cmd(:break, token(node.keyword_loc), nil, visit(node.arguments) || [], nil)
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 352
def visit_call_and_write_node(node)
  call_operator_loc = node.call_operator_loc

  builder.op_assign(
    builder.call_method(
      visit(node.receiver),
      call_operator_loc.nil? ? nil : [{ "." => :dot, "&." => :anddot, "::" => "::" }.fetch(call_operator_loc.slice), srange(call_operator_loc)],
      node.message_loc ? [node.read_name, srange(node.message_loc)] : nil,
      nil,
      [],
      nil
    ),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 259
def visit_call_node(node)
  name = node.name
  arguments = node.arguments&.arguments || []
  block = node.block

  if block.is_a?(BlockArgumentNode)
    arguments = [*arguments, block]
    block = nil
  end

  if node.call_operator_loc.nil?
    case name
    when :-@
      case (receiver = node.receiver).type
      when :integer_node, :float_node, :rational_node, :imaginary_node
        return visit(numeric_negate(node.message_loc, receiver))
      end
    when :!
      return visit_block(builder.not_op(token(node.message_loc), token(node.opening_loc), visit(node.receiver), token(node.closing_loc)), block)
    when :=~
      if (receiver = node.receiver).is_a?(RegularExpressionNode)
        return builder.match_op(visit(receiver), token(node.message_loc), visit(node.arguments.arguments.first))
      end
    when :[]
      return visit_block(builder.index(visit(node.receiver), token(node.opening_loc), visit_all(arguments), token(node.closing_loc)), block)
    when :[]=
      if node.message != "[]=" && node.arguments && block.nil? && !node.safe_navigation?
        arguments = node.arguments.arguments[...-1]
        arguments << node.block if node.block

        return visit_block(
          builder.assign(
            builder.index_asgn(
              visit(node.receiver),
              token(node.opening_loc),
              visit_all(arguments),
              token(node.closing_loc),
            ),
            srange_find(node.message_loc.end_offset, node.arguments.arguments.last.location.start_offset, ["="]),
            visit(node.arguments.arguments.last)
          ),
          block
        )
      end
    end
  end

  message_loc = node.message_loc
  call_operator_loc = node.call_operator_loc
  call_operator = [{ "." => :dot, "&." => :anddot, "::" => "::" }.fetch(call_operator_loc.slice), srange(call_operator_loc)] if call_operator_loc

  visit_block(
    if name.end_with?("=") && !message_loc.slice.end_with?("=") && node.arguments && block.nil?
      builder.assign(
        builder.attr_asgn(visit(node.receiver), call_operator, token(message_loc)),
        srange_find(message_loc.end_offset, node.arguments.location.start_offset, ["="]),
        visit(node.arguments.arguments.last)
      )
    else
      builder.call_method(
        visit(node.receiver),
        call_operator,
        message_loc ? [node.name, srange(message_loc)] : nil,
        token(node.opening_loc),
        visit_all(arguments),
        token(node.closing_loc)
      )
    end,
    block
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 333
def visit_call_operator_write_node(node)
  call_operator_loc = node.call_operator_loc

  builder.op_assign(
    builder.call_method(
      visit(node.receiver),
      call_operator_loc.nil? ? nil : [{ "." => :dot, "&." => :anddot, "::" => "::" }.fetch(call_operator_loc.slice), srange(call_operator_loc)],
      node.message_loc ? [node.read_name, srange(node.message_loc)] : nil,
      nil,
      [],
      nil
    ),
    [node.binary_operator_loc.slice.chomp("="), srange(node.binary_operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 371
def visit_call_or_write_node(node)
  call_operator_loc = node.call_operator_loc

  builder.op_assign(
    builder.call_method(
      visit(node.receiver),
      call_operator_loc.nil? ? nil : [{ "." => :dot, "&." => :anddot, "::" => "::" }.fetch(call_operator_loc.slice), srange(call_operator_loc)],
      node.message_loc ? [node.read_name, srange(node.message_loc)] : nil,
      nil,
      [],
      nil
    ),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 390
def visit_call_target_node(node)
  call_operator_loc = node.call_operator_loc

  builder.attr_asgn(
    visit(node.receiver),
    call_operator_loc.nil? ? nil : [{ "." => :dot, "&." => :anddot, "::" => "::" }.fetch(call_operator_loc.slice), srange(call_operator_loc)],
    token(node.message_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 402
def visit_capture_pattern_node(node)
  builder.match_as(visit(node.value), token(node.operator_loc), visit(node.target))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 421
def visit_case_match_node(node)
  builder.case_match(
    token(node.case_keyword_loc),
    visit(node.predicate),
    visit_all(node.conditions),
    token(node.consequent&.else_keyword_loc),
    visit(node.consequent),
    token(node.end_keyword_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 408
def visit_case_node(node)
  builder.case(
    token(node.case_keyword_loc),
    visit(node.predicate),
    visit_all(node.conditions),
    token(node.consequent&.else_keyword_loc),
    visit(node.consequent),
    token(node.end_keyword_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 434
def visit_class_node(node)
  builder.def_class(
    token(node.class_keyword_loc),
    visit(node.constant_path),
    token(node.inheritance_operator_loc),
    visit(node.superclass),
    node.body&.accept(copy_compiler(forwarding: [])),
    token(node.end_keyword_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 473
def visit_class_variable_and_write_node(node)
  builder.op_assign(
    builder.assignable(builder.cvar(token(node.name_loc))),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 463
def visit_class_variable_operator_write_node(node)
  builder.op_assign(
    builder.assignable(builder.cvar(token(node.name_loc))),
    [node.binary_operator_loc.slice.chomp("="), srange(node.binary_operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 483
def visit_class_variable_or_write_node(node)
  builder.op_assign(
    builder.assignable(builder.cvar(token(node.name_loc))),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 447
def visit_class_variable_read_node(node)
  builder.cvar(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 493
def visit_class_variable_target_node(node)
  builder.assignable(builder.cvar(token(node.location)))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 453
def visit_class_variable_write_node(node)
  builder.assign(
    builder.assignable(builder.cvar(token(node.name_loc))),
    token(node.operator_loc),
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 524
def visit_constant_and_write_node(node)
  builder.op_assign(
    builder.assignable(builder.const([node.name, srange(node.name_loc)])),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 514
def visit_constant_operator_write_node(node)
  builder.op_assign(
    builder.assignable(builder.const([node.name, srange(node.name_loc)])),
    [node.binary_operator_loc.slice.chomp("="), srange(node.binary_operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 534
def visit_constant_or_write_node(node)
  builder.op_assign(
    builder.assignable(builder.const([node.name, srange(node.name_loc)])),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 590
def visit_constant_path_and_write_node(node)
  builder.op_assign(
    builder.assignable(visit(node.target)),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 550
def visit_constant_path_node(node)
  if node.parent.nil?
    builder.const_global(
      token(node.delimiter_loc),
      [node.name, srange(node.name_loc)]
    )
  else
    builder.const_fetch(
      visit(node.parent),
      token(node.delimiter_loc),
      [node.name, srange(node.name_loc)]
    )
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 580
def visit_constant_path_operator_write_node(node)
  builder.op_assign(
    builder.assignable(visit(node.target)),
    [node.binary_operator_loc.slice.chomp("="), srange(node.binary_operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 600
def visit_constant_path_or_write_node(node)
  builder.op_assign(
    builder.assignable(visit(node.target)),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 610
def visit_constant_path_target_node(node)
  builder.assignable(visit_constant_path_node(node))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 570
def visit_constant_path_write_node(node)
  builder.assign(
    builder.assignable(visit(node.target)),
    token(node.operator_loc),
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 499
def visit_constant_read_node(node)
  builder.const([node.name, srange(node.location)])
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 544
def visit_constant_target_node(node)
  builder.assignable(builder.const([node.name, srange(node.location)]))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 508
def visit_constant_write_node(node)
  builder.assign(builder.assignable(builder.const([node.name, srange(node.name_loc)])), token(node.operator_loc), visit(node.value))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 619
def visit_def_node(node)
  if node.equal_loc
    if node.receiver
      builder.def_endless_singleton(
        token(node.def_keyword_loc),
        visit(node.receiver.is_a?(ParenthesesNode) ? node.receiver.body : node.receiver),
        token(node.operator_loc),
        token(node.name_loc),
        builder.args(token(node.lparen_loc), visit(node.parameters) || [], token(node.rparen_loc), false),
        token(node.equal_loc),
        node.body&.accept(copy_compiler(forwarding: find_forwarding(node.parameters)))
      )
    else
      builder.def_endless_method(
        token(node.def_keyword_loc),
        token(node.name_loc),
        builder.args(token(node.lparen_loc), visit(node.parameters) || [], token(node.rparen_loc), false),
        token(node.equal_loc),
        node.body&.accept(copy_compiler(forwarding: find_forwarding(node.parameters)))
      )
    end
  elsif node.receiver
    builder.def_singleton(
      token(node.def_keyword_loc),
      visit(node.receiver.is_a?(ParenthesesNode) ? node.receiver.body : node.receiver),
      token(node.operator_loc),
      token(node.name_loc),
      builder.args(token(node.lparen_loc), visit(node.parameters) || [], token(node.rparen_loc), false),
      node.body&.accept(copy_compiler(forwarding: find_forwarding(node.parameters))),
      token(node.end_keyword_loc)
    )
  else
    builder.def_method(
      token(node.def_keyword_loc),
      token(node.name_loc),
      builder.args(token(node.lparen_loc), visit(node.parameters) || [], token(node.rparen_loc), false),
      node.body&.accept(copy_compiler(forwarding: find_forwarding(node.parameters))),
      token(node.end_keyword_loc)
    )
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 666
def visit_defined_node(node)
  builder.keyword_cmd(
    :defined?,
    token(node.keyword_loc),
    token(node.lparen_loc),
    [visit(node.value)],
    token(node.rparen_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 678
def visit_else_node(node)
  visit(node.statements)
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 684
def visit_embedded_statements_node(node)
  builder.begin(
    token(node.opening_loc),
    visit(node.statements),
    token(node.closing_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 694
def visit_embedded_variable_node(node)
  visit(node.variable)
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 700
def visit_ensure_node(node)
  raise CompilationError, "Cannot directly compile ensure nodes"
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 706
def visit_false_node(node)
  builder.false(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 712
def visit_find_pattern_node(node)
  elements = [node.left, *node.requireds, node.right]

  if node.constant
    builder.const_pattern(visit(node.constant), token(node.opening_loc), builder.find_pattern(nil, visit_all(elements), nil), token(node.closing_loc))
  else
    builder.find_pattern(token(node.opening_loc), visit_all(elements), token(node.closing_loc))
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 724
def visit_float_node(node)
  visit_numeric(node, builder.float([node.value, srange(node.location)]))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 730
def visit_for_node(node)
  builder.for(
    token(node.for_keyword_loc),
    visit(node.index),
    token(node.in_keyword_loc),
    visit(node.collection),
    if node.do_keyword_loc
      token(node.do_keyword_loc)
    else
      srange_find(node.collection.location.end_offset, (node.statements&.location || node.end_keyword_loc).start_offset, [";"])
    end,
    visit(node.statements),
    token(node.end_keyword_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 748
def visit_forwarding_arguments_node(node)
  builder.forwarded_args(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 754
def visit_forwarding_parameter_node(node)
  builder.forward_arg(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 763
def visit_forwarding_super_node(node)
  visit_block(
    builder.keyword_cmd(
      :zsuper,
      ["super", srange_offsets(node.location.start_offset, node.location.start_offset + 5)]
    ),
    node.block
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 801
def visit_global_variable_and_write_node(node)
  builder.op_assign(
    builder.assignable(builder.gvar(token(node.name_loc))),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 791
def visit_global_variable_operator_write_node(node)
  builder.op_assign(
    builder.assignable(builder.gvar(token(node.name_loc))),
    [node.binary_operator_loc.slice.chomp("="), srange(node.binary_operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 811
def visit_global_variable_or_write_node(node)
  builder.op_assign(
    builder.assignable(builder.gvar(token(node.name_loc))),
    [node.operator_loc.slice.chomp("="), srange(node.operator_loc)],
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 775
def visit_global_variable_read_node(node)
  builder.gvar(token(node.location))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 821
def visit_global_variable_target_node(node)
  builder.assignable(builder.gvar([node.slice, srange(node.location)]))
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 781
def visit_global_variable_write_node(node)
  builder.assign(
    builder.assignable(builder.gvar(token(node.name_loc))),
    token(node.operator_loc),
    visit(node.value)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 827
def visit_hash_node(node)
  builder.associate(
    token(node.opening_loc),
    visit_all(node.elements),
    token(node.closing_loc)
  )
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 837
def visit_hash_pattern_node(node)
  elements = [*node.elements, *node.rest]

  if node.constant
    builder.const_pattern(visit(node.constant), token(node.opening_loc), builder.hash_pattern(nil, visit_all(elements), nil), token(node.closing_loc))
  else
    builder.hash_pattern(token(node.opening_loc), visit_all(elements), token(node.closing_loc))
  end
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 2051
def visit_heredoc(node)
  children = Array.new
  indented = false

  # If this is a dedenting heredoc, then we need to insert the opening
  # content into the children as well.
  if node.opening.start_with?("<<~") && node.parts.length > 0 && !node.parts.first.is_a?(StringNode)
    location = node.parts.first.location
    location = location.copy(start_offset: location.start_offset - location.start_line_slice.bytesize)
    children << builder.string_internal(token(location))
    indented = true
  end

  node.parts.each do |part|
    pushing =
      if part.is_a?(StringNode) && part.unescaped.include?("\n")
        unescaped = part.unescaped.lines
        escaped = part.content.lines

        escaped_lengths = []
        normalized_lengths = []

        if node.opening.end_with?("'")
          escaped.each do |line|
            escaped_lengths << line.bytesize
            normalized_lengths << chomped_bytesize(line)
          end
        else
          escaped
            .chunk_while { |before, after| before.match?(/(?<!\\)\\\r?\n$/) }
            .each do |lines|
              escaped_lengths << lines.sum(&:bytesize)
              normalized_lengths << lines.sum { |line| chomped_bytesize(line) }
            end
        end

        start_offset = part.location.start_offset

        unescaped.map.with_index do |unescaped_line, index|
          inner_part = builder.string_internal([unescaped_line, srange_offsets(start_offset, start_offset + normalized_lengths.fetch(index, 0))])
          start_offset += escaped_lengths.fetch(index, 0)
          inner_part
        end
      else
        [visit(part)]
      end

    pushing.each do |child|
      if child.type == :str && child.children.last == ""
        # nothing
      elsif child.type == :str && children.last && children.last.type == :str && !children.last.children.first.end_with?("\n")
        appendee = children[-1]

        location = appendee.loc
        location = location.with_expression(location.expression.join(child.loc.expression))

        children[-1] = appendee.updated(:str, [appendee.children.first << child.children.first], location: location)
      else
        children << child
      end
    end
  end

  closing = node.closing
  closing_t = [closing.chomp, srange_offsets(node.closing_loc.start_offset, node.closing_loc.end_offset - (closing[/\s+$/]&.length || 0))]
  composed = yield children, closing_t

  composed = composed.updated(nil, children[1..-1]) if indented
  composed
end

          
        

          
            
# File tmp/rubies/ruby-master/lib/prism/translation/parser/compiler.rb, line 855
def visit_if_node(node)
  if !node.if_keyword_loc
    builder.ternary(
      visit(node.predicate),
      token(node.then_keyword_loc),
      visit(node.statements),
      token(node.consequent.else_keyword_loc),
      visit(node.consequent)
    )
  elsif node.if_keyword_loc.start_offset == node.location.start_offset
    builder.condition(
      token(node.if_keyword_loc),
      visit(node.predicate),
      if node.then_keyword_loc
        token(node.then_keyword_loc)
      else
        srange_find(node.predicate.location.end_offset, (node.statements&.location || node.consequent&.location || node.end_keyword_loc).start_offset, [";"])
      end,
      visit(node.statements),
      case node.consequent
      when IfNode
        token(node.consequent.if_keyword_loc)
      when ElseNode
        token(node.consequent.else_keyword_loc)
      end,
      visit(node.consequent),
      if node.if_keyword != "elsif"
        token(node.end_keyword_loc)
      end
    )
  else
    builder.condition_mod(
      visit(node.statements),
      visit(node.consequent),
      token(node.if_keyword_loc),
      visit(node.predicate)
    )
  end
end